@timmeck/brain 1.8.1 → 1.8.3
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/BRAIN_PLAN.md +3324 -3324
- package/LICENSE +21 -21
- package/dist/cli/commands/dashboard.js +595 -595
- package/dist/cli/commands/doctor.js +6 -1
- package/dist/cli/commands/doctor.js.map +1 -1
- package/dist/dashboard/server.js +25 -25
- package/dist/db/migrations/001_core_schema.js +115 -115
- package/dist/db/migrations/002_learning_schema.js +33 -33
- package/dist/db/migrations/003_code_schema.js +48 -48
- package/dist/db/migrations/004_synapses_schema.js +52 -52
- package/dist/db/migrations/005_fts_indexes.js +73 -73
- package/dist/db/migrations/007_feedback.js +8 -8
- package/dist/db/migrations/008_git_integration.js +33 -33
- package/dist/db/migrations/009_embeddings.js +3 -3
- package/dist/db/repositories/antipattern.repository.js +3 -3
- package/dist/db/repositories/code-module.repository.js +32 -32
- package/dist/db/repositories/notification.repository.js +3 -3
- package/dist/db/repositories/project.repository.js +21 -21
- package/dist/db/repositories/rule.repository.js +24 -24
- package/dist/db/repositories/solution.repository.js +50 -50
- package/dist/db/repositories/synapse.repository.js +18 -18
- package/dist/db/repositories/terminal.repository.js +24 -24
- package/dist/ipc/server.d.ts +8 -0
- package/dist/ipc/server.js +67 -1
- package/dist/ipc/server.js.map +1 -1
- package/dist/matching/error-matcher.js +5 -5
- package/dist/matching/fingerprint.js +6 -1
- package/dist/matching/fingerprint.js.map +1 -1
- package/dist/services/error.service.js +4 -3
- package/dist/services/error.service.js.map +1 -1
- package/dist/services/git.service.js +14 -14
- package/package.json +49 -49
- package/src/api/server.ts +395 -395
- package/src/brain.ts +266 -266
- package/src/cli/colors.ts +116 -116
- package/src/cli/commands/config.ts +169 -169
- package/src/cli/commands/dashboard.ts +755 -755
- package/src/cli/commands/doctor.ts +124 -118
- package/src/cli/commands/explain.ts +83 -83
- package/src/cli/commands/export.ts +31 -31
- package/src/cli/commands/import.ts +199 -199
- package/src/cli/commands/insights.ts +65 -65
- package/src/cli/commands/learn.ts +24 -24
- package/src/cli/commands/modules.ts +53 -53
- package/src/cli/commands/network.ts +67 -67
- package/src/cli/commands/projects.ts +42 -42
- package/src/cli/commands/query.ts +120 -120
- package/src/cli/commands/start.ts +62 -62
- package/src/cli/commands/status.ts +75 -75
- package/src/cli/commands/stop.ts +34 -34
- package/src/cli/ipc-helper.ts +22 -22
- package/src/cli/update-check.ts +63 -63
- package/src/code/fingerprint.ts +87 -87
- package/src/code/parsers/generic.ts +29 -29
- package/src/code/parsers/python.ts +54 -54
- package/src/code/parsers/typescript.ts +65 -65
- package/src/code/registry.ts +60 -60
- package/src/dashboard/server.ts +142 -142
- package/src/db/connection.ts +22 -22
- package/src/db/migrations/001_core_schema.ts +120 -120
- package/src/db/migrations/002_learning_schema.ts +38 -38
- package/src/db/migrations/003_code_schema.ts +53 -53
- package/src/db/migrations/004_synapses_schema.ts +57 -57
- package/src/db/migrations/005_fts_indexes.ts +78 -78
- package/src/db/migrations/006_synapses_phase3.ts +17 -17
- package/src/db/migrations/007_feedback.ts +13 -13
- package/src/db/migrations/008_git_integration.ts +38 -38
- package/src/db/migrations/009_embeddings.ts +8 -8
- package/src/db/repositories/antipattern.repository.ts +66 -66
- package/src/db/repositories/code-module.repository.ts +142 -142
- package/src/db/repositories/notification.repository.ts +66 -66
- package/src/db/repositories/project.repository.ts +93 -93
- package/src/db/repositories/rule.repository.ts +108 -108
- package/src/db/repositories/solution.repository.ts +154 -154
- package/src/db/repositories/synapse.repository.ts +153 -153
- package/src/db/repositories/terminal.repository.ts +101 -101
- package/src/embeddings/engine.ts +238 -238
- package/src/index.ts +63 -63
- package/src/ipc/client.ts +118 -118
- package/src/ipc/protocol.ts +35 -35
- package/src/ipc/router.ts +133 -133
- package/src/ipc/server.ts +176 -110
- package/src/learning/decay.ts +46 -46
- package/src/learning/pattern-extractor.ts +90 -90
- package/src/learning/rule-generator.ts +74 -74
- package/src/matching/error-matcher.ts +5 -5
- package/src/matching/fingerprint.ts +34 -29
- package/src/matching/similarity.ts +61 -61
- package/src/matching/tfidf.ts +74 -74
- package/src/matching/tokenizer.ts +41 -41
- package/src/mcp/auto-detect.ts +93 -93
- package/src/mcp/http-server.ts +140 -140
- package/src/mcp/server.ts +73 -73
- package/src/parsing/error-parser.ts +28 -28
- package/src/parsing/parsers/compiler.ts +93 -93
- package/src/parsing/parsers/generic.ts +28 -28
- package/src/parsing/parsers/go.ts +97 -97
- package/src/parsing/parsers/node.ts +69 -69
- package/src/parsing/parsers/python.ts +62 -62
- package/src/parsing/parsers/rust.ts +50 -50
- package/src/parsing/parsers/shell.ts +42 -42
- package/src/parsing/types.ts +47 -47
- package/src/research/gap-analyzer.ts +135 -135
- package/src/research/insight-generator.ts +123 -123
- package/src/research/research-engine.ts +116 -116
- package/src/research/synergy-detector.ts +126 -126
- package/src/research/template-extractor.ts +130 -130
- package/src/research/trend-analyzer.ts +127 -127
- package/src/services/code.service.ts +271 -271
- package/src/services/error.service.ts +4 -3
- package/src/services/git.service.ts +132 -132
- package/src/services/notification.service.ts +41 -41
- package/src/services/synapse.service.ts +59 -59
- package/src/services/terminal.service.ts +81 -81
- package/src/synapses/activation.ts +80 -80
- package/src/synapses/decay.ts +38 -38
- package/src/synapses/hebbian.ts +69 -69
- package/src/synapses/pathfinder.ts +81 -81
- package/src/synapses/synapse-manager.ts +109 -109
- package/src/types/code.types.ts +52 -52
- package/src/types/error.types.ts +67 -67
- package/src/types/ipc.types.ts +8 -8
- package/src/types/mcp.types.ts +53 -53
- package/src/types/research.types.ts +28 -28
- package/src/types/solution.types.ts +30 -30
- package/src/utils/events.ts +45 -45
- package/src/utils/hash.ts +5 -5
- package/src/utils/logger.ts +48 -48
- package/src/utils/paths.ts +19 -19
- package/tests/e2e/test_code_intelligence.py +1015 -0
- package/tests/e2e/test_error_memory.py +451 -0
- package/tests/e2e/test_full_integration.py +534 -0
- package/tests/fixtures/code-modules/modules.ts +83 -83
- package/tests/fixtures/errors/go.ts +9 -9
- package/tests/fixtures/errors/node.ts +24 -24
- package/tests/fixtures/errors/python.ts +21 -21
- package/tests/fixtures/errors/rust.ts +25 -25
- package/tests/fixtures/errors/shell.ts +15 -15
- package/tests/fixtures/solutions/solutions.ts +27 -27
- package/tests/helpers/setup-db.ts +52 -52
- package/tests/integration/code-flow.test.ts +86 -86
- package/tests/integration/error-flow.test.ts +83 -83
- package/tests/integration/ipc-flow.test.ts +166 -166
- package/tests/integration/learning-cycle.test.ts +82 -82
- package/tests/integration/synapse-flow.test.ts +117 -117
- package/tests/unit/code/analyzer.test.ts +58 -58
- package/tests/unit/code/fingerprint.test.ts +51 -51
- package/tests/unit/code/scorer.test.ts +55 -55
- package/tests/unit/learning/confidence-scorer.test.ts +60 -60
- package/tests/unit/learning/decay.test.ts +45 -45
- package/tests/unit/learning/pattern-extractor.test.ts +50 -50
- package/tests/unit/matching/error-matcher.test.ts +69 -69
- package/tests/unit/matching/fingerprint.test.ts +47 -47
- package/tests/unit/matching/similarity.test.ts +65 -65
- package/tests/unit/matching/tfidf.test.ts +71 -71
- package/tests/unit/matching/tokenizer.test.ts +83 -83
- package/tests/unit/parsing/parsers.test.ts +113 -113
- package/tests/unit/research/gap-analyzer.test.ts +45 -45
- package/tests/unit/research/trend-analyzer.test.ts +45 -45
- package/tests/unit/synapses/activation.test.ts +80 -80
- package/tests/unit/synapses/decay.test.ts +27 -27
- package/tests/unit/synapses/hebbian.test.ts +96 -96
- package/tests/unit/synapses/pathfinder.test.ts +72 -72
- package/tsconfig.json +18 -18
package/src/mcp/auto-detect.ts
CHANGED
|
@@ -1,93 +1,93 @@
|
|
|
1
|
-
import { IpcClient } from '../ipc/client.js';
|
|
2
|
-
import { getPipeName } from '../utils/paths.js';
|
|
3
|
-
|
|
4
|
-
interface HookInput {
|
|
5
|
-
tool_name: string;
|
|
6
|
-
tool_input: { command?: string; file_path?: string };
|
|
7
|
-
tool_output: string;
|
|
8
|
-
exit_code?: number;
|
|
9
|
-
}
|
|
10
|
-
|
|
11
|
-
const ERROR_PATTERNS = [
|
|
12
|
-
/Error:/i,
|
|
13
|
-
/error\[E\d+\]/, // Rust errors
|
|
14
|
-
/Traceback \(most recent call last\)/, // Python
|
|
15
|
-
/FATAL|PANIC/i,
|
|
16
|
-
/npm ERR!/,
|
|
17
|
-
/SyntaxError|TypeError|ReferenceError|RangeError/,
|
|
18
|
-
/ENOENT|EACCES|ECONNREFUSED|ETIMEDOUT/,
|
|
19
|
-
/ModuleNotFoundError|ImportError/,
|
|
20
|
-
/failed to compile/i,
|
|
21
|
-
/BUILD FAILED/i,
|
|
22
|
-
/Cannot find module/,
|
|
23
|
-
/command not found/,
|
|
24
|
-
/Permission denied/,
|
|
25
|
-
];
|
|
26
|
-
|
|
27
|
-
function isError(input: HookInput): boolean {
|
|
28
|
-
if (input.exit_code !== undefined && input.exit_code !== 0) return true;
|
|
29
|
-
return ERROR_PATTERNS.some(p => p.test(input.tool_output));
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
function readStdin(): Promise<string> {
|
|
33
|
-
return new Promise((resolve) => {
|
|
34
|
-
let data = '';
|
|
35
|
-
process.stdin.setEncoding('utf8');
|
|
36
|
-
process.stdin.on('data', (chunk) => { data += chunk; });
|
|
37
|
-
process.stdin.on('end', () => resolve(data));
|
|
38
|
-
});
|
|
39
|
-
}
|
|
40
|
-
|
|
41
|
-
async function main(): Promise<void> {
|
|
42
|
-
const raw = await readStdin();
|
|
43
|
-
if (!raw.trim()) return;
|
|
44
|
-
|
|
45
|
-
let input: HookInput;
|
|
46
|
-
try {
|
|
47
|
-
input = JSON.parse(raw);
|
|
48
|
-
} catch {
|
|
49
|
-
return;
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
if (!isError(input)) return;
|
|
53
|
-
|
|
54
|
-
const client = new IpcClient(getPipeName(), 3000);
|
|
55
|
-
try {
|
|
56
|
-
await client.connect();
|
|
57
|
-
|
|
58
|
-
// Report error to Brain
|
|
59
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
60
|
-
const result: any = await client.request('error.report', {
|
|
61
|
-
project: 'auto-detected',
|
|
62
|
-
errorOutput: input.tool_output,
|
|
63
|
-
command: input.tool_input.command,
|
|
64
|
-
autoDetected: true,
|
|
65
|
-
});
|
|
66
|
-
|
|
67
|
-
// If Brain knows a solution, output as feedback
|
|
68
|
-
if (result.matches?.length > 0) {
|
|
69
|
-
const best = result.matches[0];
|
|
70
|
-
process.stderr.write(`Brain: Similar error found (#${best.errorId}, ${Math.round(best.score * 100)}% match)\n`);
|
|
71
|
-
if (best.solutions?.length > 0) {
|
|
72
|
-
process.stderr.write(`Brain: Solution available — use brain_query_error to see details\n`);
|
|
73
|
-
}
|
|
74
|
-
}
|
|
75
|
-
|
|
76
|
-
// Anti-pattern check
|
|
77
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
78
|
-
const antipatterns: any = await client.request('prevention.antipatterns', {
|
|
79
|
-
errorType: '',
|
|
80
|
-
message: input.tool_output,
|
|
81
|
-
});
|
|
82
|
-
if (antipatterns?.length > 0 && antipatterns[0].matched) {
|
|
83
|
-
process.stderr.write(`Brain WARNING: Known anti-pattern detected: ${antipatterns[0].description}\n`);
|
|
84
|
-
}
|
|
85
|
-
|
|
86
|
-
} catch {
|
|
87
|
-
// Hook must never block the workflow — silent failure
|
|
88
|
-
} finally {
|
|
89
|
-
client.disconnect();
|
|
90
|
-
}
|
|
91
|
-
}
|
|
92
|
-
|
|
93
|
-
main();
|
|
1
|
+
import { IpcClient } from '../ipc/client.js';
|
|
2
|
+
import { getPipeName } from '../utils/paths.js';
|
|
3
|
+
|
|
4
|
+
interface HookInput {
|
|
5
|
+
tool_name: string;
|
|
6
|
+
tool_input: { command?: string; file_path?: string };
|
|
7
|
+
tool_output: string;
|
|
8
|
+
exit_code?: number;
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
const ERROR_PATTERNS = [
|
|
12
|
+
/Error:/i,
|
|
13
|
+
/error\[E\d+\]/, // Rust errors
|
|
14
|
+
/Traceback \(most recent call last\)/, // Python
|
|
15
|
+
/FATAL|PANIC/i,
|
|
16
|
+
/npm ERR!/,
|
|
17
|
+
/SyntaxError|TypeError|ReferenceError|RangeError/,
|
|
18
|
+
/ENOENT|EACCES|ECONNREFUSED|ETIMEDOUT/,
|
|
19
|
+
/ModuleNotFoundError|ImportError/,
|
|
20
|
+
/failed to compile/i,
|
|
21
|
+
/BUILD FAILED/i,
|
|
22
|
+
/Cannot find module/,
|
|
23
|
+
/command not found/,
|
|
24
|
+
/Permission denied/,
|
|
25
|
+
];
|
|
26
|
+
|
|
27
|
+
function isError(input: HookInput): boolean {
|
|
28
|
+
if (input.exit_code !== undefined && input.exit_code !== 0) return true;
|
|
29
|
+
return ERROR_PATTERNS.some(p => p.test(input.tool_output));
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
function readStdin(): Promise<string> {
|
|
33
|
+
return new Promise((resolve) => {
|
|
34
|
+
let data = '';
|
|
35
|
+
process.stdin.setEncoding('utf8');
|
|
36
|
+
process.stdin.on('data', (chunk) => { data += chunk; });
|
|
37
|
+
process.stdin.on('end', () => resolve(data));
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
async function main(): Promise<void> {
|
|
42
|
+
const raw = await readStdin();
|
|
43
|
+
if (!raw.trim()) return;
|
|
44
|
+
|
|
45
|
+
let input: HookInput;
|
|
46
|
+
try {
|
|
47
|
+
input = JSON.parse(raw);
|
|
48
|
+
} catch {
|
|
49
|
+
return;
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
if (!isError(input)) return;
|
|
53
|
+
|
|
54
|
+
const client = new IpcClient(getPipeName(), 3000);
|
|
55
|
+
try {
|
|
56
|
+
await client.connect();
|
|
57
|
+
|
|
58
|
+
// Report error to Brain
|
|
59
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
60
|
+
const result: any = await client.request('error.report', {
|
|
61
|
+
project: 'auto-detected',
|
|
62
|
+
errorOutput: input.tool_output,
|
|
63
|
+
command: input.tool_input.command,
|
|
64
|
+
autoDetected: true,
|
|
65
|
+
});
|
|
66
|
+
|
|
67
|
+
// If Brain knows a solution, output as feedback
|
|
68
|
+
if (result.matches?.length > 0) {
|
|
69
|
+
const best = result.matches[0];
|
|
70
|
+
process.stderr.write(`Brain: Similar error found (#${best.errorId}, ${Math.round(best.score * 100)}% match)\n`);
|
|
71
|
+
if (best.solutions?.length > 0) {
|
|
72
|
+
process.stderr.write(`Brain: Solution available — use brain_query_error to see details\n`);
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
|
|
76
|
+
// Anti-pattern check
|
|
77
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
78
|
+
const antipatterns: any = await client.request('prevention.antipatterns', {
|
|
79
|
+
errorType: '',
|
|
80
|
+
message: input.tool_output,
|
|
81
|
+
});
|
|
82
|
+
if (antipatterns?.length > 0 && antipatterns[0].matched) {
|
|
83
|
+
process.stderr.write(`Brain WARNING: Known anti-pattern detected: ${antipatterns[0].description}\n`);
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
} catch {
|
|
87
|
+
// Hook must never block the workflow — silent failure
|
|
88
|
+
} finally {
|
|
89
|
+
client.disconnect();
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
main();
|
package/src/mcp/http-server.ts
CHANGED
|
@@ -1,140 +1,140 @@
|
|
|
1
|
-
import http from 'node:http';
|
|
2
|
-
import { randomUUID } from 'node:crypto';
|
|
3
|
-
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
4
|
-
import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
|
|
5
|
-
import { getLogger } from '../utils/logger.js';
|
|
6
|
-
import type { IpcRouter } from '../ipc/router.js';
|
|
7
|
-
import { registerToolsDirect } from './tools.js';
|
|
8
|
-
|
|
9
|
-
export class McpHttpServer {
|
|
10
|
-
private server: http.Server | null = null;
|
|
11
|
-
private transports = new Map<string, SSEServerTransport>();
|
|
12
|
-
private logger = getLogger();
|
|
13
|
-
|
|
14
|
-
constructor(
|
|
15
|
-
private port: number,
|
|
16
|
-
private router: IpcRouter,
|
|
17
|
-
) {}
|
|
18
|
-
|
|
19
|
-
start(): void {
|
|
20
|
-
this.server = http.createServer((req, res) => {
|
|
21
|
-
res.setHeader('Access-Control-Allow-Origin', '*');
|
|
22
|
-
res.setHeader('Access-Control-Allow-Methods', 'GET, POST, DELETE, OPTIONS');
|
|
23
|
-
res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
|
|
24
|
-
|
|
25
|
-
if (req.method === 'OPTIONS') {
|
|
26
|
-
res.writeHead(204);
|
|
27
|
-
res.end();
|
|
28
|
-
return;
|
|
29
|
-
}
|
|
30
|
-
|
|
31
|
-
const url = new URL(req.url ?? '/', `http://localhost:${this.port}`);
|
|
32
|
-
|
|
33
|
-
if (url.pathname === '/sse' && req.method === 'GET') {
|
|
34
|
-
this.handleSSE(res);
|
|
35
|
-
return;
|
|
36
|
-
}
|
|
37
|
-
|
|
38
|
-
if (url.pathname === '/messages' && req.method === 'POST') {
|
|
39
|
-
this.handleMessage(req, res, url);
|
|
40
|
-
return;
|
|
41
|
-
}
|
|
42
|
-
|
|
43
|
-
if (url.pathname === '/' && req.method === 'GET') {
|
|
44
|
-
res.writeHead(200, { 'Content-Type': 'application/json' });
|
|
45
|
-
res.end(JSON.stringify({
|
|
46
|
-
name: 'brain',
|
|
47
|
-
version: '1.8.0',
|
|
48
|
-
protocol: 'MCP',
|
|
49
|
-
transport: 'sse',
|
|
50
|
-
endpoints: {
|
|
51
|
-
sse: '/sse',
|
|
52
|
-
messages: '/messages',
|
|
53
|
-
},
|
|
54
|
-
clients: this.transports.size,
|
|
55
|
-
}));
|
|
56
|
-
return;
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
res.writeHead(404, { 'Content-Type': 'text/plain' });
|
|
60
|
-
res.end('Not Found');
|
|
61
|
-
});
|
|
62
|
-
|
|
63
|
-
this.server.listen(this.port, () => {
|
|
64
|
-
this.logger.info(`MCP HTTP server (SSE) started on http://localhost:${this.port}`);
|
|
65
|
-
});
|
|
66
|
-
}
|
|
67
|
-
|
|
68
|
-
stop(): void {
|
|
69
|
-
for (const transport of this.transports.values()) {
|
|
70
|
-
try { transport.close?.(); } catch { /* ignore */ }
|
|
71
|
-
}
|
|
72
|
-
this.transports.clear();
|
|
73
|
-
this.server?.close();
|
|
74
|
-
this.server = null;
|
|
75
|
-
this.logger.info('MCP HTTP server stopped');
|
|
76
|
-
}
|
|
77
|
-
|
|
78
|
-
getClientCount(): number {
|
|
79
|
-
return this.transports.size;
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
private async handleSSE(res: http.ServerResponse): Promise<void> {
|
|
83
|
-
try {
|
|
84
|
-
const transport = new SSEServerTransport('/messages', res);
|
|
85
|
-
const sessionId = transport.sessionId ?? randomUUID();
|
|
86
|
-
this.transports.set(sessionId, transport);
|
|
87
|
-
|
|
88
|
-
const server = new McpServer({
|
|
89
|
-
name: 'brain',
|
|
90
|
-
version: '1.8.0',
|
|
91
|
-
});
|
|
92
|
-
|
|
93
|
-
registerToolsDirect(server, this.router);
|
|
94
|
-
|
|
95
|
-
res.on('close', () => {
|
|
96
|
-
this.transports.delete(sessionId);
|
|
97
|
-
this.logger.debug(`MCP SSE client disconnected: ${sessionId}`);
|
|
98
|
-
});
|
|
99
|
-
|
|
100
|
-
await server.connect(transport);
|
|
101
|
-
this.logger.info(`MCP SSE client connected: ${sessionId}`);
|
|
102
|
-
} catch (err) {
|
|
103
|
-
this.logger.error('MCP SSE connection error:', err);
|
|
104
|
-
if (!res.headersSent) {
|
|
105
|
-
res.writeHead(500, { 'Content-Type': 'text/plain' });
|
|
106
|
-
res.end('Internal Server Error');
|
|
107
|
-
}
|
|
108
|
-
}
|
|
109
|
-
}
|
|
110
|
-
|
|
111
|
-
private async handleMessage(
|
|
112
|
-
req: http.IncomingMessage,
|
|
113
|
-
res: http.ServerResponse,
|
|
114
|
-
url: URL,
|
|
115
|
-
): Promise<void> {
|
|
116
|
-
try {
|
|
117
|
-
const sessionId = url.searchParams.get('sessionId');
|
|
118
|
-
if (!sessionId) {
|
|
119
|
-
res.writeHead(400, { 'Content-Type': 'text/plain' });
|
|
120
|
-
res.end('Missing sessionId parameter');
|
|
121
|
-
return;
|
|
122
|
-
}
|
|
123
|
-
|
|
124
|
-
const transport = this.transports.get(sessionId);
|
|
125
|
-
if (!transport) {
|
|
126
|
-
res.writeHead(404, { 'Content-Type': 'text/plain' });
|
|
127
|
-
res.end('Session not found. Connect to /sse first.');
|
|
128
|
-
return;
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
await transport.handlePostMessage(req, res);
|
|
132
|
-
} catch (err) {
|
|
133
|
-
this.logger.error('MCP message error:', err);
|
|
134
|
-
if (!res.headersSent) {
|
|
135
|
-
res.writeHead(500, { 'Content-Type': 'text/plain' });
|
|
136
|
-
res.end('Internal Server Error');
|
|
137
|
-
}
|
|
138
|
-
}
|
|
139
|
-
}
|
|
140
|
-
}
|
|
1
|
+
import http from 'node:http';
|
|
2
|
+
import { randomUUID } from 'node:crypto';
|
|
3
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
4
|
+
import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
|
|
5
|
+
import { getLogger } from '../utils/logger.js';
|
|
6
|
+
import type { IpcRouter } from '../ipc/router.js';
|
|
7
|
+
import { registerToolsDirect } from './tools.js';
|
|
8
|
+
|
|
9
|
+
export class McpHttpServer {
|
|
10
|
+
private server: http.Server | null = null;
|
|
11
|
+
private transports = new Map<string, SSEServerTransport>();
|
|
12
|
+
private logger = getLogger();
|
|
13
|
+
|
|
14
|
+
constructor(
|
|
15
|
+
private port: number,
|
|
16
|
+
private router: IpcRouter,
|
|
17
|
+
) {}
|
|
18
|
+
|
|
19
|
+
start(): void {
|
|
20
|
+
this.server = http.createServer((req, res) => {
|
|
21
|
+
res.setHeader('Access-Control-Allow-Origin', '*');
|
|
22
|
+
res.setHeader('Access-Control-Allow-Methods', 'GET, POST, DELETE, OPTIONS');
|
|
23
|
+
res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
|
|
24
|
+
|
|
25
|
+
if (req.method === 'OPTIONS') {
|
|
26
|
+
res.writeHead(204);
|
|
27
|
+
res.end();
|
|
28
|
+
return;
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
const url = new URL(req.url ?? '/', `http://localhost:${this.port}`);
|
|
32
|
+
|
|
33
|
+
if (url.pathname === '/sse' && req.method === 'GET') {
|
|
34
|
+
this.handleSSE(res);
|
|
35
|
+
return;
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
if (url.pathname === '/messages' && req.method === 'POST') {
|
|
39
|
+
this.handleMessage(req, res, url);
|
|
40
|
+
return;
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
if (url.pathname === '/' && req.method === 'GET') {
|
|
44
|
+
res.writeHead(200, { 'Content-Type': 'application/json' });
|
|
45
|
+
res.end(JSON.stringify({
|
|
46
|
+
name: 'brain',
|
|
47
|
+
version: '1.8.0',
|
|
48
|
+
protocol: 'MCP',
|
|
49
|
+
transport: 'sse',
|
|
50
|
+
endpoints: {
|
|
51
|
+
sse: '/sse',
|
|
52
|
+
messages: '/messages',
|
|
53
|
+
},
|
|
54
|
+
clients: this.transports.size,
|
|
55
|
+
}));
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
res.writeHead(404, { 'Content-Type': 'text/plain' });
|
|
60
|
+
res.end('Not Found');
|
|
61
|
+
});
|
|
62
|
+
|
|
63
|
+
this.server.listen(this.port, () => {
|
|
64
|
+
this.logger.info(`MCP HTTP server (SSE) started on http://localhost:${this.port}`);
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
|
|
68
|
+
stop(): void {
|
|
69
|
+
for (const transport of this.transports.values()) {
|
|
70
|
+
try { transport.close?.(); } catch { /* ignore */ }
|
|
71
|
+
}
|
|
72
|
+
this.transports.clear();
|
|
73
|
+
this.server?.close();
|
|
74
|
+
this.server = null;
|
|
75
|
+
this.logger.info('MCP HTTP server stopped');
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
getClientCount(): number {
|
|
79
|
+
return this.transports.size;
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
private async handleSSE(res: http.ServerResponse): Promise<void> {
|
|
83
|
+
try {
|
|
84
|
+
const transport = new SSEServerTransport('/messages', res);
|
|
85
|
+
const sessionId = transport.sessionId ?? randomUUID();
|
|
86
|
+
this.transports.set(sessionId, transport);
|
|
87
|
+
|
|
88
|
+
const server = new McpServer({
|
|
89
|
+
name: 'brain',
|
|
90
|
+
version: '1.8.0',
|
|
91
|
+
});
|
|
92
|
+
|
|
93
|
+
registerToolsDirect(server, this.router);
|
|
94
|
+
|
|
95
|
+
res.on('close', () => {
|
|
96
|
+
this.transports.delete(sessionId);
|
|
97
|
+
this.logger.debug(`MCP SSE client disconnected: ${sessionId}`);
|
|
98
|
+
});
|
|
99
|
+
|
|
100
|
+
await server.connect(transport);
|
|
101
|
+
this.logger.info(`MCP SSE client connected: ${sessionId}`);
|
|
102
|
+
} catch (err) {
|
|
103
|
+
this.logger.error('MCP SSE connection error:', err);
|
|
104
|
+
if (!res.headersSent) {
|
|
105
|
+
res.writeHead(500, { 'Content-Type': 'text/plain' });
|
|
106
|
+
res.end('Internal Server Error');
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
private async handleMessage(
|
|
112
|
+
req: http.IncomingMessage,
|
|
113
|
+
res: http.ServerResponse,
|
|
114
|
+
url: URL,
|
|
115
|
+
): Promise<void> {
|
|
116
|
+
try {
|
|
117
|
+
const sessionId = url.searchParams.get('sessionId');
|
|
118
|
+
if (!sessionId) {
|
|
119
|
+
res.writeHead(400, { 'Content-Type': 'text/plain' });
|
|
120
|
+
res.end('Missing sessionId parameter');
|
|
121
|
+
return;
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
const transport = this.transports.get(sessionId);
|
|
125
|
+
if (!transport) {
|
|
126
|
+
res.writeHead(404, { 'Content-Type': 'text/plain' });
|
|
127
|
+
res.end('Session not found. Connect to /sse first.');
|
|
128
|
+
return;
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
await transport.handlePostMessage(req, res);
|
|
132
|
+
} catch (err) {
|
|
133
|
+
this.logger.error('MCP message error:', err);
|
|
134
|
+
if (!res.headersSent) {
|
|
135
|
+
res.writeHead(500, { 'Content-Type': 'text/plain' });
|
|
136
|
+
res.end('Internal Server Error');
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
}
|
package/src/mcp/server.ts
CHANGED
|
@@ -1,73 +1,73 @@
|
|
|
1
|
-
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
|
-
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
3
|
-
import { spawn } from 'node:child_process';
|
|
4
|
-
import path from 'node:path';
|
|
5
|
-
import { IpcClient } from '../ipc/client.js';
|
|
6
|
-
import { getPipeName } from '../utils/paths.js';
|
|
7
|
-
import { registerTools } from './tools.js';
|
|
8
|
-
|
|
9
|
-
function spawnDaemon(): void {
|
|
10
|
-
const entryPoint = path.resolve(import.meta.dirname, '../index.ts');
|
|
11
|
-
const child = spawn(process.execPath, [
|
|
12
|
-
'--import', 'tsx',
|
|
13
|
-
entryPoint, 'daemon',
|
|
14
|
-
], {
|
|
15
|
-
detached: true,
|
|
16
|
-
stdio: 'ignore',
|
|
17
|
-
cwd: path.resolve(import.meta.dirname, '../..'),
|
|
18
|
-
});
|
|
19
|
-
child.unref();
|
|
20
|
-
process.stderr.write(`Brain: Auto-started daemon (PID: ${child.pid})\n`);
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
async function connectWithRetry(ipc: IpcClient, retries: number, delayMs: number): Promise<void> {
|
|
24
|
-
for (let i = 0; i < retries; i++) {
|
|
25
|
-
try {
|
|
26
|
-
await ipc.connect();
|
|
27
|
-
return;
|
|
28
|
-
} catch {
|
|
29
|
-
if (i < retries - 1) {
|
|
30
|
-
await new Promise(r => setTimeout(r, delayMs));
|
|
31
|
-
}
|
|
32
|
-
}
|
|
33
|
-
}
|
|
34
|
-
throw new Error('Could not connect to daemon after retries');
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
export async function startMcpServer(): Promise<void> {
|
|
38
|
-
const server = new McpServer({
|
|
39
|
-
name: 'brain',
|
|
40
|
-
version: '1.0.0',
|
|
41
|
-
});
|
|
42
|
-
|
|
43
|
-
const ipc = new IpcClient(getPipeName());
|
|
44
|
-
|
|
45
|
-
try {
|
|
46
|
-
await ipc.connect();
|
|
47
|
-
} catch {
|
|
48
|
-
// Daemon not running — auto-start it
|
|
49
|
-
process.stderr.write('Brain: Daemon not running, starting automatically...\n');
|
|
50
|
-
spawnDaemon();
|
|
51
|
-
try {
|
|
52
|
-
await connectWithRetry(ipc, 10, 500);
|
|
53
|
-
} catch {
|
|
54
|
-
process.stderr.write('Brain: Could not connect to daemon after auto-start. Check logs.\n');
|
|
55
|
-
process.exit(1);
|
|
56
|
-
}
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
registerTools(server, ipc);
|
|
60
|
-
|
|
61
|
-
const transport = new StdioServerTransport();
|
|
62
|
-
await server.connect(transport);
|
|
63
|
-
|
|
64
|
-
process.on('SIGINT', () => {
|
|
65
|
-
ipc.disconnect();
|
|
66
|
-
process.exit(0);
|
|
67
|
-
});
|
|
68
|
-
|
|
69
|
-
process.on('SIGTERM', () => {
|
|
70
|
-
ipc.disconnect();
|
|
71
|
-
process.exit(0);
|
|
72
|
-
});
|
|
73
|
-
}
|
|
1
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
|
+
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
3
|
+
import { spawn } from 'node:child_process';
|
|
4
|
+
import path from 'node:path';
|
|
5
|
+
import { IpcClient } from '../ipc/client.js';
|
|
6
|
+
import { getPipeName } from '../utils/paths.js';
|
|
7
|
+
import { registerTools } from './tools.js';
|
|
8
|
+
|
|
9
|
+
function spawnDaemon(): void {
|
|
10
|
+
const entryPoint = path.resolve(import.meta.dirname, '../index.ts');
|
|
11
|
+
const child = spawn(process.execPath, [
|
|
12
|
+
'--import', 'tsx',
|
|
13
|
+
entryPoint, 'daemon',
|
|
14
|
+
], {
|
|
15
|
+
detached: true,
|
|
16
|
+
stdio: 'ignore',
|
|
17
|
+
cwd: path.resolve(import.meta.dirname, '../..'),
|
|
18
|
+
});
|
|
19
|
+
child.unref();
|
|
20
|
+
process.stderr.write(`Brain: Auto-started daemon (PID: ${child.pid})\n`);
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
async function connectWithRetry(ipc: IpcClient, retries: number, delayMs: number): Promise<void> {
|
|
24
|
+
for (let i = 0; i < retries; i++) {
|
|
25
|
+
try {
|
|
26
|
+
await ipc.connect();
|
|
27
|
+
return;
|
|
28
|
+
} catch {
|
|
29
|
+
if (i < retries - 1) {
|
|
30
|
+
await new Promise(r => setTimeout(r, delayMs));
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
throw new Error('Could not connect to daemon after retries');
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
export async function startMcpServer(): Promise<void> {
|
|
38
|
+
const server = new McpServer({
|
|
39
|
+
name: 'brain',
|
|
40
|
+
version: '1.0.0',
|
|
41
|
+
});
|
|
42
|
+
|
|
43
|
+
const ipc = new IpcClient(getPipeName());
|
|
44
|
+
|
|
45
|
+
try {
|
|
46
|
+
await ipc.connect();
|
|
47
|
+
} catch {
|
|
48
|
+
// Daemon not running — auto-start it
|
|
49
|
+
process.stderr.write('Brain: Daemon not running, starting automatically...\n');
|
|
50
|
+
spawnDaemon();
|
|
51
|
+
try {
|
|
52
|
+
await connectWithRetry(ipc, 10, 500);
|
|
53
|
+
} catch {
|
|
54
|
+
process.stderr.write('Brain: Could not connect to daemon after auto-start. Check logs.\n');
|
|
55
|
+
process.exit(1);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
registerTools(server, ipc);
|
|
60
|
+
|
|
61
|
+
const transport = new StdioServerTransport();
|
|
62
|
+
await server.connect(transport);
|
|
63
|
+
|
|
64
|
+
process.on('SIGINT', () => {
|
|
65
|
+
ipc.disconnect();
|
|
66
|
+
process.exit(0);
|
|
67
|
+
});
|
|
68
|
+
|
|
69
|
+
process.on('SIGTERM', () => {
|
|
70
|
+
ipc.disconnect();
|
|
71
|
+
process.exit(0);
|
|
72
|
+
});
|
|
73
|
+
}
|
|
@@ -1,28 +1,28 @@
|
|
|
1
|
-
import { ErrorParserRegistry } from './types.js';
|
|
2
|
-
import { nodeParser } from './parsers/node.js';
|
|
3
|
-
import { pythonParser } from './parsers/python.js';
|
|
4
|
-
import { rustParser } from './parsers/rust.js';
|
|
5
|
-
import { goParser } from './parsers/go.js';
|
|
6
|
-
import { shellParser } from './parsers/shell.js';
|
|
7
|
-
import { compilerParser } from './parsers/compiler.js';
|
|
8
|
-
import { genericParser } from './parsers/generic.js';
|
|
9
|
-
|
|
10
|
-
let registryInstance: ErrorParserRegistry | null = null;
|
|
11
|
-
|
|
12
|
-
export function getParserRegistry(): ErrorParserRegistry {
|
|
13
|
-
if (!registryInstance) {
|
|
14
|
-
registryInstance = new ErrorParserRegistry();
|
|
15
|
-
registryInstance.register(nodeParser);
|
|
16
|
-
registryInstance.register(pythonParser);
|
|
17
|
-
registryInstance.register(rustParser);
|
|
18
|
-
registryInstance.register(goParser);
|
|
19
|
-
registryInstance.register(shellParser);
|
|
20
|
-
registryInstance.register(compilerParser);
|
|
21
|
-
registryInstance.register(genericParser);
|
|
22
|
-
}
|
|
23
|
-
return registryInstance;
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
export function parseError(input: string) {
|
|
27
|
-
return getParserRegistry().parse(input);
|
|
28
|
-
}
|
|
1
|
+
import { ErrorParserRegistry } from './types.js';
|
|
2
|
+
import { nodeParser } from './parsers/node.js';
|
|
3
|
+
import { pythonParser } from './parsers/python.js';
|
|
4
|
+
import { rustParser } from './parsers/rust.js';
|
|
5
|
+
import { goParser } from './parsers/go.js';
|
|
6
|
+
import { shellParser } from './parsers/shell.js';
|
|
7
|
+
import { compilerParser } from './parsers/compiler.js';
|
|
8
|
+
import { genericParser } from './parsers/generic.js';
|
|
9
|
+
|
|
10
|
+
let registryInstance: ErrorParserRegistry | null = null;
|
|
11
|
+
|
|
12
|
+
export function getParserRegistry(): ErrorParserRegistry {
|
|
13
|
+
if (!registryInstance) {
|
|
14
|
+
registryInstance = new ErrorParserRegistry();
|
|
15
|
+
registryInstance.register(nodeParser);
|
|
16
|
+
registryInstance.register(pythonParser);
|
|
17
|
+
registryInstance.register(rustParser);
|
|
18
|
+
registryInstance.register(goParser);
|
|
19
|
+
registryInstance.register(shellParser);
|
|
20
|
+
registryInstance.register(compilerParser);
|
|
21
|
+
registryInstance.register(genericParser);
|
|
22
|
+
}
|
|
23
|
+
return registryInstance;
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
export function parseError(input: string) {
|
|
27
|
+
return getParserRegistry().parse(input);
|
|
28
|
+
}
|