feihong-code 0.2.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.github/ISSUE_TEMPLATE/bug_report.md +37 -0
- package/.github/ISSUE_TEMPLATE/config.yml +14 -0
- package/.github/ISSUE_TEMPLATE/feature_request.md +28 -0
- package/.github/PULL_REQUEST_TEMPLATE.md +46 -0
- package/.github/SECURITY.md +67 -0
- package/.github/workflows/ci.yml +140 -0
- package/AGENT-GUIDE.md +237 -0
- package/CHANGELOG.md +73 -0
- package/CODE_OF_CONDUCT.md +56 -0
- package/CONTRIBUTING.md +68 -0
- package/LICENSE +22 -0
- package/README.md +500 -0
- package/dist/agent/code-review.js +115 -0
- package/dist/agent/code-review.js.map +1 -0
- package/dist/agent/code-writer.js +227 -0
- package/dist/agent/code-writer.js.map +1 -0
- package/dist/agent/context-compactor.js +108 -0
- package/dist/agent/context-compactor.js.map +1 -0
- package/dist/agent/experience.js +190 -0
- package/dist/agent/experience.js.map +1 -0
- package/dist/agent/orchestrator.js +212 -0
- package/dist/agent/orchestrator.js.map +1 -0
- package/dist/agent/parallel-orchestrator.js +94 -0
- package/dist/agent/parallel-orchestrator.js.map +1 -0
- package/dist/agent/planner.js +53 -0
- package/dist/agent/planner.js.map +1 -0
- package/dist/agent/prompts.js +26 -0
- package/dist/agent/prompts.js.map +1 -0
- package/dist/agent/quality-gate.js +143 -0
- package/dist/agent/quality-gate.js.map +1 -0
- package/dist/agent/repo-reader.js +384 -0
- package/dist/agent/repo-reader.js.map +1 -0
- package/dist/agent/repo-underwriter.js +132 -0
- package/dist/agent/repo-underwriter.js.map +1 -0
- package/dist/agent/self-heal.js +134 -0
- package/dist/agent/self-heal.js.map +1 -0
- package/dist/agent/self-improver.js +148 -0
- package/dist/agent/self-improver.js.map +1 -0
- package/dist/agent/subagent.js +75 -0
- package/dist/agent/subagent.js.map +1 -0
- package/dist/agent/swe-agent.js +162 -0
- package/dist/agent/swe-agent.js.map +1 -0
- package/dist/agent/swe-planner.js +155 -0
- package/dist/agent/swe-planner.js.map +1 -0
- package/dist/agent/swe-verifier.js +113 -0
- package/dist/agent/swe-verifier.js.map +1 -0
- package/dist/cli/commands.js +167 -0
- package/dist/cli/commands.js.map +1 -0
- package/dist/cli/index.js +179 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/repl.js +67 -0
- package/dist/cli/repl.js.map +1 -0
- package/dist/cli/run.js +762 -0
- package/dist/cli/run.js.map +1 -0
- package/dist/cli/version.js +14 -0
- package/dist/cli/version.js.map +1 -0
- package/dist/enterprise/audit.js +294 -0
- package/dist/enterprise/audit.js.map +1 -0
- package/dist/enterprise/guard.js +58 -0
- package/dist/enterprise/guard.js.map +1 -0
- package/dist/enterprise/index.js +99 -0
- package/dist/enterprise/index.js.map +1 -0
- package/dist/enterprise/policy.js +242 -0
- package/dist/enterprise/policy.js.map +1 -0
- package/dist/enterprise/quota.js +58 -0
- package/dist/enterprise/quota.js.map +1 -0
- package/dist/enterprise/tenant.js +143 -0
- package/dist/enterprise/tenant.js.map +1 -0
- package/dist/models/cost.js +13 -0
- package/dist/models/cost.js.map +1 -0
- package/dist/models/model-router.js +170 -0
- package/dist/models/model-router.js.map +1 -0
- package/dist/models/model.dto.js +61 -0
- package/dist/models/model.dto.js.map +1 -0
- package/dist/models/model.interface.js +3 -0
- package/dist/models/model.interface.js.map +1 -0
- package/dist/models/providers/mock.provider.js +35 -0
- package/dist/models/providers/mock.provider.js.map +1 -0
- package/dist/models/providers/ollama.provider.js +88 -0
- package/dist/models/providers/ollama.provider.js.map +1 -0
- package/dist/models/providers/openai-compatible.provider.js +106 -0
- package/dist/models/providers/openai-compatible.provider.js.map +1 -0
- package/dist/runtime/event-log.js +42 -0
- package/dist/runtime/event-log.js.map +1 -0
- package/dist/runtime/git.js +106 -0
- package/dist/runtime/git.js.map +1 -0
- package/dist/runtime/session-persist.js +60 -0
- package/dist/runtime/session-persist.js.map +1 -0
- package/dist/runtime/session-store.js +34 -0
- package/dist/runtime/session-store.js.map +1 -0
- package/dist/runtime/worktree.js +117 -0
- package/dist/runtime/worktree.js.map +1 -0
- package/dist/shared/config.js +184 -0
- package/dist/shared/config.js.map +1 -0
- package/dist/shared/errors.js +66 -0
- package/dist/shared/errors.js.map +1 -0
- package/dist/shared/logger.js +51 -0
- package/dist/shared/logger.js.map +1 -0
- package/dist/shared/types.js +9 -0
- package/dist/shared/types.js.map +1 -0
- package/dist/skills/goal.js +60 -0
- package/dist/skills/goal.js.map +1 -0
- package/dist/skills/grill.js +129 -0
- package/dist/skills/grill.js.map +1 -0
- package/dist/skills/plan.js +40 -0
- package/dist/skills/plan.js.map +1 -0
- package/dist/tools/analysis/code-analyzer.js +136 -0
- package/dist/tools/analysis/code-analyzer.js.map +1 -0
- package/dist/tools/file/edit.tool.js +43 -0
- package/dist/tools/file/edit.tool.js.map +1 -0
- package/dist/tools/file/list.tool.js +36 -0
- package/dist/tools/file/list.tool.js.map +1 -0
- package/dist/tools/file/read.tool.js +34 -0
- package/dist/tools/file/read.tool.js.map +1 -0
- package/dist/tools/file/write.tool.js +39 -0
- package/dist/tools/file/write.tool.js.map +1 -0
- package/dist/tools/generator/code-generator.js +119 -0
- package/dist/tools/generator/code-generator.js.map +1 -0
- package/dist/tools/generator/test-generator.js +65 -0
- package/dist/tools/generator/test-generator.js.map +1 -0
- package/dist/tools/index.js +53 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/safe-path.js +36 -0
- package/dist/tools/safe-path.js.map +1 -0
- package/dist/tools/search/grep.tool.js +81 -0
- package/dist/tools/search/grep.tool.js.map +1 -0
- package/dist/tools/shell/exec.js +50 -0
- package/dist/tools/shell/exec.js.map +1 -0
- package/dist/tools/shell/run-shell.tool.js +49 -0
- package/dist/tools/shell/run-shell.tool.js.map +1 -0
- package/dist/tools/tool.interface.js +8 -0
- package/dist/tools/tool.interface.js.map +1 -0
- package/dist/tools/tool.registry.js +75 -0
- package/dist/tools/tool.registry.js.map +1 -0
- package/dist/tools/verify/build-check.tool.js +37 -0
- package/dist/tools/verify/build-check.tool.js.map +1 -0
- package/dist/tools/verify/test-run.tool.js +37 -0
- package/dist/tools/verify/test-run.tool.js.map +1 -0
- package/dist/web/auth.js +25 -0
- package/dist/web/auth.js.map +1 -0
- package/dist/web/public/index.html +39 -0
- package/dist/web/server.js +60 -0
- package/dist/web/server.js.map +1 -0
- package/docs//344/272/247/345/223/201/345/274/200/345/217/221/346/226/207/346/241/243.md +614 -0
- package/docs//344/274/201/344/270/232/351/203/250/347/275/262/344/270/216/345/220/210/350/247/204.md +267 -0
- package/docs//344/275/277/347/224/250/350/257/264/346/230/216/344/271/246.md +358 -0
- package/docs//345/270/270/350/247/201/351/227/256/351/242/230/344/270/216/346/225/205/351/232/234/346/216/222/346/237/245.md +130 -0
- package/docs//346/212/200/346/234/257/350/257/264/346/230/216/344/271/246.md +303 -0
- package/docs//346/236/266/346/236/204/344/270/216API.md +238 -0
- package/docs//347/224/250/346/210/267/346/211/213/345/206/214.md +196 -0
- package/docs//351/203/250/347/275/262/346/214/207/345/215/227.md +165 -0
- package/docs//351/205/215/347/275/256/345/217/202/350/200/203.md +127 -0
- package/package.json +83 -0
- package/tool-schema.json +127 -0
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.reviewCode = reviewCode;
|
|
4
|
+
exports.reviewDirectory = reviewDirectory;
|
|
5
|
+
exports.loadReviewRules = loadReviewRules;
|
|
6
|
+
exports.applyFixes = applyFixes;
|
|
7
|
+
/**
|
|
8
|
+
* 飞虹 Code (Muse Code 参照复刻)
|
|
9
|
+
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
10
|
+
*
|
|
11
|
+
* AI 代码审查器(M7):
|
|
12
|
+
* - 读取代码 → 分析风险 → 提出修复建议 → 应用修复
|
|
13
|
+
* - 安全检查:硬编码密钥、SQL 注入、路径穿越
|
|
14
|
+
* - 性能审查:N+1 查询、未使用索引、内存泄漏
|
|
15
|
+
* - 可配置审查规则(.fhcode-review.yml)
|
|
16
|
+
*/
|
|
17
|
+
const fs_1 = require("fs");
|
|
18
|
+
/** 默认审查规则 */
|
|
19
|
+
const DEFAULT_RULES = [
|
|
20
|
+
{
|
|
21
|
+
id: 'hardcoded-secret',
|
|
22
|
+
name: '硬编码密钥',
|
|
23
|
+
pattern: /['"`](password|secret|key|token|api[_-]?key)[^'"`]*['"`]/gi,
|
|
24
|
+
severity: 'high',
|
|
25
|
+
message: '检测到可能硬编码的密钥/密码',
|
|
26
|
+
fix: '使用环境变量 FH_* 管理',
|
|
27
|
+
},
|
|
28
|
+
{
|
|
29
|
+
id: 'sql-injection',
|
|
30
|
+
name: 'SQL 注入风险',
|
|
31
|
+
pattern: /['"]\s*(SELECT|INSERT|UPDATE|DELETE)\s+.*['"]\s*\+\s*\w+/i,
|
|
32
|
+
severity: 'high',
|
|
33
|
+
message: '可能的 SQL 注入',
|
|
34
|
+
fix: '使用参数化查询',
|
|
35
|
+
},
|
|
36
|
+
{
|
|
37
|
+
id: 'path-traversal',
|
|
38
|
+
name: '路径穿越',
|
|
39
|
+
pattern: /join\s*\(\s*[^)]+\s*,\s*[^)]*\)/,
|
|
40
|
+
severity: 'medium',
|
|
41
|
+
message: '检查路径拼接安全性',
|
|
42
|
+
fix: '使用 safePath 工具校验',
|
|
43
|
+
},
|
|
44
|
+
{
|
|
45
|
+
id: 'console-log',
|
|
46
|
+
name: 'Console 输出残留',
|
|
47
|
+
pattern: /console\.(log|warn|error)\(/,
|
|
48
|
+
severity: 'low',
|
|
49
|
+
message: '生产环境不应有 console 输出',
|
|
50
|
+
fix: '使用 logger 替代',
|
|
51
|
+
},
|
|
52
|
+
];
|
|
53
|
+
/** 执行代码审查 */
|
|
54
|
+
function reviewCode(filePath, content, rules = DEFAULT_RULES) {
|
|
55
|
+
const issues = [];
|
|
56
|
+
const lines = content.split('\n');
|
|
57
|
+
for (let i = 0; i < lines.length; i++) {
|
|
58
|
+
const line = lines[i];
|
|
59
|
+
for (const rule of rules) {
|
|
60
|
+
if (rule.pattern.test(line)) {
|
|
61
|
+
issues.push({
|
|
62
|
+
ruleId: rule.id,
|
|
63
|
+
line: i + 1,
|
|
64
|
+
severity: rule.severity,
|
|
65
|
+
message: rule.message,
|
|
66
|
+
suggestion: rule.fix || '请手动修复',
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
return {
|
|
72
|
+
file: filePath,
|
|
73
|
+
issues,
|
|
74
|
+
passed: issues.filter(i => i.severity === 'high').length === 0,
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
/** 批量审查目录 */
|
|
78
|
+
function reviewDirectory(dirPath, _rules) {
|
|
79
|
+
const { analyzeDirectory } = require('../../tools/analysis/code-analyzer');
|
|
80
|
+
const results = analyzeDirectory(dirPath, 50);
|
|
81
|
+
return results.map((r) => ({
|
|
82
|
+
file: r.file,
|
|
83
|
+
issues: r.issues.map((i) => ({
|
|
84
|
+
ruleId: 'auto',
|
|
85
|
+
line: undefined,
|
|
86
|
+
severity: i.severity,
|
|
87
|
+
message: i.message,
|
|
88
|
+
suggestion: i.suggestion || '',
|
|
89
|
+
})),
|
|
90
|
+
passed: r.issues.filter((i) => i.severity === 'high').length === 0,
|
|
91
|
+
}));
|
|
92
|
+
}
|
|
93
|
+
/** 加载自定义规则 */
|
|
94
|
+
function loadReviewRules(configPath) {
|
|
95
|
+
if (!(0, fs_1.existsSync)(configPath))
|
|
96
|
+
return DEFAULT_RULES;
|
|
97
|
+
try {
|
|
98
|
+
const config = JSON.parse((0, fs_1.readFileSync)(configPath, 'utf8'));
|
|
99
|
+
return config.rules || DEFAULT_RULES;
|
|
100
|
+
}
|
|
101
|
+
catch {
|
|
102
|
+
return DEFAULT_RULES;
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
/** 应用修复建议(简化版) */
|
|
106
|
+
function applyFixes(filePath, content, fixes) {
|
|
107
|
+
let result = content;
|
|
108
|
+
for (const fix of fixes) {
|
|
109
|
+
result = result.replace(fix.pattern, fix.replacement);
|
|
110
|
+
}
|
|
111
|
+
(0, fs_1.writeFileSync)(filePath, result, 'utf8');
|
|
112
|
+
console.log('[M7 Review] 修复已应用', { file: filePath, count: fixes.length });
|
|
113
|
+
return result;
|
|
114
|
+
}
|
|
115
|
+
//# sourceMappingURL=code-review.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"code-review.js","sourceRoot":"","sources":["../../src/agent/code-review.ts"],"names":[],"mappings":";;AAwEA,gCAwBC;AAGD,0CAcC;AAGD,0CAQC;AAGD,gCAQC;AAvID;;;;;;;;;GASG;AACH,2BAA6D;AAyB7D,aAAa;AACb,MAAM,aAAa,GAAiB;IAClC;QACE,EAAE,EAAE,kBAAkB;QACtB,IAAI,EAAE,OAAO;QACb,OAAO,EAAE,4DAA4D;QACrE,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,gBAAgB;QACzB,GAAG,EAAE,gBAAgB;KACtB;IACD;QACE,EAAE,EAAE,eAAe;QACnB,IAAI,EAAE,UAAU;QAChB,OAAO,EAAE,2DAA2D;QACpE,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,YAAY;QACrB,GAAG,EAAE,SAAS;KACf;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,iCAAiC;QAC1C,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,WAAW;QACpB,GAAG,EAAE,kBAAkB;KACxB;IACD;QACE,EAAE,EAAE,aAAa;QACjB,IAAI,EAAE,cAAc;QACpB,OAAO,EAAE,6BAA6B;QACtC,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,oBAAoB;QAC7B,GAAG,EAAE,cAAc;KACpB;CACF,CAAC;AAEF,aAAa;AACb,SAAgB,UAAU,CAAC,QAAgB,EAAE,OAAe,EAAE,QAAsB,aAAa;IAC/F,MAAM,MAAM,GAAkB,EAAE,CAAC;IAEjC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,MAAM,CAAC,IAAI,CAAC;oBACV,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,IAAI,EAAE,CAAC,GAAG,CAAC;oBACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,UAAU,EAAE,IAAI,CAAC,GAAG,IAAI,OAAO;iBAChC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,MAAM;QACN,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC;KAC/D,CAAC;AACJ,CAAC;AAED,aAAa;AACb,SAAgB,eAAe,CAAC,OAAe,EAAE,MAAqB;IACpE,MAAM,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,oCAAoC,CAAC,CAAC;IAC3E,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC9C,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;QAC9B,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YAChC,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;SAC/B,CAAC,CAAC;QACH,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC;KACxE,CAAC,CAAC,CAAC;AACN,CAAC;AAED,cAAc;AACd,SAAgB,eAAe,CAAC,UAAkB;IAChD,IAAI,CAAC,IAAA,eAAU,EAAC,UAAU,CAAC;QAAE,OAAO,aAAa,CAAC;IAClD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,iBAAY,EAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;QAC5D,OAAQ,MAAM,CAAC,KAAsB,IAAI,aAAa,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,aAAa,CAAC;IACvB,CAAC;AACH,CAAC;AAED,kBAAkB;AAClB,SAAgB,UAAU,CAAC,QAAgB,EAAE,OAAe,EAAE,KAAsD;IAClH,IAAI,MAAM,GAAG,OAAO,CAAC;IACrB,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;QACxB,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC;IACD,IAAA,kBAAa,EAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1E,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,227 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.CodeWriter = void 0;
|
|
4
|
+
exports.createCodeWriter = createCodeWriter;
|
|
5
|
+
/**
|
|
6
|
+
* 飞虹 Code (Muse Code 参照复刻)
|
|
7
|
+
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
8
|
+
*
|
|
9
|
+
* 自主代码编写器(M8):
|
|
10
|
+
* - 任务规划 → 代码编写 → 测试生成 → 审查反馈 → 迭代优化
|
|
11
|
+
* - 内置代码生成模板(API 路由、Model、工具骨架)
|
|
12
|
+
* - 自动测试生成(Jest/Vitest)
|
|
13
|
+
* - 质量门禁集成(审查规则 + 分析结果)
|
|
14
|
+
*/
|
|
15
|
+
const fs_1 = require("fs");
|
|
16
|
+
const path_1 = require("path");
|
|
17
|
+
const code_review_1 = require("./code-review");
|
|
18
|
+
const code_analyzer_1 = require("../tools/analysis/code-analyzer");
|
|
19
|
+
const test_generator_1 = require("../tools/generator/test-generator");
|
|
20
|
+
const code_generator_1 = require("../tools/generator/code-generator");
|
|
21
|
+
/** 自主编写器状态机 */
|
|
22
|
+
class CodeWriter {
|
|
23
|
+
steps = [];
|
|
24
|
+
filesCreated = [];
|
|
25
|
+
issuesFound = 0;
|
|
26
|
+
issuesFixed = 0;
|
|
27
|
+
rules;
|
|
28
|
+
cwd;
|
|
29
|
+
constructor(cwd, rules = []) {
|
|
30
|
+
this.cwd = cwd;
|
|
31
|
+
this.rules = rules;
|
|
32
|
+
}
|
|
33
|
+
/** 阶段 1:规划 */
|
|
34
|
+
plan(goal) {
|
|
35
|
+
const step = {
|
|
36
|
+
type: 'plan',
|
|
37
|
+
content: this.buildPlan(goal),
|
|
38
|
+
};
|
|
39
|
+
this.steps.push(step);
|
|
40
|
+
console.log(`[M8 Plan] ${step.content}`);
|
|
41
|
+
return step;
|
|
42
|
+
}
|
|
43
|
+
/** 阶段 2:编写代码 */
|
|
44
|
+
write(code, filePath) {
|
|
45
|
+
const absPath = (0, path_1.join)(this.cwd, filePath);
|
|
46
|
+
(0, fs_1.mkdirSync)((0, path_1.dirname)(absPath), { recursive: true });
|
|
47
|
+
(0, fs_1.writeFileSync)(absPath, code, 'utf8');
|
|
48
|
+
this.filesCreated.push(filePath);
|
|
49
|
+
const step = {
|
|
50
|
+
type: 'write',
|
|
51
|
+
content: `已生成文件: ${filePath} (${code.length} 字符)`,
|
|
52
|
+
file: filePath,
|
|
53
|
+
};
|
|
54
|
+
this.steps.push(step);
|
|
55
|
+
console.log(`[M8 Write] ${step.content}`);
|
|
56
|
+
return step;
|
|
57
|
+
}
|
|
58
|
+
/** 阶段 2b:使用模板生成代码 */
|
|
59
|
+
writeTemplate(template, params) {
|
|
60
|
+
let code;
|
|
61
|
+
let filename;
|
|
62
|
+
const result = template === 'api-route'
|
|
63
|
+
? (0, code_generator_1.generateApiRoute)(params.method, params.path, params.controller)
|
|
64
|
+
: template === 'model'
|
|
65
|
+
? (0, code_generator_1.generateModel)(params.name, JSON.parse(params.fields || '{}'))
|
|
66
|
+
: (0, code_generator_1.generateToolTemplate)(params.name);
|
|
67
|
+
if (!result.success || !result.content) {
|
|
68
|
+
return { type: 'write', content: `模板生成失败: ${result.error || '未知错误'}` };
|
|
69
|
+
}
|
|
70
|
+
code = result.content;
|
|
71
|
+
if (template === 'api-route') {
|
|
72
|
+
filename = params.path.replace(/^\//, '').replace(/\//g, '-') + '.ts';
|
|
73
|
+
}
|
|
74
|
+
else if (template === 'model') {
|
|
75
|
+
filename = params.name + '.ts';
|
|
76
|
+
}
|
|
77
|
+
else {
|
|
78
|
+
filename = params.name + '.tool.ts';
|
|
79
|
+
}
|
|
80
|
+
return this.write(code, (0, path_1.join)('generated', filename));
|
|
81
|
+
}
|
|
82
|
+
/** 阶段 3:生成测试 */
|
|
83
|
+
test(targetFile, functionName) {
|
|
84
|
+
const absPath = (0, path_1.join)(this.cwd, targetFile);
|
|
85
|
+
if (!(0, fs_1.existsSync)(absPath)) {
|
|
86
|
+
return { type: 'test', content: `跳过测试生成:文件不存在 ${targetFile}` };
|
|
87
|
+
}
|
|
88
|
+
const content = (0, fs_1.readFileSync)(absPath, 'utf8');
|
|
89
|
+
const funcName = functionName || this.inferFunctionName(content);
|
|
90
|
+
const testCases = (0, test_generator_1.inferTestCases)(funcName, content.slice(0, 500));
|
|
91
|
+
const result = (0, test_generator_1.generateJestTest)(targetFile, funcName, testCases);
|
|
92
|
+
const testFile = targetFile.replace('.ts', '.test.ts');
|
|
93
|
+
return this.write(result.content, testFile);
|
|
94
|
+
}
|
|
95
|
+
/** 阶段 4:审查 */
|
|
96
|
+
review(filePath) {
|
|
97
|
+
const absPath = (0, path_1.join)(this.cwd, filePath);
|
|
98
|
+
if (!(0, fs_1.existsSync)(absPath)) {
|
|
99
|
+
return { type: 'review', content: `跳过审查:文件不存在 ${filePath}` };
|
|
100
|
+
}
|
|
101
|
+
const content = (0, fs_1.readFileSync)(absPath, 'utf8');
|
|
102
|
+
const result = (0, code_review_1.reviewCode)(filePath, content, this.rules);
|
|
103
|
+
this.issuesFound += result.issues.length;
|
|
104
|
+
const step = {
|
|
105
|
+
type: 'review',
|
|
106
|
+
content: this.formatReviewResult(result),
|
|
107
|
+
file: filePath,
|
|
108
|
+
};
|
|
109
|
+
this.steps.push(step);
|
|
110
|
+
console.log(`[M8 Review] ${step.content}`);
|
|
111
|
+
return step;
|
|
112
|
+
}
|
|
113
|
+
/** 阶段 5:修复高优先级问题 */
|
|
114
|
+
fix(filePath, maxFixes = 3) {
|
|
115
|
+
const absPath = (0, path_1.join)(this.cwd, filePath);
|
|
116
|
+
if (!(0, fs_1.existsSync)(absPath)) {
|
|
117
|
+
return { type: 'fix', content: `文件不存在 ${filePath}` };
|
|
118
|
+
}
|
|
119
|
+
const content = (0, fs_1.readFileSync)(absPath, 'utf8');
|
|
120
|
+
const result = (0, code_review_1.reviewCode)(filePath, content, this.rules);
|
|
121
|
+
const highIssues = result.issues.filter((i) => i.severity === 'high').slice(0, maxFixes);
|
|
122
|
+
if (highIssues.length === 0) {
|
|
123
|
+
return { type: 'fix', content: '无高优先级问题需要修复' };
|
|
124
|
+
}
|
|
125
|
+
let newContent = content;
|
|
126
|
+
for (const issue of highIssues) {
|
|
127
|
+
const fix = this.issueToFix(issue);
|
|
128
|
+
newContent = newContent.replace(fix.pattern, fix.replacement);
|
|
129
|
+
}
|
|
130
|
+
(0, fs_1.writeFileSync)(absPath, newContent, 'utf8');
|
|
131
|
+
this.issuesFixed += highIssues.length;
|
|
132
|
+
return {
|
|
133
|
+
type: 'fix',
|
|
134
|
+
content: `已修复 ${highIssues.length} 个高优先级问题: ${highIssues.map((i) => i.message).join(', ')}`,
|
|
135
|
+
file: filePath,
|
|
136
|
+
};
|
|
137
|
+
}
|
|
138
|
+
/** 阶段 6:分析代码质量 */
|
|
139
|
+
analyze(filePath) {
|
|
140
|
+
const absPath = (0, path_1.join)(this.cwd, filePath);
|
|
141
|
+
if (!(0, fs_1.existsSync)(absPath)) {
|
|
142
|
+
return { type: 'review', content: `文件不存在 ${filePath}` };
|
|
143
|
+
}
|
|
144
|
+
const content = (0, fs_1.readFileSync)(absPath, 'utf8');
|
|
145
|
+
const analysis = (0, code_analyzer_1.analyzeFile)(filePath, content);
|
|
146
|
+
return {
|
|
147
|
+
type: 'review',
|
|
148
|
+
content: this.formatAnalysisResult(analysis),
|
|
149
|
+
file: filePath,
|
|
150
|
+
};
|
|
151
|
+
}
|
|
152
|
+
/** 总结 */
|
|
153
|
+
summary() {
|
|
154
|
+
const step = {
|
|
155
|
+
type: 'summary',
|
|
156
|
+
content: this.buildSummary(),
|
|
157
|
+
};
|
|
158
|
+
this.steps.push(step);
|
|
159
|
+
return step;
|
|
160
|
+
}
|
|
161
|
+
/** 完整流程:规划 → 编写 → 测试 → 审查 → 修复 → 总结 */
|
|
162
|
+
async run(goal, code, filePath) {
|
|
163
|
+
this.plan(goal);
|
|
164
|
+
this.write(code, filePath);
|
|
165
|
+
this.test(filePath);
|
|
166
|
+
this.review(filePath);
|
|
167
|
+
this.analyze(filePath);
|
|
168
|
+
this.fix(filePath);
|
|
169
|
+
this.summary();
|
|
170
|
+
return this.getResult();
|
|
171
|
+
}
|
|
172
|
+
/** 批量编写多个文件 */
|
|
173
|
+
async runBatch(tasks) {
|
|
174
|
+
for (const task of tasks) {
|
|
175
|
+
await this.run(task.goal, task.code, task.filePath);
|
|
176
|
+
}
|
|
177
|
+
return this.getResult();
|
|
178
|
+
}
|
|
179
|
+
getResult() {
|
|
180
|
+
return {
|
|
181
|
+
success: this.issuesFound - this.issuesFixed === 0,
|
|
182
|
+
steps: this.steps,
|
|
183
|
+
finalFiles: this.filesCreated,
|
|
184
|
+
issuesFound: this.issuesFound,
|
|
185
|
+
issuesFixed: this.issuesFixed,
|
|
186
|
+
summary: this.buildSummary(),
|
|
187
|
+
};
|
|
188
|
+
}
|
|
189
|
+
buildPlan(goal) {
|
|
190
|
+
return `任务规划:${goal}\n预期输出文件: 待编写\n策略: 先规划结构,再逐步实现`;
|
|
191
|
+
}
|
|
192
|
+
formatReviewResult(result) {
|
|
193
|
+
const high = result.issues.filter((i) => i.severity === 'high').length;
|
|
194
|
+
const med = result.issues.filter((i) => i.severity === 'medium').length;
|
|
195
|
+
const low = result.issues.filter((i) => i.severity === 'low').length;
|
|
196
|
+
return `${result.file}: ${result.issues.length} 个问题 (高:${high} 中:${med} 低:${low}) ${result.passed ? '✅ 通过' : '❌ 未通过'}`;
|
|
197
|
+
}
|
|
198
|
+
formatAnalysisResult(analysis) {
|
|
199
|
+
const highIssues = analysis.issues.filter((i) => i.severity === 'high').length;
|
|
200
|
+
return `代码质量分析: 复杂度=${analysis.metrics.complexity} 高优问题=${highIssues} 总问题=${analysis.issues.length}`;
|
|
201
|
+
}
|
|
202
|
+
buildSummary() {
|
|
203
|
+
return `M8 自主编写完成: ${this.filesCreated.length} 文件, ${this.issuesFound} 问题发现, ${this.issuesFixed} 问题修复`;
|
|
204
|
+
}
|
|
205
|
+
inferFunctionName(content) {
|
|
206
|
+
const match = content.match(/export\s+(?:async\s+)?function\s+(\w+)/);
|
|
207
|
+
return match?.[1] || 'main';
|
|
208
|
+
}
|
|
209
|
+
issueToFix(issue) {
|
|
210
|
+
if (issue.message.includes('硬编码')) {
|
|
211
|
+
return {
|
|
212
|
+
pattern: /['"`][^'"`]*(password|secret|key|token)[^'"`]*['"`]/gi,
|
|
213
|
+
replacement: (_sub, p1) => `process.env.FH_${p1?.toUpperCase() ?? 'UNKNOWN'}`,
|
|
214
|
+
};
|
|
215
|
+
}
|
|
216
|
+
if (issue.message.includes('SQL')) {
|
|
217
|
+
return { pattern: /['"][\s\S]*?SELECT[\s\S]*?['"]/gi, replacement: '// 使用参数化查询' };
|
|
218
|
+
}
|
|
219
|
+
return { pattern: /.*/, replacement: issue.suggestion };
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
exports.CodeWriter = CodeWriter;
|
|
223
|
+
/** 便捷函数 */
|
|
224
|
+
function createCodeWriter(cwd, rules) {
|
|
225
|
+
return new CodeWriter(cwd, rules);
|
|
226
|
+
}
|
|
227
|
+
//# sourceMappingURL=code-writer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"code-writer.js","sourceRoot":"","sources":["../../src/agent/code-writer.ts"],"names":[],"mappings":";;;AAsQA,4CAEC;AAxQD;;;;;;;;;GASG;AACH,2BAAwE;AACxE,+BAAqC;AAErC,+CAA2C;AAC3C,mEAA8D;AAC9D,sEAAqF;AACrF,sEAI2C;AAiB3C,eAAe;AACf,MAAa,UAAU;IACb,KAAK,GAAqB,EAAE,CAAC;IAC7B,YAAY,GAAa,EAAE,CAAC;IAC5B,WAAW,GAAG,CAAC,CAAC;IAChB,WAAW,GAAG,CAAC,CAAC;IACP,KAAK,CAAe;IACpB,GAAG,CAAS;IAE7B,YAAY,GAAW,EAAE,QAAsB,EAAE;QAC/C,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,cAAc;IACd,IAAI,CAAC,IAAY;QACf,MAAM,IAAI,GAAmB;YAC3B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC9B,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB;IAChB,KAAK,CAAC,IAAY,EAAE,QAAgB;QAClC,MAAM,OAAO,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACzC,IAAA,cAAS,EAAC,IAAA,cAAO,EAAC,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjD,IAAA,kBAAa,EAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,IAAI,GAAmB;YAC3B,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,UAAU,QAAQ,KAAK,IAAI,CAAC,MAAM,MAAM;YACjD,IAAI,EAAE,QAAQ;SACf,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qBAAqB;IACrB,aAAa,CACX,QAAwC,EACxC,MAA8B;QAE9B,IAAI,IAAY,CAAC;QACjB,IAAI,QAAgB,CAAC;QACrB,MAAM,MAAM,GACV,QAAQ,KAAK,WAAW;YACtB,CAAC,CAAC,IAAA,iCAAgB,EAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,CAAC;YACjE,CAAC,CAAC,QAAQ,KAAK,OAAO;gBACpB,CAAC,CAAC,IAAA,8BAAa,EAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;gBAC/D,CAAC,CAAC,IAAA,qCAAoB,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACvC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,MAAM,CAAC,KAAK,IAAI,MAAM,EAAE,EAAE,CAAC;QACzE,CAAC;QACD,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC;QACtB,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;YAC7B,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC;QACxE,CAAC;aAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YAChC,QAAQ,GAAG,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,CAAC,IAAI,GAAG,UAAU,CAAC;QACtC,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,IAAA,WAAI,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,gBAAgB;IAChB,IAAI,CAAC,UAAkB,EAAE,YAAqB;QAC5C,MAAM,OAAO,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,UAAU,EAAE,EAAE,CAAC;QACjE,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,iBAAY,EAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,YAAY,IAAI,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,SAAS,GAAG,IAAA,+BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,IAAA,iCAAgB,EAAC,UAAU,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,cAAc;IACd,MAAM,CAAC,QAAgB;QACrB,MAAM,OAAO,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,cAAc,QAAQ,EAAE,EAAE,CAAC;QAC/D,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,iBAAY,EAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAA,wBAAU,EAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC;QACzC,MAAM,IAAI,GAAmB;YAC3B,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC;YACxC,IAAI,EAAE,QAAQ;SACf,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oBAAoB;IACpB,GAAG,CAAC,QAAgB,EAAE,QAAQ,GAAG,CAAC;QAChC,MAAM,OAAO,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,QAAQ,EAAE,EAAE,CAAC;QACvD,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,iBAAY,EAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAA,wBAAU,EAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QACzF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACjD,CAAC;QACD,IAAI,UAAU,GAAG,OAAO,CAAC;QACzB,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACnC,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,WAAkB,CAAC,CAAC;QACvE,CAAC;QACD,IAAA,kBAAa,EAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,WAAW,IAAI,UAAU,CAAC,MAAM,CAAC;QACtC,OAAO;YACL,IAAI,EAAE,KAAK;YACX,OAAO,EAAE,OAAO,UAAU,CAAC,MAAM,aAAa,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC3F,IAAI,EAAE,QAAQ;SACf,CAAC;IACJ,CAAC;IAED,kBAAkB;IAClB,OAAO,CAAC,QAAgB;QACtB,MAAM,OAAO,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,QAAQ,EAAE,EAAE,CAAC;QAC1D,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,iBAAY,EAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,IAAA,2BAAW,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChD,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC;YAC5C,IAAI,EAAE,QAAQ;SACf,CAAC;IACJ,CAAC;IAED,SAAS;IACT,OAAO;QACL,MAAM,IAAI,GAAmB;YAC3B,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;SAC7B,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,uCAAuC;IACvC,KAAK,CAAC,GAAG,CAAC,IAAY,EAAE,IAAY,EAAE,QAAgB;QACpD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtB,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnB,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC;IAED,eAAe;IACf,KAAK,CAAC,QAAQ,CACZ,KAA8D;QAE9D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC;IAEO,SAAS;QACf,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,KAAK,CAAC;YAClD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,IAAI,CAAC,YAAY;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;SAC7B,CAAC;IACJ,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,OAAO,QAAQ,IAAI,gCAAgC,CAAC;IACtD,CAAC;IAEO,kBAAkB,CAAC,MAA0F;QACnH,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,MAAM,CAAC;QACvE,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;QACxE,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,MAAM,CAAC;QACrE,OAAO,GAAG,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,MAAM,CAAC,MAAM,WAAW,IAAI,MAAM,GAAG,MAAM,GAAG,KAAK,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACzH,CAAC;IAEO,oBAAoB,CAAC,QAA4G;QACvI,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,MAAM,CAAC;QAC/E,OAAO,eAAe,QAAQ,CAAC,OAAO,CAAC,UAAU,SAAS,UAAU,QAAQ,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;IACvG,CAAC;IAEO,YAAY;QAClB,OAAO,cAAc,IAAI,CAAC,YAAY,CAAC,MAAM,QAAQ,IAAI,CAAC,WAAW,UAAU,IAAI,CAAC,WAAW,OAAO,CAAC;IACzG,CAAC;IAEO,iBAAiB,CAAC,OAAe;QACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACtE,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC;IAC9B,CAAC;IAEO,UAAU,CAAC,KAA8C;QAC/D,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO;gBACL,OAAO,EAAE,uDAAuD;gBAChE,WAAW,EAAE,CAAC,IAAY,EAAE,EAAU,EAAE,EAAE,CAAC,kBAAkB,EAAE,EAAE,WAAW,EAAE,IAAI,SAAS,EAAE;aAC9F,CAAC;QACJ,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,EAAE,OAAO,EAAE,kCAAkC,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;QACpF,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;IAC1D,CAAC;CACF;AA7ND,gCA6NC;AAED,WAAW;AACX,SAAgB,gBAAgB,CAAC,GAAW,EAAE,KAAoB;IAChE,OAAO,IAAI,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;AACpC,CAAC"}
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.compactContext = compactContext;
|
|
4
|
+
exports.shouldCompact = shouldCompact;
|
|
5
|
+
exports.getCompactionThreshold = getCompactionThreshold;
|
|
6
|
+
const logger_1 = require("../shared/logger");
|
|
7
|
+
/** 从早期消息中提取结构化摘要 */
|
|
8
|
+
function extractSummary(messages) {
|
|
9
|
+
const modifiedFiles = new Set();
|
|
10
|
+
const decisionPoints = [];
|
|
11
|
+
const keyInsights = [];
|
|
12
|
+
for (const msg of messages) {
|
|
13
|
+
// 提取文件路径
|
|
14
|
+
if (msg.role === 'assistant' && msg.toolCalls) {
|
|
15
|
+
for (const tc of msg.toolCalls) {
|
|
16
|
+
if (tc.name === 'write_file' || tc.name === 'edit_file') {
|
|
17
|
+
const path = tc.arguments?.path;
|
|
18
|
+
if (path)
|
|
19
|
+
modifiedFiles.add(path);
|
|
20
|
+
}
|
|
21
|
+
if (tc.name === 'run_shell') {
|
|
22
|
+
const cmd = tc.arguments?.command || '';
|
|
23
|
+
if (cmd.includes('npm test') || cmd.includes('npm run build')) {
|
|
24
|
+
decisionPoints.push(`执行验证: ${cmd.slice(0, 50)}`);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
// 提取关键洞察
|
|
30
|
+
if (msg.role === 'assistant' && msg.content) {
|
|
31
|
+
const content = msg.content;
|
|
32
|
+
if (content.includes('完成') || content.includes('成功') || content.includes('注意')) {
|
|
33
|
+
keyInsights.push(content.slice(0, 100));
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
return {
|
|
38
|
+
decisionPoints: [...new Set(decisionPoints)].slice(0, 5),
|
|
39
|
+
modifiedFiles: [...modifiedFiles].slice(0, 10),
|
|
40
|
+
keyInsights: [...new Set(keyInsights)].slice(0, 3),
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
/** 生成压缩后的 system prompt */
|
|
44
|
+
function generateCompactionPrompt(summary, preservedCount, totalRounds) {
|
|
45
|
+
const parts = ['📋 任务进展摘要(上下文压缩)'];
|
|
46
|
+
if (summary.modifiedFiles.length > 0) {
|
|
47
|
+
parts.push(`\n**已修改文件** (${summary.modifiedFiles.length} 个):\n${summary.modifiedFiles.map((f) => `- ${f}`).join('\n')}`);
|
|
48
|
+
}
|
|
49
|
+
if (summary.decisionPoints.length > 0) {
|
|
50
|
+
parts.push(`\n**关键决策点**:\n${summary.decisionPoints.map((d) => `- ${d}`).join('\n')}`);
|
|
51
|
+
}
|
|
52
|
+
if (summary.keyInsights.length > 0) {
|
|
53
|
+
parts.push(`\n**关键洞察**:\n${summary.keyInsights.map((k) => `- ${k}`).join('\n')}`);
|
|
54
|
+
}
|
|
55
|
+
parts.push(`\n**上下文状态**: 已压缩 ${totalRounds - preservedCount} 轮早期消息,保留最近 ${preservedCount} 轮完整对话`);
|
|
56
|
+
return parts.join('\n');
|
|
57
|
+
}
|
|
58
|
+
/**
|
|
59
|
+
* 压缩上下文:
|
|
60
|
+
* - 保留最近 preservedCount 轮完整消息
|
|
61
|
+
* - 压缩早期消息为结构化摘要
|
|
62
|
+
* - 返回新的消息列表和压缩统计
|
|
63
|
+
*/
|
|
64
|
+
function compactContext(messages, preservedCount = 10) {
|
|
65
|
+
// 始终保留首条 system 指令(messages[0]),避免压缩后丢失系统级提示导致行为退化
|
|
66
|
+
const systemMsg = messages[0]?.role === 'system' ? messages[0] : null;
|
|
67
|
+
const body = systemMsg ? messages.slice(1) : messages;
|
|
68
|
+
const totalRounds = Math.floor(body.length / 2); // 假设每轮 2 条消息(assistant + tool)
|
|
69
|
+
const preservedMessages = Math.min(preservedCount * 2, body.length);
|
|
70
|
+
const earlyMessages = body.slice(0, body.length - preservedMessages);
|
|
71
|
+
const recentMessages = body.slice(body.length - preservedMessages);
|
|
72
|
+
if (earlyMessages.length === 0) {
|
|
73
|
+
return { messages, stats: { originalLength: messages.length, compressedLength: messages.length, preservedMessages, compressedRounds: 0, timestamp: new Date().toISOString() } };
|
|
74
|
+
}
|
|
75
|
+
// 提取摘要
|
|
76
|
+
const summary = extractSummary(earlyMessages);
|
|
77
|
+
const compactionPrompt = generateCompactionPrompt(summary, preservedMessages, totalRounds);
|
|
78
|
+
// 压缩摘要作为 system message 注入;原 system 指令(如有)始终保留在首位
|
|
79
|
+
const systemHint = {
|
|
80
|
+
role: 'system',
|
|
81
|
+
content: compactionPrompt,
|
|
82
|
+
};
|
|
83
|
+
const prefix = systemMsg ? [systemMsg, systemHint] : [systemHint];
|
|
84
|
+
const compacted = [...prefix, ...recentMessages];
|
|
85
|
+
const stats = {
|
|
86
|
+
originalLength: messages.length,
|
|
87
|
+
compressedLength: compacted.length,
|
|
88
|
+
preservedMessages,
|
|
89
|
+
compressedRounds: totalRounds - Math.floor(preservedMessages / 2),
|
|
90
|
+
timestamp: new Date().toISOString(),
|
|
91
|
+
};
|
|
92
|
+
logger_1.logger.info('context compaction applied', {
|
|
93
|
+
originalLength: stats.originalLength,
|
|
94
|
+
compressedLength: stats.compressedLength,
|
|
95
|
+
preservedMessages: stats.preservedMessages,
|
|
96
|
+
compressedRounds: stats.compressedRounds,
|
|
97
|
+
});
|
|
98
|
+
return { messages: compacted, stats };
|
|
99
|
+
}
|
|
100
|
+
/** 检查是否需要触发压缩 */
|
|
101
|
+
function shouldCompact(messages, threshold) {
|
|
102
|
+
return messages.length >= threshold;
|
|
103
|
+
}
|
|
104
|
+
/** 从配置读取压缩阈值(默认每 30 条消息触发一次) */
|
|
105
|
+
function getCompactionThreshold(config = {}) {
|
|
106
|
+
return config.compactEvery ?? 30;
|
|
107
|
+
}
|
|
108
|
+
//# sourceMappingURL=context-compactor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-compactor.js","sourceRoot":"","sources":["../../src/agent/context-compactor.ts"],"names":[],"mappings":";;AA8FA,wCA6CC;AAGD,sCAEC;AAGD,wDAEC;AA1ID,6CAA0C;AAU1C,oBAAoB;AACpB,SAAS,cAAc,CAAC,QAAuB;IAK7C,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,SAAS;QACT,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAC9C,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;gBAC/B,IAAI,EAAE,CAAC,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACxD,MAAM,IAAI,GAAI,EAAE,CAAC,SAA+B,EAAE,IAAI,CAAC;oBACvD,IAAI,IAAI;wBAAE,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACpC,CAAC;gBACD,IAAI,EAAE,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC5B,MAAM,GAAG,GAAI,EAAE,CAAC,SAAkC,EAAE,OAAO,IAAI,EAAE,CAAC;oBAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;wBAC9D,cAAc,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;oBACnD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QACD,SAAS;QACT,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAC5C,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC5B,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/E,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,cAAc,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;QACxD,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;QAC9C,WAAW,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;KACnD,CAAC;AACJ,CAAC;AAED,2BAA2B;AAC3B,SAAS,wBAAwB,CAAC,OAIjC,EAAE,cAAsB,EAAE,WAAmB;IAC5C,MAAM,KAAK,GAAa,CAAC,kBAAkB,CAAC,CAAC;IAE7C,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrC,KAAK,CAAC,IAAI,CAAC,gBAAgB,OAAO,CAAC,aAAa,CAAC,MAAM,SAAS,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnI,CAAC;IAED,IAAI,OAAO,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,iBAAiB,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxF,CAAC;IAED,IAAI,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,gBAAgB,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,oBAAoB,WAAW,GAAG,cAAc,eAAe,cAAc,QAAQ,CAAC,CAAC;IAElG,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;;;;GAKG;AACH,SAAgB,cAAc,CAC5B,QAAuB,EACvB,iBAAyB,EAAE;IAE3B,mDAAmD;IACnD,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACtE,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAEtD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,+BAA+B;IAChF,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,GAAG,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACpE,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,GAAG,iBAAiB,CAAC,CAAC;IACrE,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,iBAAiB,CAAC,CAAC;IAEnE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,QAAQ,CAAC,MAAM,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC;IAClL,CAAC;IAED,OAAO;IACP,MAAM,OAAO,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;IAC9C,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,OAAO,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC;IAE3F,kDAAkD;IAClD,MAAM,UAAU,GAAgB;QAC9B,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,gBAAgB;KAC1B,CAAC;IACF,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IAElE,MAAM,SAAS,GAAkB,CAAC,GAAG,MAAM,EAAE,GAAG,cAAc,CAAC,CAAC;IAChE,MAAM,KAAK,GAAoB;QAC7B,cAAc,EAAE,QAAQ,CAAC,MAAM;QAC/B,gBAAgB,EAAE,SAAS,CAAC,MAAM;QAClC,iBAAiB;QACjB,gBAAgB,EAAE,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,GAAG,CAAC,CAAC;QACjE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;IAEF,eAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;QACxC,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;QACxC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;QAC1C,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;KACzC,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;AACxC,CAAC;AAED,iBAAiB;AACjB,SAAgB,aAAa,CAAC,QAAuB,EAAE,SAAiB;IACtE,OAAO,QAAQ,CAAC,MAAM,IAAI,SAAS,CAAC;AACtC,CAAC;AAED,gCAAgC;AAChC,SAAgB,sBAAsB,CAAC,SAAoC,EAAE;IAC3E,OAAO,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;AACnC,CAAC"}
|