code to snippet from code node

Author: Calvinn Ng

extensions/vscode/src/util/expandSnippet.ts

@@ -1,8 +1,9 @@
 import { Chunk, IDE } from "core";
+import { languageForFilepath } from "core/autocomplete/constructPrompt";
 import { DEFAULT_IGNORE_DIRS } from "core/indexing/ignore";
 import { deduplicateArray } from "core/util";
 import { getParserForFile } from "core/util/treeSitter";
-import { SyntaxNode } from "web-tree-sitter";
+import type { SyntaxNode } from "web-tree-sitter";
 import { getDefinitionsForNode } from "../autocomplete/lsp";
 
 export async function expandSnippet(
@@ -20,8 +21,8 @@ export async function expandSnippet(
   const root: SyntaxNode = parser.parse(fullFileContents).rootNode;
 
   // Find all nodes contained in the range
-  let containedInRange: SyntaxNode[] = [];
-  let toExplore: SyntaxNode[] = [root];
+  const containedInRange: SyntaxNode[] = [];
+  const toExplore: SyntaxNode[] = [root];
   while (toExplore.length > 0) {
     const node = toExplore.pop()!;
     for (const child of node.namedChildren) {
@@ -49,7 +50,12 @@ export async function expandSnippet(
   let callExpressionDefinitions = (
     await Promise.all(
       callExpressions.map(async (node) => {
-        return getDefinitionsForNode(filepath, node);
+        return getDefinitionsForNode(
+          filepath,
+          node,
+          ide,
+          languageForFilepath(filepath),
+        );
       }),
     )
   ).flat();
@@ -103,4 +109,4 @@ export async function expandSnippet(
     }),
   );
   return chunks;
-}
+}