micro-models-agent 0.44.0 → 0.45.0
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/README.md +312 -312
- package/bin/mma.mjs +41 -41
- package/dist/main.js +241 -56
- package/dist/skills/builtin/git.md +36 -36
- package/dist/skills/builtin/typescript.md +35 -35
- package/package.json +1 -1
- package/dist/cli/commands.js +0 -333
- package/dist/cli/completer.js +0 -168
- package/dist/cli/index.js +0 -2
- package/dist/cli/main.js +0 -140
- package/dist/cli/repl-commands.js +0 -633
- package/dist/cli/repl.js +0 -486
- package/dist/cli/security-commands.js +0 -166
- package/dist/cli/setup.js +0 -249
- package/dist/config/config.js +0 -202
- package/dist/config/defaults.js +0 -100
- package/dist/config/experts.js +0 -15
- package/dist/config/index.js +0 -3
- package/dist/config/security.js +0 -200
- package/dist/config/types.js +0 -1
- package/dist/core/agent-moe.js +0 -110
- package/dist/core/agent.js +0 -695
- package/dist/core/bootstrap.js +0 -337
- package/dist/core/index.js +0 -2
- package/dist/core/prompt-builder.js +0 -55
- package/dist/core/session-logger.js +0 -155
- package/dist/core/types.js +0 -1
- package/dist/core/workspace.js +0 -76
- package/dist/i18n/en.json +0 -525
- package/dist/i18n/index.js +0 -46
- package/dist/i18n/ru.json +0 -525
- package/dist/index.js +0 -22
- package/dist/llm/image-utils.js +0 -144
- package/dist/llm/index.js +0 -4
- package/dist/llm/model-loader.js +0 -78
- package/dist/llm/openai-compat.js +0 -353
- package/dist/llm/orchestrator.js +0 -194
- package/dist/llm/provider.js +0 -10
- package/dist/llm/response.js +0 -39
- package/dist/llm/token-counter.js +0 -39
- package/dist/llm/types.js +0 -1
- package/dist/logger/app-logger.js +0 -143
- package/dist/logger/file-log.js +0 -151
- package/dist/logger/index.js +0 -1
- package/dist/migration/backup.js +0 -45
- package/dist/migration/detect.js +0 -50
- package/dist/migration/index.js +0 -2
- package/dist/modules/browser/actions.js +0 -46
- package/dist/modules/browser/cookie-store.js +0 -24
- package/dist/modules/browser/index.js +0 -5
- package/dist/modules/browser/module.js +0 -28
- package/dist/modules/browser/session.js +0 -335
- package/dist/modules/browser/snapshot.js +0 -114
- package/dist/modules/browser/types.js +0 -9
- package/dist/modules/certification/cli.js +0 -176
- package/dist/modules/certification/fact-checker.js +0 -84
- package/dist/modules/certification/loader.js +0 -111
- package/dist/modules/certification/manifest.js +0 -50
- package/dist/modules/certification/runner.js +0 -162
- package/dist/modules/certification/scenarios.js +0 -124
- package/dist/modules/certification/types.js +0 -1
- package/dist/modules/context/index.js +0 -1
- package/dist/modules/context/manager.js +0 -349
- package/dist/modules/execution/auditor.js +0 -66
- package/dist/modules/execution/index.js +0 -8
- package/dist/modules/execution/module.js +0 -779
- package/dist/modules/execution/moe-executor.js +0 -266
- package/dist/modules/execution/plan-coverage.js +0 -68
- package/dist/modules/execution/plan-persister.js +0 -46
- package/dist/modules/execution/plan-store.js +0 -159
- package/dist/modules/execution/plan-validator.js +0 -153
- package/dist/modules/execution/planner.js +0 -85
- package/dist/modules/execution/stuck-detector.js +0 -347
- package/dist/modules/execution/tracker.js +0 -67
- package/dist/modules/execution/types.js +0 -1
- package/dist/modules/execution/verifier.js +0 -178
- package/dist/modules/hallucination/confidence.js +0 -59
- package/dist/modules/hallucination/consistency.js +0 -26
- package/dist/modules/hallucination/detector.js +0 -46
- package/dist/modules/hallucination/factual.js +0 -190
- package/dist/modules/hallucination/index.js +0 -5
- package/dist/modules/hallucination/js-identifiers.js +0 -72
- package/dist/modules/hallucination/llm-judge.js +0 -103
- package/dist/modules/index.js +0 -5
- package/dist/modules/indexer/cache.js +0 -38
- package/dist/modules/indexer/index.js +0 -3
- package/dist/modules/indexer/module.js +0 -192
- package/dist/modules/indexer/walker.js +0 -101
- package/dist/modules/lsp/client.js +0 -235
- package/dist/modules/lsp/config.js +0 -81
- package/dist/modules/lsp/index.js +0 -3
- package/dist/modules/lsp/module.js +0 -68
- package/dist/modules/lsp/types.js +0 -1
- package/dist/modules/mcp/client.js +0 -399
- package/dist/modules/mcp/index.js +0 -3
- package/dist/modules/mcp/module.js +0 -146
- package/dist/modules/mcp/registry.js +0 -15
- package/dist/modules/memory/index.js +0 -1
- package/dist/modules/memory/module.js +0 -48
- package/dist/modules/memory/search.js +0 -40
- package/dist/modules/memory/store.js +0 -69
- package/dist/modules/pipelines/engine.js +0 -60
- package/dist/modules/pipelines/index.js +0 -3
- package/dist/modules/pipelines/parser.js +0 -53
- package/dist/modules/pipelines/template.js +0 -14
- package/dist/modules/plugins/builtin/lint-on-write.js +0 -226
- package/dist/modules/plugins/builtin/notify.js +0 -8
- package/dist/modules/plugins/index.js +0 -1
- package/dist/modules/plugins/loader.js +0 -28
- package/dist/modules/plugins/manager.js +0 -161
- package/dist/modules/plugins/types.js +0 -1
- package/dist/modules/processes/index.js +0 -2
- package/dist/modules/processes/registry.js +0 -238
- package/dist/modules/processes/runner.js +0 -23
- package/dist/modules/registry.js +0 -45
- package/dist/modules/security/audit-log.js +0 -136
- package/dist/modules/security/audit-notifier.js +0 -292
- package/dist/modules/security/command-validator.js +0 -211
- package/dist/modules/security/content-scanner.js +0 -53
- package/dist/modules/security/data-sanitizer.js +0 -97
- package/dist/modules/security/encryption.js +0 -240
- package/dist/modules/security/index.js +0 -14
- package/dist/modules/security/network-validator.js +0 -79
- package/dist/modules/security/path-validator.js +0 -209
- package/dist/modules/security/rate-limiter.js +0 -119
- package/dist/modules/security/security-policies.js +0 -547
- package/dist/modules/security/session-encryption.js +0 -210
- package/dist/modules/security/session-isolation.js +0 -95
- package/dist/modules/session/index.js +0 -3
- package/dist/modules/session/manager.js +0 -172
- package/dist/modules/session/module.js +0 -24
- package/dist/modules/session/store.js +0 -228
- package/dist/modules/session/types.js +0 -1
- package/dist/modules/skills/index.js +0 -2
- package/dist/modules/skills/loader.js +0 -72
- package/dist/modules/skills/module.js +0 -130
- package/dist/modules/types.js +0 -1
- package/dist/modules/updater/checker.js +0 -32
- package/dist/modules/updater/index.js +0 -1
- package/dist/modules/user-profile/compressor.js +0 -16
- package/dist/modules/user-profile/index.js +0 -1
- package/dist/modules/user-profile/profile.js +0 -68
- package/dist/tools/approve.js +0 -32
- package/dist/tools/attach-image.js +0 -89
- package/dist/tools/bash.js +0 -337
- package/dist/tools/browser.js +0 -97
- package/dist/tools/create-dir.js +0 -55
- package/dist/tools/delete-file.js +0 -62
- package/dist/tools/edit-file.js +0 -79
- package/dist/tools/executor.js +0 -145
- package/dist/tools/file-info.js +0 -45
- package/dist/tools/filter-tools.js +0 -10
- package/dist/tools/glob-tool.js +0 -26
- package/dist/tools/grep-tool.js +0 -86
- package/dist/tools/index.js +0 -67
- package/dist/tools/list-dir.js +0 -47
- package/dist/tools/load-skill.js +0 -44
- package/dist/tools/mcp-call.js +0 -68
- package/dist/tools/move-file.js +0 -85
- package/dist/tools/path-utils.js +0 -51
- package/dist/tools/pipeline-run.js +0 -144
- package/dist/tools/preview.js +0 -2
- package/dist/tools/process-kill.js +0 -29
- package/dist/tools/process-list.js +0 -38
- package/dist/tools/process-log.js +0 -41
- package/dist/tools/question.js +0 -142
- package/dist/tools/read-file.js +0 -83
- package/dist/tools/recall.js +0 -110
- package/dist/tools/registry.js +0 -36
- package/dist/tools/remember.js +0 -67
- package/dist/tools/scope-check.js +0 -30
- package/dist/tools/search-history.js +0 -84
- package/dist/tools/subagent.js +0 -151
- package/dist/tools/types.js +0 -1
- package/dist/tools/user-input.js +0 -123
- package/dist/tools/web-browse.js +0 -86
- package/dist/tools/web-fetch.js +0 -98
- package/dist/tools/web-search.js +0 -78
- package/dist/tools/write-file.js +0 -83
- package/dist/ui/box.js +0 -81
- package/dist/ui/colors.js +0 -4
- package/dist/ui/diff.js +0 -178
- package/dist/ui/index.js +0 -6
- package/dist/ui/md-formatter.js +0 -212
- package/dist/ui/output.js +0 -13
- package/dist/ui/renderer.js +0 -204
- package/dist/ui/spinner.js +0 -70
- package/dist/ui/table.js +0 -144
|
@@ -1,192 +0,0 @@
|
|
|
1
|
-
import { dirname } from 'path';
|
|
2
|
-
import { Indexer } from './walker';
|
|
3
|
-
import { IndexCache } from './cache';
|
|
4
|
-
import { t } from '../../i18n/index';
|
|
5
|
-
export class IndexerModule {
|
|
6
|
-
name = 'indexer';
|
|
7
|
-
indexer;
|
|
8
|
-
cache;
|
|
9
|
-
baseDir;
|
|
10
|
-
index = null;
|
|
11
|
-
rebuildTimeout = null;
|
|
12
|
-
constructor(opts) {
|
|
13
|
-
this.baseDir = opts.baseDir;
|
|
14
|
-
this.indexer = new Indexer(opts.baseDir);
|
|
15
|
-
this.cache = new IndexCache(opts.cacheDir);
|
|
16
|
-
}
|
|
17
|
-
async buildIndex() {
|
|
18
|
-
if (this.index)
|
|
19
|
-
return this.index;
|
|
20
|
-
const cached = this.cache.load();
|
|
21
|
-
if (cached) {
|
|
22
|
-
this.index = cached;
|
|
23
|
-
return cached;
|
|
24
|
-
}
|
|
25
|
-
try {
|
|
26
|
-
const result = await this.indexer.walk();
|
|
27
|
-
this.cache.save(result);
|
|
28
|
-
this.index = result;
|
|
29
|
-
return result;
|
|
30
|
-
}
|
|
31
|
-
catch {
|
|
32
|
-
return null;
|
|
33
|
-
}
|
|
34
|
-
}
|
|
35
|
-
async refresh() {
|
|
36
|
-
this.cache.invalidate();
|
|
37
|
-
this.index = null;
|
|
38
|
-
return this.buildIndex();
|
|
39
|
-
}
|
|
40
|
-
find(query) {
|
|
41
|
-
if (!this.index)
|
|
42
|
-
return [];
|
|
43
|
-
const q = query.toLowerCase();
|
|
44
|
-
return this.index.files.filter((f) => {
|
|
45
|
-
const path = f.path.toLowerCase();
|
|
46
|
-
return path.includes(q) || f.exports.some((e) => e.toLowerCase().includes(q));
|
|
47
|
-
});
|
|
48
|
-
}
|
|
49
|
-
watch() {
|
|
50
|
-
this.unwatch();
|
|
51
|
-
try {
|
|
52
|
-
this.indexer.watch((_event, filename) => {
|
|
53
|
-
if (!filename)
|
|
54
|
-
return;
|
|
55
|
-
if (this.rebuildTimeout)
|
|
56
|
-
clearTimeout(this.rebuildTimeout);
|
|
57
|
-
this.rebuildTimeout = setTimeout(() => {
|
|
58
|
-
this.refresh().catch(() => { });
|
|
59
|
-
}, 2000);
|
|
60
|
-
});
|
|
61
|
-
}
|
|
62
|
-
catch {
|
|
63
|
-
// Watching is best-effort; manual refresh still works.
|
|
64
|
-
}
|
|
65
|
-
}
|
|
66
|
-
unwatch() {
|
|
67
|
-
if (this.rebuildTimeout) {
|
|
68
|
-
clearTimeout(this.rebuildTimeout);
|
|
69
|
-
this.rebuildTimeout = null;
|
|
70
|
-
}
|
|
71
|
-
this.indexer.unwatch();
|
|
72
|
-
}
|
|
73
|
-
getSystemPromptBlock() {
|
|
74
|
-
if (!this.index)
|
|
75
|
-
return null;
|
|
76
|
-
const content = this.formatMap(this.index);
|
|
77
|
-
return {
|
|
78
|
-
content,
|
|
79
|
-
priority: 'normal',
|
|
80
|
-
essential: false,
|
|
81
|
-
estimatedTokens: this.estimateTokens(content),
|
|
82
|
-
};
|
|
83
|
-
}
|
|
84
|
-
getToolDefinitions() {
|
|
85
|
-
return [this.createProjectMapTool()];
|
|
86
|
-
}
|
|
87
|
-
getPlugin() {
|
|
88
|
-
return {
|
|
89
|
-
name: 'indexer',
|
|
90
|
-
onSessionStart: () => {
|
|
91
|
-
this.buildIndex().catch(() => { });
|
|
92
|
-
this.watch();
|
|
93
|
-
},
|
|
94
|
-
onSessionEnd: () => {
|
|
95
|
-
this.unwatch();
|
|
96
|
-
},
|
|
97
|
-
};
|
|
98
|
-
}
|
|
99
|
-
formatMap(result) {
|
|
100
|
-
const summary = this.indexer.summarize(result);
|
|
101
|
-
const dirCounts = this.getDirectoryCounts(result);
|
|
102
|
-
const topDirs = Object.entries(dirCounts)
|
|
103
|
-
.sort((a, b) => b[1] - a[1])
|
|
104
|
-
.slice(0, 10)
|
|
105
|
-
.map(([dir, count]) => `${dir} (${count})`)
|
|
106
|
-
.join(', ') || '-';
|
|
107
|
-
const fileLines = result.files.slice(0, 100).map((f) => {
|
|
108
|
-
const path = f.path.replace(/\\/g, '/');
|
|
109
|
-
const exports = f.exports.length > 0 ? `: ${f.exports.join(', ')}` : '';
|
|
110
|
-
return `- ${path}${exports}`;
|
|
111
|
-
});
|
|
112
|
-
const more = result.files.length > 100
|
|
113
|
-
? `\n${t('indexer.and_more', { count: result.files.length - 100 })}`
|
|
114
|
-
: '';
|
|
115
|
-
return [
|
|
116
|
-
`${t('indexer.map_header')} (${this.baseDir})`,
|
|
117
|
-
summary,
|
|
118
|
-
`${t('indexer.top_directories')}: ${topDirs}`,
|
|
119
|
-
`${t('indexer.files')}:` + (fileLines.length > 0 ? '' : ' ' + t('indexer.empty')),
|
|
120
|
-
...fileLines,
|
|
121
|
-
more,
|
|
122
|
-
].join('\n');
|
|
123
|
-
}
|
|
124
|
-
getDirectoryCounts(result) {
|
|
125
|
-
const counts = {};
|
|
126
|
-
for (const f of result.files) {
|
|
127
|
-
const normalized = f.path.replace(/\\/g, '/');
|
|
128
|
-
const dir = dirname(normalized);
|
|
129
|
-
const key = dir === '.' ? '(root)' : dir;
|
|
130
|
-
counts[key] = (counts[key] || 0) + 1;
|
|
131
|
-
}
|
|
132
|
-
return counts;
|
|
133
|
-
}
|
|
134
|
-
estimateTokens(content) {
|
|
135
|
-
return Math.ceil(content.length / 4);
|
|
136
|
-
}
|
|
137
|
-
createProjectMapTool() {
|
|
138
|
-
return {
|
|
139
|
-
name: 'project_map',
|
|
140
|
-
description: t('indexer.project_map_desc'),
|
|
141
|
-
parameters: {
|
|
142
|
-
type: 'object',
|
|
143
|
-
properties: {
|
|
144
|
-
action: {
|
|
145
|
-
type: 'string',
|
|
146
|
-
enum: ['summary', 'refresh', 'find'],
|
|
147
|
-
description: t('indexer.project_map_action_desc'),
|
|
148
|
-
},
|
|
149
|
-
query: {
|
|
150
|
-
type: 'string',
|
|
151
|
-
description: t('indexer.project_map_query_desc'),
|
|
152
|
-
},
|
|
153
|
-
},
|
|
154
|
-
required: ['action'],
|
|
155
|
-
},
|
|
156
|
-
handler: async (_ctx, args) => {
|
|
157
|
-
const action = String(args.action || 'summary');
|
|
158
|
-
if (action === 'refresh') {
|
|
159
|
-
const result = await this.refresh();
|
|
160
|
-
const summary = result ? this.indexer.summarize(result) : t('indexer.empty');
|
|
161
|
-
return this.makeResult(t('indexer.refreshed', { summary }));
|
|
162
|
-
}
|
|
163
|
-
if (action === 'find') {
|
|
164
|
-
const query = String(args.query || '').toLowerCase();
|
|
165
|
-
if (!query)
|
|
166
|
-
return { success: false, output: t('tool.invalid_params') };
|
|
167
|
-
const matches = this.find(query);
|
|
168
|
-
if (matches.length === 0) {
|
|
169
|
-
return this.makeResult(t('indexer.no_matches', { query }));
|
|
170
|
-
}
|
|
171
|
-
const lines = matches.map((f) => {
|
|
172
|
-
const path = f.path.replace(/\\/g, '/');
|
|
173
|
-
const exports = f.exports.length > 0 ? `: ${f.exports.join(', ')}` : '';
|
|
174
|
-
return `- ${path}${exports}`;
|
|
175
|
-
}).join('\n');
|
|
176
|
-
return this.makeResult(t('indexer.find_results', { count: matches.length, results: lines }));
|
|
177
|
-
}
|
|
178
|
-
if (!this.index) {
|
|
179
|
-
await this.buildIndex();
|
|
180
|
-
}
|
|
181
|
-
if (!this.index) {
|
|
182
|
-
return this.makeResult(t('indexer.not_indexed'));
|
|
183
|
-
}
|
|
184
|
-
const summary = this.indexer.summarize(this.index);
|
|
185
|
-
return this.makeResult(t('indexer.summary', { summary }));
|
|
186
|
-
},
|
|
187
|
-
};
|
|
188
|
-
}
|
|
189
|
-
makeResult(output) {
|
|
190
|
-
return { success: true, output, display: output };
|
|
191
|
-
}
|
|
192
|
-
}
|
|
@@ -1,101 +0,0 @@
|
|
|
1
|
-
import { readdirSync, readFileSync, statSync, existsSync, watch } from 'fs';
|
|
2
|
-
import { join, relative, extname } from 'path';
|
|
3
|
-
const LANGUAGES = {
|
|
4
|
-
'.ts': 'typescript',
|
|
5
|
-
'.tsx': 'typescript-react',
|
|
6
|
-
'.js': 'javascript',
|
|
7
|
-
'.jsx': 'javascript-react',
|
|
8
|
-
'.json': 'json',
|
|
9
|
-
'.md': 'markdown',
|
|
10
|
-
'.yaml': 'yaml',
|
|
11
|
-
'.yml': 'yaml',
|
|
12
|
-
'.py': 'python',
|
|
13
|
-
'.rs': 'rust',
|
|
14
|
-
'.go': 'go',
|
|
15
|
-
};
|
|
16
|
-
const IGNORE_DIRS = new Set(['node_modules', '.git', 'dist', 'build', '.mma', 'coverage']);
|
|
17
|
-
export class Indexer {
|
|
18
|
-
baseDir;
|
|
19
|
-
MAX_FILES = 1000;
|
|
20
|
-
watcher = null;
|
|
21
|
-
constructor(baseDir) {
|
|
22
|
-
this.baseDir = baseDir;
|
|
23
|
-
}
|
|
24
|
-
watch(callback) {
|
|
25
|
-
this.watcher = watch(this.baseDir, { recursive: true }, (event, filename) => {
|
|
26
|
-
if (filename && !Array.from(IGNORE_DIRS).some((dir) => filename.includes(dir))) {
|
|
27
|
-
callback(event, filename);
|
|
28
|
-
}
|
|
29
|
-
});
|
|
30
|
-
}
|
|
31
|
-
unwatch() {
|
|
32
|
-
if (this.watcher) {
|
|
33
|
-
this.watcher.close();
|
|
34
|
-
this.watcher = null;
|
|
35
|
-
}
|
|
36
|
-
}
|
|
37
|
-
async walk() {
|
|
38
|
-
const files = [];
|
|
39
|
-
let totalSize = 0;
|
|
40
|
-
let count = 0;
|
|
41
|
-
const walkDir = (dir) => {
|
|
42
|
-
if (!existsSync(dir))
|
|
43
|
-
return;
|
|
44
|
-
let entries;
|
|
45
|
-
try {
|
|
46
|
-
entries = readdirSync(dir);
|
|
47
|
-
}
|
|
48
|
-
catch {
|
|
49
|
-
return;
|
|
50
|
-
}
|
|
51
|
-
for (const entry of entries) {
|
|
52
|
-
if (count >= this.MAX_FILES)
|
|
53
|
-
return;
|
|
54
|
-
const fullPath = join(dir, entry);
|
|
55
|
-
const relPath = relative(this.baseDir, fullPath);
|
|
56
|
-
const stat = statSync(fullPath);
|
|
57
|
-
if (stat.isDirectory()) {
|
|
58
|
-
if (!IGNORE_DIRS.has(entry)) {
|
|
59
|
-
walkDir(fullPath);
|
|
60
|
-
}
|
|
61
|
-
}
|
|
62
|
-
else if (stat.isFile()) {
|
|
63
|
-
const ext = extname(entry).toLowerCase();
|
|
64
|
-
const language = LANGUAGES[ext];
|
|
65
|
-
if (language) {
|
|
66
|
-
const content = readFileSync(fullPath, 'utf-8');
|
|
67
|
-
const exports = this.extractExports(content, language);
|
|
68
|
-
files.push({ path: relPath, language, exports, size: stat.size });
|
|
69
|
-
totalSize += stat.size;
|
|
70
|
-
count++;
|
|
71
|
-
}
|
|
72
|
-
}
|
|
73
|
-
}
|
|
74
|
-
};
|
|
75
|
-
walkDir(this.baseDir);
|
|
76
|
-
return { files, totalSize };
|
|
77
|
-
}
|
|
78
|
-
extractExports(content, language) {
|
|
79
|
-
if (language === 'typescript' || language === 'javascript') {
|
|
80
|
-
const exports = [];
|
|
81
|
-
const exportRegex = /export\s+(?:const|function|class|interface|type|enum|default\s+(?:class|function))\s+(\w+)/g;
|
|
82
|
-
let match;
|
|
83
|
-
while ((match = exportRegex.exec(content)) !== null) {
|
|
84
|
-
exports.push(match[1]);
|
|
85
|
-
}
|
|
86
|
-
return exports;
|
|
87
|
-
}
|
|
88
|
-
return [];
|
|
89
|
-
}
|
|
90
|
-
summarize(result) {
|
|
91
|
-
const byLang = {};
|
|
92
|
-
for (const f of result.files) {
|
|
93
|
-
byLang[f.language] = (byLang[f.language] || 0) + 1;
|
|
94
|
-
}
|
|
95
|
-
const langs = Object.entries(byLang)
|
|
96
|
-
.sort((a, b) => b[1] - a[1])
|
|
97
|
-
.map(([lang, count]) => `${lang}:${count}`)
|
|
98
|
-
.join(', ');
|
|
99
|
-
return `[Index: ${result.files.length} files, ${(result.totalSize / 1024).toFixed(1)}KB (${langs})]`;
|
|
100
|
-
}
|
|
101
|
-
}
|
|
@@ -1,235 +0,0 @@
|
|
|
1
|
-
import { spawn, execSync } from "child_process";
|
|
2
|
-
import { resolve } from "path";
|
|
3
|
-
const DEFAULT_TIMEOUT = 15000;
|
|
4
|
-
export class LspClient {
|
|
5
|
-
process = null;
|
|
6
|
-
requestId = 0;
|
|
7
|
-
pending = new Map();
|
|
8
|
-
buffer = "";
|
|
9
|
-
contentLength = -1;
|
|
10
|
-
diagnostics = [];
|
|
11
|
-
diagnosticsResolve = null;
|
|
12
|
-
diagnosticsTimer = null;
|
|
13
|
-
initialized = false;
|
|
14
|
-
async checkFile(filePath, baseDir, config) {
|
|
15
|
-
const timeout = config.timeout ?? DEFAULT_TIMEOUT;
|
|
16
|
-
try {
|
|
17
|
-
await this.startServer(config, baseDir);
|
|
18
|
-
const rootUri = this.pathToUri(resolve(baseDir));
|
|
19
|
-
const initResult = await this.sendRequest("initialize", {
|
|
20
|
-
processId: process.pid,
|
|
21
|
-
rootUri,
|
|
22
|
-
workspaceFolders: [{ uri: rootUri, name: "workspace" }],
|
|
23
|
-
capabilities: { textDocument: { publishDiagnostics: {} } },
|
|
24
|
-
}, timeout);
|
|
25
|
-
this.initialized = true;
|
|
26
|
-
this.sendNotification("initialized", {});
|
|
27
|
-
const uri = this.pathToUri(resolve(filePath));
|
|
28
|
-
const fs = await import("fs");
|
|
29
|
-
const content = fs.readFileSync(filePath, "utf-8");
|
|
30
|
-
const diagPromise = new Promise((resolve) => {
|
|
31
|
-
this.diagnosticsResolve = resolve;
|
|
32
|
-
this.diagnostics = [];
|
|
33
|
-
this.diagnosticsTimer = setTimeout(() => {
|
|
34
|
-
if (this.diagnosticsResolve) {
|
|
35
|
-
this.diagnosticsResolve([]);
|
|
36
|
-
this.diagnosticsResolve = null;
|
|
37
|
-
}
|
|
38
|
-
}, timeout);
|
|
39
|
-
});
|
|
40
|
-
this.sendNotification("textDocument/didOpen", {
|
|
41
|
-
textDocument: {
|
|
42
|
-
uri,
|
|
43
|
-
languageId: this.languageFromPath(filePath),
|
|
44
|
-
version: 1,
|
|
45
|
-
text: content,
|
|
46
|
-
},
|
|
47
|
-
});
|
|
48
|
-
return await diagPromise;
|
|
49
|
-
}
|
|
50
|
-
catch {
|
|
51
|
-
return [];
|
|
52
|
-
}
|
|
53
|
-
finally {
|
|
54
|
-
await this.shutdown();
|
|
55
|
-
}
|
|
56
|
-
}
|
|
57
|
-
async startServer(config, baseDir) {
|
|
58
|
-
if (config.autoInstall === false) {
|
|
59
|
-
try {
|
|
60
|
-
execSync(`where ${config.command}`, { stdio: "pipe", timeout: 3000 });
|
|
61
|
-
}
|
|
62
|
-
catch {
|
|
63
|
-
throw new Error(`${config.command} not found in PATH`);
|
|
64
|
-
}
|
|
65
|
-
}
|
|
66
|
-
return new Promise((resolve, reject) => {
|
|
67
|
-
const args = config.args ?? [];
|
|
68
|
-
const proc = spawn(config.command, args, {
|
|
69
|
-
stdio: ["pipe", "pipe", "pipe"],
|
|
70
|
-
env: { ...process.env, ...config.env },
|
|
71
|
-
cwd: baseDir,
|
|
72
|
-
});
|
|
73
|
-
proc.on("error", reject);
|
|
74
|
-
proc.stdout.on("data", (chunk) => {
|
|
75
|
-
this.handleData(chunk);
|
|
76
|
-
});
|
|
77
|
-
proc.stderr.on("data", () => { });
|
|
78
|
-
proc.once("spawn", () => {
|
|
79
|
-
resolve();
|
|
80
|
-
});
|
|
81
|
-
this.process = proc;
|
|
82
|
-
setTimeout(() => {
|
|
83
|
-
if (!this.initialized && this.process) {
|
|
84
|
-
reject(new Error("LSP server start timeout"));
|
|
85
|
-
}
|
|
86
|
-
}, config.timeout ?? 10000);
|
|
87
|
-
});
|
|
88
|
-
}
|
|
89
|
-
handleData(chunk) {
|
|
90
|
-
this.buffer += chunk.toString();
|
|
91
|
-
this.parseMessages();
|
|
92
|
-
}
|
|
93
|
-
parseMessages() {
|
|
94
|
-
while (true) {
|
|
95
|
-
if (this.contentLength === -1) {
|
|
96
|
-
const headerEnd = this.buffer.indexOf("\r\n\r\n");
|
|
97
|
-
if (headerEnd === -1)
|
|
98
|
-
return;
|
|
99
|
-
const header = this.buffer.slice(0, headerEnd);
|
|
100
|
-
const match = header.match(/Content-Length: (\d+)/i);
|
|
101
|
-
if (!match) {
|
|
102
|
-
this.buffer = this.buffer.slice(headerEnd + 4);
|
|
103
|
-
this.contentLength = -1;
|
|
104
|
-
continue;
|
|
105
|
-
}
|
|
106
|
-
this.contentLength = parseInt(match[1], 10);
|
|
107
|
-
// Guard against malicious/buggy servers sending absurd Content-Length
|
|
108
|
-
if (this.contentLength > 10 * 1024 * 1024) {
|
|
109
|
-
this.buffer = this.buffer.slice(headerEnd + 4);
|
|
110
|
-
this.contentLength = -1;
|
|
111
|
-
continue;
|
|
112
|
-
}
|
|
113
|
-
this.buffer = this.buffer.slice(headerEnd + 4);
|
|
114
|
-
}
|
|
115
|
-
if (this.buffer.length < this.contentLength)
|
|
116
|
-
return;
|
|
117
|
-
const body = this.buffer.slice(0, this.contentLength);
|
|
118
|
-
this.buffer = this.buffer.slice(this.contentLength);
|
|
119
|
-
this.contentLength = -1;
|
|
120
|
-
try {
|
|
121
|
-
const msg = JSON.parse(body);
|
|
122
|
-
this.handleMessage(msg);
|
|
123
|
-
}
|
|
124
|
-
catch { }
|
|
125
|
-
}
|
|
126
|
-
}
|
|
127
|
-
handleMessage(msg) {
|
|
128
|
-
if (msg.method === "textDocument/publishDiagnostics") {
|
|
129
|
-
const params = msg.params;
|
|
130
|
-
if (params?.diagnostics) {
|
|
131
|
-
this.diagnostics = params.diagnostics;
|
|
132
|
-
if (this.diagnosticsTimer) {
|
|
133
|
-
clearTimeout(this.diagnosticsTimer);
|
|
134
|
-
this.diagnosticsTimer = null;
|
|
135
|
-
}
|
|
136
|
-
if (this.diagnosticsResolve) {
|
|
137
|
-
this.diagnosticsResolve(this.diagnostics);
|
|
138
|
-
this.diagnosticsResolve = null;
|
|
139
|
-
}
|
|
140
|
-
}
|
|
141
|
-
return;
|
|
142
|
-
}
|
|
143
|
-
if (msg.id !== undefined && msg.id !== null) {
|
|
144
|
-
const pending = this.pending.get(msg.id);
|
|
145
|
-
if (pending) {
|
|
146
|
-
this.pending.delete(msg.id);
|
|
147
|
-
if (msg.error) {
|
|
148
|
-
pending.reject(new Error(String(msg.error.message ?? "LSP error")));
|
|
149
|
-
}
|
|
150
|
-
else {
|
|
151
|
-
pending.resolve(msg.result);
|
|
152
|
-
}
|
|
153
|
-
}
|
|
154
|
-
}
|
|
155
|
-
}
|
|
156
|
-
sendRequest(method, params, timeout) {
|
|
157
|
-
return new Promise((resolve, reject) => {
|
|
158
|
-
const id = ++this.requestId;
|
|
159
|
-
this.pending.set(id, { resolve, reject });
|
|
160
|
-
const message = JSON.stringify({ jsonrpc: "2.0", id, method, params });
|
|
161
|
-
this.write(message);
|
|
162
|
-
setTimeout(() => {
|
|
163
|
-
if (this.pending.has(id)) {
|
|
164
|
-
this.pending.delete(id);
|
|
165
|
-
reject(new Error(`LSP request timeout: ${method}`));
|
|
166
|
-
}
|
|
167
|
-
}, timeout);
|
|
168
|
-
});
|
|
169
|
-
}
|
|
170
|
-
sendNotification(method, params) {
|
|
171
|
-
const message = JSON.stringify({ jsonrpc: "2.0", method, params });
|
|
172
|
-
this.write(message);
|
|
173
|
-
}
|
|
174
|
-
write(message) {
|
|
175
|
-
if (!this.process?.stdin?.writable)
|
|
176
|
-
return;
|
|
177
|
-
const header = `Content-Length: ${Buffer.byteLength(message)}\r\n\r\n`;
|
|
178
|
-
try {
|
|
179
|
-
this.process.stdin.write(header + message);
|
|
180
|
-
}
|
|
181
|
-
catch { }
|
|
182
|
-
}
|
|
183
|
-
async shutdown() {
|
|
184
|
-
try {
|
|
185
|
-
if (this.process && this.initialized && this.process.stdin?.writable) {
|
|
186
|
-
this.sendRequest("shutdown", null, 3000).catch(() => { });
|
|
187
|
-
this.sendNotification("exit", null);
|
|
188
|
-
await new Promise((r) => setTimeout(r, 200));
|
|
189
|
-
}
|
|
190
|
-
}
|
|
191
|
-
catch {
|
|
192
|
-
}
|
|
193
|
-
finally {
|
|
194
|
-
if (this.process) {
|
|
195
|
-
try {
|
|
196
|
-
this.process.kill();
|
|
197
|
-
}
|
|
198
|
-
catch { }
|
|
199
|
-
this.process = null;
|
|
200
|
-
}
|
|
201
|
-
this.initialized = false;
|
|
202
|
-
if (this.diagnosticsTimer) {
|
|
203
|
-
clearTimeout(this.diagnosticsTimer);
|
|
204
|
-
this.diagnosticsTimer = null;
|
|
205
|
-
}
|
|
206
|
-
}
|
|
207
|
-
}
|
|
208
|
-
pathToUri(filePath) {
|
|
209
|
-
const normalized = filePath.replace(/\\/g, "/");
|
|
210
|
-
if (/^[a-zA-Z]:/.test(normalized)) {
|
|
211
|
-
return `file:///${normalized}`;
|
|
212
|
-
}
|
|
213
|
-
return `file://${normalized}`;
|
|
214
|
-
}
|
|
215
|
-
languageFromPath(filePath) {
|
|
216
|
-
const ext = filePath.split(".").pop()?.toLowerCase();
|
|
217
|
-
const map = {
|
|
218
|
-
ts: "typescript",
|
|
219
|
-
tsx: "typescriptreact",
|
|
220
|
-
js: "javascript",
|
|
221
|
-
jsx: "javascriptreact",
|
|
222
|
-
py: "python",
|
|
223
|
-
go: "go",
|
|
224
|
-
rs: "rust",
|
|
225
|
-
json: "json",
|
|
226
|
-
html: "html",
|
|
227
|
-
css: "css",
|
|
228
|
-
scss: "scss",
|
|
229
|
-
md: "markdown",
|
|
230
|
-
yaml: "yaml",
|
|
231
|
-
yml: "yaml",
|
|
232
|
-
};
|
|
233
|
-
return map[ext ?? ""] ?? "plaintext";
|
|
234
|
-
}
|
|
235
|
-
}
|
|
@@ -1,81 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* LSP server defaults.
|
|
3
|
-
*
|
|
4
|
-
* Strategy: prefer `npx` (auto-downloads) over globally installed binaries.
|
|
5
|
-
* Servers using system tools (rust-analyzer, gopls) have `autoInstall: false`
|
|
6
|
-
* and are skipped silently when not found.
|
|
7
|
-
*/
|
|
8
|
-
export const DEFAULT_LSP_CONFIG = {
|
|
9
|
-
enabled: true,
|
|
10
|
-
timeout: 15000,
|
|
11
|
-
servers: {
|
|
12
|
-
typescript: {
|
|
13
|
-
command: 'npx',
|
|
14
|
-
args: ['typescript-language-server', '--stdio'],
|
|
15
|
-
timeout: 20000,
|
|
16
|
-
autoInstall: true,
|
|
17
|
-
},
|
|
18
|
-
javascript: {
|
|
19
|
-
command: 'npx',
|
|
20
|
-
args: ['typescript-language-server', '--stdio'],
|
|
21
|
-
timeout: 15000,
|
|
22
|
-
autoInstall: true,
|
|
23
|
-
},
|
|
24
|
-
python: {
|
|
25
|
-
command: 'npx',
|
|
26
|
-
args: ['pyright', '--stdio'],
|
|
27
|
-
timeout: 15000,
|
|
28
|
-
autoInstall: true,
|
|
29
|
-
},
|
|
30
|
-
rust: {
|
|
31
|
-
command: 'rust-analyzer',
|
|
32
|
-
timeout: 15000,
|
|
33
|
-
autoInstall: false,
|
|
34
|
-
},
|
|
35
|
-
go: {
|
|
36
|
-
command: 'gopls',
|
|
37
|
-
timeout: 15000,
|
|
38
|
-
autoInstall: false,
|
|
39
|
-
},
|
|
40
|
-
json: {
|
|
41
|
-
command: 'npx',
|
|
42
|
-
args: ['vscode-json-languageserver', '--stdio'],
|
|
43
|
-
timeout: 10000,
|
|
44
|
-
autoInstall: true,
|
|
45
|
-
},
|
|
46
|
-
html: {
|
|
47
|
-
command: 'npx',
|
|
48
|
-
args: ['vscode-html-languageserver', '--stdio'],
|
|
49
|
-
timeout: 10000,
|
|
50
|
-
autoInstall: true,
|
|
51
|
-
},
|
|
52
|
-
css: {
|
|
53
|
-
command: 'npx',
|
|
54
|
-
args: ['vscode-css-languageserver', '--stdio'],
|
|
55
|
-
timeout: 10000,
|
|
56
|
-
autoInstall: true,
|
|
57
|
-
},
|
|
58
|
-
},
|
|
59
|
-
};
|
|
60
|
-
const LANGUAGE_MAP = {
|
|
61
|
-
ts: 'typescript',
|
|
62
|
-
tsx: 'typescript',
|
|
63
|
-
js: 'javascript',
|
|
64
|
-
jsx: 'javascript',
|
|
65
|
-
py: 'python',
|
|
66
|
-
rs: 'rust',
|
|
67
|
-
go: 'go',
|
|
68
|
-
json: 'json',
|
|
69
|
-
html: 'html',
|
|
70
|
-
css: 'css',
|
|
71
|
-
scss: 'css',
|
|
72
|
-
};
|
|
73
|
-
export function getServerForFile(filePath, config) {
|
|
74
|
-
const ext = filePath.split('.').pop()?.toLowerCase();
|
|
75
|
-
if (!ext)
|
|
76
|
-
return null;
|
|
77
|
-
const lang = LANGUAGE_MAP[ext];
|
|
78
|
-
if (!lang)
|
|
79
|
-
return null;
|
|
80
|
-
return config.servers[lang] ?? null;
|
|
81
|
-
}
|
|
@@ -1,68 +0,0 @@
|
|
|
1
|
-
import { existsSync } from 'fs';
|
|
2
|
-
import { resolve } from 'path';
|
|
3
|
-
import { LspClient } from './client';
|
|
4
|
-
import { getServerForFile, DEFAULT_LSP_CONFIG } from './config';
|
|
5
|
-
const severityLabels = {
|
|
6
|
-
1: 'error',
|
|
7
|
-
2: 'warning',
|
|
8
|
-
3: 'info',
|
|
9
|
-
4: 'hint',
|
|
10
|
-
};
|
|
11
|
-
export class LspModule {
|
|
12
|
-
name = 'lsp';
|
|
13
|
-
config;
|
|
14
|
-
client = new LspClient();
|
|
15
|
-
constructor(config) {
|
|
16
|
-
this.config = { ...DEFAULT_LSP_CONFIG, ...config };
|
|
17
|
-
}
|
|
18
|
-
getPlugin() {
|
|
19
|
-
const self = this;
|
|
20
|
-
return {
|
|
21
|
-
name: 'lsp-check',
|
|
22
|
-
priority: 15,
|
|
23
|
-
onAfterTool: async (_ctx, call, result) => {
|
|
24
|
-
if (!self.config.enabled)
|
|
25
|
-
return;
|
|
26
|
-
if (call.name !== 'write_file' && call.name !== 'edit_file')
|
|
27
|
-
return;
|
|
28
|
-
if (!result.success)
|
|
29
|
-
return;
|
|
30
|
-
const filePath = String(call.arguments.path ?? '');
|
|
31
|
-
if (!filePath)
|
|
32
|
-
return;
|
|
33
|
-
const fullPath = resolve(_ctx.baseDir, filePath);
|
|
34
|
-
if (!existsSync(fullPath))
|
|
35
|
-
return;
|
|
36
|
-
const serverConfig = getServerForFile(fullPath, self.config);
|
|
37
|
-
if (!serverConfig)
|
|
38
|
-
return;
|
|
39
|
-
try {
|
|
40
|
-
const diagnostics = await self.client.checkFile(fullPath, _ctx.baseDir, serverConfig);
|
|
41
|
-
const errors = diagnostics.filter((d) => d.severity === 1);
|
|
42
|
-
const warnings = diagnostics.filter((d) => d.severity === 2);
|
|
43
|
-
if (errors.length > 0) {
|
|
44
|
-
const items = formatDiagnostics(errors, fullPath);
|
|
45
|
-
result.output += `\n\n[LSP errors]:\n${items}`;
|
|
46
|
-
}
|
|
47
|
-
if (warnings.length > 0) {
|
|
48
|
-
const items = formatDiagnostics(warnings, fullPath);
|
|
49
|
-
result.output += `\n\n[LSP warnings]:\n${items}`;
|
|
50
|
-
}
|
|
51
|
-
}
|
|
52
|
-
catch { }
|
|
53
|
-
},
|
|
54
|
-
};
|
|
55
|
-
}
|
|
56
|
-
}
|
|
57
|
-
function formatDiagnostics(diagnostics, filePath) {
|
|
58
|
-
const filename = filePath.replace(/\\/g, '/').split('/').pop() ?? filePath;
|
|
59
|
-
return diagnostics
|
|
60
|
-
.slice(0, 5)
|
|
61
|
-
.map((d) => {
|
|
62
|
-
const line = d.range.start.line + 1;
|
|
63
|
-
const col = d.range.start.character + 1;
|
|
64
|
-
const sev = severityLabels[d.severity] ?? 'unknown';
|
|
65
|
-
return ` ${filename}(${line},${col}): ${sev}: ${d.message}`;
|
|
66
|
-
})
|
|
67
|
-
.join('\n');
|
|
68
|
-
}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|