@sdsrs/code-graph 0.7.14 → 0.7.16

package/claude-plugin/scripts/user-prompt-context.js

@@ -69,7 +69,7 @@ const cwd = process.cwd();
 const dbPath = path.join(cwd, '.code-graph', 'index.db');
 if (!fs.existsSync(dbPath)) process.exit(0);
 
-// --- Constants ---
+// --- Pure logic (exported for testing) ---
 
 const STOP_WORDS = new Set([
   'this', 'that', 'with', 'from', 'what', 'when', 'which', 'there',
@@ -79,100 +79,109 @@ const STOP_WORDS = new Set([
   'being', 'through', 'default', 'function', 'method', 'class',
 ]);
 
-// --- Detect intent + entities ---
+const PLAIN_WORD_EXCLUDE = /^(possible|together|actually|something|different|important|following|available|necessary|currently|implement|operation|otherwise|beginning|knowledge|attention|according|certainly|sometimes|direction|recommend|structure|describe|question|complete|generate|anything|continue|consider|response|approach|happened|recently|probably|expected|previous|original|specific|directly|received|required|supposed|separate|designed|finished|provided|included|prepared|combined|properly|remember|whatever|although|document|handling|existing|everyone|standard|research|personal|relative|absolute|practice|language|thousand|national|evidence|refactor|understand|validate|analysis|debugging|configure|improving|resolving|creating|building|checking|updating|removing|changing|searching|cleaning|optimize|migration|overview|introduce|reviewing|thinking|managing|starting|yourself|features|problems|breaking|requires|argument|settings|includes|examples|comments|patterns|tutorial|concepts|supports|priority|organize|scenario|tracking|internal|external|abstract|concrete|strategy|evaluate|diagnose|platform|variable|optional|multiple)$/;
 
-// Skip non-code prompts (commit, push, simple confirmations, chat, instructions, etc.)
-const trimmed = message.trim();
-if (/^(yes|no|ok|commit|push|y|n|done|thanks|thank you|继续|确认|好的|好|是的|不|可以|行|对|提交|推送|没问题|谢谢|发布|更新|编译|安装|卸载|重启|重连|清理)\s*[.!?。！？]?\s*$/i.test(trimmed)) {
-  process.exit(0);
+function shouldSkip(msg) {
+  const trimmed = msg.trim();
+  if (/^(yes|no|ok|commit|push|y|n|done|thanks|thank you|继续|确认|好的|好|是的|不|可以|行|对|提交|推送|没问题|谢谢|发布|更新|编译|安装|卸载|重启|重连|清理)\s*[.!?。！？]?\s*$/i.test(trimmed)) return 'simple';
+  if (/^(修复|实施|执行|开始|按|实测|进入|用|重新)/.test(trimmed) && !/[a-zA-Z_]{3,}/.test(trimmed)) return 'action-only';
+  return false;
 }
-// Skip action-only prompts without code entities (修复这些问题, 按优先级实施, etc.)
-if (/^(修复|优化|实施|执行|开始|按|实测|帮我|进入|用|重新)/.test(trimmed) && !/[a-zA-Z_]{4,}/.test(trimmed)) {
-  process.exit(0);
+
+function extractFilePaths(msg) {
+  return (msg.match(/(?:src|lib|test|pkg|cmd|internal|app|components?)\/[\w/.-]+/g) || []).slice(0, 2);
 }
 
-// Extract file paths from message
-const filePaths = (message.match(/(?:src|lib|test|pkg|cmd|internal|app|components?)\/[\w/.-]+/g) || [])
-  .slice(0, 2);
-
-// Extract potential symbol names (camelCase, snake_case, PascalCase, qualified like Foo::bar, Foo.bar, Foo::bar::baz)
-const symbolCandidates = (message.match(/\b(?:[A-Z]\w*(?:(?:::|\.)\w+)+|[a-z]\w*(?:_\w+){1,}|[a-z]\w*(?:[A-Z]\w*)+|[A-Z][a-z]+(?:[A-Z][a-z]+)+)\b/g) || [])
-  .filter(s => s.length > 4)
-  .filter(s => !STOP_WORDS.has(s.toLowerCase()))
-  .slice(0, 3);
-
-// Fallback: extract backtick-quoted symbols (common in mixed Chinese+code: "修改 `parse_code` 函数")
-if (symbolCandidates.length === 0) {
-  const backtickSymbols = (message.match(/`([a-zA-Z_]\w{2,})`/g) || [])
-    .map(s => s.replace(/`/g, ''))
-    .filter(s => s.length >= 3 && !STOP_WORDS.has(s.toLowerCase()));
-  symbolCandidates.push(...backtickSymbols.slice(0, 3));
+function extractSymbols(msg) {
+  const candidates = (msg.match(/\b(?:[A-Z]\w*(?:(?:::|\.)\w+)+|[a-z]\w*(?:_\w+){1,}|[a-z]\w*(?:[A-Z]\w*)+|[A-Z][a-z]+(?:[A-Z][a-z]+)+)\b/g) || [])
+    .filter(s => s.length > 4)
+    .filter(s => !STOP_WORDS.has(s.toLowerCase()))
+    .slice(0, 3);
+
+  if (candidates.length === 0) {
+    const backtickSymbols = (msg.match(/`([a-zA-Z_]\w{2,})`/g) || [])
+      .map(s => s.replace(/`/g, ''))
+      .filter(s => s.length >= 3 && !STOP_WORDS.has(s.toLowerCase()));
+    candidates.push(...backtickSymbols.slice(0, 3));
+  }
+
+  let lowConfidence = false;
+  if (candidates.length === 0) {
+    const plain = (msg.match(/\b[a-z][a-z]{7,}\b/g) || [])
+      .filter(s => !STOP_WORDS.has(s))
+      .filter(s => !PLAIN_WORD_EXCLUDE.test(s));
+    candidates.push(...plain.slice(0, 2));
+    if (candidates.length > 0) lowConfidence = true;
+  }
+
+  return { symbols: candidates, lowConfidence };
 }
 
-// Fallback: plain lowercase words (8+ chars) likely to be function/type names.
-// Only when strict patterns found nothing — avoids false positives from English prose.
-// Minimum 8 chars filters most common English words while keeping technical terms
-// (authenticate, serialize, initialize, dispatch, resolver, etc.)
-if (symbolCandidates.length === 0) {
-  const plain = (message.match(/\b[a-z][a-z]{7,}\b/g) || [])
-    .filter(s => !STOP_WORDS.has(s))
-    .filter(s => !/^(possible|together|actually|something|different|important|following|available|necessary|currently|implement|operation|otherwise|beginning|knowledge|attention|according|certainly|sometimes|direction|recommend|structure|describe|question|complete|generate|anything|continue|consider|response|approach|happened|recently|probably|expected|previous|original|specific|directly|received|required|supposed|separate|designed|finished|provided|included|prepared|combined|properly|remember|whatever|although|document|handling|existing|everyone|standard|research|personal|relative|absolute|practice|language|thousand|national|evidence)$/.test(s));
-  symbolCandidates.push(...plain.slice(0, 2));
+function detectIntents(msg) {
+  return {
+    impact: /(?:impact|影响|修改前|改之前|blast radius|before (?:edit|chang|modif)|risk|风险|改动范围|波及|问题在|bug|干扰|冲突|卡)/i.test(msg),
+    modify: /(?:改(?!变)|修改|修复|重构|优化|简化|精简|适配|统一|修正|调整|去掉|整理|清理|解耦|更新|\brefactor\b|\bchange\b|\brename\b|\bfix\b|移动|\bmove\b|删(?!除文件)|\bremove\b|替换|\breplace\b|\bupdate\b|升级|\bmigrate\b|迁移|拆分|\bsplit\b|合并|\bmerge\b|提取|\bextract\b|改成|改为|换成|转为|异步|同步)/i.test(msg),
+    implement: /(?:\badd\b|\bimplement\b|\bcreate\b|\bbuild\b|\bwrite\b|新增|添加|实现|创建|编写|开发|增加|加上|加个|写|做个|搭建|补充|引入|支持|封装|接入|对接|配置)/i.test(msg),
+    understand: /(?:how does|怎么工作|怎么实现|怎么做|什么|理解|看看|看一下|了解|分析|explain|understand|架构|architecture|structure|overview|模块|概览|干什么|做什么|工作原理|逻辑|机制|流程|功能|结合度|效率|评估|调研|是什么|有什么|能用不|高效不|达标|起作用|科学|深入思考|源码|检查|审核|审查|验证|诊断)/i.test(msg),
+    callgraph: /(?:who calls|what calls|调用|call(?:graph|er|ee)|trace|链路|追踪|谁调|被谁调|调了谁|上下游|依赖关系|触发|路径|覆盖|介入)/i.test(msg),
+    search: /(?:where is|在哪|find|search|搜索|找|locate|哪里用|哪里定义|定义在|实现在|处理没|在源码|加不加)/i.test(msg),
+  };
 }
 
-// Detect intent keywords (EN + ZH, derived from user's actual prompt history)
-const intentImpact = /(?:impact|影响|修改前|改之前|blast radius|before (?:edit|chang|modif)|risk|风险|改动范围|波及|问题在|bug|干扰|冲突|卡)/i.test(message);
-const intentModify = /(?:改(?!变)|修改|重构|\brefactor\b|\bchange\b|\brename\b|移动|\bmove\b|删(?!除文件)|\bremove\b|替换|\breplace\b|\bupdate\b|升级|\bmigrate\b|迁移|拆分|\bsplit\b|合并|\bmerge\b|提取|\bextract\b|改成|改为|换成|转为|异步|同步)/i.test(message);
-const intentUnderstand = /(?:how does|怎么工作|怎么实现|怎么做|什么|理解|看看|看一下|了解|分析|explain|understand|架构|architecture|structure|overview|模块|概览|干什么|做什么|工作原理|逻辑|机制|流程|功能|结合度|效率|评估|调研|是什么|有什么|能用不|高效不|达标|起作用|科学|深入思考|源码)/i.test(message);
-const intentCallgraph = /(?:who calls|what calls|调用|call(?:graph|er|ee)|trace|链路|追踪|谁调|被谁调|调了谁|上下游|依赖关系|触发|路径|覆盖|介入)/i.test(message);
-const intentSearch = /(?:where is|在哪|find|search|搜索|找|locate|哪里用|哪里定义|定义在|实现在|处理没|在源码|加不加)/i.test(message);
-
-// Need entities AND intent, or strong entity signal (qualified names like Foo::bar)
-const hasQualifiedSymbol = symbolCandidates.some(s => s.includes('::'));
-const hasIntent = intentImpact || intentModify || intentUnderstand || intentCallgraph || intentSearch;
-if (!hasIntent && !hasQualifiedSymbol && filePaths.length === 0) {
-  process.exit(0);
+function determineQueryType(intents, symbols, filePaths, isCoolingDownFn) {
+  const hasStrict = symbols.symbols.length > 0 && !symbols.lowConfidence;
+  const hasQualified = symbols.symbols.some(s => s.includes('::'));
+  const hasAny = intents.impact || intents.modify || intents.implement || intents.understand || intents.callgraph || intents.search;
+
+  // Gate: need intent, qualified symbol, file path, or any symbol
+  if (!hasAny && !hasQualified && filePaths.length === 0 && symbols.symbols.length === 0) return null;
+
+  const cd = isCoolingDownFn || (() => false);
+
+  if ((intents.impact || intents.modify) && hasStrict && !cd('impact')) return { type: 'impact', symbol: symbols.symbols[0] };
+  if (intents.callgraph && hasStrict && !cd('callgraph')) return { type: 'callgraph', symbol: symbols.symbols[0] };
+  if (filePaths.length > 0 && !cd('overview')) return { type: 'overview', path: filePaths[0].replace(/\/[^/]+$/, '/') };
+  if ((intents.search || intents.implement || hasQualified) && symbols.symbols.length > 0 && !cd('search')) return { type: 'search', symbol: symbols.symbols[0] };
+  if ((intents.understand || !hasAny) && symbols.symbols.length > 0 && !cd('search')) return { type: 'search', symbol: symbols.symbols[0] };
+
+  return null;
 }
 
-// --- Semantic output prefixes ---
-const PREFIXES = {
-  impact:    '[code-graph:impact] Blast radius — review before editing:',
-  overview:  '[code-graph:structure] Module structure:',
-  callgraph: '[code-graph:callgraph] Call relationships:',
-  search:    '[code-graph:search] Relevant code:',
-};
+// --- Main execution (only when run directly) ---
+if (require.main === module) {
+  if (shouldSkip(message)) process.exit(0);
 
-// --- Run ONE targeted CLI query (per-type cooldown allows different types to fire) ---
-let queryType = null;
-let result = '';
-try {
-  // Priority: impact/modify > callgraph > understand/overview > search
-  // intentModify + symbol → inject impact so Claude knows blast radius before editing
-  if ((intentImpact || intentModify) && symbolCandidates.length > 0 && !isCoolingDown('impact')) {
-    queryType = 'impact';
-    result = run('code-graph-mcp', ['impact', symbolCandidates[0]]);
-  } else if (intentCallgraph && symbolCandidates.length > 0 && !isCoolingDown('callgraph')) {
-    queryType = 'callgraph';
-    result = run('code-graph-mcp', ['callgraph', symbolCandidates[0], '--depth', '2']);
-  } else if (filePaths.length > 0 && (intentUnderstand || !hasIntent) && !isCoolingDown('overview')) {
-    queryType = 'overview';
-    const dir = filePaths[0].replace(/\/[^/]+$/, '/');
-    result = run('code-graph-mcp', ['overview', dir]);
-  } else if ((intentSearch || hasQualifiedSymbol) && symbolCandidates.length > 0 && !isCoolingDown('search')) {
-    queryType = 'search';
-    result = run('code-graph-mcp', ['search', symbolCandidates[0], '--limit', '8']);
-  } else if (intentUnderstand && symbolCandidates.length > 0 && !isCoolingDown('search')) {
-    queryType = 'search';
-    result = run('code-graph-mcp', ['search', symbolCandidates[0], '--limit', '8']);
+  const filePaths = extractFilePaths(message);
+  const symbols = extractSymbols(message);
+  const intents = detectIntents(message);
+  const query = determineQueryType(intents, symbols, filePaths, isCoolingDown);
+
+  if (!query) process.exit(0);
+
+  const PREFIXES = {
+    impact:    '[code-graph:impact] Blast radius — review before editing:',
+    overview:  '[code-graph:structure] Module structure:',
+    callgraph: '[code-graph:callgraph] Call relationships:',
+    search:    '[code-graph:search] Relevant code:',
+  };
+
+  try {
+    let result = '';
+    if (query.type === 'impact') result = run('code-graph-mcp', ['impact', query.symbol]);
+    else if (query.type === 'callgraph') result = run('code-graph-mcp', ['callgraph', query.symbol, '--depth', '2']);
+    else if (query.type === 'overview') result = run('code-graph-mcp', ['overview', query.path]);
+    else if (query.type === 'search') result = run('code-graph-mcp', ['search', query.symbol, '--limit', '8']);
+
+    if (result && result.trim()) {
+      markCooldown(query.type);
+      process.stdout.write(`${PREFIXES[query.type]}\n${result.trim()}\n`);
+    }
+  } catch {
+    process.exit(0);
   }
-} catch {
-  process.exit(0);
 }
 
-if (result && result.trim() && queryType) {
-  markCooldown(queryType);
-  process.stdout.write(`${PREFIXES[queryType]}\n${result.trim()}\n`);
-}
+module.exports = { shouldSkip, extractFilePaths, extractSymbols, detectIntents, determineQueryType, STOP_WORDS, PLAIN_WORD_EXCLUDE };
 
 // --- Helpers ---