@timmeck/brain 1.0.0 → 1.1.1

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 (202) hide show
  1. package/BRAIN_PLAN.md +3324 -3324
  2. package/LICENSE +21 -21
  3. package/README.md +194 -188
  4. package/dist/brain.js +2 -0
  5. package/dist/brain.js.map +1 -1
  6. package/dist/cli/colors.d.ts +50 -0
  7. package/dist/cli/colors.js +106 -0
  8. package/dist/cli/colors.js.map +1 -0
  9. package/dist/cli/commands/config.d.ts +2 -0
  10. package/dist/cli/commands/config.js +165 -0
  11. package/dist/cli/commands/config.js.map +1 -0
  12. package/dist/cli/commands/dashboard.js +222 -8
  13. package/dist/cli/commands/dashboard.js.map +1 -1
  14. package/dist/cli/commands/export.js +3 -0
  15. package/dist/cli/commands/export.js.map +1 -1
  16. package/dist/cli/commands/import.js +24 -15
  17. package/dist/cli/commands/import.js.map +1 -1
  18. package/dist/cli/commands/insights.js +33 -6
  19. package/dist/cli/commands/insights.js.map +1 -1
  20. package/dist/cli/commands/learn.d.ts +2 -0
  21. package/dist/cli/commands/learn.js +22 -0
  22. package/dist/cli/commands/learn.js.map +1 -0
  23. package/dist/cli/commands/modules.js +25 -6
  24. package/dist/cli/commands/modules.js.map +1 -1
  25. package/dist/cli/commands/network.js +15 -9
  26. package/dist/cli/commands/network.js.map +1 -1
  27. package/dist/cli/commands/query.js +92 -25
  28. package/dist/cli/commands/query.js.map +1 -1
  29. package/dist/cli/commands/start.js +8 -5
  30. package/dist/cli/commands/start.js.map +1 -1
  31. package/dist/cli/commands/status.js +21 -16
  32. package/dist/cli/commands/status.js.map +1 -1
  33. package/dist/cli/commands/stop.js +5 -4
  34. package/dist/cli/commands/stop.js.map +1 -1
  35. package/dist/cli/ipc-helper.js +4 -3
  36. package/dist/cli/ipc-helper.js.map +1 -1
  37. package/dist/cli/update-check.d.ts +2 -0
  38. package/dist/cli/update-check.js +58 -0
  39. package/dist/cli/update-check.js.map +1 -0
  40. package/dist/db/migrations/001_core_schema.js +115 -115
  41. package/dist/db/migrations/002_learning_schema.js +33 -33
  42. package/dist/db/migrations/003_code_schema.js +48 -48
  43. package/dist/db/migrations/004_synapses_schema.js +52 -52
  44. package/dist/db/migrations/005_fts_indexes.js +73 -73
  45. package/dist/db/migrations/index.js +6 -6
  46. package/dist/db/repositories/antipattern.repository.js +3 -3
  47. package/dist/db/repositories/code-module.repository.d.ts +1 -0
  48. package/dist/db/repositories/code-module.repository.js +8 -0
  49. package/dist/db/repositories/code-module.repository.js.map +1 -1
  50. package/dist/db/repositories/error.repository.js +46 -46
  51. package/dist/db/repositories/insight.repository.js +3 -3
  52. package/dist/db/repositories/notification.repository.js +3 -3
  53. package/dist/db/repositories/project.repository.js +21 -21
  54. package/dist/db/repositories/rule.repository.js +24 -24
  55. package/dist/db/repositories/solution.repository.js +50 -50
  56. package/dist/db/repositories/synapse.repository.js +18 -18
  57. package/dist/db/repositories/terminal.repository.js +24 -24
  58. package/dist/index.js +4 -0
  59. package/dist/index.js.map +1 -1
  60. package/dist/ipc/router.d.ts +2 -0
  61. package/dist/ipc/router.js +7 -1
  62. package/dist/ipc/router.js.map +1 -1
  63. package/dist/services/code.service.d.ts +1 -1
  64. package/dist/services/code.service.js +5 -2
  65. package/dist/services/code.service.js.map +1 -1
  66. package/package.json +5 -4
  67. package/src/brain.ts +3 -0
  68. package/src/cli/colors.ts +116 -0
  69. package/src/cli/commands/config.ts +169 -0
  70. package/src/cli/commands/dashboard.ts +231 -8
  71. package/src/cli/commands/export.ts +4 -0
  72. package/src/cli/commands/import.ts +24 -15
  73. package/src/cli/commands/insights.ts +37 -5
  74. package/src/cli/commands/learn.ts +24 -0
  75. package/src/cli/commands/modules.ts +28 -5
  76. package/src/cli/commands/network.ts +15 -9
  77. package/src/cli/commands/query.ts +103 -26
  78. package/src/cli/commands/start.ts +8 -5
  79. package/src/cli/commands/status.ts +22 -16
  80. package/src/cli/commands/stop.ts +5 -4
  81. package/src/cli/ipc-helper.ts +4 -3
  82. package/src/cli/update-check.ts +63 -0
  83. package/src/code/analyzer.ts +77 -77
  84. package/src/code/fingerprint.ts +87 -87
  85. package/src/code/matcher.ts +64 -64
  86. package/src/code/parsers/generic.ts +29 -29
  87. package/src/code/parsers/python.ts +54 -54
  88. package/src/code/parsers/typescript.ts +65 -65
  89. package/src/code/registry.ts +60 -60
  90. package/src/code/scorer.ts +108 -108
  91. package/src/config.ts +111 -111
  92. package/src/db/connection.ts +22 -22
  93. package/src/db/migrations/001_core_schema.ts +120 -120
  94. package/src/db/migrations/002_learning_schema.ts +38 -38
  95. package/src/db/migrations/003_code_schema.ts +53 -53
  96. package/src/db/migrations/004_synapses_schema.ts +57 -57
  97. package/src/db/migrations/005_fts_indexes.ts +78 -78
  98. package/src/db/migrations/006_synapses_phase3.ts +17 -17
  99. package/src/db/migrations/index.ts +64 -64
  100. package/src/db/repositories/antipattern.repository.ts +66 -66
  101. package/src/db/repositories/code-module.repository.ts +9 -0
  102. package/src/db/repositories/error.repository.ts +149 -149
  103. package/src/db/repositories/insight.repository.ts +78 -78
  104. package/src/db/repositories/notification.repository.ts +66 -66
  105. package/src/db/repositories/project.repository.ts +93 -93
  106. package/src/db/repositories/rule.repository.ts +108 -108
  107. package/src/db/repositories/solution.repository.ts +154 -154
  108. package/src/db/repositories/synapse.repository.ts +153 -153
  109. package/src/db/repositories/terminal.repository.ts +101 -101
  110. package/src/hooks/post-tool-use.ts +90 -90
  111. package/src/hooks/post-write.ts +117 -117
  112. package/src/index.ts +4 -0
  113. package/src/ipc/client.ts +118 -118
  114. package/src/ipc/protocol.ts +35 -35
  115. package/src/ipc/router.ts +9 -1
  116. package/src/ipc/server.ts +110 -110
  117. package/src/learning/confidence-scorer.ts +47 -47
  118. package/src/learning/decay.ts +46 -46
  119. package/src/learning/learning-engine.ts +162 -162
  120. package/src/learning/pattern-extractor.ts +90 -90
  121. package/src/learning/rule-generator.ts +74 -74
  122. package/src/matching/error-matcher.ts +115 -115
  123. package/src/matching/fingerprint.ts +29 -29
  124. package/src/matching/similarity.ts +61 -61
  125. package/src/matching/tfidf.ts +74 -74
  126. package/src/matching/tokenizer.ts +41 -41
  127. package/src/mcp/auto-detect.ts +93 -93
  128. package/src/mcp/server.ts +73 -73
  129. package/src/mcp/tools.ts +290 -290
  130. package/src/parsing/error-parser.ts +28 -28
  131. package/src/parsing/parsers/compiler.ts +93 -93
  132. package/src/parsing/parsers/generic.ts +28 -28
  133. package/src/parsing/parsers/go.ts +97 -97
  134. package/src/parsing/parsers/node.ts +69 -69
  135. package/src/parsing/parsers/python.ts +62 -62
  136. package/src/parsing/parsers/rust.ts +50 -50
  137. package/src/parsing/parsers/shell.ts +42 -42
  138. package/src/parsing/types.ts +47 -47
  139. package/src/research/gap-analyzer.ts +135 -135
  140. package/src/research/insight-generator.ts +123 -123
  141. package/src/research/research-engine.ts +116 -116
  142. package/src/research/synergy-detector.ts +126 -126
  143. package/src/research/template-extractor.ts +130 -130
  144. package/src/research/trend-analyzer.ts +127 -127
  145. package/src/services/analytics.service.ts +87 -87
  146. package/src/services/code.service.ts +5 -2
  147. package/src/services/error.service.ts +164 -164
  148. package/src/services/notification.service.ts +41 -41
  149. package/src/services/prevention.service.ts +119 -119
  150. package/src/services/research.service.ts +93 -93
  151. package/src/services/solution.service.ts +116 -116
  152. package/src/services/synapse.service.ts +59 -59
  153. package/src/services/terminal.service.ts +81 -81
  154. package/src/synapses/activation.ts +80 -80
  155. package/src/synapses/decay.ts +38 -38
  156. package/src/synapses/hebbian.ts +69 -69
  157. package/src/synapses/pathfinder.ts +81 -81
  158. package/src/synapses/synapse-manager.ts +109 -109
  159. package/src/types/code.types.ts +52 -52
  160. package/src/types/config.types.ts +79 -79
  161. package/src/types/error.types.ts +67 -67
  162. package/src/types/ipc.types.ts +8 -8
  163. package/src/types/mcp.types.ts +53 -53
  164. package/src/types/research.types.ts +28 -28
  165. package/src/types/solution.types.ts +30 -30
  166. package/src/types/synapse.types.ts +49 -49
  167. package/src/utils/events.ts +45 -45
  168. package/src/utils/hash.ts +5 -5
  169. package/src/utils/logger.ts +48 -48
  170. package/src/utils/paths.ts +19 -19
  171. package/tests/fixtures/code-modules/modules.ts +83 -83
  172. package/tests/fixtures/errors/go.ts +9 -9
  173. package/tests/fixtures/errors/node.ts +24 -24
  174. package/tests/fixtures/errors/python.ts +21 -21
  175. package/tests/fixtures/errors/rust.ts +25 -25
  176. package/tests/fixtures/errors/shell.ts +15 -15
  177. package/tests/fixtures/solutions/solutions.ts +27 -27
  178. package/tests/helpers/setup-db.ts +52 -52
  179. package/tests/integration/code-flow.test.ts +86 -86
  180. package/tests/integration/error-flow.test.ts +83 -83
  181. package/tests/integration/ipc-flow.test.ts +166 -166
  182. package/tests/integration/learning-cycle.test.ts +82 -82
  183. package/tests/integration/synapse-flow.test.ts +117 -117
  184. package/tests/unit/code/analyzer.test.ts +58 -58
  185. package/tests/unit/code/fingerprint.test.ts +51 -51
  186. package/tests/unit/code/scorer.test.ts +55 -55
  187. package/tests/unit/learning/confidence-scorer.test.ts +60 -60
  188. package/tests/unit/learning/decay.test.ts +45 -45
  189. package/tests/unit/learning/pattern-extractor.test.ts +50 -50
  190. package/tests/unit/matching/error-matcher.test.ts +69 -69
  191. package/tests/unit/matching/fingerprint.test.ts +47 -47
  192. package/tests/unit/matching/similarity.test.ts +65 -65
  193. package/tests/unit/matching/tfidf.test.ts +71 -71
  194. package/tests/unit/matching/tokenizer.test.ts +83 -83
  195. package/tests/unit/parsing/parsers.test.ts +113 -113
  196. package/tests/unit/research/gap-analyzer.test.ts +45 -45
  197. package/tests/unit/research/trend-analyzer.test.ts +45 -45
  198. package/tests/unit/synapses/activation.test.ts +80 -80
  199. package/tests/unit/synapses/decay.test.ts +27 -27
  200. package/tests/unit/synapses/hebbian.test.ts +96 -96
  201. package/tests/unit/synapses/pathfinder.test.ts +72 -72
  202. 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
+ }