@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,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
+ }