@timmeck/brain 1.8.0 → 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.
- package/BRAIN_PLAN.md +3324 -3324
- package/LICENSE +21 -21
- package/dist/api/server.d.ts +4 -0
- package/dist/api/server.js +73 -0
- package/dist/api/server.js.map +1 -1
- package/dist/brain.js +2 -1
- package/dist/brain.js.map +1 -1
- package/dist/cli/commands/dashboard.js +606 -572
- package/dist/cli/commands/dashboard.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/embeddings/engine.d.ts +2 -2
- package/dist/embeddings/engine.js +17 -4
- package/dist/embeddings/engine.js.map +1 -1
- package/dist/index.js +1 -1
- 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/mcp/http-server.js +8 -2
- package/dist/mcp/http-server.js.map +1 -1
- package/dist/services/code.service.d.ts +3 -0
- package/dist/services/code.service.js +33 -4
- package/dist/services/code.service.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 -321
- package/src/brain.ts +266 -265
- package/src/cli/colors.ts +116 -116
- package/src/cli/commands/config.ts +169 -169
- package/src/cli/commands/dashboard.ts +755 -720
- package/src/cli/commands/doctor.ts +118 -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 -217
- 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 -137
- 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 -238
- 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/index.ts
CHANGED
|
@@ -1,63 +1,63 @@
|
|
|
1
|
-
#!/usr/bin/env node
|
|
2
|
-
|
|
3
|
-
import { Command } from 'commander';
|
|
4
|
-
import { startCommand } from './cli/commands/start.js';
|
|
5
|
-
import { stopCommand } from './cli/commands/stop.js';
|
|
6
|
-
import { statusCommand } from './cli/commands/status.js';
|
|
7
|
-
import { queryCommand } from './cli/commands/query.js';
|
|
8
|
-
import { modulesCommand } from './cli/commands/modules.js';
|
|
9
|
-
import { insightsCommand } from './cli/commands/insights.js';
|
|
10
|
-
import { networkCommand } from './cli/commands/network.js';
|
|
11
|
-
import { exportCommand } from './cli/commands/export.js';
|
|
12
|
-
import { importCommand } from './cli/commands/import.js';
|
|
13
|
-
import { dashboardCommand } from './cli/commands/dashboard.js';
|
|
14
|
-
import { learnCommand } from './cli/commands/learn.js';
|
|
15
|
-
import { configCommand } from './cli/commands/config.js';
|
|
16
|
-
import { projectsCommand } from './cli/commands/projects.js';
|
|
17
|
-
import { doctorCommand } from './cli/commands/doctor.js';
|
|
18
|
-
import { explainCommand } from './cli/commands/explain.js';
|
|
19
|
-
|
|
20
|
-
const program = new Command();
|
|
21
|
-
|
|
22
|
-
program
|
|
23
|
-
.name('brain')
|
|
24
|
-
.description('Brain — Adaptive Error Memory & Code Intelligence System')
|
|
25
|
-
.version('1.8.
|
|
26
|
-
|
|
27
|
-
program.addCommand(startCommand());
|
|
28
|
-
program.addCommand(stopCommand());
|
|
29
|
-
program.addCommand(statusCommand());
|
|
30
|
-
program.addCommand(queryCommand());
|
|
31
|
-
program.addCommand(modulesCommand());
|
|
32
|
-
program.addCommand(insightsCommand());
|
|
33
|
-
program.addCommand(networkCommand());
|
|
34
|
-
program.addCommand(exportCommand());
|
|
35
|
-
program.addCommand(importCommand());
|
|
36
|
-
program.addCommand(dashboardCommand());
|
|
37
|
-
program.addCommand(learnCommand());
|
|
38
|
-
program.addCommand(configCommand());
|
|
39
|
-
program.addCommand(projectsCommand());
|
|
40
|
-
program.addCommand(doctorCommand());
|
|
41
|
-
program.addCommand(explainCommand());
|
|
42
|
-
|
|
43
|
-
// Hidden command: run MCP server (called by Claude Code)
|
|
44
|
-
program
|
|
45
|
-
.command('mcp-server')
|
|
46
|
-
.description('Start MCP server (stdio transport, used by Claude Code)')
|
|
47
|
-
.action(async () => {
|
|
48
|
-
const { startMcpServer } = await import('./mcp/server.js');
|
|
49
|
-
await startMcpServer();
|
|
50
|
-
});
|
|
51
|
-
|
|
52
|
-
// Hidden command: run daemon in foreground (called by start command)
|
|
53
|
-
program
|
|
54
|
-
.command('daemon')
|
|
55
|
-
.description('Run daemon in foreground')
|
|
56
|
-
.option('-c, --config <path>', 'Config file path')
|
|
57
|
-
.action(async (opts) => {
|
|
58
|
-
const { BrainCore } = await import('./brain.js');
|
|
59
|
-
const core = new BrainCore();
|
|
60
|
-
core.start(opts.config);
|
|
61
|
-
});
|
|
62
|
-
|
|
63
|
-
program.parse();
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
|
|
3
|
+
import { Command } from 'commander';
|
|
4
|
+
import { startCommand } from './cli/commands/start.js';
|
|
5
|
+
import { stopCommand } from './cli/commands/stop.js';
|
|
6
|
+
import { statusCommand } from './cli/commands/status.js';
|
|
7
|
+
import { queryCommand } from './cli/commands/query.js';
|
|
8
|
+
import { modulesCommand } from './cli/commands/modules.js';
|
|
9
|
+
import { insightsCommand } from './cli/commands/insights.js';
|
|
10
|
+
import { networkCommand } from './cli/commands/network.js';
|
|
11
|
+
import { exportCommand } from './cli/commands/export.js';
|
|
12
|
+
import { importCommand } from './cli/commands/import.js';
|
|
13
|
+
import { dashboardCommand } from './cli/commands/dashboard.js';
|
|
14
|
+
import { learnCommand } from './cli/commands/learn.js';
|
|
15
|
+
import { configCommand } from './cli/commands/config.js';
|
|
16
|
+
import { projectsCommand } from './cli/commands/projects.js';
|
|
17
|
+
import { doctorCommand } from './cli/commands/doctor.js';
|
|
18
|
+
import { explainCommand } from './cli/commands/explain.js';
|
|
19
|
+
|
|
20
|
+
const program = new Command();
|
|
21
|
+
|
|
22
|
+
program
|
|
23
|
+
.name('brain')
|
|
24
|
+
.description('Brain — Adaptive Error Memory & Code Intelligence System')
|
|
25
|
+
.version('1.8.1');
|
|
26
|
+
|
|
27
|
+
program.addCommand(startCommand());
|
|
28
|
+
program.addCommand(stopCommand());
|
|
29
|
+
program.addCommand(statusCommand());
|
|
30
|
+
program.addCommand(queryCommand());
|
|
31
|
+
program.addCommand(modulesCommand());
|
|
32
|
+
program.addCommand(insightsCommand());
|
|
33
|
+
program.addCommand(networkCommand());
|
|
34
|
+
program.addCommand(exportCommand());
|
|
35
|
+
program.addCommand(importCommand());
|
|
36
|
+
program.addCommand(dashboardCommand());
|
|
37
|
+
program.addCommand(learnCommand());
|
|
38
|
+
program.addCommand(configCommand());
|
|
39
|
+
program.addCommand(projectsCommand());
|
|
40
|
+
program.addCommand(doctorCommand());
|
|
41
|
+
program.addCommand(explainCommand());
|
|
42
|
+
|
|
43
|
+
// Hidden command: run MCP server (called by Claude Code)
|
|
44
|
+
program
|
|
45
|
+
.command('mcp-server')
|
|
46
|
+
.description('Start MCP server (stdio transport, used by Claude Code)')
|
|
47
|
+
.action(async () => {
|
|
48
|
+
const { startMcpServer } = await import('./mcp/server.js');
|
|
49
|
+
await startMcpServer();
|
|
50
|
+
});
|
|
51
|
+
|
|
52
|
+
// Hidden command: run daemon in foreground (called by start command)
|
|
53
|
+
program
|
|
54
|
+
.command('daemon')
|
|
55
|
+
.description('Run daemon in foreground')
|
|
56
|
+
.option('-c, --config <path>', 'Config file path')
|
|
57
|
+
.action(async (opts) => {
|
|
58
|
+
const { BrainCore } = await import('./brain.js');
|
|
59
|
+
const core = new BrainCore();
|
|
60
|
+
core.start(opts.config);
|
|
61
|
+
});
|
|
62
|
+
|
|
63
|
+
program.parse();
|
package/src/ipc/client.ts
CHANGED
|
@@ -1,118 +1,118 @@
|
|
|
1
|
-
import net from 'node:net';
|
|
2
|
-
import { randomUUID } from 'node:crypto';
|
|
3
|
-
import type { IpcMessage } from '../types/ipc.types.js';
|
|
4
|
-
import { encodeMessage, MessageDecoder } from './protocol.js';
|
|
5
|
-
import { getPipeName } from '../utils/paths.js';
|
|
6
|
-
|
|
7
|
-
interface PendingRequest {
|
|
8
|
-
resolve: (result: unknown) => void;
|
|
9
|
-
reject: (err: Error) => void;
|
|
10
|
-
timer: ReturnType<typeof setTimeout>;
|
|
11
|
-
}
|
|
12
|
-
|
|
13
|
-
export class IpcClient {
|
|
14
|
-
private socket: net.Socket | null = null;
|
|
15
|
-
private decoder = new MessageDecoder();
|
|
16
|
-
private pending = new Map<string, PendingRequest>();
|
|
17
|
-
private onNotification?: (msg: IpcMessage) => void;
|
|
18
|
-
|
|
19
|
-
constructor(
|
|
20
|
-
private pipeName: string = getPipeName(),
|
|
21
|
-
private timeout: number = 5000,
|
|
22
|
-
) {}
|
|
23
|
-
|
|
24
|
-
connect(): Promise<void> {
|
|
25
|
-
return new Promise((resolve, reject) => {
|
|
26
|
-
this.socket = net.createConnection(this.pipeName, () => {
|
|
27
|
-
resolve();
|
|
28
|
-
});
|
|
29
|
-
|
|
30
|
-
this.socket.on('data', (chunk) => {
|
|
31
|
-
const messages = this.decoder.feed(chunk);
|
|
32
|
-
for (const msg of messages) {
|
|
33
|
-
this.handleMessage(msg);
|
|
34
|
-
}
|
|
35
|
-
});
|
|
36
|
-
|
|
37
|
-
this.socket.on('error', (err) => {
|
|
38
|
-
reject(err);
|
|
39
|
-
// Reject all pending requests
|
|
40
|
-
for (const [id, req] of this.pending) {
|
|
41
|
-
clearTimeout(req.timer);
|
|
42
|
-
req.reject(new Error(`Connection error: ${err.message}`));
|
|
43
|
-
this.pending.delete(id);
|
|
44
|
-
}
|
|
45
|
-
});
|
|
46
|
-
|
|
47
|
-
this.socket.on('close', () => {
|
|
48
|
-
for (const [id, req] of this.pending) {
|
|
49
|
-
clearTimeout(req.timer);
|
|
50
|
-
req.reject(new Error('Connection closed'));
|
|
51
|
-
this.pending.delete(id);
|
|
52
|
-
}
|
|
53
|
-
this.socket = null;
|
|
54
|
-
});
|
|
55
|
-
});
|
|
56
|
-
}
|
|
57
|
-
|
|
58
|
-
request(method: string, params?: unknown): Promise<unknown> {
|
|
59
|
-
return new Promise((resolve, reject) => {
|
|
60
|
-
if (!this.socket || this.socket.destroyed) {
|
|
61
|
-
return reject(new Error('Not connected'));
|
|
62
|
-
}
|
|
63
|
-
|
|
64
|
-
const id = randomUUID();
|
|
65
|
-
const timer = setTimeout(() => {
|
|
66
|
-
this.pending.delete(id);
|
|
67
|
-
reject(new Error(`Request timeout: ${method} (${this.timeout}ms)`));
|
|
68
|
-
}, this.timeout);
|
|
69
|
-
|
|
70
|
-
this.pending.set(id, { resolve, reject, timer });
|
|
71
|
-
|
|
72
|
-
const msg: IpcMessage = {
|
|
73
|
-
id,
|
|
74
|
-
type: 'request',
|
|
75
|
-
method,
|
|
76
|
-
params,
|
|
77
|
-
};
|
|
78
|
-
this.socket.write(encodeMessage(msg));
|
|
79
|
-
});
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
setNotificationHandler(handler: (msg: IpcMessage) => void): void {
|
|
83
|
-
this.onNotification = handler;
|
|
84
|
-
}
|
|
85
|
-
|
|
86
|
-
disconnect(): void {
|
|
87
|
-
for (const [id, req] of this.pending) {
|
|
88
|
-
clearTimeout(req.timer);
|
|
89
|
-
req.reject(new Error('Client disconnecting'));
|
|
90
|
-
this.pending.delete(id);
|
|
91
|
-
}
|
|
92
|
-
this.socket?.destroy();
|
|
93
|
-
this.socket = null;
|
|
94
|
-
this.decoder.reset();
|
|
95
|
-
}
|
|
96
|
-
|
|
97
|
-
get connected(): boolean {
|
|
98
|
-
return this.socket !== null && !this.socket.destroyed;
|
|
99
|
-
}
|
|
100
|
-
|
|
101
|
-
private handleMessage(msg: IpcMessage): void {
|
|
102
|
-
if (msg.type === 'response') {
|
|
103
|
-
const req = this.pending.get(msg.id);
|
|
104
|
-
if (!req) return;
|
|
105
|
-
|
|
106
|
-
clearTimeout(req.timer);
|
|
107
|
-
this.pending.delete(msg.id);
|
|
108
|
-
|
|
109
|
-
if (msg.error) {
|
|
110
|
-
req.reject(new Error(msg.error.message));
|
|
111
|
-
} else {
|
|
112
|
-
req.resolve(msg.result);
|
|
113
|
-
}
|
|
114
|
-
} else if (msg.type === 'notification') {
|
|
115
|
-
this.onNotification?.(msg);
|
|
116
|
-
}
|
|
117
|
-
}
|
|
118
|
-
}
|
|
1
|
+
import net from 'node:net';
|
|
2
|
+
import { randomUUID } from 'node:crypto';
|
|
3
|
+
import type { IpcMessage } from '../types/ipc.types.js';
|
|
4
|
+
import { encodeMessage, MessageDecoder } from './protocol.js';
|
|
5
|
+
import { getPipeName } from '../utils/paths.js';
|
|
6
|
+
|
|
7
|
+
interface PendingRequest {
|
|
8
|
+
resolve: (result: unknown) => void;
|
|
9
|
+
reject: (err: Error) => void;
|
|
10
|
+
timer: ReturnType<typeof setTimeout>;
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
export class IpcClient {
|
|
14
|
+
private socket: net.Socket | null = null;
|
|
15
|
+
private decoder = new MessageDecoder();
|
|
16
|
+
private pending = new Map<string, PendingRequest>();
|
|
17
|
+
private onNotification?: (msg: IpcMessage) => void;
|
|
18
|
+
|
|
19
|
+
constructor(
|
|
20
|
+
private pipeName: string = getPipeName(),
|
|
21
|
+
private timeout: number = 5000,
|
|
22
|
+
) {}
|
|
23
|
+
|
|
24
|
+
connect(): Promise<void> {
|
|
25
|
+
return new Promise((resolve, reject) => {
|
|
26
|
+
this.socket = net.createConnection(this.pipeName, () => {
|
|
27
|
+
resolve();
|
|
28
|
+
});
|
|
29
|
+
|
|
30
|
+
this.socket.on('data', (chunk) => {
|
|
31
|
+
const messages = this.decoder.feed(chunk);
|
|
32
|
+
for (const msg of messages) {
|
|
33
|
+
this.handleMessage(msg);
|
|
34
|
+
}
|
|
35
|
+
});
|
|
36
|
+
|
|
37
|
+
this.socket.on('error', (err) => {
|
|
38
|
+
reject(err);
|
|
39
|
+
// Reject all pending requests
|
|
40
|
+
for (const [id, req] of this.pending) {
|
|
41
|
+
clearTimeout(req.timer);
|
|
42
|
+
req.reject(new Error(`Connection error: ${err.message}`));
|
|
43
|
+
this.pending.delete(id);
|
|
44
|
+
}
|
|
45
|
+
});
|
|
46
|
+
|
|
47
|
+
this.socket.on('close', () => {
|
|
48
|
+
for (const [id, req] of this.pending) {
|
|
49
|
+
clearTimeout(req.timer);
|
|
50
|
+
req.reject(new Error('Connection closed'));
|
|
51
|
+
this.pending.delete(id);
|
|
52
|
+
}
|
|
53
|
+
this.socket = null;
|
|
54
|
+
});
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
request(method: string, params?: unknown): Promise<unknown> {
|
|
59
|
+
return new Promise((resolve, reject) => {
|
|
60
|
+
if (!this.socket || this.socket.destroyed) {
|
|
61
|
+
return reject(new Error('Not connected'));
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
const id = randomUUID();
|
|
65
|
+
const timer = setTimeout(() => {
|
|
66
|
+
this.pending.delete(id);
|
|
67
|
+
reject(new Error(`Request timeout: ${method} (${this.timeout}ms)`));
|
|
68
|
+
}, this.timeout);
|
|
69
|
+
|
|
70
|
+
this.pending.set(id, { resolve, reject, timer });
|
|
71
|
+
|
|
72
|
+
const msg: IpcMessage = {
|
|
73
|
+
id,
|
|
74
|
+
type: 'request',
|
|
75
|
+
method,
|
|
76
|
+
params,
|
|
77
|
+
};
|
|
78
|
+
this.socket.write(encodeMessage(msg));
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
setNotificationHandler(handler: (msg: IpcMessage) => void): void {
|
|
83
|
+
this.onNotification = handler;
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
disconnect(): void {
|
|
87
|
+
for (const [id, req] of this.pending) {
|
|
88
|
+
clearTimeout(req.timer);
|
|
89
|
+
req.reject(new Error('Client disconnecting'));
|
|
90
|
+
this.pending.delete(id);
|
|
91
|
+
}
|
|
92
|
+
this.socket?.destroy();
|
|
93
|
+
this.socket = null;
|
|
94
|
+
this.decoder.reset();
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
get connected(): boolean {
|
|
98
|
+
return this.socket !== null && !this.socket.destroyed;
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
private handleMessage(msg: IpcMessage): void {
|
|
102
|
+
if (msg.type === 'response') {
|
|
103
|
+
const req = this.pending.get(msg.id);
|
|
104
|
+
if (!req) return;
|
|
105
|
+
|
|
106
|
+
clearTimeout(req.timer);
|
|
107
|
+
this.pending.delete(msg.id);
|
|
108
|
+
|
|
109
|
+
if (msg.error) {
|
|
110
|
+
req.reject(new Error(msg.error.message));
|
|
111
|
+
} else {
|
|
112
|
+
req.resolve(msg.result);
|
|
113
|
+
}
|
|
114
|
+
} else if (msg.type === 'notification') {
|
|
115
|
+
this.onNotification?.(msg);
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
}
|
package/src/ipc/protocol.ts
CHANGED
|
@@ -1,35 +1,35 @@
|
|
|
1
|
-
import { Buffer } from 'node:buffer';
|
|
2
|
-
import type { IpcMessage } from '../types/ipc.types.js';
|
|
3
|
-
|
|
4
|
-
export function encodeMessage(msg: IpcMessage): Buffer {
|
|
5
|
-
const json = JSON.stringify(msg);
|
|
6
|
-
const payload = Buffer.from(json, 'utf8');
|
|
7
|
-
const frame = Buffer.alloc(4 + payload.length);
|
|
8
|
-
frame.writeUInt32BE(payload.length, 0);
|
|
9
|
-
payload.copy(frame, 4);
|
|
10
|
-
return frame;
|
|
11
|
-
}
|
|
12
|
-
|
|
13
|
-
export class MessageDecoder {
|
|
14
|
-
private buffer = Buffer.alloc(0);
|
|
15
|
-
|
|
16
|
-
feed(chunk: Buffer): IpcMessage[] {
|
|
17
|
-
this.buffer = Buffer.concat([this.buffer, chunk]);
|
|
18
|
-
const messages: IpcMessage[] = [];
|
|
19
|
-
|
|
20
|
-
while (this.buffer.length >= 4) {
|
|
21
|
-
const length = this.buffer.readUInt32BE(0);
|
|
22
|
-
if (this.buffer.length < 4 + length) break;
|
|
23
|
-
|
|
24
|
-
const json = this.buffer.subarray(4, 4 + length).toString('utf8');
|
|
25
|
-
this.buffer = this.buffer.subarray(4 + length);
|
|
26
|
-
messages.push(JSON.parse(json) as IpcMessage);
|
|
27
|
-
}
|
|
28
|
-
|
|
29
|
-
return messages;
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
reset(): void {
|
|
33
|
-
this.buffer = Buffer.alloc(0);
|
|
34
|
-
}
|
|
35
|
-
}
|
|
1
|
+
import { Buffer } from 'node:buffer';
|
|
2
|
+
import type { IpcMessage } from '../types/ipc.types.js';
|
|
3
|
+
|
|
4
|
+
export function encodeMessage(msg: IpcMessage): Buffer {
|
|
5
|
+
const json = JSON.stringify(msg);
|
|
6
|
+
const payload = Buffer.from(json, 'utf8');
|
|
7
|
+
const frame = Buffer.alloc(4 + payload.length);
|
|
8
|
+
frame.writeUInt32BE(payload.length, 0);
|
|
9
|
+
payload.copy(frame, 4);
|
|
10
|
+
return frame;
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
export class MessageDecoder {
|
|
14
|
+
private buffer = Buffer.alloc(0);
|
|
15
|
+
|
|
16
|
+
feed(chunk: Buffer): IpcMessage[] {
|
|
17
|
+
this.buffer = Buffer.concat([this.buffer, chunk]);
|
|
18
|
+
const messages: IpcMessage[] = [];
|
|
19
|
+
|
|
20
|
+
while (this.buffer.length >= 4) {
|
|
21
|
+
const length = this.buffer.readUInt32BE(0);
|
|
22
|
+
if (this.buffer.length < 4 + length) break;
|
|
23
|
+
|
|
24
|
+
const json = this.buffer.subarray(4, 4 + length).toString('utf8');
|
|
25
|
+
this.buffer = this.buffer.subarray(4 + length);
|
|
26
|
+
messages.push(JSON.parse(json) as IpcMessage);
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
return messages;
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
reset(): void {
|
|
33
|
+
this.buffer = Buffer.alloc(0);
|
|
34
|
+
}
|
|
35
|
+
}
|