@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,66 +1,66 @@
1
- import type Database from 'better-sqlite3';
2
- import type { Statement } from 'better-sqlite3';
3
-
4
- export interface NotificationRecord {
5
- id: number;
6
- type: string;
7
- title: string;
8
- message: string;
9
- priority: number;
10
- project_id: number | null;
11
- acknowledged: number;
12
- acknowledged_at: string | null;
13
- created_at: string;
14
- }
15
-
16
- type NotificationCreate = Omit<NotificationRecord, 'id' | 'acknowledged' | 'acknowledged_at' | 'created_at'>;
17
-
18
- export class NotificationRepository {
19
- private stmts: Record<string, Statement>;
20
-
21
- constructor(private db: Database.Database) {
22
- this.stmts = {
23
- create: db.prepare(`
24
- INSERT INTO notifications (type, title, message, priority, project_id)
25
- VALUES (@type, @title, @message, @priority, @project_id)
26
- `),
27
- getById: db.prepare('SELECT * FROM notifications WHERE id = ?'),
28
- delete: db.prepare('DELETE FROM notifications WHERE id = ?'),
29
- findUnacknowledgedAll: db.prepare(
30
- 'SELECT * FROM notifications WHERE acknowledged = 0 ORDER BY priority DESC, created_at DESC'
31
- ),
32
- findUnacknowledgedByProject: db.prepare(
33
- 'SELECT * FROM notifications WHERE acknowledged = 0 AND project_id = ? ORDER BY priority DESC, created_at DESC'
34
- ),
35
- acknowledge: db.prepare(
36
- `UPDATE notifications SET acknowledged = 1, acknowledged_at = datetime('now') WHERE id = ?`
37
- ),
38
- };
39
- }
40
-
41
- create(data: NotificationCreate): number {
42
- const result = this.stmts.create.run(data);
43
- return result.lastInsertRowid as number;
44
- }
45
-
46
- getById(id: number): NotificationRecord | undefined {
47
- return this.stmts.getById.get(id) as NotificationRecord | undefined;
48
- }
49
-
50
- delete(id: number): boolean {
51
- const result = this.stmts.delete.run(id);
52
- return result.changes > 0;
53
- }
54
-
55
- findUnacknowledged(projectId?: number): NotificationRecord[] {
56
- if (projectId !== undefined) {
57
- return this.stmts.findUnacknowledgedByProject.all(projectId) as NotificationRecord[];
58
- }
59
- return this.stmts.findUnacknowledgedAll.all() as NotificationRecord[];
60
- }
61
-
62
- acknowledge(id: number): boolean {
63
- const result = this.stmts.acknowledge.run(id);
64
- return result.changes > 0;
65
- }
66
- }
1
+ import type Database from 'better-sqlite3';
2
+ import type { Statement } from 'better-sqlite3';
3
+
4
+ export interface NotificationRecord {
5
+ id: number;
6
+ type: string;
7
+ title: string;
8
+ message: string;
9
+ priority: number;
10
+ project_id: number | null;
11
+ acknowledged: number;
12
+ acknowledged_at: string | null;
13
+ created_at: string;
14
+ }
15
+
16
+ type NotificationCreate = Omit<NotificationRecord, 'id' | 'acknowledged' | 'acknowledged_at' | 'created_at'>;
17
+
18
+ export class NotificationRepository {
19
+ private stmts: Record<string, Statement>;
20
+
21
+ constructor(private db: Database.Database) {
22
+ this.stmts = {
23
+ create: db.prepare(`
24
+ INSERT INTO notifications (type, title, message, priority, project_id)
25
+ VALUES (@type, @title, @message, @priority, @project_id)
26
+ `),
27
+ getById: db.prepare('SELECT * FROM notifications WHERE id = ?'),
28
+ delete: db.prepare('DELETE FROM notifications WHERE id = ?'),
29
+ findUnacknowledgedAll: db.prepare(
30
+ 'SELECT * FROM notifications WHERE acknowledged = 0 ORDER BY priority DESC, created_at DESC'
31
+ ),
32
+ findUnacknowledgedByProject: db.prepare(
33
+ 'SELECT * FROM notifications WHERE acknowledged = 0 AND project_id = ? ORDER BY priority DESC, created_at DESC'
34
+ ),
35
+ acknowledge: db.prepare(
36
+ `UPDATE notifications SET acknowledged = 1, acknowledged_at = datetime('now') WHERE id = ?`
37
+ ),
38
+ };
39
+ }
40
+
41
+ create(data: NotificationCreate): number {
42
+ const result = this.stmts.create.run(data);
43
+ return result.lastInsertRowid as number;
44
+ }
45
+
46
+ getById(id: number): NotificationRecord | undefined {
47
+ return this.stmts.getById.get(id) as NotificationRecord | undefined;
48
+ }
49
+
50
+ delete(id: number): boolean {
51
+ const result = this.stmts.delete.run(id);
52
+ return result.changes > 0;
53
+ }
54
+
55
+ findUnacknowledged(projectId?: number): NotificationRecord[] {
56
+ if (projectId !== undefined) {
57
+ return this.stmts.findUnacknowledgedByProject.all(projectId) as NotificationRecord[];
58
+ }
59
+ return this.stmts.findUnacknowledgedAll.all() as NotificationRecord[];
60
+ }
61
+
62
+ acknowledge(id: number): boolean {
63
+ const result = this.stmts.acknowledge.run(id);
64
+ return result.changes > 0;
65
+ }
66
+ }
@@ -1,93 +1,93 @@
1
- import type Database from 'better-sqlite3';
2
- import type { Statement } from 'better-sqlite3';
3
-
4
- export interface ProjectRecord {
5
- id: number;
6
- name: string;
7
- path: string | null;
8
- language: string | null;
9
- framework: string | null;
10
- created_at: string;
11
- updated_at: string;
12
- }
13
-
14
- type CreateProjectData = Omit<ProjectRecord, 'id' | 'created_at' | 'updated_at'>;
15
- type UpdateProjectData = Partial<Omit<ProjectRecord, 'id' | 'created_at'>>;
16
-
17
- export class ProjectRepository {
18
- private stmts: Record<string, Statement>;
19
-
20
- constructor(private db: Database.Database) {
21
- this.stmts = {
22
- create: this.db.prepare(`
23
- INSERT INTO projects (name, path, language, framework)
24
- VALUES (@name, @path, @language, @framework)
25
- `),
26
- getById: this.db.prepare(`
27
- SELECT * FROM projects WHERE id = ?
28
- `),
29
- update: this.db.prepare(`
30
- UPDATE projects
31
- SET name = COALESCE(@name, name),
32
- path = COALESCE(@path, path),
33
- language = COALESCE(@language, language),
34
- framework = COALESCE(@framework, framework),
35
- updated_at = datetime('now')
36
- WHERE id = @id
37
- `),
38
- delete: this.db.prepare(`
39
- DELETE FROM projects WHERE id = ?
40
- `),
41
- findByName: this.db.prepare(`
42
- SELECT * FROM projects WHERE name = ?
43
- `),
44
- findByPath: this.db.prepare(`
45
- SELECT * FROM projects WHERE path = ?
46
- `),
47
- getAll: this.db.prepare(`
48
- SELECT * FROM projects ORDER BY name
49
- `),
50
- };
51
- }
52
-
53
- create(data: CreateProjectData): number {
54
- const result = this.stmts.create.run({
55
- name: data.name,
56
- path: data.path ?? null,
57
- language: data.language ?? null,
58
- framework: data.framework ?? null,
59
- });
60
- return result.lastInsertRowid as number;
61
- }
62
-
63
- getById(id: number): ProjectRecord | undefined {
64
- return this.stmts.getById.get(id) as ProjectRecord | undefined;
65
- }
66
-
67
- update(id: number, data: UpdateProjectData): void {
68
- this.stmts.update.run({
69
- id,
70
- name: data.name ?? null,
71
- path: data.path ?? null,
72
- language: data.language ?? null,
73
- framework: data.framework ?? null,
74
- updated_at: data.updated_at ?? null,
75
- });
76
- }
77
-
78
- delete(id: number): void {
79
- this.stmts.delete.run(id);
80
- }
81
-
82
- findByName(name: string): ProjectRecord | undefined {
83
- return this.stmts.findByName.get(name) as ProjectRecord | undefined;
84
- }
85
-
86
- findByPath(path: string): ProjectRecord | undefined {
87
- return this.stmts.findByPath.get(path) as ProjectRecord | undefined;
88
- }
89
-
90
- getAll(): ProjectRecord[] {
91
- return this.stmts.getAll.all() as ProjectRecord[];
92
- }
93
- }
1
+ import type Database from 'better-sqlite3';
2
+ import type { Statement } from 'better-sqlite3';
3
+
4
+ export interface ProjectRecord {
5
+ id: number;
6
+ name: string;
7
+ path: string | null;
8
+ language: string | null;
9
+ framework: string | null;
10
+ created_at: string;
11
+ updated_at: string;
12
+ }
13
+
14
+ type CreateProjectData = Omit<ProjectRecord, 'id' | 'created_at' | 'updated_at'>;
15
+ type UpdateProjectData = Partial<Omit<ProjectRecord, 'id' | 'created_at'>>;
16
+
17
+ export class ProjectRepository {
18
+ private stmts: Record<string, Statement>;
19
+
20
+ constructor(private db: Database.Database) {
21
+ this.stmts = {
22
+ create: this.db.prepare(`
23
+ INSERT INTO projects (name, path, language, framework)
24
+ VALUES (@name, @path, @language, @framework)
25
+ `),
26
+ getById: this.db.prepare(`
27
+ SELECT * FROM projects WHERE id = ?
28
+ `),
29
+ update: this.db.prepare(`
30
+ UPDATE projects
31
+ SET name = COALESCE(@name, name),
32
+ path = COALESCE(@path, path),
33
+ language = COALESCE(@language, language),
34
+ framework = COALESCE(@framework, framework),
35
+ updated_at = datetime('now')
36
+ WHERE id = @id
37
+ `),
38
+ delete: this.db.prepare(`
39
+ DELETE FROM projects WHERE id = ?
40
+ `),
41
+ findByName: this.db.prepare(`
42
+ SELECT * FROM projects WHERE name = ?
43
+ `),
44
+ findByPath: this.db.prepare(`
45
+ SELECT * FROM projects WHERE path = ?
46
+ `),
47
+ getAll: this.db.prepare(`
48
+ SELECT * FROM projects ORDER BY name
49
+ `),
50
+ };
51
+ }
52
+
53
+ create(data: CreateProjectData): number {
54
+ const result = this.stmts.create.run({
55
+ name: data.name,
56
+ path: data.path ?? null,
57
+ language: data.language ?? null,
58
+ framework: data.framework ?? null,
59
+ });
60
+ return result.lastInsertRowid as number;
61
+ }
62
+
63
+ getById(id: number): ProjectRecord | undefined {
64
+ return this.stmts.getById.get(id) as ProjectRecord | undefined;
65
+ }
66
+
67
+ update(id: number, data: UpdateProjectData): void {
68
+ this.stmts.update.run({
69
+ id,
70
+ name: data.name ?? null,
71
+ path: data.path ?? null,
72
+ language: data.language ?? null,
73
+ framework: data.framework ?? null,
74
+ updated_at: data.updated_at ?? null,
75
+ });
76
+ }
77
+
78
+ delete(id: number): void {
79
+ this.stmts.delete.run(id);
80
+ }
81
+
82
+ findByName(name: string): ProjectRecord | undefined {
83
+ return this.stmts.findByName.get(name) as ProjectRecord | undefined;
84
+ }
85
+
86
+ findByPath(path: string): ProjectRecord | undefined {
87
+ return this.stmts.findByPath.get(path) as ProjectRecord | undefined;
88
+ }
89
+
90
+ getAll(): ProjectRecord[] {
91
+ return this.stmts.getAll.all() as ProjectRecord[];
92
+ }
93
+ }
@@ -1,108 +1,108 @@
1
- import type Database from 'better-sqlite3';
2
- import type { Statement } from 'better-sqlite3';
3
-
4
- export interface RuleRecord {
5
- id: number;
6
- pattern: string;
7
- action: string;
8
- description: string | null;
9
- confidence: number;
10
- occurrences: number;
11
- active: number;
12
- project_id: number | null;
13
- created_at: string;
14
- updated_at: string;
15
- }
16
-
17
- type CreateRuleData = Omit<RuleRecord, 'id' | 'confidence' | 'occurrences' | 'active' | 'created_at' | 'updated_at'> & {
18
- confidence?: number;
19
- occurrences?: number;
20
- active?: number;
21
- };
22
- type UpdateRuleData = Partial<Omit<RuleRecord, 'id' | 'created_at'>>;
23
-
24
- export class RuleRepository {
25
- private stmts: Record<string, Statement>;
26
-
27
- constructor(private db: Database.Database) {
28
- this.stmts = {
29
- create: this.db.prepare(`
30
- INSERT INTO rules (pattern, action, description, confidence, occurrences, active, project_id)
31
- VALUES (@pattern, @action, @description, @confidence, @occurrences, @active, @project_id)
32
- `),
33
- getById: this.db.prepare(`
34
- SELECT * FROM rules WHERE id = ?
35
- `),
36
- update: this.db.prepare(`
37
- UPDATE rules
38
- SET pattern = COALESCE(@pattern, pattern),
39
- action = COALESCE(@action, action),
40
- description = COALESCE(@description, description),
41
- confidence = COALESCE(@confidence, confidence),
42
- occurrences = COALESCE(@occurrences, occurrences),
43
- active = COALESCE(@active, active),
44
- project_id = COALESCE(@project_id, project_id),
45
- updated_at = datetime('now')
46
- WHERE id = @id
47
- `),
48
- delete: this.db.prepare(`
49
- DELETE FROM rules WHERE id = ?
50
- `),
51
- findActiveAll: this.db.prepare(`
52
- SELECT * FROM rules WHERE active = 1 ORDER BY confidence DESC
53
- `),
54
- findActiveByProject: this.db.prepare(`
55
- SELECT * FROM rules WHERE active = 1 AND (project_id = ? OR project_id IS NULL) ORDER BY confidence DESC
56
- `),
57
- findByPattern: this.db.prepare(`
58
- SELECT * FROM rules WHERE pattern = ?
59
- `),
60
- };
61
- }
62
-
63
- create(data: CreateRuleData): number {
64
- const result = this.stmts.create.run({
65
- pattern: data.pattern,
66
- action: data.action,
67
- description: data.description ?? null,
68
- confidence: data.confidence ?? 0.5,
69
- occurrences: data.occurrences ?? 0,
70
- active: data.active ?? 1,
71
- project_id: data.project_id ?? null,
72
- });
73
- return result.lastInsertRowid as number;
74
- }
75
-
76
- getById(id: number): RuleRecord | undefined {
77
- return this.stmts.getById.get(id) as RuleRecord | undefined;
78
- }
79
-
80
- update(id: number, data: UpdateRuleData): void {
81
- this.stmts.update.run({
82
- id,
83
- pattern: data.pattern ?? null,
84
- action: data.action ?? null,
85
- description: data.description ?? null,
86
- confidence: data.confidence ?? null,
87
- occurrences: data.occurrences ?? null,
88
- active: data.active ?? null,
89
- project_id: data.project_id ?? null,
90
- updated_at: data.updated_at ?? null,
91
- });
92
- }
93
-
94
- delete(id: number): void {
95
- this.stmts.delete.run(id);
96
- }
97
-
98
- findActive(projectId?: number): RuleRecord[] {
99
- if (projectId !== undefined) {
100
- return this.stmts.findActiveByProject.all(projectId) as RuleRecord[];
101
- }
102
- return this.stmts.findActiveAll.all() as RuleRecord[];
103
- }
104
-
105
- findByPattern(pattern: string): RuleRecord[] {
106
- return this.stmts.findByPattern.all(pattern) as RuleRecord[];
107
- }
108
- }
1
+ import type Database from 'better-sqlite3';
2
+ import type { Statement } from 'better-sqlite3';
3
+
4
+ export interface RuleRecord {
5
+ id: number;
6
+ pattern: string;
7
+ action: string;
8
+ description: string | null;
9
+ confidence: number;
10
+ occurrences: number;
11
+ active: number;
12
+ project_id: number | null;
13
+ created_at: string;
14
+ updated_at: string;
15
+ }
16
+
17
+ type CreateRuleData = Omit<RuleRecord, 'id' | 'confidence' | 'occurrences' | 'active' | 'created_at' | 'updated_at'> & {
18
+ confidence?: number;
19
+ occurrences?: number;
20
+ active?: number;
21
+ };
22
+ type UpdateRuleData = Partial<Omit<RuleRecord, 'id' | 'created_at'>>;
23
+
24
+ export class RuleRepository {
25
+ private stmts: Record<string, Statement>;
26
+
27
+ constructor(private db: Database.Database) {
28
+ this.stmts = {
29
+ create: this.db.prepare(`
30
+ INSERT INTO rules (pattern, action, description, confidence, occurrences, active, project_id)
31
+ VALUES (@pattern, @action, @description, @confidence, @occurrences, @active, @project_id)
32
+ `),
33
+ getById: this.db.prepare(`
34
+ SELECT * FROM rules WHERE id = ?
35
+ `),
36
+ update: this.db.prepare(`
37
+ UPDATE rules
38
+ SET pattern = COALESCE(@pattern, pattern),
39
+ action = COALESCE(@action, action),
40
+ description = COALESCE(@description, description),
41
+ confidence = COALESCE(@confidence, confidence),
42
+ occurrences = COALESCE(@occurrences, occurrences),
43
+ active = COALESCE(@active, active),
44
+ project_id = COALESCE(@project_id, project_id),
45
+ updated_at = datetime('now')
46
+ WHERE id = @id
47
+ `),
48
+ delete: this.db.prepare(`
49
+ DELETE FROM rules WHERE id = ?
50
+ `),
51
+ findActiveAll: this.db.prepare(`
52
+ SELECT * FROM rules WHERE active = 1 ORDER BY confidence DESC
53
+ `),
54
+ findActiveByProject: this.db.prepare(`
55
+ SELECT * FROM rules WHERE active = 1 AND (project_id = ? OR project_id IS NULL) ORDER BY confidence DESC
56
+ `),
57
+ findByPattern: this.db.prepare(`
58
+ SELECT * FROM rules WHERE pattern = ?
59
+ `),
60
+ };
61
+ }
62
+
63
+ create(data: CreateRuleData): number {
64
+ const result = this.stmts.create.run({
65
+ pattern: data.pattern,
66
+ action: data.action,
67
+ description: data.description ?? null,
68
+ confidence: data.confidence ?? 0.5,
69
+ occurrences: data.occurrences ?? 0,
70
+ active: data.active ?? 1,
71
+ project_id: data.project_id ?? null,
72
+ });
73
+ return result.lastInsertRowid as number;
74
+ }
75
+
76
+ getById(id: number): RuleRecord | undefined {
77
+ return this.stmts.getById.get(id) as RuleRecord | undefined;
78
+ }
79
+
80
+ update(id: number, data: UpdateRuleData): void {
81
+ this.stmts.update.run({
82
+ id,
83
+ pattern: data.pattern ?? null,
84
+ action: data.action ?? null,
85
+ description: data.description ?? null,
86
+ confidence: data.confidence ?? null,
87
+ occurrences: data.occurrences ?? null,
88
+ active: data.active ?? null,
89
+ project_id: data.project_id ?? null,
90
+ updated_at: data.updated_at ?? null,
91
+ });
92
+ }
93
+
94
+ delete(id: number): void {
95
+ this.stmts.delete.run(id);
96
+ }
97
+
98
+ findActive(projectId?: number): RuleRecord[] {
99
+ if (projectId !== undefined) {
100
+ return this.stmts.findActiveByProject.all(projectId) as RuleRecord[];
101
+ }
102
+ return this.stmts.findActiveAll.all() as RuleRecord[];
103
+ }
104
+
105
+ findByPattern(pattern: string): RuleRecord[] {
106
+ return this.stmts.findByPattern.all(pattern) as RuleRecord[];
107
+ }
108
+ }