nexus-agents 2.169.0 → 2.171.0

package/dist/{chunk-XGNB7LQB.js → chunk-QBOTSJ7S.js}

@@ -1,19 +1,19 @@
 import {
   resolveToken
-} from "./chunk-3JH3FMEJ.js";
+} from "./chunk-Y7JDXVJA.js";
 import {
   REGISTRY_PATH,
   getProjectRoot,
   loadPapersRegistry,
   loadTechniquesRegistry,
   savePapersRegistry
-} from "./chunk-HAVBBSYN.js";
+} from "./chunk-ZSWXEDI6.js";
 import {
   resolveInsideRoot
 } from "./chunk-NUBSJGQZ.js";
 import {
   getToolMemory
-} from "./chunk-Q3UMNYV2.js";
+} from "./chunk-LFKCJQ7G.js";
 import {
   capitalizeWords,
   truncateSentence
@@ -38,7 +38,7 @@ import {
   getTimeProvider,
   ok,
   registerPersistentDistillerFactory
-} from "./chunk-QXSR5KPG.js";
+} from "./chunk-QBCWORAK.js";
 import {
   ensureLearningDir,
   getRulesFile
@@ -3337,11 +3337,195 @@ function buildIndex(files) {
   };
 }
 
+// src/context/repo-map-callsites.ts
+import { readFileSync as readFileSync3 } from "fs";
+import { resolve as resolve4 } from "path";
+
+// src/indexer/usage-ast.ts
+import { extname } from "path";
+import { Lang, parse as parse2 } from "@ast-grep/napi";
+var LANG_VALUES = ["typescript", "tsx", "javascript"];
+var MAX_SNIPPET_CHARS = 200;
+var IDENTIFIER_RE = /^[A-Za-z_$][A-Za-z0-9_$]*$/;
+var CALL_USAGE_KINDS = /* @__PURE__ */ new Set(["call", "method-call", "new"]);
+var DECLARATION_PARENT_KINDS = /* @__PURE__ */ new Set([
+  "function_declaration",
+  "generator_function_declaration",
+  "class_declaration",
+  "interface_declaration",
+  "type_alias_declaration",
+  "enum_declaration",
+  "function_signature",
+  "method_definition",
+  "abstract_method_signature"
+]);
+var IMPORT_PARENT_KINDS = /* @__PURE__ */ new Set([
+  "import_specifier",
+  "namespace_import",
+  "import_clause"
+]);
+function langToNapi(lang) {
+  if (lang === "tsx") return Lang.Tsx;
+  if (lang === "javascript") return Lang.JavaScript;
+  return Lang.TypeScript;
+}
+function inferLang(file) {
+  const ext = extname(file).toLowerCase();
+  if (ext === ".tsx" || ext === ".jsx") return "tsx";
+  if (ext === ".js" || ext === ".mjs" || ext === ".cjs") return "javascript";
+  return "typescript";
+}
+function escapeRegex(value) {
+  return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
+function isDeclaredName(declarator, node) {
+  const name = declarator.field("name");
+  return name !== null && name.id() === node.id();
+}
+function classifyIdentifier(node) {
+  const parent = node.parent();
+  if (parent === null) return "reference";
+  const pk = String(parent.kind());
+  if (DECLARATION_PARENT_KINDS.has(pk)) return null;
+  if (pk === "variable_declarator") return isDeclaredName(parent, node) ? null : "reference";
+  if (IMPORT_PARENT_KINDS.has(pk)) return "import";
+  if (pk === "call_expression") return "call";
+  if (pk === "new_expression") return "new";
+  return "reference";
+}
+function classifyProperty(node) {
+  const member = node.parent();
+  const grandparent = member === null ? null : member.parent();
+  if (grandparent !== null && String(grandparent.kind()) === "call_expression")
+    return "method-call";
+  return "reference";
+}
+function toMatch(node, kind, lines) {
+  const { start } = node.range();
+  const rawLine = lines[start.line] ?? node.text();
+  return {
+    line: start.line + 1,
+    column: start.column + 1,
+    kind,
+    snippet: rawLine.trim().slice(0, MAX_SNIPPET_CHARS)
+  };
+}
+function findUsagesInSource(symbol, src, lang) {
+  if (!IDENTIFIER_RE.test(symbol)) return [];
+  const root = parse2(langToNapi(lang), src).root();
+  const lines = src.split("\n");
+  const out = [];
+  for (const node of root.findAll(symbol)) {
+    const kind = classifyIdentifier(node);
+    if (kind !== null) out.push(toMatch(node, kind, lines));
+  }
+  const propRule = {
+    rule: { kind: "property_identifier", regex: `^${escapeRegex(symbol)}$` }
+  };
+  for (const node of root.findAll(propRule)) {
+    out.push(toMatch(node, classifyProperty(node), lines));
+  }
+  return out.sort((a, b) => a.line - b.line || a.column - b.column);
+}
+function bump(counts, symbols, name) {
+  if (symbols.has(name)) counts.set(name, (counts.get(name) ?? 0) + 1);
+}
+function tallyIdentifierCalls(root, symbols, counts) {
+  for (const node of root.findAll({ rule: { kind: "identifier" } })) {
+    const kind = classifyIdentifier(node);
+    if (kind !== null && CALL_USAGE_KINDS.has(kind)) bump(counts, symbols, node.text());
+  }
+}
+function tallyPropertyCalls(root, symbols, counts) {
+  for (const node of root.findAll({ rule: { kind: "property_identifier" } })) {
+    if (CALL_USAGE_KINDS.has(classifyProperty(node))) bump(counts, symbols, node.text());
+  }
+}
+function countCallSitesInSource(symbols, src, lang) {
+  const counts = /* @__PURE__ */ new Map();
+  if (symbols.size === 0) return counts;
+  const root = parse2(langToNapi(lang), src).root();
+  tallyIdentifierCalls(root, symbols, counts);
+  tallyPropertyCalls(root, symbols, counts);
+  return counts;
+}
+
+// src/context/repo-map-callsites.ts
+var CALLSITE_TOP_N_MODULES = 25;
+var MAX_PROBE_SYMBOLS = 60;
+var MAX_CALLSITE_FILES = 4e3;
+function defaultSourceRoot() {
+  return resolve4(process.cwd(), "src");
+}
+function addModuleExports(mod, name, owner) {
+  for (const file of mod.files) {
+    for (const exp of file.exports) {
+      if (exp.isReExport) continue;
+      if (!owner.has(exp.name)) owner.set(exp.name, name);
+      if (owner.size >= MAX_PROBE_SYMBOLS) return;
+    }
+  }
+}
+function buildProbeSet(index, topModuleNames) {
+  const owner = /* @__PURE__ */ new Map();
+  for (const name of topModuleNames) {
+    const mod = index.modules[name];
+    if (mod === void 0) continue;
+    addModuleExports(mod, name, owner);
+    if (owner.size >= MAX_PROBE_SYMBOLS) break;
+  }
+  return { symbols: new Set(owner.keys()), owner };
+}
+function collectFiles(index, sourceRoot, maxFiles) {
+  const refs = [];
+  const seen = /* @__PURE__ */ new Set();
+  for (const mod of Object.values(index.modules)) {
+    for (const file of mod.files) {
+      if (seen.has(file.path)) continue;
+      seen.add(file.path);
+      refs.push({ abs: resolve4(sourceRoot, file.path), rel: file.path });
+      if (refs.length >= maxFiles) return refs;
+    }
+  }
+  return refs;
+}
+function tallyFile(probe, ref, counts, log) {
+  let src;
+  try {
+    src = readFileSync3(ref.abs, "utf8");
+  } catch {
+    log.debug("repo-map call-site: unreadable file", { file: ref.rel });
+    return;
+  }
+  for (const [sym, n] of countCallSitesInSource(probe.symbols, src, inferLang(ref.rel))) {
+    const owner = probe.owner.get(sym);
+    if (owner !== void 0) counts.set(owner, (counts.get(owner) ?? 0) + n);
+  }
+}
+function computeCallSiteCounts(index, topModuleNames, options = {}) {
+  const log = options.logger ?? createLogger({ component: "RepoMapCallSites" });
+  const counts = /* @__PURE__ */ new Map();
+  try {
+    const probe = buildProbeSet(index, topModuleNames);
+    if (probe.symbols.size === 0) return counts;
+    const sourceRoot = options.sourceRoot ?? defaultSourceRoot();
+    const maxFiles = options.maxFiles ?? MAX_CALLSITE_FILES;
+    for (const ref of collectFiles(index, sourceRoot, maxFiles)) {
+      tallyFile(probe, ref, counts, log);
+    }
+  } catch (error) {
+    log.debug("repo-map call-site counting failed", {
+      error: error instanceof Error ? error.message : String(error)
+    });
+  }
+  return counts;
+}
+
 // src/context/repo-map.ts
 var REPO_MAP_FLAG = "NEXUS_REPO_MAP";
 var DEFAULT_REPO_MAP_TOKEN_BUDGET = 400;
-var REPO_MAP_CAVEAT = "Import-graph only: edges are module import dependencies + declaration names, NOT call-site/usage data. Centrality reflects import structure, not runtime call frequency \u2014 do not treat this as a call graph (call-site edges tracked in #4249-A).";
-var REPO_MAP_HEADER = "## Repo Map (module import graph \u2014 PageRank-ranked)";
+var REPO_MAP_CAVEAT = "Ranking blends import-graph PageRank with structural call-site frequency (ast-grep, sampled over the top import-ranked modules). Call-site matching is syntactic, not type-aware: dynamic dispatch, computed/string-keyed calls, and same-named members on unrelated objects are not resolved \u2014 treat centrality as an approximate structural+usage signal, not an exact call graph.";
+var REPO_MAP_HEADER = "## Repo Map (module import graph + call-site usage \u2014 ranked)";
 var PAGERANK_DAMPING = 0.85;
 var PAGERANK_MAX_ITERATIONS = 100;
 var PAGERANK_EPSILON = 1e-6;
@@ -3396,24 +3580,43 @@ function l1Delta(a, b) {
   for (const [k, v] of a) d += Math.abs(v - (b.get(k) ?? 0));
   return d;
 }
-function rankRepoMapEntries(index) {
+var CALL_SITE_WEIGHT = 2;
+function blendRankScore(centrality, callSites, maxCallSites) {
+  if (maxCallSites <= 0) return centrality;
+  return centrality * (1 + CALL_SITE_WEIGHT * (callSites / maxCallSites));
+}
+function maxCount(counts) {
+  let max = 0;
+  for (const v of counts.values()) if (v > max) max = v;
+  return max;
+}
+function rankRepoMapEntries(index, callSiteCounts) {
   const ranks = computeModulePageRank(index.modules);
-  const entries = Object.entries(index.modules).map(([name, mod]) => ({
-    module: name,
-    centrality: ranks.get(name) ?? 0,
-    purpose: mod.purpose,
-    dependsOn: mod.dependsOn,
-    fileCount: mod.stats.fileCount
-  }));
-  return entries.sort(
-    (a, b) => b.centrality !== a.centrality ? b.centrality - a.centrality : a.module.localeCompare(b.module)
+  const counts = callSiteCounts ?? /* @__PURE__ */ new Map();
+  const max = maxCount(counts);
+  const scored = Object.entries(index.modules).map(([name, mod]) => {
+    const centrality = ranks.get(name) ?? 0;
+    const callSites = counts.get(name) ?? 0;
+    const entry = {
+      module: name,
+      centrality,
+      callSites,
+      purpose: mod.purpose,
+      dependsOn: mod.dependsOn,
+      fileCount: mod.stats.fileCount
+    };
+    return { entry, score: blendRankScore(centrality, callSites, max) };
+  });
+  scored.sort(
+    (a, b) => b.score !== a.score ? b.score - a.score : a.entry.module.localeCompare(b.entry.module)
   );
+  return scored.map((s) => s.entry);
 }
 var CHARS_PER_TOKEN3 = 4;
 var MAX_DEPS_PER_LINE = 6;
 var MAX_LINE_CHARS = 200;
 function buildRepoMap(index, options = {}) {
-  const entries = rankRepoMapEntries(index);
+  const entries = rankRepoMapEntries(index, options.callSiteCounts);
   if (entries.length === 0) return "";
   const budget = options.budgetTokens ?? DEFAULT_REPO_MAP_TOKEN_BUDGET;
   return renderRepoMap(entries, budget);
@@ -3434,7 +3637,8 @@ function formatEntryLine(e) {
   const shownDeps = e.dependsOn.slice(0, MAX_DEPS_PER_LINE);
   const more = e.dependsOn.length > MAX_DEPS_PER_LINE ? ", \u2026" : "";
   const deps = shownDeps.length > 0 ? ` \u2192 ${shownDeps.join(", ")}${more}` : "";
-  const line = `- ${e.module} (centrality ${e.centrality.toFixed(3)}, ${String(e.fileCount)} files): ${oneLine2(e.purpose)}${deps}`;
+  const calls = e.callSites > 0 ? `, ${String(e.callSites)} call-sites` : "";
+  const line = `- ${e.module} (centrality ${e.centrality.toFixed(3)}${calls}, ${String(e.fileCount)} files): ${oneLine2(e.purpose)}${deps}`;
   return line.slice(0, MAX_LINE_CHARS);
 }
 function fitLinesWithinBudget(lines, budgetTokens) {
@@ -3472,8 +3676,10 @@ function getRepoMapForTask(options) {
   const log = options.logger ?? createLogger({ component: "RepoMap" });
   try {
     const index = (options.indexProvider ?? buildLiveIndex)();
+    const callSiteCounts = options.callSiteCounts ?? liveCallSiteCounts(index, log);
     const map = buildRepoMap(index, {
-      budgetTokens: options.budgetTokens ?? DEFAULT_REPO_MAP_TOKEN_BUDGET
+      budgetTokens: options.budgetTokens ?? DEFAULT_REPO_MAP_TOKEN_BUDGET,
+      callSiteCounts
     });
     return map === "" ? void 0 : map;
   } catch (error) {
@@ -3483,6 +3689,10 @@ function getRepoMapForTask(options) {
     return void 0;
   }
 }
+function liveCallSiteCounts(index, log) {
+  const topModules = rankRepoMapEntries(index).slice(0, CALLSITE_TOP_N_MODULES).map((e) => e.module);
+  return computeCallSiteCounts(index, topModules, { logger: log });
+}
 function buildLiveIndex() {
   const extraction = extractProject({ extractDescriptions: false });
   return buildIndex(extraction.files);
@@ -3811,6 +4021,11 @@ export {
   CodebaseIndexSchema,
   extractProject,
   buildIndex,
+  LANG_VALUES,
+  IDENTIFIER_RE,
+  langToNapi,
+  inferLang,
+  findUsagesInSource,
   getContextForTask,
   selectRelevantResearch,
   getResearchInsightsForTask,
@@ -3819,4 +4034,4 @@ export {
   DEFAULT_CONTEXT_BUDGET_TOKENS,
   getContextPromptPrefix
 };
-//# sourceMappingURL=chunk-XGNB7LQB.js.map
+//# sourceMappingURL=chunk-QBOTSJ7S.js.map