codegen-gql.mjs

#!/usr/bin/env zx

import * as fs from 'fs';

const NM_RUN_PATH = 'node_modules/.bin/';
const SDK_FILE_PATH = './src/sdk.ts';

process.env.NODE_TLS_REJECT_UNAUTHORIZED = 0;

await $`${NM_RUN_PATH}graphql-codegen --config ./codegen.ts`;
// 生成 hooks 的额外参数
const hooksExtraParam = process.env.HOOKS_EXTRA_PARAM;
const hooksExtraParamDefaultValue = process.env.HOOKS_EXTRA_PARAM_DEFAULT_VALUE;

let sdkContent = fs
  .readFileSync(SDK_FILE_PATH)
  .toString()
  .replace(
    'import useSWR,',
    `import useSWR from './useSWR';
import`
  )
  .replace(
    `import { GraphQLClientRequestHeaders } from 'graphql-request/build/cjs/types';`,
    `import { RequestConfig } from 'graphql-request/src/types';`
  )
  .replaceAll('GraphQLClientRequestHeaders', `RequestConfig['headers']`)
  .replace(/graphql\-request\/dist\//g, 'graphql-request/src/')
  .replace(
    /\s*id:\sstring,\s*fieldName:\s*keyof\s*Variables,\s*fieldValue:\s*Variables\[\s*typeof\sfieldName\s*\]\s*/,
    `[
        id,
        fieldName,
        fieldValue
      ]: [SWRKeyInterface, keyof Variables, Variables[keyof Variables]]`
  );
if (hooksExtraParam) {
  sdkContent = sdkContent
    .replace(
      'export function getSdkWithHooks(client: GraphQLClient,',
      `export function getSdkWithHooks(client: GraphQLClient, ${hooksExtraParam}${hooksExtraParamDefaultValue ? ` = '${hooksExtraParamDefaultValue}'` : ''},`
    )
    .replace(
      '(name: string, object: V = {} as V): SWRKeyInterface => [',
      `(name: string, object: V = {} as V): SWRKeyInterface => [ ${hooksExtraParam},`
    );
}
fs.writeFileSync(SDK_FILE_PATH, sdkContent);

await $`${NM_RUN_PATH}prettier --write \"./**/*.ts\"`;