claude-brain 0.15.2 → 0.17.0

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 (250) hide show
  1. package/README.md +191 -191
  2. package/VERSION +1 -1
  3. package/assets/CLAUDE-unified.md +11 -11
  4. package/assets/CLAUDE.md +29 -11
  5. package/bunfig.toml +8 -8
  6. package/package.json +82 -82
  7. package/packs/backend/node.json +173 -173
  8. package/packs/core/javascript.json +176 -176
  9. package/packs/core/typescript.json +222 -222
  10. package/packs/frontend/react.json +254 -254
  11. package/packs/meta/testing.json +172 -172
  12. package/scripts/postinstall.mjs +341 -341
  13. package/src/automation/auto-context.ts +240 -240
  14. package/src/automation/decision-detector.ts +452 -452
  15. package/src/automation/index.ts +11 -11
  16. package/src/automation/phase12-manager.ts +456 -456
  17. package/src/automation/proactive-recall.ts +373 -373
  18. package/src/automation/project-detector.ts +310 -310
  19. package/src/automation/repo-scanner.ts +205 -205
  20. package/src/cli/auto-setup.ts +82 -82
  21. package/src/cli/bin.ts +209 -202
  22. package/src/cli/commands/chroma.ts +573 -573
  23. package/src/cli/commands/git-hook.ts +189 -189
  24. package/src/cli/commands/hooks.ts +213 -213
  25. package/src/cli/commands/init.ts +122 -122
  26. package/src/cli/commands/install-mcp.ts +92 -92
  27. package/src/cli/commands/pack.ts +197 -197
  28. package/src/cli/commands/refresh.ts +323 -0
  29. package/src/cli/commands/serve.ts +167 -173
  30. package/src/cli/commands/start.ts +42 -42
  31. package/src/cli/commands/uninstall-mcp.ts +41 -41
  32. package/src/cli/commands/update.ts +124 -121
  33. package/src/cli/diagnose.ts +4 -4
  34. package/src/cli/health-check.ts +4 -4
  35. package/src/cli/migrate-chroma.ts +106 -106
  36. package/src/cli/setup.ts +4 -4
  37. package/src/cli/ui/animations.ts +80 -80
  38. package/src/cli/ui/components.ts +82 -82
  39. package/src/cli/ui/index.ts +4 -4
  40. package/src/cli/ui/logo.ts +36 -36
  41. package/src/cli/ui/theme.ts +55 -55
  42. package/src/config/defaults.ts +50 -50
  43. package/src/config/home.ts +55 -55
  44. package/src/config/index.ts +7 -7
  45. package/src/config/loader.ts +166 -166
  46. package/src/config/migration.ts +76 -76
  47. package/src/config/schema.ts +360 -360
  48. package/src/config/validator.ts +184 -184
  49. package/src/config/watcher.ts +86 -86
  50. package/src/context/assembler.ts +398 -398
  51. package/src/context/cache-manager.ts +101 -101
  52. package/src/context/formatter.ts +84 -84
  53. package/src/context/hierarchy.ts +85 -85
  54. package/src/context/index.ts +83 -83
  55. package/src/context/progress-tracker.ts +174 -174
  56. package/src/context/standards-manager.ts +287 -287
  57. package/src/context/types.ts +252 -252
  58. package/src/context/validator.ts +58 -58
  59. package/src/diagnostics/index.ts +123 -123
  60. package/src/health/index.ts +229 -229
  61. package/src/hooks/brain-hook.ts +128 -112
  62. package/src/hooks/capture.ts +168 -205
  63. package/src/hooks/context-hook.ts +137 -0
  64. package/src/hooks/deduplicator.ts +72 -72
  65. package/src/hooks/git-capture.ts +109 -109
  66. package/src/hooks/git-hook-installer.ts +207 -207
  67. package/src/hooks/index.ts +20 -20
  68. package/src/hooks/installer.ts +244 -194
  69. package/src/hooks/passive-classifier.ts +404 -723
  70. package/src/hooks/queue.ts +129 -129
  71. package/src/hooks/session-tracker.ts +312 -275
  72. package/src/hooks/types.ts +52 -47
  73. package/src/index.ts +7 -7
  74. package/src/intelligence/cross-project/affinity.ts +162 -162
  75. package/src/intelligence/cross-project/generalizer.ts +283 -283
  76. package/src/intelligence/cross-project/index.ts +13 -13
  77. package/src/intelligence/cross-project/transfer.ts +201 -201
  78. package/src/intelligence/index.ts +24 -24
  79. package/src/intelligence/optimization/index.ts +10 -10
  80. package/src/intelligence/optimization/precompute.ts +202 -202
  81. package/src/intelligence/optimization/semantic-cache.ts +207 -207
  82. package/src/intelligence/prediction/context-anticipator.ts +198 -198
  83. package/src/intelligence/prediction/decision-predictor.ts +184 -184
  84. package/src/intelligence/prediction/index.ts +13 -13
  85. package/src/intelligence/prediction/recommender.ts +268 -268
  86. package/src/intelligence/reasoning/chain-retrieval.ts +247 -247
  87. package/src/intelligence/reasoning/counterfactual.ts +248 -248
  88. package/src/intelligence/reasoning/index.ts +13 -13
  89. package/src/intelligence/reasoning/synthesizer.ts +169 -169
  90. package/src/intelligence/temporal/evolution.ts +197 -197
  91. package/src/intelligence/temporal/index.ts +16 -16
  92. package/src/intelligence/temporal/query-processor.ts +190 -190
  93. package/src/intelligence/temporal/timeline.ts +259 -259
  94. package/src/intelligence/temporal/trends.ts +263 -263
  95. package/src/knowledge/entity-extractor.ts +416 -416
  96. package/src/knowledge/graph/builder.ts +185 -185
  97. package/src/knowledge/graph/linker.ts +201 -201
  98. package/src/knowledge/graph/memory-graph.ts +359 -359
  99. package/src/knowledge/graph/schema.ts +99 -99
  100. package/src/knowledge/graph/search.ts +168 -168
  101. package/src/knowledge/relationship-extractor.ts +108 -108
  102. package/src/memory/chroma/client.ts +174 -174
  103. package/src/memory/chroma/collection-manager.ts +94 -94
  104. package/src/memory/chroma/config.ts +57 -57
  105. package/src/memory/chroma/embeddings.ts +155 -155
  106. package/src/memory/chroma/index.ts +82 -82
  107. package/src/memory/chroma/migration.ts +270 -270
  108. package/src/memory/chroma/schemas.ts +69 -69
  109. package/src/memory/chroma/search.ts +315 -315
  110. package/src/memory/chroma/store.ts +741 -741
  111. package/src/memory/consolidation/archiver.ts +164 -164
  112. package/src/memory/consolidation/merger.ts +186 -186
  113. package/src/memory/consolidation/scorer.ts +138 -138
  114. package/src/memory/context-builder.ts +236 -236
  115. package/src/memory/database.ts +169 -169
  116. package/src/memory/embedding-utils.ts +156 -156
  117. package/src/memory/embeddings.ts +226 -226
  118. package/src/memory/episodic/detector.ts +108 -108
  119. package/src/memory/episodic/manager.ts +351 -351
  120. package/src/memory/episodic/summarizer.ts +179 -179
  121. package/src/memory/episodic/types.ts +52 -52
  122. package/src/memory/index.ts +582 -582
  123. package/src/memory/knowledge-extractor.ts +455 -455
  124. package/src/memory/learning.ts +378 -378
  125. package/src/memory/patterns.ts +396 -396
  126. package/src/memory/schema.ts +88 -88
  127. package/src/memory/search.ts +309 -309
  128. package/src/memory/store.ts +787 -787
  129. package/src/memory/types.ts +121 -121
  130. package/src/orchestrator/coordinator.ts +272 -272
  131. package/src/orchestrator/decision-logger.ts +228 -228
  132. package/src/orchestrator/event-emitter.ts +198 -198
  133. package/src/orchestrator/event-queue.ts +184 -184
  134. package/src/orchestrator/handlers/base-handler.ts +70 -70
  135. package/src/orchestrator/handlers/context-handler.ts +73 -73
  136. package/src/orchestrator/handlers/decision-handler.ts +204 -204
  137. package/src/orchestrator/handlers/index.ts +10 -10
  138. package/src/orchestrator/handlers/status-handler.ts +131 -131
  139. package/src/orchestrator/handlers/task-handler.ts +171 -171
  140. package/src/orchestrator/index.ts +275 -275
  141. package/src/orchestrator/task-parser.ts +284 -284
  142. package/src/orchestrator/types.ts +98 -98
  143. package/src/packs/index.ts +9 -9
  144. package/src/packs/loader.ts +134 -134
  145. package/src/packs/manager.ts +204 -204
  146. package/src/packs/ranker.ts +78 -78
  147. package/src/packs/types.ts +81 -81
  148. package/src/phase12/index.ts +5 -5
  149. package/src/retrieval/bm25/index.ts +300 -300
  150. package/src/retrieval/bm25/tokenizer.ts +184 -184
  151. package/src/retrieval/feedback/adaptive.ts +223 -223
  152. package/src/retrieval/feedback/index.ts +16 -16
  153. package/src/retrieval/feedback/metrics.ts +223 -223
  154. package/src/retrieval/feedback/store.ts +283 -283
  155. package/src/retrieval/fusion/index.ts +194 -194
  156. package/src/retrieval/fusion/rrf.ts +163 -163
  157. package/src/retrieval/index.ts +12 -12
  158. package/src/retrieval/pipeline.ts +375 -375
  159. package/src/retrieval/query/expander.ts +198 -198
  160. package/src/retrieval/query/index.ts +27 -27
  161. package/src/retrieval/query/intent-classifier.ts +236 -236
  162. package/src/retrieval/query/temporal-parser.ts +295 -295
  163. package/src/retrieval/reranker/index.ts +188 -188
  164. package/src/retrieval/reranker/model.ts +95 -95
  165. package/src/retrieval/service.ts +125 -125
  166. package/src/retrieval/types.ts +162 -162
  167. package/src/routing/entity-extractor.ts +428 -428
  168. package/src/routing/intent-classifier.ts +450 -436
  169. package/src/routing/response-filter.ts +261 -258
  170. package/src/routing/router.ts +1441 -1322
  171. package/src/routing/search-engine.ts +515 -475
  172. package/src/routing/types.ts +94 -94
  173. package/src/scripts/health-check.ts +118 -118
  174. package/src/scripts/setup.ts +122 -122
  175. package/src/server/handlers/call-tool.ts +156 -156
  176. package/src/server/handlers/index.ts +9 -9
  177. package/src/server/handlers/list-tools.ts +35 -35
  178. package/src/server/handlers/tools/analyze-decision-evolution.ts +151 -151
  179. package/src/server/handlers/tools/auto-remember.ts +200 -200
  180. package/src/server/handlers/tools/brain.ts +85 -85
  181. package/src/server/handlers/tools/create-project.ts +135 -135
  182. package/src/server/handlers/tools/detect-trends.ts +144 -144
  183. package/src/server/handlers/tools/find-cross-project-patterns.ts +168 -168
  184. package/src/server/handlers/tools/get-activity-log.ts +194 -194
  185. package/src/server/handlers/tools/get-code-standards.ts +124 -124
  186. package/src/server/handlers/tools/get-corrections.ts +154 -154
  187. package/src/server/handlers/tools/get-decision-timeline.ts +172 -172
  188. package/src/server/handlers/tools/get-episode.ts +103 -103
  189. package/src/server/handlers/tools/get-patterns.ts +158 -158
  190. package/src/server/handlers/tools/get-phase12-status.ts +63 -63
  191. package/src/server/handlers/tools/get-project-context.ts +75 -75
  192. package/src/server/handlers/tools/get-recommendations.ts +145 -145
  193. package/src/server/handlers/tools/index.ts +31 -31
  194. package/src/server/handlers/tools/init-project.ts +757 -757
  195. package/src/server/handlers/tools/list-episodes.ts +90 -90
  196. package/src/server/handlers/tools/list-projects.ts +125 -125
  197. package/src/server/handlers/tools/rate-memory.ts +101 -101
  198. package/src/server/handlers/tools/recall-similar.ts +87 -87
  199. package/src/server/handlers/tools/recognize-pattern.ts +126 -126
  200. package/src/server/handlers/tools/record-correction.ts +125 -125
  201. package/src/server/handlers/tools/remember-decision.ts +153 -153
  202. package/src/server/handlers/tools/schemas.ts +253 -253
  203. package/src/server/handlers/tools/search-knowledge-graph.ts +102 -102
  204. package/src/server/handlers/tools/smart-context.ts +146 -146
  205. package/src/server/handlers/tools/update-progress.ts +131 -131
  206. package/src/server/handlers/tools/what-if-analysis.ts +135 -135
  207. package/src/server/http-api.ts +761 -693
  208. package/src/server/index.ts +40 -40
  209. package/src/server/mcp-server.ts +283 -283
  210. package/src/server/providers/index.ts +7 -7
  211. package/src/server/providers/prompts.ts +327 -327
  212. package/src/server/providers/resources.ts +622 -622
  213. package/src/server/services.ts +468 -468
  214. package/src/server/types.ts +39 -39
  215. package/src/server/utils/error-handler.ts +155 -155
  216. package/src/server/utils/index.ts +13 -13
  217. package/src/server/utils/memory-indicator.ts +83 -83
  218. package/src/server/utils/request-context.ts +122 -122
  219. package/src/server/utils/response-formatter.ts +129 -129
  220. package/src/server/utils/validators.ts +210 -210
  221. package/src/setup/index.ts +48 -48
  222. package/src/setup/wizard.ts +461 -461
  223. package/src/tools/index.ts +24 -24
  224. package/src/tools/registry.ts +115 -115
  225. package/src/tools/schemas.test.ts +30 -30
  226. package/src/tools/schemas.ts +617 -617
  227. package/src/tools/types.ts +412 -412
  228. package/src/utils/circuit-breaker.ts +130 -130
  229. package/src/utils/cleanup.ts +34 -34
  230. package/src/utils/error-handler.ts +132 -132
  231. package/src/utils/error-messages.ts +60 -60
  232. package/src/utils/fallback.ts +45 -45
  233. package/src/utils/index.ts +54 -54
  234. package/src/utils/logger-utils.ts +80 -80
  235. package/src/utils/logger.ts +88 -88
  236. package/src/utils/phase12-helper.ts +56 -56
  237. package/src/utils/retry.ts +94 -94
  238. package/src/utils/timing.ts +47 -47
  239. package/src/utils/transaction.ts +63 -63
  240. package/src/vault/frontmatter.ts +264 -264
  241. package/src/vault/index.ts +318 -318
  242. package/src/vault/paths.ts +106 -106
  243. package/src/vault/query.ts +422 -422
  244. package/src/vault/reader.ts +264 -264
  245. package/src/vault/templates.ts +186 -186
  246. package/src/vault/types.ts +73 -73
  247. package/src/vault/watcher.ts +277 -277
  248. package/src/vault/writer.ts +413 -413
  249. package/tsconfig.json +30 -30
  250. package/src/cli/auto-update.ts +0 -157
@@ -1,461 +1,461 @@
1
- import prompts from 'prompts'
2
- import fs from 'fs/promises'
3
- import { existsSync } from 'fs'
4
- import { execSync } from 'child_process'
5
- import path from 'path'
6
- import os from 'os'
7
- import { fileURLToPath } from 'url'
8
- import type { Logger } from 'pino'
9
- import { getHomePaths } from '@/config/home'
10
- import {
11
- theme, heading, successText, errorText, warningText, dimText,
12
- box, stepIndicator, summaryPanel,
13
- withSpinner, transition,
14
- } from '@/cli/ui/index.js'
15
-
16
- const __filename = fileURLToPath(import.meta.url)
17
- const __dirname = path.dirname(__filename)
18
- // Package root is two levels up from src/setup/
19
- const PACKAGE_ROOT = path.resolve(__dirname, '..', '..')
20
-
21
- export interface SetupAnswers {
22
- vaultPath: string
23
- logLevel: string
24
- enableFileWatch: boolean
25
- enableChromaDB: boolean
26
- createSampleProject: boolean
27
- installClaudeMd: boolean
28
- }
29
-
30
- export class SetupWizard {
31
- private logger: Logger
32
-
33
- constructor(logger: Logger) {
34
- this.logger = logger.child({ component: 'setup-wizard' })
35
- }
36
-
37
- async run(): Promise<SetupAnswers> {
38
- // Step 1: Detect Vaults
39
- console.log(stepIndicator(1, 6, 'Detecting Obsidian Vaults'))
40
- await transition()
41
-
42
- const suggestedPaths = await withSpinner('Scanning for Obsidian vaults', () =>
43
- this.detectVaultLocations()
44
- )
45
-
46
- if (suggestedPaths.length > 0) {
47
- console.log(successText(`Found ${suggestedPaths.length} vault${suggestedPaths.length > 1 ? 's' : ''}`))
48
- } else {
49
- console.log(warningText('No vaults auto-detected — enter path manually'))
50
- }
51
- console.log()
52
-
53
- // Step 2: Vault Configuration
54
- console.log(stepIndicator(2, 6, 'Vault Configuration'))
55
- await transition()
56
-
57
- const vaultAnswers = await prompts([
58
- {
59
- type: 'select',
60
- name: 'vaultPathChoice',
61
- message: 'Where is your Obsidian vault?',
62
- choices: [
63
- ...suggestedPaths.map(p => ({ title: p, value: p })),
64
- { title: 'Enter custom path', value: 'custom' }
65
- ],
66
- initial: 0
67
- },
68
- {
69
- type: (prev) => prev === 'custom' ? 'text' : null,
70
- name: 'vaultPath',
71
- message: 'Enter vault path:',
72
- validate: async (value: string) => {
73
- try {
74
- const stat = await fs.stat(value)
75
- return stat.isDirectory() || 'Path must be a directory'
76
- } catch {
77
- return 'Path does not exist'
78
- }
79
- }
80
- },
81
- ])
82
-
83
- if (!vaultAnswers.vaultPathChoice) {
84
- console.log('\n' + errorText('Setup cancelled.'))
85
- process.exit(0)
86
- }
87
-
88
- const finalVaultPath = vaultAnswers.vaultPath || vaultAnswers.vaultPathChoice
89
-
90
- // Step 3: Logging
91
- console.log(stepIndicator(3, 6, 'Logging Configuration'))
92
- await transition()
93
-
94
- const loggingAnswers = await prompts({
95
- type: 'select',
96
- name: 'logLevel',
97
- message: 'Select log level:',
98
- choices: [
99
- { title: 'Error (production)', value: 'error' },
100
- { title: 'Warn (recommended)', value: 'warn' },
101
- { title: 'Info', value: 'info' },
102
- { title: 'Debug (verbose)', value: 'debug' }
103
- ],
104
- initial: 1
105
- })
106
-
107
- // Step 4: Features
108
- console.log(stepIndicator(4, 6, 'Feature Selection'))
109
- await transition()
110
-
111
- const featureAnswers = await prompts([
112
- {
113
- type: 'confirm',
114
- name: 'enableFileWatch',
115
- message: 'Enable automatic file watching?',
116
- initial: true
117
- },
118
- {
119
- type: 'confirm',
120
- name: 'createSampleProject',
121
- message: 'Create a sample project to test with?',
122
- initial: true
123
- },
124
- {
125
- type: 'confirm',
126
- name: 'installClaudeMd',
127
- message: 'Install CLAUDE.md protocol file to ~/.claude/CLAUDE.md?',
128
- initial: true
129
- }
130
- ])
131
-
132
- // Step 5: ChromaDB
133
- console.log(stepIndicator(5, 6, 'ChromaDB (Vector Database)'))
134
- await transition()
135
-
136
- console.log(dimText('ChromaDB enables semantic search, knowledge graph,'))
137
- console.log(dimText('and advanced intelligence features. Requires Python 3.'))
138
- console.log()
139
-
140
- const chromaInstalled = this.isChromaInstalled()
141
- if (chromaInstalled) {
142
- console.log(successText('ChromaDB CLI detected'))
143
- } else {
144
- console.log(warningText('ChromaDB CLI not found — you can install it later'))
145
- }
146
- console.log()
147
-
148
- const chromaAnswers = await prompts({
149
- type: 'confirm',
150
- name: 'enableChromaDB',
151
- message: chromaInstalled
152
- ? 'Enable ChromaDB for advanced features?'
153
- : 'Enable ChromaDB? (install later: pip install chromadb)',
154
- initial: chromaInstalled
155
- })
156
-
157
- // Step 6: Review
158
- console.log(stepIndicator(6, 6, 'Review Configuration'))
159
- await transition()
160
-
161
- const answers: SetupAnswers = {
162
- vaultPath: finalVaultPath,
163
- logLevel: loggingAnswers.logLevel ?? 'warn',
164
- enableFileWatch: featureAnswers.enableFileWatch ?? true,
165
- enableChromaDB: chromaAnswers.enableChromaDB ?? false,
166
- createSampleProject: featureAnswers.createSampleProject ?? true,
167
- installClaudeMd: featureAnswers.installClaudeMd ?? true,
168
- }
169
-
170
- console.log(summaryPanel('Configuration Summary', [
171
- { label: 'Vault Path', value: answers.vaultPath, status: 'success' },
172
- { label: 'Log Level', value: answers.logLevel, status: 'info' },
173
- { label: 'File Watching', value: answers.enableFileWatch ? 'Enabled' : 'Disabled', status: answers.enableFileWatch ? 'success' : 'warning' },
174
- { label: 'ChromaDB', value: answers.enableChromaDB ? 'Enabled' : 'Disabled', status: answers.enableChromaDB ? 'success' : 'warning' },
175
- { label: 'Sample Project', value: answers.createSampleProject ? 'Yes' : 'No', status: 'info' },
176
- { label: 'Install CLAUDE.md', value: answers.installClaudeMd ? 'Yes' : 'No', status: 'info' },
177
- ]))
178
- console.log()
179
-
180
- const { confirm } = await prompts({
181
- type: 'confirm',
182
- name: 'confirm',
183
- message: 'Apply this configuration?',
184
- initial: true
185
- })
186
-
187
- if (!confirm) {
188
- console.log('\n' + errorText('Setup cancelled.'))
189
- process.exit(0)
190
- }
191
-
192
- return answers
193
- }
194
-
195
- private async detectVaultLocations(): Promise<string[]> {
196
- const locations: string[] = []
197
- const home = os.homedir()
198
- const homePaths = getHomePaths()
199
-
200
- const commonPaths = [
201
- homePaths.vault,
202
- path.join(home, 'Documents', 'Obsidian'),
203
- path.join(home, 'Obsidian'),
204
- path.join(home, 'vault'),
205
- path.join(home, 'Documents', 'vault'),
206
- path.join(home, 'iCloud', 'Obsidian'),
207
- path.join(home, 'Library', 'Mobile Documents', 'iCloud~md~obsidian', 'Documents')
208
- ]
209
-
210
- for (const loc of commonPaths) {
211
- try {
212
- const stat = await fs.stat(loc)
213
- if (stat.isDirectory()) {
214
- locations.push(loc)
215
- }
216
- } catch {
217
- }
218
- }
219
-
220
- return locations
221
- }
222
-
223
- private isChromaInstalled(): boolean {
224
- try {
225
- execSync('chroma --version', { stdio: 'pipe', timeout: 5000 })
226
- return true
227
- } catch {
228
- return false
229
- }
230
- }
231
-
232
- async applyConfiguration(answers: SetupAnswers): Promise<void> {
233
- console.log('\n' + heading('Applying configuration...') + '\n')
234
-
235
- const homePaths = getHomePaths()
236
-
237
- const chromaEnvLines = answers.enableChromaDB
238
- ? [
239
- '',
240
- '# ChromaDB Configuration',
241
- 'CHROMA_MODE=client-server',
242
- 'CHROMA_HOST=localhost',
243
- 'CHROMA_PORT=8000',
244
- 'CHROMA_EMBEDDING_PROVIDER=transformers',
245
- ].join('\n')
246
- : ''
247
-
248
- const envContent = `# Claude Brain Configuration
249
- VAULT_PATH=${answers.vaultPath}
250
- LOG_LEVEL=${answers.logLevel}
251
- ENABLE_FILE_WATCH=${answers.enableFileWatch}
252
- DB_PATH=./data/memory.db
253
- LOG_FILE_PATH=./logs/claude-brain.log
254
- SERVER_NAME=claude-brain
255
- ${chromaEnvLines}
256
- `
257
-
258
- await withSpinner('Writing .env configuration', async () => {
259
- await fs.writeFile(homePaths.env, envContent, 'utf-8')
260
- })
261
-
262
- await withSpinner('Creating data and log directories', async () => {
263
- await fs.mkdir(homePaths.data, { recursive: true })
264
- await fs.mkdir(homePaths.chroma, { recursive: true })
265
- await fs.mkdir(homePaths.logs, { recursive: true })
266
- })
267
-
268
- if (answers.createSampleProject) {
269
- await withSpinner('Creating sample project in vault', async () => {
270
- await this.createSampleProject(answers.vaultPath)
271
- })
272
- }
273
-
274
- if (answers.installClaudeMd) {
275
- const shouldInstall = await this.confirmClaudeMdInstall()
276
- if (shouldInstall) {
277
- await withSpinner('Installing CLAUDE.md', async () => {
278
- await this.copyClaudeMd()
279
- })
280
- }
281
- }
282
-
283
- const nextSteps: string[] = [
284
- heading('Setup complete!'),
285
- '',
286
- dimText('Next steps:'),
287
- ` ${theme.primary('1.')} ${theme.bold('claude-brain install')}`,
288
- ` ${dimText('Register as MCP server')}`,
289
- ]
290
-
291
- if (answers.enableChromaDB) {
292
- nextSteps.push(
293
- ` ${theme.primary('2.')} ${theme.bold('claude-brain chroma install')}`,
294
- ` ${dimText('Install ChromaDB (if not already installed)')}`,
295
- ` ${theme.primary('3.')} ${theme.bold('claude-brain chroma start')}`,
296
- ` ${dimText('Start the ChromaDB server')}`,
297
- ` ${theme.primary('4.')} ${theme.bold('claude-brain health')}`,
298
- ` ${dimText('Verify everything works')}`,
299
- )
300
- } else {
301
- nextSteps.push(
302
- ` ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
303
- ` ${dimText('Verify everything works')}`,
304
- '',
305
- dimText('Optional: Enable ChromaDB later for advanced features:'),
306
- ` ${dimText('claude-brain chroma install && claude-brain chroma start')}`,
307
- )
308
- }
309
-
310
- console.log()
311
- console.log(box(nextSteps.join('\n'), 'Done'))
312
- console.log()
313
- }
314
-
315
- private async confirmClaudeMdInstall(): Promise<boolean> {
316
- const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
317
- const destPath = path.join(os.homedir(), '.claude', 'CLAUDE.md')
318
-
319
- if (!existsSync(sourcePath)) {
320
- console.log(warningText('CLAUDE.md asset not found, skipping'))
321
- return false
322
- }
323
-
324
- if (existsSync(destPath)) {
325
- const { overwrite } = await prompts({
326
- type: 'confirm',
327
- name: 'overwrite',
328
- message: '~/.claude/CLAUDE.md already exists. Overwrite?',
329
- initial: false,
330
- })
331
- if (!overwrite) {
332
- console.log(dimText(' Skipped CLAUDE.md installation'))
333
- return false
334
- }
335
- }
336
-
337
- return true
338
- }
339
-
340
- private async copyClaudeMd(): Promise<void> {
341
- const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
342
- const claudeDir = path.join(os.homedir(), '.claude')
343
- const destPath = path.join(claudeDir, 'CLAUDE.md')
344
- await fs.mkdir(claudeDir, { recursive: true })
345
- await fs.copyFile(sourcePath, destPath)
346
- }
347
-
348
- private async createSampleProject(vaultPath: string): Promise<void> {
349
- const projectPath = path.join(vaultPath, 'Projects', 'sample-project')
350
-
351
- await fs.mkdir(projectPath, { recursive: true })
352
-
353
- await fs.writeFile(
354
- path.join(projectPath, 'context.md'),
355
- `---
356
- type: project-context
357
- project: sample-project
358
- status: active
359
- created: ${new Date().toISOString().split('T')[0]}
360
- updated: ${new Date().toISOString().split('T')[0]}
361
- tech_stack:
362
- - TypeScript
363
- - Node.js
364
- tags:
365
- - sample
366
- ---
367
-
368
- # Sample Project
369
-
370
- ## Overview
371
- This is a sample project created by Claude Brain setup wizard.
372
-
373
- ## Tech Stack
374
- - TypeScript
375
- - Node.js
376
- - Bun
377
-
378
- ## Goals
379
- - Test Claude Brain integration
380
- - Learn the workflow
381
- `,
382
- 'utf-8'
383
- )
384
-
385
- await fs.writeFile(
386
- path.join(projectPath, 'progress.md'),
387
- `---
388
- type: progress-tracker
389
- project: sample-project
390
- status: in-progress
391
- current_phase: setup
392
- completion_percentage: 0
393
- ---
394
-
395
- # Progress
396
-
397
- ## Next Steps
398
- - [ ] Configure Claude Brain
399
- - [ ] Test MCP integration
400
- - [ ] Try all tools
401
- `,
402
- 'utf-8'
403
- )
404
-
405
- await fs.writeFile(
406
- path.join(projectPath, 'decisions.md'),
407
- `---
408
- type: decision-log
409
- project: sample-project
410
- decision_count: 0
411
- ---
412
-
413
- # Decision Log
414
-
415
- This file will track important decisions for the project.
416
- `,
417
- 'utf-8'
418
- )
419
-
420
- await fs.writeFile(
421
- path.join(projectPath, 'standards.md'),
422
- `---
423
- type: coding-standards
424
- project: sample-project
425
- ---
426
-
427
- # Coding Standards
428
-
429
- ## TypeScript
430
- - Use strict mode
431
- - Prefer const over let
432
- `,
433
- 'utf-8'
434
- )
435
-
436
- const globalPath = path.join(vaultPath, 'Global')
437
- await fs.mkdir(globalPath, { recursive: true })
438
-
439
- const globalStandardsPath = path.join(globalPath, 'standards.md')
440
- try {
441
- await fs.access(globalStandardsPath)
442
- } catch {
443
- await fs.writeFile(
444
- globalStandardsPath,
445
- `---
446
- type: global-standards
447
- last_updated: ${new Date().toISOString().split('T')[0]}
448
- ---
449
-
450
- # Global Coding Standards
451
-
452
- ## TypeScript
453
- - Use strict mode
454
- - Prefer const over let
455
- - Add JSDoc comments for public APIs
456
- `,
457
- 'utf-8'
458
- )
459
- }
460
- }
461
- }
1
+ import prompts from 'prompts'
2
+ import fs from 'fs/promises'
3
+ import { existsSync } from 'fs'
4
+ import { execSync } from 'child_process'
5
+ import path from 'path'
6
+ import os from 'os'
7
+ import { fileURLToPath } from 'url'
8
+ import type { Logger } from 'pino'
9
+ import { getHomePaths } from '@/config/home'
10
+ import {
11
+ theme, heading, successText, errorText, warningText, dimText,
12
+ box, stepIndicator, summaryPanel,
13
+ withSpinner, transition,
14
+ } from '@/cli/ui/index.js'
15
+
16
+ const __filename = fileURLToPath(import.meta.url)
17
+ const __dirname = path.dirname(__filename)
18
+ // Package root is two levels up from src/setup/
19
+ const PACKAGE_ROOT = path.resolve(__dirname, '..', '..')
20
+
21
+ export interface SetupAnswers {
22
+ vaultPath: string
23
+ logLevel: string
24
+ enableFileWatch: boolean
25
+ enableChromaDB: boolean
26
+ createSampleProject: boolean
27
+ installClaudeMd: boolean
28
+ }
29
+
30
+ export class SetupWizard {
31
+ private logger: Logger
32
+
33
+ constructor(logger: Logger) {
34
+ this.logger = logger.child({ component: 'setup-wizard' })
35
+ }
36
+
37
+ async run(): Promise<SetupAnswers> {
38
+ // Step 1: Detect Vaults
39
+ console.log(stepIndicator(1, 6, 'Detecting Obsidian Vaults'))
40
+ await transition()
41
+
42
+ const suggestedPaths = await withSpinner('Scanning for Obsidian vaults', () =>
43
+ this.detectVaultLocations()
44
+ )
45
+
46
+ if (suggestedPaths.length > 0) {
47
+ console.log(successText(`Found ${suggestedPaths.length} vault${suggestedPaths.length > 1 ? 's' : ''}`))
48
+ } else {
49
+ console.log(warningText('No vaults auto-detected — enter path manually'))
50
+ }
51
+ console.log()
52
+
53
+ // Step 2: Vault Configuration
54
+ console.log(stepIndicator(2, 6, 'Vault Configuration'))
55
+ await transition()
56
+
57
+ const vaultAnswers = await prompts([
58
+ {
59
+ type: 'select',
60
+ name: 'vaultPathChoice',
61
+ message: 'Where is your Obsidian vault?',
62
+ choices: [
63
+ ...suggestedPaths.map(p => ({ title: p, value: p })),
64
+ { title: 'Enter custom path', value: 'custom' }
65
+ ],
66
+ initial: 0
67
+ },
68
+ {
69
+ type: (prev) => prev === 'custom' ? 'text' : null,
70
+ name: 'vaultPath',
71
+ message: 'Enter vault path:',
72
+ validate: async (value: string) => {
73
+ try {
74
+ const stat = await fs.stat(value)
75
+ return stat.isDirectory() || 'Path must be a directory'
76
+ } catch {
77
+ return 'Path does not exist'
78
+ }
79
+ }
80
+ },
81
+ ])
82
+
83
+ if (!vaultAnswers.vaultPathChoice) {
84
+ console.log('\n' + errorText('Setup cancelled.'))
85
+ process.exit(0)
86
+ }
87
+
88
+ const finalVaultPath = vaultAnswers.vaultPath || vaultAnswers.vaultPathChoice
89
+
90
+ // Step 3: Logging
91
+ console.log(stepIndicator(3, 6, 'Logging Configuration'))
92
+ await transition()
93
+
94
+ const loggingAnswers = await prompts({
95
+ type: 'select',
96
+ name: 'logLevel',
97
+ message: 'Select log level:',
98
+ choices: [
99
+ { title: 'Error (production)', value: 'error' },
100
+ { title: 'Warn (recommended)', value: 'warn' },
101
+ { title: 'Info', value: 'info' },
102
+ { title: 'Debug (verbose)', value: 'debug' }
103
+ ],
104
+ initial: 1
105
+ })
106
+
107
+ // Step 4: Features
108
+ console.log(stepIndicator(4, 6, 'Feature Selection'))
109
+ await transition()
110
+
111
+ const featureAnswers = await prompts([
112
+ {
113
+ type: 'confirm',
114
+ name: 'enableFileWatch',
115
+ message: 'Enable automatic file watching?',
116
+ initial: true
117
+ },
118
+ {
119
+ type: 'confirm',
120
+ name: 'createSampleProject',
121
+ message: 'Create a sample project to test with?',
122
+ initial: true
123
+ },
124
+ {
125
+ type: 'confirm',
126
+ name: 'installClaudeMd',
127
+ message: 'Install CLAUDE.md protocol file to ~/.claude/CLAUDE.md?',
128
+ initial: true
129
+ }
130
+ ])
131
+
132
+ // Step 5: ChromaDB
133
+ console.log(stepIndicator(5, 6, 'ChromaDB (Vector Database)'))
134
+ await transition()
135
+
136
+ console.log(dimText('ChromaDB enables semantic search, knowledge graph,'))
137
+ console.log(dimText('and advanced intelligence features. Requires Python 3.'))
138
+ console.log()
139
+
140
+ const chromaInstalled = this.isChromaInstalled()
141
+ if (chromaInstalled) {
142
+ console.log(successText('ChromaDB CLI detected'))
143
+ } else {
144
+ console.log(warningText('ChromaDB CLI not found — you can install it later'))
145
+ }
146
+ console.log()
147
+
148
+ const chromaAnswers = await prompts({
149
+ type: 'confirm',
150
+ name: 'enableChromaDB',
151
+ message: chromaInstalled
152
+ ? 'Enable ChromaDB for advanced features?'
153
+ : 'Enable ChromaDB? (install later: pip install chromadb)',
154
+ initial: chromaInstalled
155
+ })
156
+
157
+ // Step 6: Review
158
+ console.log(stepIndicator(6, 6, 'Review Configuration'))
159
+ await transition()
160
+
161
+ const answers: SetupAnswers = {
162
+ vaultPath: finalVaultPath,
163
+ logLevel: loggingAnswers.logLevel ?? 'warn',
164
+ enableFileWatch: featureAnswers.enableFileWatch ?? true,
165
+ enableChromaDB: chromaAnswers.enableChromaDB ?? false,
166
+ createSampleProject: featureAnswers.createSampleProject ?? true,
167
+ installClaudeMd: featureAnswers.installClaudeMd ?? true,
168
+ }
169
+
170
+ console.log(summaryPanel('Configuration Summary', [
171
+ { label: 'Vault Path', value: answers.vaultPath, status: 'success' },
172
+ { label: 'Log Level', value: answers.logLevel, status: 'info' },
173
+ { label: 'File Watching', value: answers.enableFileWatch ? 'Enabled' : 'Disabled', status: answers.enableFileWatch ? 'success' : 'warning' },
174
+ { label: 'ChromaDB', value: answers.enableChromaDB ? 'Enabled' : 'Disabled', status: answers.enableChromaDB ? 'success' : 'warning' },
175
+ { label: 'Sample Project', value: answers.createSampleProject ? 'Yes' : 'No', status: 'info' },
176
+ { label: 'Install CLAUDE.md', value: answers.installClaudeMd ? 'Yes' : 'No', status: 'info' },
177
+ ]))
178
+ console.log()
179
+
180
+ const { confirm } = await prompts({
181
+ type: 'confirm',
182
+ name: 'confirm',
183
+ message: 'Apply this configuration?',
184
+ initial: true
185
+ })
186
+
187
+ if (!confirm) {
188
+ console.log('\n' + errorText('Setup cancelled.'))
189
+ process.exit(0)
190
+ }
191
+
192
+ return answers
193
+ }
194
+
195
+ private async detectVaultLocations(): Promise<string[]> {
196
+ const locations: string[] = []
197
+ const home = os.homedir()
198
+ const homePaths = getHomePaths()
199
+
200
+ const commonPaths = [
201
+ homePaths.vault,
202
+ path.join(home, 'Documents', 'Obsidian'),
203
+ path.join(home, 'Obsidian'),
204
+ path.join(home, 'vault'),
205
+ path.join(home, 'Documents', 'vault'),
206
+ path.join(home, 'iCloud', 'Obsidian'),
207
+ path.join(home, 'Library', 'Mobile Documents', 'iCloud~md~obsidian', 'Documents')
208
+ ]
209
+
210
+ for (const loc of commonPaths) {
211
+ try {
212
+ const stat = await fs.stat(loc)
213
+ if (stat.isDirectory()) {
214
+ locations.push(loc)
215
+ }
216
+ } catch {
217
+ }
218
+ }
219
+
220
+ return locations
221
+ }
222
+
223
+ private isChromaInstalled(): boolean {
224
+ try {
225
+ execSync('chroma --version', { stdio: 'pipe', timeout: 5000 })
226
+ return true
227
+ } catch {
228
+ return false
229
+ }
230
+ }
231
+
232
+ async applyConfiguration(answers: SetupAnswers): Promise<void> {
233
+ console.log('\n' + heading('Applying configuration...') + '\n')
234
+
235
+ const homePaths = getHomePaths()
236
+
237
+ const chromaEnvLines = answers.enableChromaDB
238
+ ? [
239
+ '',
240
+ '# ChromaDB Configuration',
241
+ 'CHROMA_MODE=client-server',
242
+ 'CHROMA_HOST=localhost',
243
+ 'CHROMA_PORT=8000',
244
+ 'CHROMA_EMBEDDING_PROVIDER=transformers',
245
+ ].join('\n')
246
+ : ''
247
+
248
+ const envContent = `# Claude Brain Configuration
249
+ VAULT_PATH=${answers.vaultPath}
250
+ LOG_LEVEL=${answers.logLevel}
251
+ ENABLE_FILE_WATCH=${answers.enableFileWatch}
252
+ DB_PATH=./data/memory.db
253
+ LOG_FILE_PATH=./logs/claude-brain.log
254
+ SERVER_NAME=claude-brain
255
+ ${chromaEnvLines}
256
+ `
257
+
258
+ await withSpinner('Writing .env configuration', async () => {
259
+ await fs.writeFile(homePaths.env, envContent, 'utf-8')
260
+ })
261
+
262
+ await withSpinner('Creating data and log directories', async () => {
263
+ await fs.mkdir(homePaths.data, { recursive: true })
264
+ await fs.mkdir(homePaths.chroma, { recursive: true })
265
+ await fs.mkdir(homePaths.logs, { recursive: true })
266
+ })
267
+
268
+ if (answers.createSampleProject) {
269
+ await withSpinner('Creating sample project in vault', async () => {
270
+ await this.createSampleProject(answers.vaultPath)
271
+ })
272
+ }
273
+
274
+ if (answers.installClaudeMd) {
275
+ const shouldInstall = await this.confirmClaudeMdInstall()
276
+ if (shouldInstall) {
277
+ await withSpinner('Installing CLAUDE.md', async () => {
278
+ await this.copyClaudeMd()
279
+ })
280
+ }
281
+ }
282
+
283
+ const nextSteps: string[] = [
284
+ heading('Setup complete!'),
285
+ '',
286
+ dimText('Next steps:'),
287
+ ` ${theme.primary('1.')} ${theme.bold('claude-brain install')}`,
288
+ ` ${dimText('Register as MCP server')}`,
289
+ ]
290
+
291
+ if (answers.enableChromaDB) {
292
+ nextSteps.push(
293
+ ` ${theme.primary('2.')} ${theme.bold('claude-brain chroma install')}`,
294
+ ` ${dimText('Install ChromaDB (if not already installed)')}`,
295
+ ` ${theme.primary('3.')} ${theme.bold('claude-brain chroma start')}`,
296
+ ` ${dimText('Start the ChromaDB server')}`,
297
+ ` ${theme.primary('4.')} ${theme.bold('claude-brain health')}`,
298
+ ` ${dimText('Verify everything works')}`,
299
+ )
300
+ } else {
301
+ nextSteps.push(
302
+ ` ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
303
+ ` ${dimText('Verify everything works')}`,
304
+ '',
305
+ dimText('Optional: Enable ChromaDB later for advanced features:'),
306
+ ` ${dimText('claude-brain chroma install && claude-brain chroma start')}`,
307
+ )
308
+ }
309
+
310
+ console.log()
311
+ console.log(box(nextSteps.join('\n'), 'Done'))
312
+ console.log()
313
+ }
314
+
315
+ private async confirmClaudeMdInstall(): Promise<boolean> {
316
+ const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
317
+ const destPath = path.join(os.homedir(), '.claude', 'CLAUDE.md')
318
+
319
+ if (!existsSync(sourcePath)) {
320
+ console.log(warningText('CLAUDE.md asset not found, skipping'))
321
+ return false
322
+ }
323
+
324
+ if (existsSync(destPath)) {
325
+ const { overwrite } = await prompts({
326
+ type: 'confirm',
327
+ name: 'overwrite',
328
+ message: '~/.claude/CLAUDE.md already exists. Overwrite?',
329
+ initial: false,
330
+ })
331
+ if (!overwrite) {
332
+ console.log(dimText(' Skipped CLAUDE.md installation'))
333
+ return false
334
+ }
335
+ }
336
+
337
+ return true
338
+ }
339
+
340
+ private async copyClaudeMd(): Promise<void> {
341
+ const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
342
+ const claudeDir = path.join(os.homedir(), '.claude')
343
+ const destPath = path.join(claudeDir, 'CLAUDE.md')
344
+ await fs.mkdir(claudeDir, { recursive: true })
345
+ await fs.copyFile(sourcePath, destPath)
346
+ }
347
+
348
+ private async createSampleProject(vaultPath: string): Promise<void> {
349
+ const projectPath = path.join(vaultPath, 'Projects', 'sample-project')
350
+
351
+ await fs.mkdir(projectPath, { recursive: true })
352
+
353
+ await fs.writeFile(
354
+ path.join(projectPath, 'context.md'),
355
+ `---
356
+ type: project-context
357
+ project: sample-project
358
+ status: active
359
+ created: ${new Date().toISOString().split('T')[0]}
360
+ updated: ${new Date().toISOString().split('T')[0]}
361
+ tech_stack:
362
+ - TypeScript
363
+ - Node.js
364
+ tags:
365
+ - sample
366
+ ---
367
+
368
+ # Sample Project
369
+
370
+ ## Overview
371
+ This is a sample project created by Claude Brain setup wizard.
372
+
373
+ ## Tech Stack
374
+ - TypeScript
375
+ - Node.js
376
+ - Bun
377
+
378
+ ## Goals
379
+ - Test Claude Brain integration
380
+ - Learn the workflow
381
+ `,
382
+ 'utf-8'
383
+ )
384
+
385
+ await fs.writeFile(
386
+ path.join(projectPath, 'progress.md'),
387
+ `---
388
+ type: progress-tracker
389
+ project: sample-project
390
+ status: in-progress
391
+ current_phase: setup
392
+ completion_percentage: 0
393
+ ---
394
+
395
+ # Progress
396
+
397
+ ## Next Steps
398
+ - [ ] Configure Claude Brain
399
+ - [ ] Test MCP integration
400
+ - [ ] Try all tools
401
+ `,
402
+ 'utf-8'
403
+ )
404
+
405
+ await fs.writeFile(
406
+ path.join(projectPath, 'decisions.md'),
407
+ `---
408
+ type: decision-log
409
+ project: sample-project
410
+ decision_count: 0
411
+ ---
412
+
413
+ # Decision Log
414
+
415
+ This file will track important decisions for the project.
416
+ `,
417
+ 'utf-8'
418
+ )
419
+
420
+ await fs.writeFile(
421
+ path.join(projectPath, 'standards.md'),
422
+ `---
423
+ type: coding-standards
424
+ project: sample-project
425
+ ---
426
+
427
+ # Coding Standards
428
+
429
+ ## TypeScript
430
+ - Use strict mode
431
+ - Prefer const over let
432
+ `,
433
+ 'utf-8'
434
+ )
435
+
436
+ const globalPath = path.join(vaultPath, 'Global')
437
+ await fs.mkdir(globalPath, { recursive: true })
438
+
439
+ const globalStandardsPath = path.join(globalPath, 'standards.md')
440
+ try {
441
+ await fs.access(globalStandardsPath)
442
+ } catch {
443
+ await fs.writeFile(
444
+ globalStandardsPath,
445
+ `---
446
+ type: global-standards
447
+ last_updated: ${new Date().toISOString().split('T')[0]}
448
+ ---
449
+
450
+ # Global Coding Standards
451
+
452
+ ## TypeScript
453
+ - Use strict mode
454
+ - Prefer const over let
455
+ - Add JSDoc comments for public APIs
456
+ `,
457
+ 'utf-8'
458
+ )
459
+ }
460
+ }
461
+ }