@cesarandreslopez/occ 0.10.0 → 0.12.0

package/dist/src/code/tree-sitter.js.map

@@ -1 +1 @@
-{"version":3,"file":"tree-sitter.js","sourceRoot":"","sources":["../../../src/code/tree-sitter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAoBzC,IAAI,uBAA8D,CAAC;AACnE,MAAM,aAAa,GAAG,IAAI,GAAG,EAA8C,CAAC;AAC5E,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2C,CAAC;AAEtE,KAAK,UAAU,oBAAoB;IACjC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,uBAAuB,GAAG,MAAM,CAAC,iBAAiB,CAAC;aAChD,IAAI,CAAC,KAAK,EAAC,MAAM,EAAC,EAAE;YACnB,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACxB,uBAAuB,GAAG,SAAS,CAAC;YACpC,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,2DAA2D,MAAM,EAAE,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAe,EAAE,QAAgB;IACtE,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1D,MAAM,UAAU,GAAG;QACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,0BAA0B,EAAE,OAAO,EAAE,QAAQ,CAAC;QACjE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,6BAA6B,EAAE,OAAO,EAAE,QAAQ,CAAC;QACpE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,oBAAoB,EAAE,OAAO,EAAE,QAAQ,CAAC;KACrE,CAAC;IACF,OAAO,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,IAAI,CAAC;AACrE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAAe;IAC1D,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC1C,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;QAC1B,MAAM,QAAQ,GAAG,sBAAsB,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QACjE,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAC3B,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;QAC5C,IAAI,CAAC;YACH,OAAO,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,OAAe;IACpD,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACvC,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;QAC1B,MAAM,SAAS,GAAG,sBAAsB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC;QAC5B,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnD,oBAAoB,EAAE;YACtB,sBAAsB,CAAC,OAAO,CAAC;YAC/B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;SAC5B,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACzD,IAAI,CAAC;YACH,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACjC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAAe,EAAE,OAAe;IAC3E,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACvD,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3B,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAC5C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;IACnC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC7B,iFAAiF;IACjF,kFAAkF;IAClF,kFAAkF;IAClF,IAAI,CAAC;QACH,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,MAAM,EAAE,CAAC;IAClB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,OAAe,EACf,IAAoB;IAEpB,MAAM,KAAK,GAAG,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAC;IACrD,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1C,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,IAAoB;IAC9D,MAAM,MAAM,GAA0B,EAAE,CAAC;IAEzC,MAAM,KAAK,GAAG,CAAC,IAAoB,EAAQ,EAAE;QAC3C,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gBAChC,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC;aACtC,CAAC,CAAC;QACL,CAAC;QACD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC;YACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAChC,IAAI,KAAK;gBAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,CAAC,IAAI,CAAC,CAAC;IACZ,OAAO,MAAM,CAAC;AAChB,CAAC"}
+{"version":3,"file":"tree-sitter.js","sourceRoot":"","sources":["../../../src/code/tree-sitter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAsBzC,IAAI,uBAA8D,CAAC;AACnE,IAAI,YAA0C,CAAC;AAC/C,MAAM,aAAa,GAAG,IAAI,GAAG,EAA8C,CAAC;AAC5E,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2C,CAAC;AAEtE,KAAK,UAAU,oBAAoB;IACjC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,uBAAuB,GAAG,MAAM,CAAC,iBAAiB,CAAC;aAChD,IAAI,CAAC,KAAK,EAAC,MAAM,EAAC,EAAE;YACnB,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACxB,uBAAuB,GAAG,SAAS,CAAC;YACpC,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,2DAA2D,MAAM,EAAE,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAe,EAAE,QAAgB;IACtE,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1D,MAAM,UAAU,GAAG;QACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,0BAA0B,EAAE,OAAO,EAAE,QAAQ,CAAC;QACjE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,6BAA6B,EAAE,OAAO,EAAE,QAAQ,CAAC;QACpE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,oBAAoB,EAAE,OAAO,EAAE,QAAQ,CAAC;KACrE,CAAC;IACF,OAAO,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,IAAI,CAAC;AACrE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAAe;IAC1D,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC1C,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;QAC1B,MAAM,QAAQ,GAAG,sBAAsB,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QACjE,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAC3B,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;QAC5C,IAAI,CAAC;YACH,OAAO,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,OAAe;IACpD,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACvC,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;QAC1B,MAAM,SAAS,GAAG,sBAAsB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC;QAC5B,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnD,oBAAoB,EAAE;YACtB,sBAAsB,CAAC,OAAO,CAAC;YAC/B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;SAC5B,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACzD,IAAI,CAAC;YACH,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACjC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAAe,EAAE,OAAe;IAC3E,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACvD,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3B,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAC5C,+EAA+E;IAC/E,+EAA+E;IAC/E,gFAAgF;IAChF,+EAA+E;IAC/E,+EAA+E;IAC/E,0DAA0D;IAC1D,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;IACrC,CAAC;IACD,YAAY,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,OAAO,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,OAAe,EACf,IAAoB;IAEpB,MAAM,KAAK,GAAG,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAC;IACrD,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1C,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,IAAoB;IAC9D,MAAM,MAAM,GAA0B,EAAE,CAAC;IAEzC,MAAM,KAAK,GAAG,CAAC,IAAoB,EAAQ,EAAE;QAC3C,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gBAChC,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC;aACtC,CAAC,CAAC;QACL,CAAC;QACD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC;YACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAChC,IAAI,KAAK;gBAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,CAAC,IAAI,CAAC,CAAC;IACZ,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/code/tree.d.ts

@@ -0,0 +1,7 @@
+import type { CodebaseIndex, DirectoryTreeOptions, DirectoryTreeResult } from './types.js';
+export interface BuildDirectoryTreeOptions extends DirectoryTreeOptions {
+    countTokens?: (text: string) => number;
+}
+export declare function directoryTreeHeader(result: Pick<DirectoryTreeResult, 'root' | 'mode' | 'totalFiles'>): string;
+export declare function directoryTreeFooter(result: Pick<DirectoryTreeResult, 'truncated' | 'truncatedPaths'>): string | null;
+export declare function buildDirectoryTree(index: CodebaseIndex, options?: BuildDirectoryTreeOptions): DirectoryTreeResult;

package/dist/src/code/tree.js

@@ -0,0 +1,364 @@
+import { estimateTokenCount } from '../tokens.js';
+import { rankNodesCached } from './rank.js';
+import { toNormalizedSymbolKind } from './types.js';
+const DEFAULT_TREE_MODE = 'compact';
+const DEFAULT_MAX_DEPTH = 6;
+const DEFAULT_MAX_ENTRIES_PER_DIR = 50;
+const DEFAULT_TOKEN_BUDGET = 4096;
+const DEFAULT_SYMBOLS_PER_DIRECTORY = 3;
+function normalizeTreePath(value) {
+    return value.replace(/\\/g, '/').replace(/^\.\//, '').replace(/\/+/g, '/').replace(/\/$/, '');
+}
+function normalizeFocusPath(value) {
+    if (!value)
+        return undefined;
+    const normalized = normalizeTreePath(value.trim());
+    if (!normalized || normalized === '.')
+        return undefined;
+    return normalized;
+}
+function createDirectory(name, path) {
+    return {
+        name,
+        path,
+        directories: new Map(),
+        files: [],
+        fileCount: 0,
+    };
+}
+function isInFocus(relativePath, focusPath) {
+    if (!focusPath)
+        return true;
+    return relativePath === focusPath || relativePath.startsWith(`${focusPath}/`);
+}
+function pathWithinFocus(relativePath, focusPath) {
+    if (!focusPath)
+        return relativePath;
+    if (relativePath === focusPath)
+        return relativePath.split('/').at(-1) ?? relativePath;
+    return relativePath.slice(focusPath.length + 1);
+}
+function joinTreePath(base, segment) {
+    if (!base || base === '.')
+        return segment;
+    return `${base}/${segment}`;
+}
+function insertFile(root, file, focusPath) {
+    const relativePath = normalizeTreePath(file.relativePath);
+    const localPath = pathWithinFocus(relativePath, focusPath);
+    const parts = localPath.split('/').filter(Boolean);
+    if (parts.length === 0)
+        return;
+    root.fileCount++;
+    let current = root;
+    let currentPath = focusPath ?? '';
+    for (const directoryName of parts.slice(0, -1)) {
+        currentPath = joinTreePath(currentPath, directoryName);
+        let child = current.directories.get(directoryName);
+        if (!child) {
+            child = createDirectory(directoryName, currentPath);
+            current.directories.set(directoryName, child);
+        }
+        child.fileCount++;
+        current = child;
+    }
+    current.files.push(file);
+}
+function symbolIdentity(parts) {
+    return [parts.type, parts.name, parts.line ?? '', parts.containerName ?? ''].join('|');
+}
+function symbolsForFile(file, rankedNodes) {
+    const byKey = new Map();
+    for (const symbol of file.symbols) {
+        byKey.set(symbolIdentity({ type: symbol.type, name: symbol.name, line: symbol.line, containerName: symbol.containerName }), symbol);
+    }
+    const symbols = [];
+    const seen = new Set();
+    for (const ranked of rankedNodes) {
+        const key = symbolIdentity({
+            type: ranked.type,
+            name: ranked.name,
+            line: ranked.line,
+            containerName: ranked.containerName,
+        });
+        if (seen.has(key))
+            continue;
+        const parsed = byKey.get(key);
+        if (!parsed)
+            continue;
+        seen.add(key);
+        symbols.push({
+            nodeId: parsed.nodeId,
+            name: parsed.name,
+            kind: toNormalizedSymbolKind(parsed.type, parsed.containerName),
+            signature: parsed.fullSignature ?? parsed.signature ?? parsed.name,
+            line: parsed.line,
+            relativePath: file.relativePath,
+        });
+    }
+    return symbols;
+}
+function computeBestFiles(directory, fileScoresByPath) {
+    let bestScore;
+    let bestFile;
+    for (const file of directory.files) {
+        const score = fileScoresByPath.get(file.path);
+        if (!score)
+            continue;
+        if (!bestScore ||
+            score.score > bestScore.score ||
+            (score.score === bestScore.score && file.relativePath.localeCompare(bestScore.relativePath) < 0)) {
+            bestScore = score;
+            bestFile = file;
+        }
+    }
+    for (const child of directory.directories.values()) {
+        const childScore = computeBestFiles(child, fileScoresByPath);
+        if (!childScore || !child.bestFile)
+            continue;
+        if (!bestScore ||
+            childScore.score > bestScore.score ||
+            (childScore.score === bestScore.score && child.bestFile.relativePath.localeCompare(bestScore.relativePath) < 0)) {
+            bestScore = childScore;
+            bestFile = child.bestFile;
+        }
+    }
+    directory.bestFile = bestFile;
+    directory.bestFileScore = bestScore;
+    return bestScore;
+}
+function sortedDirectories(directory) {
+    return [...directory.directories.values()].sort((a, b) => a.name.localeCompare(b.name));
+}
+function sortedFiles(directory) {
+    return [...directory.files].sort((a, b) => a.relativePath.localeCompare(b.relativePath));
+}
+function collapseDirectory(directory, mode) {
+    if (mode === 'verbose')
+        return directory;
+    let current = directory;
+    while (current.files.length === 0 && current.directories.size === 1) {
+        const child = [...current.directories.values()][0];
+        if (!child)
+            break;
+        current = child;
+    }
+    return current;
+}
+function displayNameForDirectory(directory, focusPath) {
+    if (!focusPath)
+        return directory.path;
+    if (directory.path === focusPath)
+        return directory.name;
+    return directory.path.startsWith(`${focusPath}/`) ? directory.path.slice(focusPath.length + 1) : directory.path;
+}
+function directoryEntry(directory, depth, mode, focusPath, rankedSymbolsByPath) {
+    const entry = {
+        kind: 'directory',
+        path: directory.path,
+        name: displayNameForDirectory(directory, focusPath),
+        depth,
+        fileCount: directory.fileCount,
+    };
+    if (mode === 'map' && directory.bestFile && directory.bestFileScore) {
+        entry.score = directory.bestFileScore.score;
+        entry.normalizedScore = directory.bestFileScore.normalizedScore;
+        const symbols = symbolsForFile(directory.bestFile, rankedSymbolsByPath.get(directory.bestFile.path) ?? []).slice(0, DEFAULT_SYMBOLS_PER_DIRECTORY);
+        if (symbols.length > 0)
+            entry.symbols = symbols;
+    }
+    return entry;
+}
+function fileEntry(file, depth, fileScoresByPath) {
+    const score = fileScoresByPath.get(file.path);
+    const entry = {
+        kind: 'file',
+        path: file.relativePath,
+        name: file.relativePath.split('/').at(-1) ?? file.relativePath,
+        depth,
+        language: file.language,
+    };
+    if (score) {
+        entry.score = score.score;
+        entry.normalizedScore = score.normalizedScore;
+    }
+    return entry;
+}
+function renderEntryLines(entry) {
+    const indent = '  '.repeat(entry.depth);
+    if (entry.kind === 'file') {
+        const language = entry.language ? ` [${entry.language}]` : '';
+        return [`${indent}- ${entry.name}${language}`];
+    }
+    const count = entry.fileCount === 1 ? '1 file' : `${entry.fileCount ?? 0} files`;
+    const lines = [`${indent}+ ${entry.name}/ (${count})`];
+    for (const symbol of entry.symbols ?? []) {
+        const location = symbol.line ? `${symbol.relativePath}:${symbol.line}` : symbol.relativePath;
+        lines.push(`${indent}  - ${location} ${symbol.signature}`);
+    }
+    return lines;
+}
+function addTruncatedPath(paths, treePath) {
+    if (treePath.length > 0)
+        paths.add(treePath);
+}
+function collectDisplayEntries(directory, options, depth = 0) {
+    const displays = [];
+    const directoryItems = sortedDirectories(directory).map(child => ({
+        kind: 'directory',
+        directory: collapseDirectory(child, options.mode),
+        sortPath: child.path,
+    }));
+    const fileItems = sortedFiles(directory).map(file => ({
+        kind: 'file',
+        file,
+        sortPath: file.relativePath,
+    }));
+    const children = [...directoryItems, ...fileItems].sort((a, b) => {
+        if (a.kind !== b.kind)
+            return a.kind === 'directory' ? -1 : 1;
+        return a.sortPath.localeCompare(b.sortPath);
+    });
+    const admitted = children.slice(0, options.maxEntriesPerDir);
+    for (const omitted of children.slice(options.maxEntriesPerDir)) {
+        addTruncatedPath(options.truncatedPaths, omitted.sortPath);
+    }
+    for (const child of admitted) {
+        if (depth > options.maxDepth) {
+            addTruncatedPath(options.truncatedPaths, child.sortPath);
+            continue;
+        }
+        if (child.kind === 'file') {
+            const entry = fileEntry(child.file, depth, options.fileScoresByPath);
+            displays.push({ entry, lines: renderEntryLines(entry) });
+            continue;
+        }
+        const target = child.directory;
+        const entry = directoryEntry(target, depth, options.mode, options.focusPath, options.rankedSymbolsByPath);
+        displays.push({ entry, lines: renderEntryLines(entry) });
+        if (depth === options.maxDepth) {
+            if (target.fileCount > 0)
+                addTruncatedPath(options.truncatedPaths, target.path);
+            continue;
+        }
+        displays.push(...collectDisplayEntries(target, options, depth + 1));
+    }
+    return displays;
+}
+export function directoryTreeHeader(result) {
+    return `Directory Tree: ${result.root} (${result.mode}, ${result.totalFiles} files)`;
+}
+export function directoryTreeFooter(result) {
+    if (!result.truncated)
+        return null;
+    return `truncated: ${result.truncatedPaths.length} path(s)`;
+}
+function fixedBudgetReserve(root, mode, totalFiles, countTokens) {
+    const headerTokens = countTokens(directoryTreeHeader({ root, mode, totalFiles }));
+    const footerTokens = countTokens('truncated: 000 path(s)');
+    const newlineTokens = countTokens('\n\n');
+    return headerTokens + footerTokens + newlineTokens;
+}
+function enforceBudgets(candidates, truncatedPaths, options) {
+    const entries = [];
+    const lines = [];
+    const tokenBudget = options.tokenBudget;
+    const byteBudget = options.byteBudget;
+    const tokenReserve = tokenBudget == null ? 0 : fixedBudgetReserve(options.root, options.mode, options.totalFiles, options.countTokens);
+    const byteReserve = byteBudget == null
+        ? 0
+        : Buffer.byteLength(`${directoryTreeHeader({ root: options.root, mode: options.mode, totalFiles: options.totalFiles })}\ntruncated: 000 path(s)\n`, 'utf8');
+    let tokensUsed = tokenReserve;
+    let bytesUsed = byteReserve;
+    for (const candidate of candidates) {
+        const text = candidate.lines.join('\n');
+        const tokenCost = options.countTokens(`${text}\n`);
+        const byteCost = Buffer.byteLength(`${text}\n`, 'utf8');
+        const exceedsTokens = tokenBudget != null && tokensUsed + tokenCost > tokenBudget;
+        const exceedsBytes = byteBudget != null && bytesUsed + byteCost > byteBudget;
+        if (exceedsTokens || exceedsBytes) {
+            addTruncatedPath(truncatedPaths, candidate.entry.path);
+            continue;
+        }
+        entries.push(candidate.entry);
+        lines.push(...candidate.lines);
+        tokensUsed += tokenCost;
+        bytesUsed += byteCost;
+    }
+    if (tokenBudget == null)
+        tokensUsed = options.countTokens(lines.join('\n'));
+    if (byteBudget == null)
+        bytesUsed = Buffer.byteLength(lines.join('\n'), 'utf8');
+    return { entries, lines, tokensUsed, bytesUsed };
+}
+export function buildDirectoryTree(index, options = {}) {
+    const mode = options.mode ?? DEFAULT_TREE_MODE;
+    const maxDepth = options.maxDepth ?? DEFAULT_MAX_DEPTH;
+    const maxEntriesPerDir = options.maxEntriesPerDir ?? DEFAULT_MAX_ENTRIES_PER_DIR;
+    const tokenBudget = options.tokenBudget ?? DEFAULT_TOKEN_BUDGET;
+    const byteBudget = options.byteBudget;
+    const countTokens = options.countTokens ?? estimateTokenCount;
+    const focusPath = normalizeFocusPath(options.focusPath);
+    const root = createDirectory(focusPath ?? '.', focusPath ?? '.');
+    for (const file of [...index.files].sort((a, b) => a.relativePath.localeCompare(b.relativePath))) {
+        const relativePath = normalizeTreePath(file.relativePath);
+        if (!isInFocus(relativePath, focusPath))
+            continue;
+        insertFile(root, file, focusPath);
+    }
+    const ranking = rankNodesCached(index);
+    const fileScoresByPath = new Map(ranking.fileScores.map(score => [score.path, score]));
+    const rankedSymbolsByPath = new Map();
+    for (const node of ranking.nodes) {
+        if (node.type === 'file' || node.type === 'module')
+            continue;
+        const list = rankedSymbolsByPath.get(node.path);
+        if (list)
+            list.push(node);
+        else
+            rankedSymbolsByPath.set(node.path, [node]);
+    }
+    computeBestFiles(root, fileScoresByPath);
+    const truncated = new Set();
+    const candidates = collectDisplayEntries(root, {
+        mode,
+        focusPath,
+        maxDepth,
+        maxEntriesPerDir,
+        fileScoresByPath,
+        rankedSymbolsByPath,
+        truncatedPaths: truncated,
+    });
+    const budgeted = enforceBudgets(candidates, truncated, {
+        root: root.path,
+        mode,
+        totalFiles: root.fileCount,
+        tokenBudget,
+        byteBudget,
+        countTokens,
+    });
+    return {
+        root: root.path,
+        focusPath,
+        mode,
+        maxDepth,
+        maxEntriesPerDir,
+        tokenBudget,
+        tokensUsed: budgeted.tokensUsed,
+        byteBudget,
+        bytesUsed: budgeted.bytesUsed,
+        totalFiles: root.fileCount,
+        totalDirectories: countDirectories(root),
+        entries: budgeted.entries,
+        lines: budgeted.lines,
+        truncated: truncated.size > 0,
+        truncatedPaths: [...truncated].sort((a, b) => a.localeCompare(b)),
+    };
+}
+function countDirectories(directory) {
+    let count = directory.path === '.' ? 0 : 1;
+    for (const child of directory.directories.values())
+        count += countDirectories(child);
+    return count;
+}
+//# sourceMappingURL=tree.js.map

package/dist/src/code/tree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree.js","sourceRoot":"","sources":["../../../src/code/tree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,eAAe,EAAoC,MAAM,WAAW,CAAC;AAC9E,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AA+BpD,MAAM,iBAAiB,GAAsB,SAAS,CAAC;AACvD,MAAM,iBAAiB,GAAG,CAAC,CAAC;AAC5B,MAAM,2BAA2B,GAAG,EAAE,CAAC;AACvC,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAClC,MAAM,6BAA6B,GAAG,CAAC,CAAC;AAExC,SAAS,iBAAiB,CAAC,KAAa;IACtC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AAChG,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAyB;IACnD,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,UAAU,GAAG,iBAAiB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACnD,IAAI,CAAC,UAAU,IAAI,UAAU,KAAK,GAAG;QAAE,OAAO,SAAS,CAAC;IACxD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,eAAe,CAAC,IAAY,EAAE,IAAY;IACjD,OAAO;QACL,IAAI;QACJ,IAAI;QACJ,WAAW,EAAE,IAAI,GAAG,EAAE;QACtB,KAAK,EAAE,EAAE;QACT,SAAS,EAAE,CAAC;KACb,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,YAAoB,EAAE,SAA6B;IACpE,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO,YAAY,KAAK,SAAS,IAAI,YAAY,CAAC,UAAU,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,eAAe,CAAC,YAAoB,EAAE,SAA6B;IAC1E,IAAI,CAAC,SAAS;QAAE,OAAO,YAAY,CAAC;IACpC,IAAI,YAAY,KAAK,SAAS;QAAE,OAAO,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC;IACtF,OAAO,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,YAAY,CAAC,IAAY,EAAE,OAAe;IACjD,IAAI,CAAC,IAAI,IAAI,IAAI,KAAK,GAAG;QAAE,OAAO,OAAO,CAAC;IAC1C,OAAO,GAAG,IAAI,IAAI,OAAO,EAAE,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,IAAsB,EAAE,IAAgB,EAAE,SAA6B;IACzF,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC1D,MAAM,SAAS,GAAG,eAAe,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAC3D,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACnD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAE/B,IAAI,CAAC,SAAS,EAAE,CAAC;IACjB,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,IAAI,WAAW,GAAG,SAAS,IAAI,EAAE,CAAC;IAClC,KAAK,MAAM,aAAa,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,WAAW,GAAG,YAAY,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;QACvD,IAAI,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,eAAe,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;YACpD,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QACD,KAAK,CAAC,SAAS,EAAE,CAAC;QAClB,OAAO,GAAG,KAAK,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,cAAc,CAAC,KAA4E;IAClG,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE,EAAE,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzF,CAAC;AAED,SAAS,cAAc,CAAC,IAAgB,EAAE,WAAkC;IAC1E,MAAM,KAAK,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC9C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,KAAK,CAAC,GAAG,CACP,cAAc,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,EAChH,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAA0B,EAAE,CAAC;IAC1C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,cAAc,CAAC;YACzB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,aAAa,EAAE,MAAM,CAAC,aAAa;SACpC,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,SAAS;QAC5B,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM;YAAE,SAAS;QACtB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACd,OAAO,CAAC,IAAI,CAAC;YACX,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,sBAAsB,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC;YAC/D,SAAS,EAAE,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI;YAClE,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,gBAAgB,CACvB,SAA2B,EAC3B,gBAAiD;IAEjD,IAAI,SAAiC,CAAC;IACtC,IAAI,QAAgC,CAAC;IAErC,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IACE,CAAC,SAAS;YACV,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK;YAC7B,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAChG,CAAC;YACD,SAAS,GAAG,KAAK,CAAC;YAClB,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;IACH,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;QAC7D,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,QAAQ;YAAE,SAAS;QAC7C,IACE,CAAC,SAAS;YACV,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK;YAClC,CAAC,UAAU,CAAC,KAAK,KAAK,SAAS,CAAC,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAC/G,CAAC;YACD,SAAS,GAAG,UAAU,CAAC;YACvB,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC9B,SAAS,CAAC,aAAa,GAAG,SAAS,CAAC;IACpC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,SAA2B;IACpD,OAAO,CAAC,GAAG,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AAC1F,CAAC;AAED,SAAS,WAAW,CAAC,SAA2B;IAC9C,OAAO,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;AAC3F,CAAC;AAED,SAAS,iBAAiB,CAAC,SAA2B,EAAE,IAAuB;IAC7E,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACzC,IAAI,OAAO,GAAG,SAAS,CAAC;IACxB,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACpE,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK;YAAE,MAAM;QAClB,OAAO,GAAG,KAAK,CAAC;IAClB,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,uBAAuB,CAAC,SAA2B,EAAE,SAA6B;IACzF,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC,IAAI,CAAC;IACtC,IAAI,SAAS,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC,IAAI,CAAC;IACxD,OAAO,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC;AAClH,CAAC;AAED,SAAS,cAAc,CACrB,SAA2B,EAC3B,KAAa,EACb,IAAuB,EACvB,SAA6B,EAC7B,mBAAsD;IAEtD,MAAM,KAAK,GAAuB;QAChC,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,IAAI,EAAE,uBAAuB,CAAC,SAAS,EAAE,SAAS,CAAC;QACnD,KAAK;QACL,SAAS,EAAE,SAAS,CAAC,SAAS;KAC/B,CAAC;IAEF,IAAI,IAAI,KAAK,KAAK,IAAI,SAAS,CAAC,QAAQ,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;QACpE,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC;QAC5C,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC,aAAa,CAAC,eAAe,CAAC;QAChE,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,CAAC,QAAQ,EAAE,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAC9G,CAAC,EACD,6BAA6B,CAC9B,CAAC;QACF,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;IAClD,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,SAAS,CAChB,IAAgB,EAChB,KAAa,EACb,gBAAiD;IAEjD,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAuB;QAChC,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,IAAI,CAAC,YAAY;QACvB,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,YAAY;QAC9D,KAAK;QACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;KACxB,CAAC;IACF,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,KAAK,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;IAChD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAyB;IACjD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,OAAO,CAAC,GAAG,MAAM,KAAK,KAAK,CAAC,IAAI,GAAG,QAAQ,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,SAAS,IAAI,CAAC,QAAQ,CAAC;IACjF,MAAM,KAAK,GAAG,CAAC,GAAG,MAAM,KAAK,KAAK,CAAC,IAAI,MAAM,KAAK,GAAG,CAAC,CAAC;IACvD,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QACzC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC;QAC7F,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,OAAO,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAkB,EAAE,QAAgB;IAC5D,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;QAAE,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,qBAAqB,CAC5B,SAA2B,EAC3B,OAQC,EACD,KAAK,GAAG,CAAC;IAET,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAChE,IAAI,EAAE,WAAoB;QAC1B,SAAS,EAAE,iBAAiB,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC;QACjD,QAAQ,EAAE,KAAK,CAAC,IAAI;KACrB,CAAC,CAAC,CAAC;IACJ,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpD,IAAI,EAAE,MAAe;QACrB,IAAI;QACJ,QAAQ,EAAE,IAAI,CAAC,YAAY;KAC5B,CAAC,CAAC,CAAC;IACJ,MAAM,QAAQ,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAC/D,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;YAAE,OAAO,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,OAAO,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC7D,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC/D,gBAAgB,CAAC,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC7B,IAAI,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC7B,gBAAgB,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YACzD,SAAS;QACX,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;YACrE,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACzD,SAAS;QACX,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC;QAC/B,MAAM,KAAK,GAAG,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAC1G,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAEzD,IAAI,KAAK,KAAK,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,MAAM,CAAC,SAAS,GAAG,CAAC;gBAAE,gBAAgB,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAChF,SAAS;QACX,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiE;IACnG,OAAO,mBAAmB,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,SAAS,CAAC;AACvF,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiE;IACnG,IAAI,CAAC,MAAM,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IACnC,OAAO,cAAc,MAAM,CAAC,cAAc,CAAC,MAAM,UAAU,CAAC;AAC9D,CAAC;AAED,SAAS,kBAAkB,CACzB,IAAY,EACZ,IAAuB,EACvB,UAAkB,EAClB,WAAqC;IAErC,MAAM,YAAY,GAAG,WAAW,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;IAClF,MAAM,YAAY,GAAG,WAAW,CAAC,wBAAwB,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1C,OAAO,YAAY,GAAG,YAAY,GAAG,aAAa,CAAC;AACrD,CAAC;AAED,SAAS,cAAc,CACrB,UAAmC,EACnC,cAA2B,EAC3B,OAOC;IAED,MAAM,OAAO,GAAyB,EAAE,CAAC;IACzC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IACxC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACtC,MAAM,YAAY,GAChB,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IACpH,MAAM,WAAW,GACf,UAAU,IAAI,IAAI;QAChB,CAAC,CAAC,CAAC;QACH,CAAC,CAAC,MAAM,CAAC,UAAU,CACf,GAAG,mBAAmB,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,4BAA4B,EAC9H,MAAM,CACP,CAAC;IACR,IAAI,UAAU,GAAG,YAAY,CAAC;IAC9B,IAAI,SAAS,GAAG,WAAW,CAAC;IAE5B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,IAAI,IAAI,EAAE,MAAM,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,WAAW,IAAI,IAAI,IAAI,UAAU,GAAG,SAAS,GAAG,WAAW,CAAC;QAClF,MAAM,YAAY,GAAG,UAAU,IAAI,IAAI,IAAI,SAAS,GAAG,QAAQ,GAAG,UAAU,CAAC;QAC7E,IAAI,aAAa,IAAI,YAAY,EAAE,CAAC;YAClC,gBAAgB,CAAC,cAAc,EAAE,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvD,SAAS;QACX,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,UAAU,IAAI,SAAS,CAAC;QACxB,SAAS,IAAI,QAAQ,CAAC;IACxB,CAAC;IAED,IAAI,WAAW,IAAI,IAAI;QAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC5E,IAAI,UAAU,IAAI,IAAI;QAAE,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;IAEhF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAoB,EAAE,UAAqC,EAAE;IAC9F,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,iBAAiB,CAAC;IAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,iBAAiB,CAAC;IACvD,MAAM,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,2BAA2B,CAAC;IACjF,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,oBAAoB,CAAC;IAChE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACtC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,kBAAkB,CAAC;IAC9D,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACxD,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,IAAI,GAAG,EAAE,SAAS,IAAI,GAAG,CAAC,CAAC;IAEjE,KAAK,MAAM,IAAI,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;QACjG,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,SAAS,CAAC;YAAE,SAAS;QAClD,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAU,CAAC,CAAC,CAAC;IAChG,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC5D,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;YAAE,SAAS;QAC7D,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,IAAI;YAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;;YACrB,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAClD,CAAC;IACD,gBAAgB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IAEzC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,MAAM,UAAU,GAAG,qBAAqB,CAAC,IAAI,EAAE;QAC7C,IAAI;QACJ,SAAS;QACT,QAAQ;QACR,gBAAgB;QAChB,gBAAgB;QAChB,mBAAmB;QACnB,cAAc,EAAE,SAAS;KAC1B,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,cAAc,CAAC,UAAU,EAAE,SAAS,EAAE;QACrD,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,IAAI;QACJ,UAAU,EAAE,IAAI,CAAC,SAAS;QAC1B,WAAW;QACX,UAAU;QACV,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,SAAS;QACT,IAAI;QACJ,QAAQ;QACR,gBAAgB;QAChB,WAAW;QACX,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,UAAU;QACV,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,UAAU,EAAE,IAAI,CAAC,SAAS;QAC1B,gBAAgB,EAAE,gBAAgB,CAAC,IAAI,CAAC;QACxC,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,SAAS,EAAE,SAAS,CAAC,IAAI,GAAG,CAAC;QAC7B,cAAc,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;KAClE,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,SAA2B;IACnD,IAAI,KAAK,GAAG,SAAS,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,KAAK,MAAM,KAAK,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE;QAAE,KAAK,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACrF,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/code/types.d.ts

@@ -102,6 +102,7 @@ export declare const ParsedSymbolSchema: z.ZodObject<{
         variable: "variable";
     }>;
     name: z.ZodString;
+    nodeId: z.ZodOptional<z.ZodString>;
     line: z.ZodNumber;
     endLine: z.ZodOptional<z.ZodNumber>;
     startColumn: z.ZodOptional<z.ZodNumber>;
@@ -171,6 +172,7 @@ export declare const ParsedFileSchema: z.ZodObject<{
             variable: "variable";
         }>;
         name: z.ZodString;
+        nodeId: z.ZodOptional<z.ZodString>;
         line: z.ZodNumber;
         endLine: z.ZodOptional<z.ZodNumber>;
         startColumn: z.ZodOptional<z.ZodNumber>;
@@ -333,6 +335,7 @@ export declare const CodebaseIndexSchema: z.ZodObject<{
                 variable: "variable";
             }>;
             name: z.ZodString;
+            nodeId: z.ZodOptional<z.ZodString>;
             line: z.ZodNumber;
             endLine: z.ZodOptional<z.ZodNumber>;
             startColumn: z.ZodOptional<z.ZodNumber>;
@@ -475,6 +478,134 @@ export declare const CodebaseIndexSchema: z.ZodObject<{
     }, z.core.$strip>>;
 }, z.core.$strip>;
 export type CodebaseIndex = z.infer<typeof CodebaseIndexSchema>;
+export declare const DirectoryTreeModeSchema: z.ZodEnum<{
+    map: "map";
+    compact: "compact";
+    verbose: "verbose";
+}>;
+export type DirectoryTreeMode = z.infer<typeof DirectoryTreeModeSchema>;
+export declare const DirectoryTreeOptionsSchema: z.ZodObject<{
+    mode: z.ZodOptional<z.ZodEnum<{
+        map: "map";
+        compact: "compact";
+        verbose: "verbose";
+    }>>;
+    maxDepth: z.ZodOptional<z.ZodNumber>;
+    maxEntriesPerDir: z.ZodOptional<z.ZodNumber>;
+    tokenBudget: z.ZodOptional<z.ZodNumber>;
+    byteBudget: z.ZodOptional<z.ZodNumber>;
+    focusPath: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type DirectoryTreeOptions = z.infer<typeof DirectoryTreeOptionsSchema>;
+export declare const DirectoryTreeSymbolSchema: z.ZodObject<{
+    nodeId: z.ZodOptional<z.ZodString>;
+    name: z.ZodString;
+    kind: z.ZodEnum<{
+        function: "function";
+        file: "file";
+        enum: "enum";
+        type: "type";
+        module: "module";
+        class: "class";
+        interface: "interface";
+        variable: "variable";
+        parameter: "parameter";
+        method: "method";
+        other: "other";
+    }>;
+    signature: z.ZodString;
+    line: z.ZodOptional<z.ZodNumber>;
+    relativePath: z.ZodString;
+}, z.core.$strip>;
+export type DirectoryTreeSymbol = z.infer<typeof DirectoryTreeSymbolSchema>;
+export declare const DirectoryTreeEntrySchema: z.ZodObject<{
+    kind: z.ZodEnum<{
+        file: "file";
+        directory: "directory";
+    }>;
+    path: z.ZodString;
+    name: z.ZodString;
+    depth: z.ZodNumber;
+    fileCount: z.ZodOptional<z.ZodNumber>;
+    language: z.ZodOptional<z.ZodString>;
+    score: z.ZodOptional<z.ZodNumber>;
+    normalizedScore: z.ZodOptional<z.ZodNumber>;
+    symbols: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        nodeId: z.ZodOptional<z.ZodString>;
+        name: z.ZodString;
+        kind: z.ZodEnum<{
+            function: "function";
+            file: "file";
+            enum: "enum";
+            type: "type";
+            module: "module";
+            class: "class";
+            interface: "interface";
+            variable: "variable";
+            parameter: "parameter";
+            method: "method";
+            other: "other";
+        }>;
+        signature: z.ZodString;
+        line: z.ZodOptional<z.ZodNumber>;
+        relativePath: z.ZodString;
+    }, z.core.$strip>>>;
+}, z.core.$strip>;
+export type DirectoryTreeEntry = z.infer<typeof DirectoryTreeEntrySchema>;
+export declare const DirectoryTreeResultSchema: z.ZodObject<{
+    root: z.ZodString;
+    focusPath: z.ZodOptional<z.ZodString>;
+    mode: z.ZodEnum<{
+        map: "map";
+        compact: "compact";
+        verbose: "verbose";
+    }>;
+    maxDepth: z.ZodNumber;
+    maxEntriesPerDir: z.ZodNumber;
+    tokenBudget: z.ZodOptional<z.ZodNumber>;
+    tokensUsed: z.ZodNumber;
+    byteBudget: z.ZodOptional<z.ZodNumber>;
+    bytesUsed: z.ZodNumber;
+    totalFiles: z.ZodNumber;
+    totalDirectories: z.ZodNumber;
+    entries: z.ZodArray<z.ZodObject<{
+        kind: z.ZodEnum<{
+            file: "file";
+            directory: "directory";
+        }>;
+        path: z.ZodString;
+        name: z.ZodString;
+        depth: z.ZodNumber;
+        fileCount: z.ZodOptional<z.ZodNumber>;
+        language: z.ZodOptional<z.ZodString>;
+        score: z.ZodOptional<z.ZodNumber>;
+        normalizedScore: z.ZodOptional<z.ZodNumber>;
+        symbols: z.ZodOptional<z.ZodArray<z.ZodObject<{
+            nodeId: z.ZodOptional<z.ZodString>;
+            name: z.ZodString;
+            kind: z.ZodEnum<{
+                function: "function";
+                file: "file";
+                enum: "enum";
+                type: "type";
+                module: "module";
+                class: "class";
+                interface: "interface";
+                variable: "variable";
+                parameter: "parameter";
+                method: "method";
+                other: "other";
+            }>;
+            signature: z.ZodString;
+            line: z.ZodOptional<z.ZodNumber>;
+            relativePath: z.ZodString;
+        }, z.core.$strip>>>;
+    }, z.core.$strip>>;
+    lines: z.ZodArray<z.ZodString>;
+    truncated: z.ZodBoolean;
+    truncatedPaths: z.ZodArray<z.ZodString>;
+}, z.core.$strip>;
+export type DirectoryTreeResult = z.infer<typeof DirectoryTreeResultSchema>;
 export declare const CodeSearchResultSchema: z.ZodObject<{
     node: z.ZodObject<{
         id: z.ZodString;

package/dist/src/code/types.js

@@ -80,6 +80,8 @@ export const ParsedImportSchema = z.object({
 export const ParsedSymbolSchema = z.object({
     type: z.enum(['function', 'class', 'interface', 'type-alias', 'enum', 'variable']),
     name: z.string(),
+    /** Id of this symbol's graph node; stamped during index assembly. */
+    nodeId: z.string().optional(),
     line: z.number(),
     endLine: z.number().optional(),
     startColumn: z.number().optional(),
@@ -185,6 +187,51 @@ export const CodebaseIndexSchema = z.object({
     skippedFiles: z.array(CodeSkippedFileSchema),
     truncated: IndexTruncationSchema.optional(),
 });
+export const DirectoryTreeModeSchema = z.enum(['compact', 'verbose', 'map']);
+export const DirectoryTreeOptionsSchema = z.object({
+    mode: DirectoryTreeModeSchema.optional(),
+    maxDepth: z.number().int().nonnegative().optional(),
+    maxEntriesPerDir: z.number().int().positive().optional(),
+    tokenBudget: z.number().int().positive().optional(),
+    byteBudget: z.number().int().positive().optional(),
+    focusPath: z.string().optional(),
+});
+export const DirectoryTreeSymbolSchema = z.object({
+    nodeId: z.string().optional(),
+    name: z.string(),
+    kind: NormalizedSymbolKindSchema,
+    signature: z.string(),
+    line: z.number().optional(),
+    relativePath: z.string(),
+});
+export const DirectoryTreeEntrySchema = z.object({
+    kind: z.enum(['directory', 'file']),
+    path: z.string(),
+    name: z.string(),
+    depth: z.number(),
+    fileCount: z.number().optional(),
+    language: z.string().optional(),
+    score: z.number().optional(),
+    normalizedScore: z.number().optional(),
+    symbols: z.array(DirectoryTreeSymbolSchema).optional(),
+});
+export const DirectoryTreeResultSchema = z.object({
+    root: z.string(),
+    focusPath: z.string().optional(),
+    mode: DirectoryTreeModeSchema,
+    maxDepth: z.number(),
+    maxEntriesPerDir: z.number(),
+    tokenBudget: z.number().optional(),
+    tokensUsed: z.number(),
+    byteBudget: z.number().optional(),
+    bytesUsed: z.number(),
+    totalFiles: z.number(),
+    totalDirectories: z.number(),
+    entries: z.array(DirectoryTreeEntrySchema),
+    lines: z.array(z.string()),
+    truncated: z.boolean(),
+    truncatedPaths: z.array(z.string()),
+});
 export const CodeSearchResultSchema = z.object({
     node: CodeNodeSchema,
     resolution: ResolutionStatusSchema.optional(),