gql-typegen-addon.ts

/* eslint-env node */

import ts from 'typescript'
import {TypeGenAddonFactory} from 'ts-graphql-plugin'
import {astf} from 'ts-graphql-plugin/lib/ts-ast-util'

const typedQueryDocumentAddon: TypeGenAddonFactory = ({source}) => ({
  operationDefinition({tsResultNode, tsVariableNode}) {
    const lhs = astf.createIdentifier(`${tsResultNode.name.text}Document`)
    const rhs = astf.createTypeReferenceNode(
      astf.createIdentifier('TypedDocumentNode'),
      [
        astf.createTypeReferenceNode(
          astf.createIdentifier(tsResultNode.name.text)
        ),
        astf.createTypeReferenceNode(
          astf.createIdentifier(tsVariableNode.name.text)
        ),
      ]
    )

    const modifiers = [astf.createModifier(ts.SyntaxKind.ExportKeyword)]
    const typeAliasDeclaration = astf.createTypeAliasDeclaration(
      modifiers,
      lhs,
      undefined,
      rhs
    )
    source.pushNamedImportIfNeeded(
      'TypedDocumentNode',
      '@graphql-typed-document-node/core'
    )
    source.pushStatement(typeAliasDeclaration)
  },
})

export default typedQueryDocumentAddon