Function walkTree

src/core/tree-sitter.ts:181–217

(rootNode: TSNode, defTypes: Record<string, string>, maxDepth: number = 3)

}

function walkTree(rootNode: TSNode, defTypes: Record<string, string>, maxDepth: number = 3): CodeSymbol[] {
  const symbols: CodeSymbol[] = [];

  function visit(node: TSNode, depth: number, parent: CodeSymbol | null): void {
    if (depth > maxDepth) return;

    const kindStr = defTypes[node.type];
    if (kindStr) {
      const sym: CodeSymbol = {
        name: extractName(node, kindStr),
        kind: mapKind(kindStr),
        line: node.startPosition.row + 1,
        endLine: node.endPosition.row + 1,
        signature: extractSignature(node),
        children: [],
      };

      if (parent && depth > 0) {
        parent.children.push(sym);
      } else {
        symbols.push(sym);
      }

      for (const child of node.namedChildren ?? []) {
        visit(child, depth + 1, sym);
      }
      return;
    }

    for (const child of node.namedChildren ?? []) {
      visit(child, depth, parent);
    }
  }

  visit(rootNode, 0, null);
  return symbols;
}

export function getGrammarName(ext: string): string | null {
  return EXT_TO_GRAMMAR[ext.toLowerCase()] ?? null;