fix: get TS function call type params working

Author: jedwards1211

src/internal/graphqlTypegenCore.ts

@@ -21,6 +21,7 @@ import j, {
   TSTypeParameterInstantiation,
   TSQualifiedName,
   TSTypeAliasDeclaration,
+  CallExpression,
 } from 'jscodeshift'
 import findImports from 'jscodeshift-find-imports'
 import addImports from 'jscodeshift-add-imports'
@@ -178,7 +179,7 @@ export default function graphqlTypegenCore(
     throw new Error('invalid typeAnnotation')
   }
 
-  function makeFunctionTypeArguments(
+  function makeFunctionTypeParameterInstantiation(
     data: TypeAlias | TSTypeAliasDeclaration,
     variables?: TypeAlias | TSTypeAliasDeclaration | null | undefined
   ): TypeParameterInstantiation | TSTypeParameterInstantiation {
@@ -553,6 +554,14 @@ export default function graphqlTypegenCore(
         })
     }
 
+    const setTypeParameters = (
+      node: CallExpression,
+      params: TypeParameterInstantiation | TSTypeParameterInstantiation
+    ) => {
+      if (isTS) (node as any).typeParameters = params
+      else (node as any).typeArguments = params
+    }
+
     //////////////////////////////////////////////////
     // Add types to useQuery hooks
 
@@ -572,9 +581,9 @@ export default function graphqlTypegenCore(
           const { data, variables } = onlyValue(generatedTypes.query) || {}
           if (!data || path.node.init?.type !== 'CallExpression') return
           if (useFunctionTypeArguments) {
-            ;(path.node.init as any).typeArguments = makeFunctionTypeArguments(
-              data,
-              variables
+            setTypeParameters(
+              path.node.init,
+              makeFunctionTypeParameterInstantiation(data, variables)
             )
           } else {
             if (
@@ -636,9 +645,9 @@ export default function graphqlTypegenCore(
             node: { id },
           } = path
           if (useFunctionTypeArguments) {
-            ;(path.node.init as any).typeArguments = makeFunctionTypeArguments(
-              data,
-              variables
+            setTypeParameters(
+              path.node.init,
+              makeFunctionTypeParameterInstantiation(data, variables)
             )
           } else {
             if (!mutationFunction) return
@@ -684,11 +693,11 @@ export default function graphqlTypegenCore(
         .forEach((path: ASTPath<VariableDeclarator>): void => {
           const { data, variables } =
             onlyValue(generatedTypes.subscription) || {}
-          if (!data) return
+          if (!data || path.node.init?.type !== 'CallExpression') return
           if (useFunctionTypeArguments) {
-            ;(path.node.init as any).typeArguments = makeFunctionTypeArguments(
-              data,
-              variables
+            setTypeParameters(
+              path.node.init,
+              makeFunctionTypeParameterInstantiation(data, variables)
             )
           } else {
             if (

test/graphql-typegen-async/ts/useMutationReplaceTypeParams.ts

@@ -0,0 +1,76 @@
+import * as path from 'path'
+
+export const file = 'file.tsx'
+export const normalize = false
+
+export const input = `
+
+import * as React from 'react'
+import { useMutation } from '@apollo/react-hooks'
+import gql from 'graphql-tag'
+
+const mutation = gql\`
+mutation createReview($episode: Episode!, $review: ReviewInput!) {
+  createReview(episode: $episode, review: $review) {
+    episode
+    stars
+    commentary
+  }
+}
+\`
+
+const Comp = (props: {id: string}): React.Node => {
+  const [createReview] = useMutation<X>(mutation)
+  return <div />
+}
+`
+
+export const options = {
+  addTypename: false,
+  schemaFile: path.resolve(__dirname, '../../../starwars.graphql'),
+}
+
+export const expected = `
+
+import * as React from 'react'
+import { useMutation } from '@apollo/react-hooks'
+import gql from 'graphql-tag'
+
+const mutation = gql\`
+mutation createReview($episode: Episode!, $review: ReviewInput!) {
+  createReview(episode: $episode, review: $review) {
+    episode
+    stars
+    commentary
+  }
+}
+\`
+
+// @graphql-typegen auto-generated
+type CreateReviewMutationVariables = {
+  episode: "NEWHOPE" | "EMPIRE" | "JEDI",
+  review: {
+    stars: number,
+    commentary?: string | null,
+    favorite_color?: {
+      red: number,
+      green: number,
+      blue: number
+    } | null
+  }
+};
+
+// @graphql-typegen auto-generated
+type CreateReviewMutationData = {
+  createReview: {
+    episode: "NEWHOPE" | "EMPIRE" | "JEDI" | null,
+    stars: number,
+    commentary: string | null
+  } | null
+};
+
+const Comp = (props: {id: string}): React.Node => {
+  const [createReview] = useMutation<CreateReviewMutationData, CreateReviewMutationVariables>(mutation)
+  return <div />
+}
+`