code-graph-llm 1.5.0 → 1.6.0

package/index.js

@@ -154,6 +154,7 @@ export function extractEdges(content) {
 async function generate(cwd = process.cwd()) {
   const files = [];
   const allEdges = [];
+  const incomingEdges = {}; // Map of projectPath -> count of incoming edges
 
   function walk(dir, ig) {
     let localIg = ig;
@@ -186,16 +187,26 @@ async function generate(cwd = process.cwd()) {
         if (SUPPORTED_EXTENSIONS.includes(ext)) {
           const content = fs.readFileSync(fullPath, 'utf8');
           
-          // Extract file-level description
+          // Extract file-level description and tags
           const lines = content.split('\n');
           let fileDesc = '';
-          for (let i = 0; i < Math.min(10, lines.length); i++) {
+          const tags = [];
+          for (let i = 0; i < lines.length; i++) {
             const line = lines[i].trim();
-            if (line.startsWith('#!') || line === '') continue;
-            if (line.startsWith('//') || line.startsWith('#') || line.startsWith('/*')) {
-              fileDesc += line.replace(/[\/*#]/g, '').trim() + ' ';
-            } else {
-              break; 
+            // Surface actionable tags
+            const tagMatch = line.match(/\b(TODO|FIXME|BUG|DEPRECATED):?\s*(.*)/i);
+            if (tagMatch) {
+              tags.push(`${tagMatch[1]}: ${tagMatch[2].substring(0, 50).trim()}`);
+            }
+            
+            if (i < 10) {
+              if (line.startsWith('#!') || line === '') continue;
+              if (line.startsWith('//') || line.startsWith('#') || line.startsWith('/*')) {
+                fileDesc += line.replace(/[\/*#]/g, '').trim() + ' ';
+              } else if (fileDesc) {
+                // Stop if we hit non-comment code
+                break;
+              }
             }
           }
 
@@ -204,11 +215,38 @@ async function generate(cwd = process.cwd()) {
           
           if (!fileDesc.trim() && symbols.length > 0) fileDesc = `Contains ${symbols.length} symbols.`;
           
-          files.push({ path: normalizedPath, desc: fileDesc.trim(), symbols });
+          // Identify core entry points
+          const isCore = /^(index|main|app|server|cli)\.(js|ts|py|go|rs|java|cpp)$/i.test(entry.name);
+          
+          const fileObj = { 
+            path: normalizedPath, 
+            desc: fileDesc.trim(), 
+            symbols, 
+            tags: Array.from(new Set(tags)),
+            isCore,
+            outCount: dependencies.length
+          };
+          files.push(fileObj);
           
-          // Collect Edges
+          // Collect Edges and track incoming
           dependencies.forEach(dep => {
-            allEdges.push(`[${normalizedPath}] -> [imports] -> [${dep}]`);
+            let target = dep;
+            if (dep.startsWith('.')) {
+              // Resolve relative path to project-relative
+              const resolved = path.normalize(path.join(path.dirname(normalizedPath), dep));
+              target = resolved.replace(/\\/g, '/');
+              // Basic extension guesser
+              if (!path.extname(target)) {
+                for (const ex of SUPPORTED_EXTENSIONS) {
+                  if (fs.existsSync(path.join(cwd, target + ex))) {
+                    target += ex;
+                    break;
+                  }
+                }
+              }
+            }
+            allEdges.push(`[${normalizedPath}] -> [imports] -> [${target}]`);
+            incomingEdges[target] = (incomingEdges[target] || 0) + 1;
           });
           inheritance.forEach(inh => {
             allEdges.push(`[${inh.child}] -> [inherits] -> [${inh.parent}]`);
@@ -220,17 +258,30 @@ async function generate(cwd = process.cwd()) {
 
   walk(cwd, getIgnores(cwd));
 
+  // Sort: Core files first, then by incoming edges (importance)
+  files.sort((a, b) => {
+    if (a.isCore && !b.isCore) return -1;
+    if (!a.isCore && b.isCore) return 1;
+    const inA = incomingEdges[a.path] || 0;
+    const inB = incomingEdges[b.path] || 0;
+    return inB - inA;
+  });
+
   const nodesOutput = files.map(f => {
+    const inCount = incomingEdges[f.path] || 0;
+    const coreMark = f.isCore ? '[CORE] ' : '';
+    const stats = `(↑${f.outCount} ↓${inCount})`;
+    const tagStr = f.tags.length > 0 ? ` [${f.tags.join(', ')}]` : '';
     const descStr = f.desc ? ` | desc: ${f.desc.substring(0, 100)}` : '';
     const symStr = f.symbols.length > 0 ? `\n  - syms: [${f.symbols.join(', ')}]` : '';
-    return `- ${f.path}${descStr}${symStr}`;
+    return `- ${coreMark}${f.path} ${stats}${tagStr}${descStr}${symStr}`;
   }).join('\n');
 
   const edgesOutput = allEdges.length > 0 
     ? `\n\n## GRAPH EDGES\n${Array.from(new Set(allEdges)).sort().join('\n')}` 
     : '';
 
-  const header = `# CODE_GRAPH_MAP\n> LLM_ONLY: DO NOT EDIT. COMPACT PROJECT MAP.\n\n`;
+  const header = `# CODE_GRAPH_MAP\n> LLM_ONLY: DO NOT EDIT. COMPACT PROJECT MAP.\n> Legend: [CORE] Entry Point, (↑N) Outgoing Deps, (↓M) Incoming Dependents\n> Notation: syms: [Name [Signature/Context]], desc: File Summary, [TAG: Context] Actionable items\n\n`;
   fs.writeFileSync(path.join(cwd, DEFAULT_MAP_FILE), header + nodesOutput + edgesOutput);
   console.log(`[Code-Graph] Updated ${DEFAULT_MAP_FILE}`);
 }

package/llm-code-graph.md

@@ -1,9 +1,11 @@
 # CODE_GRAPH_MAP
 > LLM_ONLY: DO NOT EDIT. COMPACT PROJECT MAP.
+> Legend: [CORE] Entry Point, (↑N) Outgoing Deps, (↓M) Incoming Dependents
+> Notation: syms: [Name [Signature/Context]], desc: File Summary, [TAG: Context] Actionable items
 
-- index.js | desc: Contains 5 symbols.
+- [CORE] index.js (↑5 ↓1) [TODO: |FIXME|BUG|DEPRECATED):?\s*(.*)/i);] | desc: Contains 5 symbols.
   - syms: [SUPPORTED_EXTENSIONS [= [], extractSymbolsAndInheritance [(content)], generate [(cwd = process.cwd()], getIgnores [(cwd, additionalLines = [])], walk [(dir, ig)]]
-- test/index.test.js
+- test/index.test.js (↑10 ↓0)
 
 ## GRAPH EDGES
 [index.js] -> [imports] -> [chokidar]
@@ -11,13 +13,13 @@
 [index.js] -> [imports] -> [ignore]
 [index.js] -> [imports] -> [path]
 [index.js] -> [imports] -> [url]
-[test/index.test.js] -> [imports] -> [../index.js]
-[test/index.test.js] -> [imports] -> [./local-file]
 [test/index.test.js] -> [imports] -> [assert]
 [test/index.test.js] -> [imports] -> [fs]
 [test/index.test.js] -> [imports] -> [header.h]
+[test/index.test.js] -> [imports] -> [index.js]
 [test/index.test.js] -> [imports] -> [node]
 [test/index.test.js] -> [imports] -> [other-module]
 [test/index.test.js] -> [imports] -> [path]
+[test/index.test.js] -> [imports] -> [test/local-file]
 [test/index.test.js] -> [imports] -> [test]
 [test/index.test.js] -> [imports] -> [url]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-llm",
-  "version": "1.5.0",
+  "version": "1.6.0",
   "description": "Compact, language-agnostic codebase mapper for LLM token efficiency.",
   "main": "index.js",
   "bin": {