fix: reference to incorrect node when computing schema name of persis…

…ted call
0no-co · Jun 11, 2024 · 1fc8914 · 1fc8914
1 parent 31c96bc
commit 1fc8914
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/packages/graphqlsp/src/ast/checks.ts b/packages/graphqlsp/src/ast/checks.ts
@@ -93,10 +93,13 @@ export const isGraphQLTag = (
 /** Retrieves the `__name` branded tag from gql.tada `graphql()` or `graphql.persisted()` calls */
 export const getSchemaName = (
   node: ts.CallExpression,
-  typeChecker: ts.TypeChecker | undefined
+  typeChecker: ts.TypeChecker | undefined,
+  isTadaPersistedCall = false
 ): string | null => {
   if (!typeChecker) return null;
-  const type = typeChecker.getTypeAtLocation(node.expression);
+  const type = typeChecker.getTypeAtLocation(
+    isTadaPersistedCall ? node.getChildAt(0).getChildAt(0) : node.expression
+  );
   if (type) {
     const brandTypeSymbol = type.getProperty('__name');
     if (brandTypeSymbol) {

diff --git a/packages/graphqlsp/src/ast/index.ts b/packages/graphqlsp/src/ast/index.ts
@@ -215,7 +215,7 @@ export function findAllPersistedCallExpressions(
     if (!checks.isTadaPersistedCall(node, typeChecker)) {
       return;
     } else if (info) {
-      const name = checks.getSchemaName(node, typeChecker);
+      const name = checks.getSchemaName(node, typeChecker, true);
       result.push({ node, schema: name });
     } else {
       result.push(node);