@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.
Files changed (177) hide show
  1. package/BRAIN_PLAN.md +3324 -3324
  2. package/LICENSE +21 -21
  3. package/dist/api/server.d.ts +4 -0
  4. package/dist/api/server.js +73 -0
  5. package/dist/api/server.js.map +1 -1
  6. package/dist/brain.js +2 -1
  7. package/dist/brain.js.map +1 -1
  8. package/dist/cli/commands/dashboard.js +606 -572
  9. package/dist/cli/commands/dashboard.js.map +1 -1
  10. package/dist/dashboard/server.js +25 -25
  11. package/dist/db/migrations/001_core_schema.js +115 -115
  12. package/dist/db/migrations/002_learning_schema.js +33 -33
  13. package/dist/db/migrations/003_code_schema.js +48 -48
  14. package/dist/db/migrations/004_synapses_schema.js +52 -52
  15. package/dist/db/migrations/005_fts_indexes.js +73 -73
  16. package/dist/db/migrations/007_feedback.js +8 -8
  17. package/dist/db/migrations/008_git_integration.js +33 -33
  18. package/dist/db/migrations/009_embeddings.js +3 -3
  19. package/dist/db/repositories/antipattern.repository.js +3 -3
  20. package/dist/db/repositories/code-module.repository.js +32 -32
  21. package/dist/db/repositories/notification.repository.js +3 -3
  22. package/dist/db/repositories/project.repository.js +21 -21
  23. package/dist/db/repositories/rule.repository.js +24 -24
  24. package/dist/db/repositories/solution.repository.js +50 -50
  25. package/dist/db/repositories/synapse.repository.js +18 -18
  26. package/dist/db/repositories/terminal.repository.js +24 -24
  27. package/dist/embeddings/engine.d.ts +2 -2
  28. package/dist/embeddings/engine.js +17 -4
  29. package/dist/embeddings/engine.js.map +1 -1
  30. package/dist/index.js +1 -1
  31. package/dist/ipc/server.d.ts +8 -0
  32. package/dist/ipc/server.js +67 -1
  33. package/dist/ipc/server.js.map +1 -1
  34. package/dist/matching/error-matcher.js +5 -5
  35. package/dist/matching/fingerprint.js +6 -1
  36. package/dist/matching/fingerprint.js.map +1 -1
  37. package/dist/mcp/http-server.js +8 -2
  38. package/dist/mcp/http-server.js.map +1 -1
  39. package/dist/services/code.service.d.ts +3 -0
  40. package/dist/services/code.service.js +33 -4
  41. package/dist/services/code.service.js.map +1 -1
  42. package/dist/services/error.service.js +4 -3
  43. package/dist/services/error.service.js.map +1 -1
  44. package/dist/services/git.service.js +14 -14
  45. package/package.json +49 -49
  46. package/src/api/server.ts +395 -321
  47. package/src/brain.ts +266 -265
  48. package/src/cli/colors.ts +116 -116
  49. package/src/cli/commands/config.ts +169 -169
  50. package/src/cli/commands/dashboard.ts +755 -720
  51. package/src/cli/commands/doctor.ts +118 -118
  52. package/src/cli/commands/explain.ts +83 -83
  53. package/src/cli/commands/export.ts +31 -31
  54. package/src/cli/commands/import.ts +199 -199
  55. package/src/cli/commands/insights.ts +65 -65
  56. package/src/cli/commands/learn.ts +24 -24
  57. package/src/cli/commands/modules.ts +53 -53
  58. package/src/cli/commands/network.ts +67 -67
  59. package/src/cli/commands/projects.ts +42 -42
  60. package/src/cli/commands/query.ts +120 -120
  61. package/src/cli/commands/start.ts +62 -62
  62. package/src/cli/commands/status.ts +75 -75
  63. package/src/cli/commands/stop.ts +34 -34
  64. package/src/cli/ipc-helper.ts +22 -22
  65. package/src/cli/update-check.ts +63 -63
  66. package/src/code/fingerprint.ts +87 -87
  67. package/src/code/parsers/generic.ts +29 -29
  68. package/src/code/parsers/python.ts +54 -54
  69. package/src/code/parsers/typescript.ts +65 -65
  70. package/src/code/registry.ts +60 -60
  71. package/src/dashboard/server.ts +142 -142
  72. package/src/db/connection.ts +22 -22
  73. package/src/db/migrations/001_core_schema.ts +120 -120
  74. package/src/db/migrations/002_learning_schema.ts +38 -38
  75. package/src/db/migrations/003_code_schema.ts +53 -53
  76. package/src/db/migrations/004_synapses_schema.ts +57 -57
  77. package/src/db/migrations/005_fts_indexes.ts +78 -78
  78. package/src/db/migrations/006_synapses_phase3.ts +17 -17
  79. package/src/db/migrations/007_feedback.ts +13 -13
  80. package/src/db/migrations/008_git_integration.ts +38 -38
  81. package/src/db/migrations/009_embeddings.ts +8 -8
  82. package/src/db/repositories/antipattern.repository.ts +66 -66
  83. package/src/db/repositories/code-module.repository.ts +142 -142
  84. package/src/db/repositories/notification.repository.ts +66 -66
  85. package/src/db/repositories/project.repository.ts +93 -93
  86. package/src/db/repositories/rule.repository.ts +108 -108
  87. package/src/db/repositories/solution.repository.ts +154 -154
  88. package/src/db/repositories/synapse.repository.ts +153 -153
  89. package/src/db/repositories/terminal.repository.ts +101 -101
  90. package/src/embeddings/engine.ts +238 -217
  91. package/src/index.ts +63 -63
  92. package/src/ipc/client.ts +118 -118
  93. package/src/ipc/protocol.ts +35 -35
  94. package/src/ipc/router.ts +133 -133
  95. package/src/ipc/server.ts +176 -110
  96. package/src/learning/decay.ts +46 -46
  97. package/src/learning/pattern-extractor.ts +90 -90
  98. package/src/learning/rule-generator.ts +74 -74
  99. package/src/matching/error-matcher.ts +5 -5
  100. package/src/matching/fingerprint.ts +34 -29
  101. package/src/matching/similarity.ts +61 -61
  102. package/src/matching/tfidf.ts +74 -74
  103. package/src/matching/tokenizer.ts +41 -41
  104. package/src/mcp/auto-detect.ts +93 -93
  105. package/src/mcp/http-server.ts +140 -137
  106. package/src/mcp/server.ts +73 -73
  107. package/src/parsing/error-parser.ts +28 -28
  108. package/src/parsing/parsers/compiler.ts +93 -93
  109. package/src/parsing/parsers/generic.ts +28 -28
  110. package/src/parsing/parsers/go.ts +97 -97
  111. package/src/parsing/parsers/node.ts +69 -69
  112. package/src/parsing/parsers/python.ts +62 -62
  113. package/src/parsing/parsers/rust.ts +50 -50
  114. package/src/parsing/parsers/shell.ts +42 -42
  115. package/src/parsing/types.ts +47 -47
  116. package/src/research/gap-analyzer.ts +135 -135
  117. package/src/research/insight-generator.ts +123 -123
  118. package/src/research/research-engine.ts +116 -116
  119. package/src/research/synergy-detector.ts +126 -126
  120. package/src/research/template-extractor.ts +130 -130
  121. package/src/research/trend-analyzer.ts +127 -127
  122. package/src/services/code.service.ts +271 -238
  123. package/src/services/error.service.ts +4 -3
  124. package/src/services/git.service.ts +132 -132
  125. package/src/services/notification.service.ts +41 -41
  126. package/src/services/synapse.service.ts +59 -59
  127. package/src/services/terminal.service.ts +81 -81
  128. package/src/synapses/activation.ts +80 -80
  129. package/src/synapses/decay.ts +38 -38
  130. package/src/synapses/hebbian.ts +69 -69
  131. package/src/synapses/pathfinder.ts +81 -81
  132. package/src/synapses/synapse-manager.ts +109 -109
  133. package/src/types/code.types.ts +52 -52
  134. package/src/types/error.types.ts +67 -67
  135. package/src/types/ipc.types.ts +8 -8
  136. package/src/types/mcp.types.ts +53 -53
  137. package/src/types/research.types.ts +28 -28
  138. package/src/types/solution.types.ts +30 -30
  139. package/src/utils/events.ts +45 -45
  140. package/src/utils/hash.ts +5 -5
  141. package/src/utils/logger.ts +48 -48
  142. package/src/utils/paths.ts +19 -19
  143. package/tests/e2e/test_code_intelligence.py +1015 -0
  144. package/tests/e2e/test_error_memory.py +451 -0
  145. package/tests/e2e/test_full_integration.py +534 -0
  146. package/tests/fixtures/code-modules/modules.ts +83 -83
  147. package/tests/fixtures/errors/go.ts +9 -9
  148. package/tests/fixtures/errors/node.ts +24 -24
  149. package/tests/fixtures/errors/python.ts +21 -21
  150. package/tests/fixtures/errors/rust.ts +25 -25
  151. package/tests/fixtures/errors/shell.ts +15 -15
  152. package/tests/fixtures/solutions/solutions.ts +27 -27
  153. package/tests/helpers/setup-db.ts +52 -52
  154. package/tests/integration/code-flow.test.ts +86 -86
  155. package/tests/integration/error-flow.test.ts +83 -83
  156. package/tests/integration/ipc-flow.test.ts +166 -166
  157. package/tests/integration/learning-cycle.test.ts +82 -82
  158. package/tests/integration/synapse-flow.test.ts +117 -117
  159. package/tests/unit/code/analyzer.test.ts +58 -58
  160. package/tests/unit/code/fingerprint.test.ts +51 -51
  161. package/tests/unit/code/scorer.test.ts +55 -55
  162. package/tests/unit/learning/confidence-scorer.test.ts +60 -60
  163. package/tests/unit/learning/decay.test.ts +45 -45
  164. package/tests/unit/learning/pattern-extractor.test.ts +50 -50
  165. package/tests/unit/matching/error-matcher.test.ts +69 -69
  166. package/tests/unit/matching/fingerprint.test.ts +47 -47
  167. package/tests/unit/matching/similarity.test.ts +65 -65
  168. package/tests/unit/matching/tfidf.test.ts +71 -71
  169. package/tests/unit/matching/tokenizer.test.ts +83 -83
  170. package/tests/unit/parsing/parsers.test.ts +113 -113
  171. package/tests/unit/research/gap-analyzer.test.ts +45 -45
  172. package/tests/unit/research/trend-analyzer.test.ts +45 -45
  173. package/tests/unit/synapses/activation.test.ts +80 -80
  174. package/tests/unit/synapses/decay.test.ts +27 -27
  175. package/tests/unit/synapses/hebbian.test.ts +96 -96
  176. package/tests/unit/synapses/pathfinder.test.ts +72 -72
  177. package/tsconfig.json +18 -18
@@ -1,154 +1,154 @@
1
- import type Database from 'better-sqlite3';
2
- import type { Statement } from 'better-sqlite3';
3
- import type { SolutionRecord, ErrorSolution, SolutionAttempt } from '../../types/solution.types.js';
4
-
5
- type CreateSolutionData = Omit<SolutionRecord, 'id' | 'success_count' | 'fail_count' | 'created_at' | 'updated_at'>;
6
- type UpdateSolutionData = Partial<Omit<SolutionRecord, 'id' | 'created_at'>>;
7
-
8
- interface AttemptData {
9
- errorSolutionId: number;
10
- terminalId?: number | null;
11
- success: number;
12
- output?: string | null;
13
- durationMs?: number | null;
14
- }
15
-
16
- export class SolutionRepository {
17
- private stmts: Record<string, Statement>;
18
-
19
- constructor(private db: Database.Database) {
20
- this.stmts = {
21
- create: this.db.prepare(`
22
- INSERT INTO solutions (description, commands, code_change, source, confidence)
23
- VALUES (@description, @commands, @code_change, @source, @confidence)
24
- `),
25
- getById: this.db.prepare(`
26
- SELECT * FROM solutions WHERE id = ?
27
- `),
28
- update: this.db.prepare(`
29
- UPDATE solutions
30
- SET description = COALESCE(@description, description),
31
- commands = COALESCE(@commands, commands),
32
- code_change = COALESCE(@code_change, code_change),
33
- source = COALESCE(@source, source),
34
- confidence = COALESCE(@confidence, confidence),
35
- success_count = COALESCE(@success_count, success_count),
36
- fail_count = COALESCE(@fail_count, fail_count),
37
- updated_at = datetime('now')
38
- WHERE id = @id
39
- `),
40
- delete: this.db.prepare(`
41
- DELETE FROM solutions WHERE id = ?
42
- `),
43
- findForError: this.db.prepare(`
44
- SELECT s.* FROM solutions s
45
- JOIN error_solutions es ON s.id = es.solution_id
46
- WHERE es.error_id = ?
47
- ORDER BY s.confidence DESC
48
- `),
49
- linkToError: this.db.prepare(`
50
- INSERT OR IGNORE INTO error_solutions (error_id, solution_id)
51
- VALUES (@error_id, @solution_id)
52
- `),
53
- recordAttempt: this.db.prepare(`
54
- INSERT INTO solution_attempts (error_solution_id, terminal_id, success, output, duration_ms)
55
- VALUES (@error_solution_id, @terminal_id, @success, @output, @duration_ms)
56
- `),
57
- getAttempts: this.db.prepare(`
58
- SELECT * FROM solution_attempts WHERE error_solution_id = ? ORDER BY attempted_at DESC
59
- `),
60
- getAll: this.db.prepare(`
61
- SELECT * FROM solutions
62
- `),
63
- successRate: this.db.prepare(`
64
- SELECT
65
- COUNT(*) as total,
66
- SUM(CASE WHEN sa.success = 1 THEN 1 ELSE 0 END) as successes
67
- FROM solution_attempts sa
68
- JOIN error_solutions es ON sa.error_solution_id = es.id
69
- WHERE es.solution_id = ?
70
- `),
71
- updateSuccessCount: this.db.prepare(`
72
- UPDATE solutions
73
- SET success_count = success_count + 1,
74
- updated_at = datetime('now')
75
- WHERE id = ?
76
- `),
77
- updateFailCount: this.db.prepare(`
78
- UPDATE solutions
79
- SET fail_count = fail_count + 1,
80
- updated_at = datetime('now')
81
- WHERE id = ?
82
- `),
83
- };
84
- }
85
-
86
- create(data: CreateSolutionData): number {
87
- const result = this.stmts.create.run({
88
- description: data.description,
89
- commands: data.commands ?? null,
90
- code_change: data.code_change ?? null,
91
- source: data.source,
92
- confidence: data.confidence,
93
- });
94
- return result.lastInsertRowid as number;
95
- }
96
-
97
- getById(id: number): SolutionRecord | undefined {
98
- return this.stmts.getById.get(id) as SolutionRecord | undefined;
99
- }
100
-
101
- update(id: number, data: UpdateSolutionData): void {
102
- this.stmts.update.run({
103
- id,
104
- description: data.description ?? null,
105
- commands: data.commands ?? null,
106
- code_change: data.code_change ?? null,
107
- source: data.source ?? null,
108
- confidence: data.confidence ?? null,
109
- success_count: data.success_count ?? null,
110
- fail_count: data.fail_count ?? null,
111
- updated_at: data.updated_at ?? null,
112
- });
113
- }
114
-
115
- delete(id: number): void {
116
- this.stmts.delete.run(id);
117
- }
118
-
119
- findForError(errorId: number): SolutionRecord[] {
120
- return this.stmts.findForError.all(errorId) as SolutionRecord[];
121
- }
122
-
123
- linkToError(errorId: number, solutionId: number): void {
124
- this.stmts.linkToError.run({
125
- error_id: errorId,
126
- solution_id: solutionId,
127
- });
128
- }
129
-
130
- recordAttempt(data: AttemptData): number {
131
- const result = this.stmts.recordAttempt.run({
132
- error_solution_id: data.errorSolutionId,
133
- terminal_id: data.terminalId ?? null,
134
- success: data.success,
135
- output: data.output ?? null,
136
- duration_ms: data.durationMs ?? null,
137
- });
138
- return result.lastInsertRowid as number;
139
- }
140
-
141
- getAttempts(errorSolutionId: number): SolutionAttempt[] {
142
- return this.stmts.getAttempts.all(errorSolutionId) as SolutionAttempt[];
143
- }
144
-
145
- getAll(): SolutionRecord[] {
146
- return this.stmts.getAll.all() as SolutionRecord[];
147
- }
148
-
149
- successRate(solutionId: number): number {
150
- const row = this.stmts.successRate.get(solutionId) as { total: number; successes: number };
151
- if (row.total === 0) return 0;
152
- return row.successes / row.total;
153
- }
154
- }
1
+ import type Database from 'better-sqlite3';
2
+ import type { Statement } from 'better-sqlite3';
3
+ import type { SolutionRecord, ErrorSolution, SolutionAttempt } from '../../types/solution.types.js';
4
+
5
+ type CreateSolutionData = Omit<SolutionRecord, 'id' | 'success_count' | 'fail_count' | 'created_at' | 'updated_at'>;
6
+ type UpdateSolutionData = Partial<Omit<SolutionRecord, 'id' | 'created_at'>>;
7
+
8
+ interface AttemptData {
9
+ errorSolutionId: number;
10
+ terminalId?: number | null;
11
+ success: number;
12
+ output?: string | null;
13
+ durationMs?: number | null;
14
+ }
15
+
16
+ export class SolutionRepository {
17
+ private stmts: Record<string, Statement>;
18
+
19
+ constructor(private db: Database.Database) {
20
+ this.stmts = {
21
+ create: this.db.prepare(`
22
+ INSERT INTO solutions (description, commands, code_change, source, confidence)
23
+ VALUES (@description, @commands, @code_change, @source, @confidence)
24
+ `),
25
+ getById: this.db.prepare(`
26
+ SELECT * FROM solutions WHERE id = ?
27
+ `),
28
+ update: this.db.prepare(`
29
+ UPDATE solutions
30
+ SET description = COALESCE(@description, description),
31
+ commands = COALESCE(@commands, commands),
32
+ code_change = COALESCE(@code_change, code_change),
33
+ source = COALESCE(@source, source),
34
+ confidence = COALESCE(@confidence, confidence),
35
+ success_count = COALESCE(@success_count, success_count),
36
+ fail_count = COALESCE(@fail_count, fail_count),
37
+ updated_at = datetime('now')
38
+ WHERE id = @id
39
+ `),
40
+ delete: this.db.prepare(`
41
+ DELETE FROM solutions WHERE id = ?
42
+ `),
43
+ findForError: this.db.prepare(`
44
+ SELECT s.* FROM solutions s
45
+ JOIN error_solutions es ON s.id = es.solution_id
46
+ WHERE es.error_id = ?
47
+ ORDER BY s.confidence DESC
48
+ `),
49
+ linkToError: this.db.prepare(`
50
+ INSERT OR IGNORE INTO error_solutions (error_id, solution_id)
51
+ VALUES (@error_id, @solution_id)
52
+ `),
53
+ recordAttempt: this.db.prepare(`
54
+ INSERT INTO solution_attempts (error_solution_id, terminal_id, success, output, duration_ms)
55
+ VALUES (@error_solution_id, @terminal_id, @success, @output, @duration_ms)
56
+ `),
57
+ getAttempts: this.db.prepare(`
58
+ SELECT * FROM solution_attempts WHERE error_solution_id = ? ORDER BY attempted_at DESC
59
+ `),
60
+ getAll: this.db.prepare(`
61
+ SELECT * FROM solutions
62
+ `),
63
+ successRate: this.db.prepare(`
64
+ SELECT
65
+ COUNT(*) as total,
66
+ SUM(CASE WHEN sa.success = 1 THEN 1 ELSE 0 END) as successes
67
+ FROM solution_attempts sa
68
+ JOIN error_solutions es ON sa.error_solution_id = es.id
69
+ WHERE es.solution_id = ?
70
+ `),
71
+ updateSuccessCount: this.db.prepare(`
72
+ UPDATE solutions
73
+ SET success_count = success_count + 1,
74
+ updated_at = datetime('now')
75
+ WHERE id = ?
76
+ `),
77
+ updateFailCount: this.db.prepare(`
78
+ UPDATE solutions
79
+ SET fail_count = fail_count + 1,
80
+ updated_at = datetime('now')
81
+ WHERE id = ?
82
+ `),
83
+ };
84
+ }
85
+
86
+ create(data: CreateSolutionData): number {
87
+ const result = this.stmts.create.run({
88
+ description: data.description,
89
+ commands: data.commands ?? null,
90
+ code_change: data.code_change ?? null,
91
+ source: data.source,
92
+ confidence: data.confidence,
93
+ });
94
+ return result.lastInsertRowid as number;
95
+ }
96
+
97
+ getById(id: number): SolutionRecord | undefined {
98
+ return this.stmts.getById.get(id) as SolutionRecord | undefined;
99
+ }
100
+
101
+ update(id: number, data: UpdateSolutionData): void {
102
+ this.stmts.update.run({
103
+ id,
104
+ description: data.description ?? null,
105
+ commands: data.commands ?? null,
106
+ code_change: data.code_change ?? null,
107
+ source: data.source ?? null,
108
+ confidence: data.confidence ?? null,
109
+ success_count: data.success_count ?? null,
110
+ fail_count: data.fail_count ?? null,
111
+ updated_at: data.updated_at ?? null,
112
+ });
113
+ }
114
+
115
+ delete(id: number): void {
116
+ this.stmts.delete.run(id);
117
+ }
118
+
119
+ findForError(errorId: number): SolutionRecord[] {
120
+ return this.stmts.findForError.all(errorId) as SolutionRecord[];
121
+ }
122
+
123
+ linkToError(errorId: number, solutionId: number): void {
124
+ this.stmts.linkToError.run({
125
+ error_id: errorId,
126
+ solution_id: solutionId,
127
+ });
128
+ }
129
+
130
+ recordAttempt(data: AttemptData): number {
131
+ const result = this.stmts.recordAttempt.run({
132
+ error_solution_id: data.errorSolutionId,
133
+ terminal_id: data.terminalId ?? null,
134
+ success: data.success,
135
+ output: data.output ?? null,
136
+ duration_ms: data.durationMs ?? null,
137
+ });
138
+ return result.lastInsertRowid as number;
139
+ }
140
+
141
+ getAttempts(errorSolutionId: number): SolutionAttempt[] {
142
+ return this.stmts.getAttempts.all(errorSolutionId) as SolutionAttempt[];
143
+ }
144
+
145
+ getAll(): SolutionRecord[] {
146
+ return this.stmts.getAll.all() as SolutionRecord[];
147
+ }
148
+
149
+ successRate(solutionId: number): number {
150
+ const row = this.stmts.successRate.get(solutionId) as { total: number; successes: number };
151
+ if (row.total === 0) return 0;
152
+ return row.successes / row.total;
153
+ }
154
+ }
@@ -1,153 +1,153 @@
1
- import type Database from 'better-sqlite3';
2
- import type { Statement } from 'better-sqlite3';
3
- import type { SynapseRecord } from '../../types/synapse.types.js';
4
-
5
- type SynapseCreate = Omit<SynapseRecord, 'id' | 'activation_count' | 'last_activated_at' | 'created_at' | 'updated_at'>;
6
- type SynapseUpdate = Partial<Omit<SynapseRecord, 'id' | 'created_at'>>;
7
-
8
- export class SynapseRepository {
9
- private stmts: Record<string, Statement>;
10
-
11
- constructor(private db: Database.Database) {
12
- this.stmts = {
13
- create: db.prepare(`
14
- INSERT INTO synapses (source_type, source_id, target_type, target_id, synapse_type, weight, metadata)
15
- VALUES (@source_type, @source_id, @target_type, @target_id, @synapse_type, @weight, @metadata)
16
- `),
17
- getById: db.prepare('SELECT * FROM synapses WHERE id = ?'),
18
- delete: db.prepare('DELETE FROM synapses WHERE id = ?'),
19
- getOutgoing: db.prepare(
20
- 'SELECT * FROM synapses WHERE source_type = ? AND source_id = ? ORDER BY weight DESC'
21
- ),
22
- getIncoming: db.prepare(
23
- 'SELECT * FROM synapses WHERE target_type = ? AND target_id = ? ORDER BY weight DESC'
24
- ),
25
- findConnected: db.prepare(`
26
- SELECT * FROM synapses
27
- WHERE (source_type = ? AND source_id = ?) OR (target_type = ? AND target_id = ?)
28
- ORDER BY weight DESC
29
- `),
30
- findBySourceTarget: db.prepare(`
31
- SELECT * FROM synapses
32
- WHERE source_type = ? AND source_id = ? AND target_type = ? AND target_id = ? AND synapse_type = ?
33
- `),
34
- findInactiveSince: db.prepare(`
35
- SELECT * FROM synapses WHERE last_activated_at < ?
36
- `),
37
- countNodes: db.prepare(`
38
- SELECT COUNT(*) as count FROM (
39
- SELECT source_type, source_id FROM synapses
40
- UNION
41
- SELECT target_type, target_id FROM synapses
42
- )
43
- `),
44
- avgWeight: db.prepare('SELECT AVG(weight) as avg_weight FROM synapses'),
45
- countByType: db.prepare('SELECT synapse_type, COUNT(*) as count FROM synapses GROUP BY synapse_type'),
46
- totalCount: db.prepare('SELECT COUNT(*) as count FROM synapses'),
47
- topByWeight: db.prepare('SELECT * FROM synapses ORDER BY weight DESC LIMIT ?'),
48
- };
49
- }
50
-
51
- create(data: SynapseCreate): number {
52
- const result = this.stmts.create.run({
53
- source_type: data.source_type,
54
- source_id: data.source_id,
55
- target_type: data.target_type,
56
- target_id: data.target_id,
57
- synapse_type: data.synapse_type,
58
- weight: data.weight,
59
- metadata: data.metadata ?? null,
60
- });
61
- return result.lastInsertRowid as number;
62
- }
63
-
64
- getById(id: number): SynapseRecord | undefined {
65
- return this.stmts.getById.get(id) as SynapseRecord | undefined;
66
- }
67
-
68
- update(id: number, data: SynapseUpdate): boolean {
69
- const fields = Object.keys(data).filter(
70
- (key) => (data as Record<string, unknown>)[key] !== undefined
71
- );
72
- if (fields.length === 0) return false;
73
-
74
- const setClauses = fields.map((field) => `${field} = @${field}`).join(', ');
75
- const stmt = this.db.prepare(
76
- `UPDATE synapses SET ${setClauses}, updated_at = datetime('now') WHERE id = @id`
77
- );
78
- const result = stmt.run({ ...data, id });
79
- return result.changes > 0;
80
- }
81
-
82
- delete(id: number): boolean {
83
- const result = this.stmts.delete.run(id);
84
- return result.changes > 0;
85
- }
86
-
87
- getOutgoing(sourceType: string, sourceId: number): SynapseRecord[] {
88
- return this.stmts.getOutgoing.all(sourceType, sourceId) as SynapseRecord[];
89
- }
90
-
91
- getIncoming(targetType: string, targetId: number): SynapseRecord[] {
92
- return this.stmts.getIncoming.all(targetType, targetId) as SynapseRecord[];
93
- }
94
-
95
- findBySourceTarget(
96
- sourceType: string,
97
- sourceId: number,
98
- targetType: string,
99
- targetId: number,
100
- synapseType: string,
101
- ): SynapseRecord | undefined {
102
- return this.stmts.findBySourceTarget.get(
103
- sourceType, sourceId, targetType, targetId, synapseType
104
- ) as SynapseRecord | undefined;
105
- }
106
-
107
- findByWeight(minWeight: number, maxWeight?: number): SynapseRecord[] {
108
- if (maxWeight !== undefined) {
109
- const stmt = this.db.prepare(
110
- 'SELECT * FROM synapses WHERE weight >= ? AND weight <= ? ORDER BY weight DESC'
111
- );
112
- return stmt.all(minWeight, maxWeight) as SynapseRecord[];
113
- }
114
- const stmt = this.db.prepare('SELECT * FROM synapses WHERE weight >= ? ORDER BY weight DESC');
115
- return stmt.all(minWeight) as SynapseRecord[];
116
- }
117
-
118
- findConnected(nodeType: string, nodeId: number): SynapseRecord[] {
119
- return this.stmts.findConnected.all(nodeType, nodeId, nodeType, nodeId) as SynapseRecord[];
120
- }
121
-
122
- findInactiveSince(cutoff: string): SynapseRecord[] {
123
- return this.stmts.findInactiveSince.all(cutoff) as SynapseRecord[];
124
- }
125
-
126
- topByWeight(limit: number): SynapseRecord[] {
127
- return this.stmts.topByWeight.all(limit) as SynapseRecord[];
128
- }
129
-
130
- countNodes(): number {
131
- const row = this.stmts.countNodes.get() as { count: number };
132
- return row.count;
133
- }
134
-
135
- totalCount(): number {
136
- const row = this.stmts.totalCount.get() as { count: number };
137
- return row.count;
138
- }
139
-
140
- avgWeight(): number {
141
- const row = this.stmts.avgWeight.get() as { avg_weight: number | null };
142
- return row.avg_weight ?? 0;
143
- }
144
-
145
- countByType(): Record<string, number> {
146
- const rows = this.stmts.countByType.all() as Array<{ synapse_type: string; count: number }>;
147
- const result: Record<string, number> = {};
148
- for (const row of rows) {
149
- result[row.synapse_type] = row.count;
150
- }
151
- return result;
152
- }
153
- }
1
+ import type Database from 'better-sqlite3';
2
+ import type { Statement } from 'better-sqlite3';
3
+ import type { SynapseRecord } from '../../types/synapse.types.js';
4
+
5
+ type SynapseCreate = Omit<SynapseRecord, 'id' | 'activation_count' | 'last_activated_at' | 'created_at' | 'updated_at'>;
6
+ type SynapseUpdate = Partial<Omit<SynapseRecord, 'id' | 'created_at'>>;
7
+
8
+ export class SynapseRepository {
9
+ private stmts: Record<string, Statement>;
10
+
11
+ constructor(private db: Database.Database) {
12
+ this.stmts = {
13
+ create: db.prepare(`
14
+ INSERT INTO synapses (source_type, source_id, target_type, target_id, synapse_type, weight, metadata)
15
+ VALUES (@source_type, @source_id, @target_type, @target_id, @synapse_type, @weight, @metadata)
16
+ `),
17
+ getById: db.prepare('SELECT * FROM synapses WHERE id = ?'),
18
+ delete: db.prepare('DELETE FROM synapses WHERE id = ?'),
19
+ getOutgoing: db.prepare(
20
+ 'SELECT * FROM synapses WHERE source_type = ? AND source_id = ? ORDER BY weight DESC'
21
+ ),
22
+ getIncoming: db.prepare(
23
+ 'SELECT * FROM synapses WHERE target_type = ? AND target_id = ? ORDER BY weight DESC'
24
+ ),
25
+ findConnected: db.prepare(`
26
+ SELECT * FROM synapses
27
+ WHERE (source_type = ? AND source_id = ?) OR (target_type = ? AND target_id = ?)
28
+ ORDER BY weight DESC
29
+ `),
30
+ findBySourceTarget: db.prepare(`
31
+ SELECT * FROM synapses
32
+ WHERE source_type = ? AND source_id = ? AND target_type = ? AND target_id = ? AND synapse_type = ?
33
+ `),
34
+ findInactiveSince: db.prepare(`
35
+ SELECT * FROM synapses WHERE last_activated_at < ?
36
+ `),
37
+ countNodes: db.prepare(`
38
+ SELECT COUNT(*) as count FROM (
39
+ SELECT source_type, source_id FROM synapses
40
+ UNION
41
+ SELECT target_type, target_id FROM synapses
42
+ )
43
+ `),
44
+ avgWeight: db.prepare('SELECT AVG(weight) as avg_weight FROM synapses'),
45
+ countByType: db.prepare('SELECT synapse_type, COUNT(*) as count FROM synapses GROUP BY synapse_type'),
46
+ totalCount: db.prepare('SELECT COUNT(*) as count FROM synapses'),
47
+ topByWeight: db.prepare('SELECT * FROM synapses ORDER BY weight DESC LIMIT ?'),
48
+ };
49
+ }
50
+
51
+ create(data: SynapseCreate): number {
52
+ const result = this.stmts.create.run({
53
+ source_type: data.source_type,
54
+ source_id: data.source_id,
55
+ target_type: data.target_type,
56
+ target_id: data.target_id,
57
+ synapse_type: data.synapse_type,
58
+ weight: data.weight,
59
+ metadata: data.metadata ?? null,
60
+ });
61
+ return result.lastInsertRowid as number;
62
+ }
63
+
64
+ getById(id: number): SynapseRecord | undefined {
65
+ return this.stmts.getById.get(id) as SynapseRecord | undefined;
66
+ }
67
+
68
+ update(id: number, data: SynapseUpdate): boolean {
69
+ const fields = Object.keys(data).filter(
70
+ (key) => (data as Record<string, unknown>)[key] !== undefined
71
+ );
72
+ if (fields.length === 0) return false;
73
+
74
+ const setClauses = fields.map((field) => `${field} = @${field}`).join(', ');
75
+ const stmt = this.db.prepare(
76
+ `UPDATE synapses SET ${setClauses}, updated_at = datetime('now') WHERE id = @id`
77
+ );
78
+ const result = stmt.run({ ...data, id });
79
+ return result.changes > 0;
80
+ }
81
+
82
+ delete(id: number): boolean {
83
+ const result = this.stmts.delete.run(id);
84
+ return result.changes > 0;
85
+ }
86
+
87
+ getOutgoing(sourceType: string, sourceId: number): SynapseRecord[] {
88
+ return this.stmts.getOutgoing.all(sourceType, sourceId) as SynapseRecord[];
89
+ }
90
+
91
+ getIncoming(targetType: string, targetId: number): SynapseRecord[] {
92
+ return this.stmts.getIncoming.all(targetType, targetId) as SynapseRecord[];
93
+ }
94
+
95
+ findBySourceTarget(
96
+ sourceType: string,
97
+ sourceId: number,
98
+ targetType: string,
99
+ targetId: number,
100
+ synapseType: string,
101
+ ): SynapseRecord | undefined {
102
+ return this.stmts.findBySourceTarget.get(
103
+ sourceType, sourceId, targetType, targetId, synapseType
104
+ ) as SynapseRecord | undefined;
105
+ }
106
+
107
+ findByWeight(minWeight: number, maxWeight?: number): SynapseRecord[] {
108
+ if (maxWeight !== undefined) {
109
+ const stmt = this.db.prepare(
110
+ 'SELECT * FROM synapses WHERE weight >= ? AND weight <= ? ORDER BY weight DESC'
111
+ );
112
+ return stmt.all(minWeight, maxWeight) as SynapseRecord[];
113
+ }
114
+ const stmt = this.db.prepare('SELECT * FROM synapses WHERE weight >= ? ORDER BY weight DESC');
115
+ return stmt.all(minWeight) as SynapseRecord[];
116
+ }
117
+
118
+ findConnected(nodeType: string, nodeId: number): SynapseRecord[] {
119
+ return this.stmts.findConnected.all(nodeType, nodeId, nodeType, nodeId) as SynapseRecord[];
120
+ }
121
+
122
+ findInactiveSince(cutoff: string): SynapseRecord[] {
123
+ return this.stmts.findInactiveSince.all(cutoff) as SynapseRecord[];
124
+ }
125
+
126
+ topByWeight(limit: number): SynapseRecord[] {
127
+ return this.stmts.topByWeight.all(limit) as SynapseRecord[];
128
+ }
129
+
130
+ countNodes(): number {
131
+ const row = this.stmts.countNodes.get() as { count: number };
132
+ return row.count;
133
+ }
134
+
135
+ totalCount(): number {
136
+ const row = this.stmts.totalCount.get() as { count: number };
137
+ return row.count;
138
+ }
139
+
140
+ avgWeight(): number {
141
+ const row = this.stmts.avgWeight.get() as { avg_weight: number | null };
142
+ return row.avg_weight ?? 0;
143
+ }
144
+
145
+ countByType(): Record<string, number> {
146
+ const rows = this.stmts.countByType.all() as Array<{ synapse_type: string; count: number }>;
147
+ const result: Record<string, number> = {};
148
+ for (const row of rows) {
149
+ result[row.synapse_type] = row.count;
150
+ }
151
+ return result;
152
+ }
153
+ }