claude-brain 0.15.2 → 0.16.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 (249) 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/deduplicator.ts +72 -72
  64. package/src/hooks/git-capture.ts +109 -109
  65. package/src/hooks/git-hook-installer.ts +207 -207
  66. package/src/hooks/index.ts +20 -20
  67. package/src/hooks/installer.ts +194 -194
  68. package/src/hooks/passive-classifier.ts +404 -723
  69. package/src/hooks/queue.ts +129 -129
  70. package/src/hooks/session-tracker.ts +312 -275
  71. package/src/hooks/types.ts +47 -47
  72. package/src/index.ts +7 -7
  73. package/src/intelligence/cross-project/affinity.ts +162 -162
  74. package/src/intelligence/cross-project/generalizer.ts +283 -283
  75. package/src/intelligence/cross-project/index.ts +13 -13
  76. package/src/intelligence/cross-project/transfer.ts +201 -201
  77. package/src/intelligence/index.ts +24 -24
  78. package/src/intelligence/optimization/index.ts +10 -10
  79. package/src/intelligence/optimization/precompute.ts +202 -202
  80. package/src/intelligence/optimization/semantic-cache.ts +207 -207
  81. package/src/intelligence/prediction/context-anticipator.ts +198 -198
  82. package/src/intelligence/prediction/decision-predictor.ts +184 -184
  83. package/src/intelligence/prediction/index.ts +13 -13
  84. package/src/intelligence/prediction/recommender.ts +268 -268
  85. package/src/intelligence/reasoning/chain-retrieval.ts +247 -247
  86. package/src/intelligence/reasoning/counterfactual.ts +248 -248
  87. package/src/intelligence/reasoning/index.ts +13 -13
  88. package/src/intelligence/reasoning/synthesizer.ts +169 -169
  89. package/src/intelligence/temporal/evolution.ts +197 -197
  90. package/src/intelligence/temporal/index.ts +16 -16
  91. package/src/intelligence/temporal/query-processor.ts +190 -190
  92. package/src/intelligence/temporal/timeline.ts +259 -259
  93. package/src/intelligence/temporal/trends.ts +263 -263
  94. package/src/knowledge/entity-extractor.ts +416 -416
  95. package/src/knowledge/graph/builder.ts +185 -185
  96. package/src/knowledge/graph/linker.ts +201 -201
  97. package/src/knowledge/graph/memory-graph.ts +359 -359
  98. package/src/knowledge/graph/schema.ts +99 -99
  99. package/src/knowledge/graph/search.ts +168 -168
  100. package/src/knowledge/relationship-extractor.ts +108 -108
  101. package/src/memory/chroma/client.ts +174 -174
  102. package/src/memory/chroma/collection-manager.ts +94 -94
  103. package/src/memory/chroma/config.ts +57 -57
  104. package/src/memory/chroma/embeddings.ts +155 -155
  105. package/src/memory/chroma/index.ts +82 -82
  106. package/src/memory/chroma/migration.ts +270 -270
  107. package/src/memory/chroma/schemas.ts +69 -69
  108. package/src/memory/chroma/search.ts +315 -315
  109. package/src/memory/chroma/store.ts +741 -741
  110. package/src/memory/consolidation/archiver.ts +164 -164
  111. package/src/memory/consolidation/merger.ts +186 -186
  112. package/src/memory/consolidation/scorer.ts +138 -138
  113. package/src/memory/context-builder.ts +236 -236
  114. package/src/memory/database.ts +169 -169
  115. package/src/memory/embedding-utils.ts +156 -156
  116. package/src/memory/embeddings.ts +226 -226
  117. package/src/memory/episodic/detector.ts +108 -108
  118. package/src/memory/episodic/manager.ts +351 -351
  119. package/src/memory/episodic/summarizer.ts +179 -179
  120. package/src/memory/episodic/types.ts +52 -52
  121. package/src/memory/index.ts +582 -582
  122. package/src/memory/knowledge-extractor.ts +455 -455
  123. package/src/memory/learning.ts +378 -378
  124. package/src/memory/patterns.ts +396 -396
  125. package/src/memory/schema.ts +88 -88
  126. package/src/memory/search.ts +309 -309
  127. package/src/memory/store.ts +787 -787
  128. package/src/memory/types.ts +121 -121
  129. package/src/orchestrator/coordinator.ts +272 -272
  130. package/src/orchestrator/decision-logger.ts +228 -228
  131. package/src/orchestrator/event-emitter.ts +198 -198
  132. package/src/orchestrator/event-queue.ts +184 -184
  133. package/src/orchestrator/handlers/base-handler.ts +70 -70
  134. package/src/orchestrator/handlers/context-handler.ts +73 -73
  135. package/src/orchestrator/handlers/decision-handler.ts +204 -204
  136. package/src/orchestrator/handlers/index.ts +10 -10
  137. package/src/orchestrator/handlers/status-handler.ts +131 -131
  138. package/src/orchestrator/handlers/task-handler.ts +171 -171
  139. package/src/orchestrator/index.ts +275 -275
  140. package/src/orchestrator/task-parser.ts +284 -284
  141. package/src/orchestrator/types.ts +98 -98
  142. package/src/packs/index.ts +9 -9
  143. package/src/packs/loader.ts +134 -134
  144. package/src/packs/manager.ts +204 -204
  145. package/src/packs/ranker.ts +78 -78
  146. package/src/packs/types.ts +81 -81
  147. package/src/phase12/index.ts +5 -5
  148. package/src/retrieval/bm25/index.ts +300 -300
  149. package/src/retrieval/bm25/tokenizer.ts +184 -184
  150. package/src/retrieval/feedback/adaptive.ts +223 -223
  151. package/src/retrieval/feedback/index.ts +16 -16
  152. package/src/retrieval/feedback/metrics.ts +223 -223
  153. package/src/retrieval/feedback/store.ts +283 -283
  154. package/src/retrieval/fusion/index.ts +194 -194
  155. package/src/retrieval/fusion/rrf.ts +163 -163
  156. package/src/retrieval/index.ts +12 -12
  157. package/src/retrieval/pipeline.ts +375 -375
  158. package/src/retrieval/query/expander.ts +198 -198
  159. package/src/retrieval/query/index.ts +27 -27
  160. package/src/retrieval/query/intent-classifier.ts +236 -236
  161. package/src/retrieval/query/temporal-parser.ts +295 -295
  162. package/src/retrieval/reranker/index.ts +188 -188
  163. package/src/retrieval/reranker/model.ts +95 -95
  164. package/src/retrieval/service.ts +125 -125
  165. package/src/retrieval/types.ts +162 -162
  166. package/src/routing/entity-extractor.ts +428 -428
  167. package/src/routing/intent-classifier.ts +450 -436
  168. package/src/routing/response-filter.ts +261 -258
  169. package/src/routing/router.ts +1441 -1322
  170. package/src/routing/search-engine.ts +515 -475
  171. package/src/routing/types.ts +94 -94
  172. package/src/scripts/health-check.ts +118 -118
  173. package/src/scripts/setup.ts +122 -122
  174. package/src/server/handlers/call-tool.ts +156 -156
  175. package/src/server/handlers/index.ts +9 -9
  176. package/src/server/handlers/list-tools.ts +35 -35
  177. package/src/server/handlers/tools/analyze-decision-evolution.ts +151 -151
  178. package/src/server/handlers/tools/auto-remember.ts +200 -200
  179. package/src/server/handlers/tools/brain.ts +85 -85
  180. package/src/server/handlers/tools/create-project.ts +135 -135
  181. package/src/server/handlers/tools/detect-trends.ts +144 -144
  182. package/src/server/handlers/tools/find-cross-project-patterns.ts +168 -168
  183. package/src/server/handlers/tools/get-activity-log.ts +194 -194
  184. package/src/server/handlers/tools/get-code-standards.ts +124 -124
  185. package/src/server/handlers/tools/get-corrections.ts +154 -154
  186. package/src/server/handlers/tools/get-decision-timeline.ts +172 -172
  187. package/src/server/handlers/tools/get-episode.ts +103 -103
  188. package/src/server/handlers/tools/get-patterns.ts +158 -158
  189. package/src/server/handlers/tools/get-phase12-status.ts +63 -63
  190. package/src/server/handlers/tools/get-project-context.ts +75 -75
  191. package/src/server/handlers/tools/get-recommendations.ts +145 -145
  192. package/src/server/handlers/tools/index.ts +31 -31
  193. package/src/server/handlers/tools/init-project.ts +757 -757
  194. package/src/server/handlers/tools/list-episodes.ts +90 -90
  195. package/src/server/handlers/tools/list-projects.ts +125 -125
  196. package/src/server/handlers/tools/rate-memory.ts +101 -101
  197. package/src/server/handlers/tools/recall-similar.ts +87 -87
  198. package/src/server/handlers/tools/recognize-pattern.ts +126 -126
  199. package/src/server/handlers/tools/record-correction.ts +125 -125
  200. package/src/server/handlers/tools/remember-decision.ts +153 -153
  201. package/src/server/handlers/tools/schemas.ts +253 -253
  202. package/src/server/handlers/tools/search-knowledge-graph.ts +102 -102
  203. package/src/server/handlers/tools/smart-context.ts +146 -146
  204. package/src/server/handlers/tools/update-progress.ts +131 -131
  205. package/src/server/handlers/tools/what-if-analysis.ts +135 -135
  206. package/src/server/http-api.ts +693 -693
  207. package/src/server/index.ts +40 -40
  208. package/src/server/mcp-server.ts +283 -283
  209. package/src/server/providers/index.ts +7 -7
  210. package/src/server/providers/prompts.ts +327 -327
  211. package/src/server/providers/resources.ts +622 -622
  212. package/src/server/services.ts +468 -468
  213. package/src/server/types.ts +39 -39
  214. package/src/server/utils/error-handler.ts +155 -155
  215. package/src/server/utils/index.ts +13 -13
  216. package/src/server/utils/memory-indicator.ts +83 -83
  217. package/src/server/utils/request-context.ts +122 -122
  218. package/src/server/utils/response-formatter.ts +129 -129
  219. package/src/server/utils/validators.ts +210 -210
  220. package/src/setup/index.ts +48 -48
  221. package/src/setup/wizard.ts +461 -461
  222. package/src/tools/index.ts +24 -24
  223. package/src/tools/registry.ts +115 -115
  224. package/src/tools/schemas.test.ts +30 -30
  225. package/src/tools/schemas.ts +617 -617
  226. package/src/tools/types.ts +412 -412
  227. package/src/utils/circuit-breaker.ts +130 -130
  228. package/src/utils/cleanup.ts +34 -34
  229. package/src/utils/error-handler.ts +132 -132
  230. package/src/utils/error-messages.ts +60 -60
  231. package/src/utils/fallback.ts +45 -45
  232. package/src/utils/index.ts +54 -54
  233. package/src/utils/logger-utils.ts +80 -80
  234. package/src/utils/logger.ts +88 -88
  235. package/src/utils/phase12-helper.ts +56 -56
  236. package/src/utils/retry.ts +94 -94
  237. package/src/utils/timing.ts +47 -47
  238. package/src/utils/transaction.ts +63 -63
  239. package/src/vault/frontmatter.ts +264 -264
  240. package/src/vault/index.ts +318 -318
  241. package/src/vault/paths.ts +106 -106
  242. package/src/vault/query.ts +422 -422
  243. package/src/vault/reader.ts +264 -264
  244. package/src/vault/templates.ts +186 -186
  245. package/src/vault/types.ts +73 -73
  246. package/src/vault/watcher.ts +277 -277
  247. package/src/vault/writer.ts +413 -413
  248. package/tsconfig.json +30 -30
  249. 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
+ }