@optave/codegraph 2.4.0 → 2.5.1

package/src/structure.js

@@ -162,6 +162,48 @@ export function buildStructure(db, fileSymbols, _rootDir, lineCountMap, director
     }
   }
 
+  // Build reverse index: file → set of ancestor directories (O(files × depth))
+  const fileToAncestorDirs = new Map();
+  for (const [dir, files] of dirFiles) {
+    for (const f of files) {
+      if (!fileToAncestorDirs.has(f)) fileToAncestorDirs.set(f, new Set());
+      fileToAncestorDirs.get(f).add(dir);
+    }
+  }
+
+  // Single O(E) pass: pre-aggregate edge counts per directory
+  const dirEdgeCounts = new Map();
+  for (const dir of allDirs) {
+    dirEdgeCounts.set(dir, { intra: 0, fanIn: 0, fanOut: 0 });
+  }
+  for (const { source_file, target_file } of importEdges) {
+    const srcDirs = fileToAncestorDirs.get(source_file);
+    const tgtDirs = fileToAncestorDirs.get(target_file);
+    if (!srcDirs && !tgtDirs) continue;
+
+    // For each directory that contains the source file
+    if (srcDirs) {
+      for (const dir of srcDirs) {
+        const counts = dirEdgeCounts.get(dir);
+        if (!counts) continue;
+        if (tgtDirs?.has(dir)) {
+          counts.intra++;
+        } else {
+          counts.fanOut++;
+        }
+      }
+    }
+    // For each directory that contains the target but NOT the source
+    if (tgtDirs) {
+      for (const dir of tgtDirs) {
+        if (srcDirs?.has(dir)) continue; // already counted as intra
+        const counts = dirEdgeCounts.get(dir);
+        if (!counts) continue;
+        counts.fanIn++;
+      }
+    }
+  }
+
   const computeDirMetrics = db.transaction(() => {
     for (const [dir, files] of dirFiles) {
       const dirRow = getNodeId.get(dir, 'directory', dir, 0);
@@ -169,9 +211,6 @@ export function buildStructure(db, fileSymbols, _rootDir, lineCountMap, director
 
       const fileCount = files.length;
       let symbolCount = 0;
-      let totalFanIn = 0;
-      let totalFanOut = 0;
-      const filesInDir = new Set(files);
 
       for (const f of files) {
         const sym = fileSymbols.get(f);
@@ -187,23 +226,10 @@ export function buildStructure(db, fileSymbols, _rootDir, lineCountMap, director
         }
       }
 
-      // Compute cross-boundary fan-in/fan-out and cohesion
-      let intraEdges = 0;
-      let crossEdges = 0;
-      for (const { source_file, target_file } of importEdges) {
-        const srcInside = filesInDir.has(source_file);
-        const tgtInside = filesInDir.has(target_file);
-        if (srcInside && tgtInside) {
-          intraEdges++;
-        } else if (srcInside || tgtInside) {
-          crossEdges++;
-          if (!srcInside && tgtInside) totalFanIn++;
-          if (srcInside && !tgtInside) totalFanOut++;
-        }
-      }
-
-      const totalEdges = intraEdges + crossEdges;
-      const cohesion = totalEdges > 0 ? intraEdges / totalEdges : null;
+      // O(1) lookup from pre-aggregated edge counts
+      const counts = dirEdgeCounts.get(dir) || { intra: 0, fanIn: 0, fanOut: 0 };
+      const totalEdges = counts.intra + counts.fanIn + counts.fanOut;
+      const cohesion = totalEdges > 0 ? counts.intra / totalEdges : null;
 
       upsertMetric.run(
         dirRow.id,
@@ -211,8 +237,8 @@ export function buildStructure(db, fileSymbols, _rootDir, lineCountMap, director
         symbolCount,
         null,
         null,
-        totalFanIn,
-        totalFanOut,
+        counts.fanIn,
+        counts.fanOut,
         cohesion,
         fileCount,
       );
@@ -226,6 +252,8 @@ export function buildStructure(db, fileSymbols, _rootDir, lineCountMap, director
 
 // ─── Node role classification ─────────────────────────────────────────
 
+export const FRAMEWORK_ENTRY_PREFIXES = ['route:', 'event:', 'command:'];
+
 function median(sorted) {
   if (sorted.length === 0) return 0;
   const mid = Math.floor(sorted.length / 2);
@@ -235,7 +263,7 @@ function median(sorted) {
 export function classifyNodeRoles(db) {
   const rows = db
     .prepare(
-      `SELECT n.id, n.kind, n.file,
+      `SELECT n.id, n.name, n.kind, n.file,
         COALESCE(fi.cnt, 0) AS fan_in,
         COALESCE(fo.cnt, 0) AS fan_out
       FROM nodes n
@@ -287,7 +315,10 @@ export function classifyNodeRoles(db) {
     const isExported = exportedIds.has(row.id);
 
     let role;
-    if (row.fan_in === 0 && !isExported) {
+    const isFrameworkEntry = FRAMEWORK_ENTRY_PREFIXES.some((p) => row.name.startsWith(p));
+    if (isFrameworkEntry) {
+      role = 'entry';
+    } else if (row.fan_in === 0 && !isExported) {
       role = 'dead';
     } else if (row.fan_in === 0 && isExported) {
       role = 'entry';
@@ -330,6 +361,8 @@ export function structureData(customDbPath, opts = {}) {
   const maxDepth = opts.depth || null;
   const sortBy = opts.sort || 'files';
   const noTests = opts.noTests || false;
+  const full = opts.full || false;
+  const fileLimit = opts.fileLimit || 25;
 
   // Get all directory nodes with their metrics
   let dirs = db
@@ -403,6 +436,33 @@ export function structureData(customDbPath, opts = {}) {
   });
 
   db.close();
+
+  // Apply global file limit unless full mode
+  if (!full) {
+    const totalFiles = result.reduce((sum, d) => sum + d.files.length, 0);
+    if (totalFiles > fileLimit) {
+      let shown = 0;
+      for (const d of result) {
+        const remaining = fileLimit - shown;
+        if (remaining <= 0) {
+          d.files = [];
+        } else if (d.files.length > remaining) {
+          d.files = d.files.slice(0, remaining);
+          shown = fileLimit;
+        } else {
+          shown += d.files.length;
+        }
+      }
+      const suppressed = totalFiles - fileLimit;
+      return {
+        directories: result,
+        count: result.length,
+        suppressed,
+        warning: `${suppressed} files omitted (showing ${fileLimit}/${totalFiles}). Use --full to show all files, or narrow with --directory.`,
+      };
+    }
+  }
+
   return { directories: result, count: result.length };
 }
 
@@ -539,6 +599,10 @@ export function formatStructure(data) {
       );
     }
   }
+  if (data.warning) {
+    lines.push('');
+    lines.push(`⚠ ${data.warning}`);
+  }
   return lines.join('\n');
 }
 

package/src/update-check.js

@@ -109,6 +109,7 @@ export async function checkForUpdates(currentVersion, options = {}) {
   if (process.env.CI) return null;
   if (process.env.NO_UPDATE_CHECK) return null;
   if (!process.stderr.isTTY) return null;
+  if (currentVersion.includes('-')) return null;
 
   const cachePath = options.cachePath || CACHE_PATH;
   const fetchFn = options._fetchLatest || fetchLatestVersion;

package/src/watcher.js

@@ -2,7 +2,7 @@ import fs from 'node:fs';
 import path from 'node:path';
 import { readFileSafe } from './builder.js';
 import { EXTENSIONS, IGNORE_DIRS, normalizePath } from './constants.js';
-import { initSchema, openDb } from './db.js';
+import { closeDb, initSchema, openDb } from './db.js';
 import { appendJournalEntries } from './journal.js';
 import { info, warn } from './logger.js';
 import { createParseTreeCache, getActiveEngine, parseFileIncremental } from './parser.js';
@@ -261,7 +261,7 @@ export async function watchProject(rootDir, opts = {}) {
       }
     }
     if (cache) cache.clear();
-    db.close();
+    closeDb(db);
     process.exit(0);
   });
 }