claude-brain 0.14.2 → 0.14.4

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