grepmax 0.17.23 → 0.18.0

package/dist/lib/daemon/watcher-manager.js

@@ -0,0 +1,484 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WatcherManager = void 0;
+const watcher = __importStar(require("@parcel/watcher"));
+const fs = __importStar(require("node:fs"));
+const path = __importStar(require("node:path"));
+const batch_processor_1 = require("../index/batch-processor");
+const watcher_1 = require("../index/watcher");
+const logger_1 = require("../utils/logger");
+const project_registry_1 = require("../utils/project-registry");
+const watcher_store_1 = require("../utils/watcher-store");
+// Watcher health windows used for FSEvents auto-recovery.
+const FSEVENTS_RECOVERY_INTERVAL_MS = 60 * 60 * 1000; // try recovery hourly
+const FSEVENTS_HEALTH_WINDOW_MS = 5 * 60 * 1000; // 5 min of quiet = "healthy"
+/**
+ * Owns per-project file watching: @parcel/watcher subscriptions, FSEvents
+ * overflow recovery, poll-mode fallback, and the offline catchup scan. Holds the
+ * watcher health/state maps; shares the `processors` and `subscriptions` maps
+ * with the daemon by reference so the streaming index ops and shutdown can still
+ * detach a project directly.
+ *
+ * Extracted from daemon.ts (Phase 2). Behavior-preserving — the daemon keeps
+ * thin watchProject/unwatchProject delegators over this manager.
+ */
+class WatcherManager {
+    constructor(deps) {
+        this.deps = deps;
+        this.pendingOps = new Set();
+        this.watcherFailCount = new Map();
+        this.pollIntervals = new Map();
+        this.pollRecoveryTimers = new Map();
+        this.lastOverflowMs = new Map();
+        this.lastCatchupEndMs = new Map();
+    }
+    watchProject(root) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.deps.processors.has(root) || this.pendingOps.has(root))
+                return;
+            const vectorDb = this.deps.getVectorDb();
+            const metaCache = this.deps.getMetaCache();
+            if (!vectorDb || !metaCache)
+                return;
+            this.pendingOps.add(root);
+            const processor = new batch_processor_1.ProjectBatchProcessor({
+                projectRoot: root,
+                vectorDb,
+                metaCache,
+                onReindex: (files, ms) => __awaiter(this, void 0, void 0, function* () {
+                    console.log(`[daemon:${path.basename(root)}] Reindexed ${files} file${files !== 1 ? "s" : ""} (${(ms / 1000).toFixed(1)}s)`);
+                    // Update project registry so gmax status shows fresh data
+                    const proj = (0, project_registry_1.getProject)(root);
+                    if (proj) {
+                        let chunkCount = proj.chunkCount;
+                        try {
+                            chunkCount = yield vectorDb.countRowsForPath(root);
+                        }
+                        catch (err) {
+                            console.warn(`[daemon:${path.basename(root)}] Failed to query chunk count: ${err}`);
+                        }
+                        (0, project_registry_1.registerProject)(Object.assign(Object.assign({}, proj), { lastIndexed: new Date().toISOString(), chunkCount }));
+                    }
+                    // Back to watching after batch completes
+                    (0, watcher_store_1.registerWatcher)({
+                        pid: process.pid,
+                        projectRoot: root,
+                        startTime: Date.now(),
+                        status: "watching",
+                        lastHeartbeat: Date.now(),
+                        lastReindex: Date.now(),
+                    });
+                }),
+                onActivity: () => {
+                    this.deps.touchActivity();
+                    // Mark as syncing while processing
+                    (0, watcher_store_1.registerWatcher)({
+                        pid: process.pid,
+                        projectRoot: root,
+                        startTime: Date.now(),
+                        status: "syncing",
+                        lastHeartbeat: Date.now(),
+                    });
+                },
+            });
+            this.deps.processors.set(root, processor);
+            // Subscribe with @parcel/watcher — native backend, no polling.
+            // If the kernel refuses (e.g. FSEvents slots stuck after a prior kill -9),
+            // fall straight through to poll mode. The retry/backoff path inside
+            // recoverWatcher is for transient overflows, not hard kernel-level
+            // subscribe failures, so we skip it on startup by priming failCount past
+            // MAX before invoking it.
+            try {
+                yield this.subscribeWatcher(root, processor);
+            }
+            catch (err) {
+                const name = path.basename(root);
+                console.error(`[daemon:${name}] Subscribe failed at startup (${err instanceof Error ? err.message : err}) — switching to poll mode`);
+                this.watcherFailCount.set(root, 1000); // > MAX_WATCHER_RETRIES
+                this.lastOverflowMs.set(root, Date.now());
+                this.recoverWatcher(root, processor);
+            }
+            (0, watcher_store_1.registerWatcher)({
+                pid: process.pid,
+                projectRoot: root,
+                startTime: Date.now(),
+                status: "watching",
+                lastHeartbeat: Date.now(),
+            });
+            // Catchup scan — find files changed while daemon was offline
+            this.catchupScan(root, processor).catch((err) => {
+                console.error(`[daemon:${path.basename(root)}] Catchup scan failed:`, err);
+            });
+            this.pendingOps.delete(root);
+            console.log(`[daemon] Watching ${root}`);
+        });
+    }
+    subscribeWatcher(root, processor) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const name = path.basename(root);
+            // Unsubscribe existing watcher if any (e.g. during recovery)
+            const existingSub = this.deps.subscriptions.get(root);
+            if (existingSub) {
+                try {
+                    yield existingSub.unsubscribe();
+                }
+                catch (_a) { }
+                this.deps.subscriptions.delete(root);
+            }
+            const sub = yield watcher.subscribe(root, (err, events) => {
+                var _a;
+                if (err) {
+                    console.error(`[daemon:${name}] Watcher error:`, err);
+                    this.recoverWatcher(root, processor);
+                    return;
+                }
+                // Only reset fail counter after sustained health (5min since last overflow)
+                const lastOverflow = (_a = this.lastOverflowMs.get(root)) !== null && _a !== void 0 ? _a : 0;
+                if (Date.now() - lastOverflow > 5 * 60 * 1000) {
+                    this.watcherFailCount.delete(root);
+                }
+                for (const event of events) {
+                    processor.handleFileEvent(event.type === "delete" ? "unlink" : "change", event.path);
+                }
+                this.deps.touchActivity();
+            }, { ignore: watcher_1.WATCHER_IGNORE_GLOBS });
+            this.deps.subscriptions.set(root, sub);
+        });
+    }
+    recoverWatcher(root, processor) {
+        var _a;
+        const name = path.basename(root);
+        if (this.deps.getShuttingDown())
+            return;
+        // Debounce: avoid multiple overlapping recovery attempts
+        const recoveryKey = `recover:${root}`;
+        if (this.pendingOps.has(recoveryKey))
+            return;
+        this.pendingOps.add(recoveryKey);
+        const fails = ((_a = this.watcherFailCount.get(root)) !== null && _a !== void 0 ? _a : 0) + 1;
+        this.watcherFailCount.set(root, fails);
+        this.lastOverflowMs.set(root, Date.now());
+        const MAX_WATCHER_RETRIES = 3;
+        const POLL_INTERVAL_MS = 5 * 60 * 1000; // 5 minutes
+        if (fails > MAX_WATCHER_RETRIES) {
+            // FSEvents can't handle this project — degrade to periodic catchup scans
+            // Always tear down the broken sub, even if poll mode is already active —
+            // this can happen if a recovery attempt resubscribed successfully then
+            // re-overflowed during the 5-min health window.
+            const sub = this.deps.subscriptions.get(root);
+            if (sub) {
+                sub.unsubscribe().catch(() => { });
+                this.deps.subscriptions.delete(root);
+            }
+            if (!this.pollIntervals.has(root)) {
+                console.error(`[daemon:${name}] FSEvents unreliable after ${fails} failures — switching to poll mode (${POLL_INTERVAL_MS / 60000}min interval)`);
+                // Run an immediate catchup, then schedule periodic ones
+                this.catchupScan(root, processor).catch((err) => {
+                    console.error(`[daemon:${name}] Poll catchup failed:`, err);
+                });
+                const interval = setInterval(() => {
+                    if (this.deps.getShuttingDown())
+                        return;
+                    this.deps.touchActivity();
+                    this.catchupScan(root, processor).catch((err) => {
+                        console.error(`[daemon:${name}] Poll catchup failed:`, err);
+                    });
+                }, POLL_INTERVAL_MS);
+                this.pollIntervals.set(root, interval);
+                // Schedule periodic attempts to climb back to native FSEvents — after
+                // a transient burst (large git checkout, npm install) the kernel
+                // buffer often calms down within an hour.
+                this.schedulePollModeRecovery(root, processor);
+                (0, watcher_store_1.registerWatcher)({
+                    pid: process.pid,
+                    projectRoot: root,
+                    startTime: Date.now(),
+                    status: "watching",
+                    lastHeartbeat: Date.now(),
+                });
+            }
+            this.pendingOps.delete(recoveryKey);
+            return;
+        }
+        // Backoff: wait before re-subscribing (3s, 6s, 12s)
+        const delayMs = 3000 * Math.pow(2, fails - 1);
+        console.error(`[daemon:${name}] Recovering watcher (attempt ${fails}/${MAX_WATCHER_RETRIES}, backoff ${delayMs}ms)...`);
+        setTimeout(() => {
+            if (this.deps.getShuttingDown()) {
+                this.pendingOps.delete(recoveryKey);
+                return;
+            }
+            (() => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                try {
+                    yield this.subscribeWatcher(root, processor);
+                    const lastCatchup = (_a = this.lastCatchupEndMs.get(root)) !== null && _a !== void 0 ? _a : 0;
+                    const CATCHUP_COOLDOWN_MS = 60000;
+                    if (Date.now() - lastCatchup < CATCHUP_COOLDOWN_MS) {
+                        console.log(`[daemon:${name}] Skipping catchup scan (last completed ${Math.round((Date.now() - lastCatchup) / 1000)}s ago)`);
+                    }
+                    else {
+                        yield this.catchupScan(root, processor);
+                    }
+                    console.log(`[daemon:${name}] Watcher recovered`);
+                }
+                catch (err) {
+                    console.error(`[daemon:${name}] Watcher recovery failed:`, err);
+                }
+                finally {
+                    this.pendingOps.delete(recoveryKey);
+                }
+            }))();
+        }, delayMs);
+    }
+    /**
+     * Once a project has fallen back to poll mode, periodically try to upgrade
+     * back to native FSEvents. The buffer overflows that triggered the fallback
+     * are usually transient (big git checkout, npm install, build output) — no
+     * point staying in 5-min poll mode forever.
+     */
+    schedulePollModeRecovery(root, processor) {
+        if (this.pollRecoveryTimers.has(root))
+            return;
+        const name = path.basename(root);
+        const timer = setInterval(() => {
+            if (this.deps.getShuttingDown())
+                return;
+            // Skip if a watcher recovery is already in flight or we're not in poll mode anymore.
+            if (!this.pollIntervals.has(root)) {
+                const t = this.pollRecoveryTimers.get(root);
+                if (t)
+                    clearInterval(t);
+                this.pollRecoveryTimers.delete(root);
+                return;
+            }
+            if (this.pendingOps.has(`recover:${root}`))
+                return;
+            void (() => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                console.log(`[daemon:${name}] Attempting to leave poll mode and reattach FSEvents...`);
+                try {
+                    // Reset failure counter so subscribeWatcher's error path treats this
+                    // as a fresh start. If it fails again, we'll fall right back into
+                    // poll mode via the same recoverWatcher path.
+                    this.watcherFailCount.delete(root);
+                    yield this.subscribeWatcher(root, processor);
+                    // Wait one health window — if the new subscription survives without
+                    // another overflow, we consider it recovered and tear down poll mode.
+                    yield new Promise((r) => setTimeout(r, FSEVENTS_HEALTH_WINDOW_MS));
+                    if (this.deps.getShuttingDown())
+                        return;
+                    const lastOverflow = (_a = this.lastOverflowMs.get(root)) !== null && _a !== void 0 ? _a : 0;
+                    if (Date.now() - lastOverflow < FSEVENTS_HEALTH_WINDOW_MS) {
+                        console.log(`[daemon:${name}] FSEvents recovery aborted — fresh overflow within health window, staying in poll mode`);
+                        return; // recoverWatcher will have re-armed poll mode if needed
+                    }
+                    // Healthy — drop poll mode.
+                    const pollInterval = this.pollIntervals.get(root);
+                    if (pollInterval) {
+                        clearInterval(pollInterval);
+                        this.pollIntervals.delete(root);
+                    }
+                    const recoveryTimer = this.pollRecoveryTimers.get(root);
+                    if (recoveryTimer) {
+                        clearInterval(recoveryTimer);
+                        this.pollRecoveryTimers.delete(root);
+                    }
+                    console.log(`[daemon:${name}] FSEvents recovered — poll mode disabled`);
+                }
+                catch (err) {
+                    console.error(`[daemon:${name}] Poll-mode recovery attempt failed:`, err);
+                }
+            }))();
+        }, FSEVENTS_RECOVERY_INTERVAL_MS);
+        timer.unref();
+        this.pollRecoveryTimers.set(root, timer);
+    }
+    catchupScan(root, processor) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_1, _b, _c;
+            const { walk } = yield Promise.resolve().then(() => __importStar(require("../index/walker")));
+            const { INDEXABLE_EXTENSIONS, MAX_FILE_SIZE_BYTES } = yield Promise.resolve().then(() => __importStar(require("../../config")));
+            const { isFileCached } = yield Promise.resolve().then(() => __importStar(require("../utils/cache-check")));
+            const metaCache = this.deps.getMetaCache();
+            const rootPrefix = root.endsWith("/") ? root : `${root}/`;
+            const cachedPaths = yield metaCache.getKeysWithPrefix(rootPrefix);
+            const seenPaths = new Set();
+            let queued = 0;
+            let skipped = 0;
+            let debugSamples = 0;
+            try {
+                for (var _d = true, _e = __asyncValues(walk(root, {
+                    additionalPatterns: ["**/.git/**", "**/.gmax/**"],
+                })), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
+                    _c = _f.value;
+                    _d = false;
+                    const relPath = _c;
+                    const absPath = path.join(root, relPath);
+                    const ext = path.extname(absPath).toLowerCase();
+                    const bn = path.basename(absPath).toLowerCase();
+                    if (!INDEXABLE_EXTENSIONS.has(ext) && !INDEXABLE_EXTENSIONS.has(bn))
+                        continue;
+                    seenPaths.add(absPath);
+                    try {
+                        const stats = yield fs.promises.stat(absPath);
+                        // Skip files that are too large or empty — they'll never be indexed
+                        if (stats.size === 0 || stats.size > MAX_FILE_SIZE_BYTES)
+                            continue;
+                        const cached = metaCache.get(absPath);
+                        if (!isFileCached(cached, stats)) {
+                            // Fast path: if only mtime changed but size is identical and we have a hash,
+                            // just verify the hash in-process instead of sending to a worker.
+                            if (cached && cached.hash && cached.size === stats.size) {
+                                const { computeBufferHash } = yield Promise.resolve().then(() => __importStar(require("../utils/file-utils")));
+                                const buf = yield fs.promises.readFile(absPath);
+                                const hash = computeBufferHash(buf);
+                                if (hash === cached.hash) {
+                                    // Content unchanged — update mtime in cache and skip worker
+                                    metaCache.put(absPath, Object.assign(Object.assign({}, cached), { mtimeMs: stats.mtimeMs }));
+                                    skipped++;
+                                    continue;
+                                }
+                            }
+                            // Debug: log first few misses to diagnose re-queue loops
+                            if (debugSamples < 5) {
+                                (0, logger_1.debug)("catchup", `miss ${relPath}: cached=${cached ? `mtime=${Math.trunc(cached.mtimeMs)} size=${cached.size}` : "null"} stat=mtime=${Math.trunc(stats.mtimeMs)} size=${stats.size}`);
+                                debugSamples++;
+                            }
+                            processor.handleFileEvent("change", absPath);
+                            queued++;
+                            // Throttle: pause periodically during large catchup scans to let the
+                            // batch processor drain and compaction run between bursts.
+                            if (queued % 500 === 0) {
+                                (0, logger_1.debug)("catchup", `${path.basename(root)}: throttle pause at ${queued} queued`);
+                                yield new Promise(r => setTimeout(r, 5000));
+                            }
+                        }
+                        else {
+                            skipped++;
+                        }
+                    }
+                    catch (_g) { }
+                }
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
+                }
+                finally { if (e_1) throw e_1.error; }
+            }
+            (0, logger_1.debug)("catchup", `${path.basename(root)}: ${queued} queued, ${skipped} skipped (cached ok), ${seenPaths.size} total`);
+            // Purge files deleted while daemon was offline
+            let purged = 0;
+            for (const cachedPath of cachedPaths) {
+                if (!seenPaths.has(cachedPath)) {
+                    processor.handleFileEvent("unlink", cachedPath);
+                    purged++;
+                }
+            }
+            if (queued > 0 || purged > 0) {
+                const parts = [];
+                if (queued > 0)
+                    parts.push(`${queued} changed`);
+                if (purged > 0)
+                    parts.push(`${purged} deleted`);
+                console.log(`[daemon:${path.basename(root)}] Catchup: ${parts.join(", ")} file(s) while offline`);
+            }
+            this.lastCatchupEndMs.set(root, Date.now());
+        });
+    }
+    unwatchProject(root) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const processor = this.deps.processors.get(root);
+            if (!processor)
+                return;
+            yield processor.close();
+            const sub = this.deps.subscriptions.get(root);
+            if (sub) {
+                yield sub.unsubscribe();
+                this.deps.subscriptions.delete(root);
+            }
+            this.deps.processors.delete(root);
+            this.deps.evictSearcher(root);
+            this.lastOverflowMs.delete(root);
+            this.lastCatchupEndMs.delete(root);
+            (0, watcher_store_1.unregisterWatcherByRoot)(root);
+            console.log(`[daemon] Unwatched ${root}`);
+        });
+    }
+    /**
+     * Stop all poll intervals + their FSEvents recovery probes and unsubscribe
+     * every watcher. Called from the daemon's shutdown after the worker pool is
+     * destroyed (the subscriptions map is shared, so it is cleared here).
+     */
+    teardown() {
+        return __awaiter(this, void 0, void 0, function* () {
+            for (const interval of this.pollIntervals.values()) {
+                clearInterval(interval);
+            }
+            this.pollIntervals.clear();
+            for (const interval of this.pollRecoveryTimers.values()) {
+                clearInterval(interval);
+            }
+            this.pollRecoveryTimers.clear();
+            for (const sub of this.deps.subscriptions.values()) {
+                try {
+                    yield sub.unsubscribe();
+                }
+                catch (_a) { }
+            }
+            this.deps.subscriptions.clear();
+        });
+    }
+}
+exports.WatcherManager = WatcherManager;

package/dist/lib/graph/graph-builder.js

@@ -17,7 +17,11 @@ class GraphBuilder {
     constructor(db, pathPrefix, excludePrefixes) {
         this.db = db;
         // Normalize to ensure trailing slash for LIKE queries
-        this.pathPrefix = pathPrefix ? (pathPrefix.endsWith("/") ? pathPrefix : `${pathPrefix}/`) : undefined;
+        this.pathPrefix = pathPrefix
+            ? pathPrefix.endsWith("/")
+                ? pathPrefix
+                : `${pathPrefix}/`
+            : undefined;
         this.excludePrefixes = (excludePrefixes !== null && excludePrefixes !== void 0 ? excludePrefixes : []).map((p) => p.endsWith("/") ? p : `${p}/`);
     }
     scopeWhere(condition) {
@@ -37,14 +41,22 @@ class GraphBuilder {
         return __awaiter(this, void 0, void 0, function* () {
             const table = yield this.db.ensureTable();
             const escaped = (0, filter_builder_1.escapeSqlString)(symbol);
-            // Find chunks where referenced_symbols contains the symbol
+            // Find chunks that reference the symbol from a call position
+            // (referenced_symbols) OR a type position (type_referenced_symbols). The
+            // two are stored separately so type edges never inflate the call-edge count
+            // used for ranking; navigation unions them.
             const rows = yield table
                 .query()
                 .select([
-                "path", "start_line", "defined_symbols", "referenced_symbols",
-                "role", "parent_symbol", "complexity",
+                "path",
+                "start_line",
+                "defined_symbols",
+                "referenced_symbols",
+                "role",
+                "parent_symbol",
+                "complexity",
             ])
-                .where(this.scopeWhere(`array_contains(referenced_symbols, '${escaped}')`))
+                .where(this.scopeWhere(`(array_contains(referenced_symbols, '${escaped}') OR array_contains(type_referenced_symbols, '${escaped}'))`))
                 .limit(100)
                 .toArray();
             return rows.map((row) => this.mapRowToNode(row, symbol, "caller"));
@@ -82,8 +94,13 @@ class GraphBuilder {
             const centerRows = yield table
                 .query()
                 .select([
-                "path", "start_line", "defined_symbols", "referenced_symbols",
-                "role", "parent_symbol", "complexity",
+                "path",
+                "start_line",
+                "defined_symbols",
+                "referenced_symbols",
+                "role",
+                "parent_symbol",
+                "complexity",
             ])
                 .where(this.scopeWhere(`array_contains(defined_symbols, '${escaped}')`))
                 .limit(1)
@@ -165,7 +182,12 @@ class GraphBuilder {
             }
             const visited = new Set([symbol]);
             const callerTree = yield this.expandCallers(graph.callers, depth - 1, visited);
-            return { center: graph.center, callerTree, callees: graph.callees, importers };
+            return {
+                center: graph.center,
+                callerTree,
+                callees: graph.callees,
+                importers,
+            };
         });
     }
     expandCallers(callers, remainingDepth, visited) {

package/dist/lib/graph/impact.js

@@ -22,8 +22,10 @@ const TEST_FILE_RE = /\.(test|spec)\.[cm]?[jt]sx?$/i;
 const NATIVE_TEST_DIR_RE = /(^|\/)\w+Tests?(\/|$)/;
 const NATIVE_TEST_FILE_RE = /Tests?\.(swift|kt|java)$/;
 function isTestPath(filePath) {
-    return TEST_DIR_RE.test(filePath) || TEST_FILE_RE.test(filePath)
-        || NATIVE_TEST_DIR_RE.test(filePath) || NATIVE_TEST_FILE_RE.test(filePath);
+    return (TEST_DIR_RE.test(filePath) ||
+        TEST_FILE_RE.test(filePath) ||
+        NATIVE_TEST_DIR_RE.test(filePath) ||
+        NATIVE_TEST_FILE_RE.test(filePath));
 }
 const arrow_1 = require("../utils/arrow");
 /**
@@ -214,7 +216,7 @@ function findDependents(symbols_1, vectorDb_1, projectRoot_1, excludePaths_1) {
             const rows = yield table
                 .query()
                 .select(["path"])
-                .where(`array_contains(referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}') AND ${pathScope}`)
+                .where(`(array_contains(referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}') OR array_contains(type_referenced_symbols, '${(0, filter_builder_1.escapeSqlString)(sym)}')) AND ${pathScope}`)
                 .limit(200)
                 .toArray();
             for (const row of rows) {