lumencode 1.3.1 → 1.3.2

package/lib/scenario.js

@@ -1,5 +1,29 @@
 // 基于工具调用 + 用户文本关键词的使用场景分类引擎
 
+export function escapeRegExp(string) {
+  return string.replace(/[-.*+?^${}()|[\]\\]/g, '\\$&');
+}
+
+export function matchesKeyword(text, keyword) {
+  if (!text || !keyword) return false;
+  const t = text.toLowerCase();
+  const k = keyword.trim().toLowerCase();
+  if (!k) return false;
+
+  // 中文关键词：使用包含匹配（中文无词分隔符，边界匹配会误杀正常匹配）
+  if (/\p{Script=Han}/u.test(k)) {
+    return t.includes(k);
+  }
+
+  // 英文关键词：词边界匹配
+  // 处理以非单词字符开头/结尾的关键词（如 /review）
+  const escaped = escapeRegExp(k);
+  const prefix = /^\w/.test(k) ? '\\b' : '(?:^|\\s)';
+  const suffix = /\w$/.test(k) ? '\\b' : '(?:$|\\s)';
+  const regex = new RegExp(`${prefix}${escaped}${suffix}`);
+  return regex.test(t);
+}
+
 const TOOL_SCENARIO_MAP = {
   // Claude Code
   Write: 'coding',
@@ -101,12 +125,9 @@ export function classifyRecord(record, scenarioKeywords) {
   const recordType = record.metadata?.type || record.type;
   const recordText = record.metadata?.text || record.text;
   if (recordType === 'user' && recordText && scenarioKeywords) {
-    const lowerText = recordText.toLowerCase();
     for (const [scenario, keywords] of Object.entries(scenarioKeywords)) {
       for (const kw of keywords) {
-        const lowerKw = kw.toLowerCase();
-        // 简化正则表达式，对于中文使用简单的包含匹配
-        if (lowerText.includes(lowerKw)) {
+        if (matchesKeyword(recordText, kw)) {
           scenarios[scenario] = (scenarios[scenario] || 0) + 1;
           break;
         }
@@ -127,6 +148,7 @@ export function mapToDisplayScenarios(internalScenarios) {
     '阅读/研究': 0,
     '规划/设计': 0,
     '代码审查': 0,
+    '重构': 0,
     '其他': 0,
   };
 
@@ -154,6 +176,9 @@ export function mapToDisplayScenarios(internalScenarios) {
       case 'review':
         display['代码审查'] += count;
         break;
+      case 'refactoring':
+        display['重构'] += count;
+        break;
       case 'execution':
         display['编码'] += count;
         break;