Function crawlTypes

extensions/vscode/src/autocomplete/lsp.ts:146–226

(
  rif: RangeInFile | RangeInFileWithContents,
  ide: IDE,
  depth: number = 1,
  results: RangeInFileWithContents[] = [],
  searchedLabels: Set<string> = new Set(),
)

}

async function crawlTypes(
  rif: RangeInFile | RangeInFileWithContents,
  ide: IDE,
  depth: number = 1,
  results: RangeInFileWithContents[] = [],
  searchedLabels: Set<string> = new Set(),
): Promise<RangeInFileWithContents[]> {
  // Get the file contents if not already attached
  const contents = isRifWithContents(rif)
    ? rif.contents
    : await ide.readFile(rif.filepath);

  // Parse AST
  const ast = await getAst(rif.filepath, contents);
  if (!ast) {
    return results;
  }
  const astLineCount = ast.rootNode.text.split("\n").length;

  // Find type identifiers
  const identifierNodes = findTypeIdentifiers(ast.rootNode).filter(
    (node) => !searchedLabels.has(node.text),
  );
  // Don't search for the same type definition more than once
  // We deduplicate below to be sure, but this saves calls to the LSP
  identifierNodes.forEach((node) => searchedLabels.add(node.text));

  // Use LSP to get the definitions of those types
  const definitions = [];

  for (const node of identifierNodes) {
    const [typeDef] = await executeGotoProvider({
      uri: vscode.Uri.parse(rif.filepath),
      // TODO: tree-sitter is zero-indexed, but there seems to be an off-by-one
      // error at least with the .ts parser sometimes
      line:
        rif.range.start.line +
        Math.min(node.startPosition.row, astLineCount - 1),
      character: rif.range.start.character + node.startPosition.column,
      name: "vscode.executeDefinitionProvider",
    });

    if (!typeDef) {
      definitions.push(undefined);
      continue;
    }

    const contents = await ide.readRangeInFile(typeDef.filepath, typeDef.range);

    definitions.push({
      ...typeDef,
      contents,
    });
  }

  // TODO: Filter out if not in our code?

  // Filter out duplicates