@timmeck/brain 1.8.1 → 1.8.2

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.
Files changed (162) hide show
  1. package/BRAIN_PLAN.md +3324 -3324
  2. package/LICENSE +21 -21
  3. package/dist/cli/commands/dashboard.js +595 -595
  4. package/dist/dashboard/server.js +25 -25
  5. package/dist/db/migrations/001_core_schema.js +115 -115
  6. package/dist/db/migrations/002_learning_schema.js +33 -33
  7. package/dist/db/migrations/003_code_schema.js +48 -48
  8. package/dist/db/migrations/004_synapses_schema.js +52 -52
  9. package/dist/db/migrations/005_fts_indexes.js +73 -73
  10. package/dist/db/migrations/007_feedback.js +8 -8
  11. package/dist/db/migrations/008_git_integration.js +33 -33
  12. package/dist/db/migrations/009_embeddings.js +3 -3
  13. package/dist/db/repositories/antipattern.repository.js +3 -3
  14. package/dist/db/repositories/code-module.repository.js +32 -32
  15. package/dist/db/repositories/notification.repository.js +3 -3
  16. package/dist/db/repositories/project.repository.js +21 -21
  17. package/dist/db/repositories/rule.repository.js +24 -24
  18. package/dist/db/repositories/solution.repository.js +50 -50
  19. package/dist/db/repositories/synapse.repository.js +18 -18
  20. package/dist/db/repositories/terminal.repository.js +24 -24
  21. package/dist/ipc/server.d.ts +8 -0
  22. package/dist/ipc/server.js +67 -1
  23. package/dist/ipc/server.js.map +1 -1
  24. package/dist/matching/error-matcher.js +5 -5
  25. package/dist/matching/fingerprint.js +6 -1
  26. package/dist/matching/fingerprint.js.map +1 -1
  27. package/dist/services/error.service.js +4 -3
  28. package/dist/services/error.service.js.map +1 -1
  29. package/dist/services/git.service.js +14 -14
  30. package/package.json +49 -49
  31. package/src/api/server.ts +395 -395
  32. package/src/brain.ts +266 -266
  33. package/src/cli/colors.ts +116 -116
  34. package/src/cli/commands/config.ts +169 -169
  35. package/src/cli/commands/dashboard.ts +755 -755
  36. package/src/cli/commands/doctor.ts +118 -118
  37. package/src/cli/commands/explain.ts +83 -83
  38. package/src/cli/commands/export.ts +31 -31
  39. package/src/cli/commands/import.ts +199 -199
  40. package/src/cli/commands/insights.ts +65 -65
  41. package/src/cli/commands/learn.ts +24 -24
  42. package/src/cli/commands/modules.ts +53 -53
  43. package/src/cli/commands/network.ts +67 -67
  44. package/src/cli/commands/projects.ts +42 -42
  45. package/src/cli/commands/query.ts +120 -120
  46. package/src/cli/commands/start.ts +62 -62
  47. package/src/cli/commands/status.ts +75 -75
  48. package/src/cli/commands/stop.ts +34 -34
  49. package/src/cli/ipc-helper.ts +22 -22
  50. package/src/cli/update-check.ts +63 -63
  51. package/src/code/fingerprint.ts +87 -87
  52. package/src/code/parsers/generic.ts +29 -29
  53. package/src/code/parsers/python.ts +54 -54
  54. package/src/code/parsers/typescript.ts +65 -65
  55. package/src/code/registry.ts +60 -60
  56. package/src/dashboard/server.ts +142 -142
  57. package/src/db/connection.ts +22 -22
  58. package/src/db/migrations/001_core_schema.ts +120 -120
  59. package/src/db/migrations/002_learning_schema.ts +38 -38
  60. package/src/db/migrations/003_code_schema.ts +53 -53
  61. package/src/db/migrations/004_synapses_schema.ts +57 -57
  62. package/src/db/migrations/005_fts_indexes.ts +78 -78
  63. package/src/db/migrations/006_synapses_phase3.ts +17 -17
  64. package/src/db/migrations/007_feedback.ts +13 -13
  65. package/src/db/migrations/008_git_integration.ts +38 -38
  66. package/src/db/migrations/009_embeddings.ts +8 -8
  67. package/src/db/repositories/antipattern.repository.ts +66 -66
  68. package/src/db/repositories/code-module.repository.ts +142 -142
  69. package/src/db/repositories/notification.repository.ts +66 -66
  70. package/src/db/repositories/project.repository.ts +93 -93
  71. package/src/db/repositories/rule.repository.ts +108 -108
  72. package/src/db/repositories/solution.repository.ts +154 -154
  73. package/src/db/repositories/synapse.repository.ts +153 -153
  74. package/src/db/repositories/terminal.repository.ts +101 -101
  75. package/src/embeddings/engine.ts +238 -238
  76. package/src/index.ts +63 -63
  77. package/src/ipc/client.ts +118 -118
  78. package/src/ipc/protocol.ts +35 -35
  79. package/src/ipc/router.ts +133 -133
  80. package/src/ipc/server.ts +176 -110
  81. package/src/learning/decay.ts +46 -46
  82. package/src/learning/pattern-extractor.ts +90 -90
  83. package/src/learning/rule-generator.ts +74 -74
  84. package/src/matching/error-matcher.ts +5 -5
  85. package/src/matching/fingerprint.ts +34 -29
  86. package/src/matching/similarity.ts +61 -61
  87. package/src/matching/tfidf.ts +74 -74
  88. package/src/matching/tokenizer.ts +41 -41
  89. package/src/mcp/auto-detect.ts +93 -93
  90. package/src/mcp/http-server.ts +140 -140
  91. package/src/mcp/server.ts +73 -73
  92. package/src/parsing/error-parser.ts +28 -28
  93. package/src/parsing/parsers/compiler.ts +93 -93
  94. package/src/parsing/parsers/generic.ts +28 -28
  95. package/src/parsing/parsers/go.ts +97 -97
  96. package/src/parsing/parsers/node.ts +69 -69
  97. package/src/parsing/parsers/python.ts +62 -62
  98. package/src/parsing/parsers/rust.ts +50 -50
  99. package/src/parsing/parsers/shell.ts +42 -42
  100. package/src/parsing/types.ts +47 -47
  101. package/src/research/gap-analyzer.ts +135 -135
  102. package/src/research/insight-generator.ts +123 -123
  103. package/src/research/research-engine.ts +116 -116
  104. package/src/research/synergy-detector.ts +126 -126
  105. package/src/research/template-extractor.ts +130 -130
  106. package/src/research/trend-analyzer.ts +127 -127
  107. package/src/services/code.service.ts +271 -271
  108. package/src/services/error.service.ts +4 -3
  109. package/src/services/git.service.ts +132 -132
  110. package/src/services/notification.service.ts +41 -41
  111. package/src/services/synapse.service.ts +59 -59
  112. package/src/services/terminal.service.ts +81 -81
  113. package/src/synapses/activation.ts +80 -80
  114. package/src/synapses/decay.ts +38 -38
  115. package/src/synapses/hebbian.ts +69 -69
  116. package/src/synapses/pathfinder.ts +81 -81
  117. package/src/synapses/synapse-manager.ts +109 -109
  118. package/src/types/code.types.ts +52 -52
  119. package/src/types/error.types.ts +67 -67
  120. package/src/types/ipc.types.ts +8 -8
  121. package/src/types/mcp.types.ts +53 -53
  122. package/src/types/research.types.ts +28 -28
  123. package/src/types/solution.types.ts +30 -30
  124. package/src/utils/events.ts +45 -45
  125. package/src/utils/hash.ts +5 -5
  126. package/src/utils/logger.ts +48 -48
  127. package/src/utils/paths.ts +19 -19
  128. package/tests/e2e/test_code_intelligence.py +1015 -0
  129. package/tests/e2e/test_error_memory.py +451 -0
  130. package/tests/e2e/test_full_integration.py +534 -0
  131. package/tests/fixtures/code-modules/modules.ts +83 -83
  132. package/tests/fixtures/errors/go.ts +9 -9
  133. package/tests/fixtures/errors/node.ts +24 -24
  134. package/tests/fixtures/errors/python.ts +21 -21
  135. package/tests/fixtures/errors/rust.ts +25 -25
  136. package/tests/fixtures/errors/shell.ts +15 -15
  137. package/tests/fixtures/solutions/solutions.ts +27 -27
  138. package/tests/helpers/setup-db.ts +52 -52
  139. package/tests/integration/code-flow.test.ts +86 -86
  140. package/tests/integration/error-flow.test.ts +83 -83
  141. package/tests/integration/ipc-flow.test.ts +166 -166
  142. package/tests/integration/learning-cycle.test.ts +82 -82
  143. package/tests/integration/synapse-flow.test.ts +117 -117
  144. package/tests/unit/code/analyzer.test.ts +58 -58
  145. package/tests/unit/code/fingerprint.test.ts +51 -51
  146. package/tests/unit/code/scorer.test.ts +55 -55
  147. package/tests/unit/learning/confidence-scorer.test.ts +60 -60
  148. package/tests/unit/learning/decay.test.ts +45 -45
  149. package/tests/unit/learning/pattern-extractor.test.ts +50 -50
  150. package/tests/unit/matching/error-matcher.test.ts +69 -69
  151. package/tests/unit/matching/fingerprint.test.ts +47 -47
  152. package/tests/unit/matching/similarity.test.ts +65 -65
  153. package/tests/unit/matching/tfidf.test.ts +71 -71
  154. package/tests/unit/matching/tokenizer.test.ts +83 -83
  155. package/tests/unit/parsing/parsers.test.ts +113 -113
  156. package/tests/unit/research/gap-analyzer.test.ts +45 -45
  157. package/tests/unit/research/trend-analyzer.test.ts +45 -45
  158. package/tests/unit/synapses/activation.test.ts +80 -80
  159. package/tests/unit/synapses/decay.test.ts +27 -27
  160. package/tests/unit/synapses/hebbian.test.ts +96 -96
  161. package/tests/unit/synapses/pathfinder.test.ts +72 -72
  162. package/tsconfig.json +18 -18
@@ -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();
@@ -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
+ }