add internal helper for tada fragment unrolling (#285)

.changeset/tough-squids-worry.md

@@ -0,0 +1,5 @@
+---
+'@0no-co/graphqlsp': patch
+---
+
+Add internal helper to unroll tada fragments

packages/graphqlsp/src/api.ts

@@ -1,4 +1,4 @@
 export { getGraphQLDiagnostics } from './diagnostics';
 export { init } from './ts';
-export { findAllPersistedCallExpressions } from './ast';
+export { findAllPersistedCallExpressions, unrollTadaFragments } from './ast';
 export { getDocumentReferenceFromTypeQuery } from './persisted';

packages/graphqlsp/src/ast/index.ts

@@ -101,6 +101,29 @@ function unrollFragment(
   return fragments;
 }
 
+export function unrollTadaFragments(
+  fragmentsArray: ts.ArrayLiteralExpression,
+  wip: FragmentDefinitionNode[],
+  info: ts.server.PluginCreateInfo
+): FragmentDefinitionNode[] {
+  fragmentsArray.elements.forEach(element => {
+    if (ts.isIdentifier(element)) {
+      wip.push(...unrollFragment(element, info));
+    } else if (ts.isPropertyAccessExpression(element)) {
+      let el = element;
+      while (ts.isPropertyAccessExpression(el.expression)) {
+        el = el.expression;
+      }
+
+      if (ts.isIdentifier(el.name)) {
+        wip.push(...unrollFragment(el.name, info));
+      }
+    }
+  });
+
+  return wip;
+}
+
 export function findAllCallExpressions(
   sourceFile: ts.SourceFile,
   info: ts.server.PluginCreateInfo,