claude-brain 0.3.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 (200) hide show
  1. package/README.md +157 -0
  2. package/VERSION +1 -0
  3. package/assets/CLAUDE.md +307 -0
  4. package/bunfig.toml +8 -0
  5. package/package.json +74 -0
  6. package/src/automation/auto-context.ts +240 -0
  7. package/src/automation/decision-detector.ts +452 -0
  8. package/src/automation/index.ts +11 -0
  9. package/src/automation/proactive-recall.ts +373 -0
  10. package/src/automation/project-detector.ts +297 -0
  11. package/src/cli/auto-setup.ts +74 -0
  12. package/src/cli/bin.ts +110 -0
  13. package/src/cli/commands/install-mcp.ts +50 -0
  14. package/src/cli/commands/serve.ts +129 -0
  15. package/src/cli/diagnose.ts +4 -0
  16. package/src/cli/health-check.ts +4 -0
  17. package/src/cli/migrate-chroma.ts +106 -0
  18. package/src/cli/setup.ts +4 -0
  19. package/src/config/defaults.ts +47 -0
  20. package/src/config/home.ts +55 -0
  21. package/src/config/index.ts +7 -0
  22. package/src/config/loader.ts +166 -0
  23. package/src/config/migration.ts +76 -0
  24. package/src/config/schema.ts +257 -0
  25. package/src/config/validator.ts +184 -0
  26. package/src/config/watcher.ts +86 -0
  27. package/src/context/assembler.ts +398 -0
  28. package/src/context/cache-manager.ts +101 -0
  29. package/src/context/formatter.ts +84 -0
  30. package/src/context/hierarchy.ts +85 -0
  31. package/src/context/index.ts +83 -0
  32. package/src/context/progress-tracker.ts +174 -0
  33. package/src/context/standards-manager.ts +267 -0
  34. package/src/context/types.ts +252 -0
  35. package/src/context/validator.ts +58 -0
  36. package/src/cross-project/affinity.ts +162 -0
  37. package/src/cross-project/generalizer.ts +283 -0
  38. package/src/cross-project/index.ts +13 -0
  39. package/src/cross-project/transfer.ts +201 -0
  40. package/src/diagnostics/index.ts +123 -0
  41. package/src/health/index.ts +229 -0
  42. package/src/index.ts +7 -0
  43. package/src/knowledge/entity-extractor.ts +416 -0
  44. package/src/knowledge/graph/builder.ts +159 -0
  45. package/src/knowledge/graph/linker.ts +201 -0
  46. package/src/knowledge/graph/memory-graph.ts +359 -0
  47. package/src/knowledge/graph/schema.ts +99 -0
  48. package/src/knowledge/graph/search.ts +168 -0
  49. package/src/knowledge/relationship-extractor.ts +108 -0
  50. package/src/memory/chroma/client.ts +169 -0
  51. package/src/memory/chroma/collection-manager.ts +94 -0
  52. package/src/memory/chroma/config.ts +46 -0
  53. package/src/memory/chroma/embeddings.ts +153 -0
  54. package/src/memory/chroma/index.ts +82 -0
  55. package/src/memory/chroma/migration.ts +270 -0
  56. package/src/memory/chroma/schemas.ts +69 -0
  57. package/src/memory/chroma/search.ts +315 -0
  58. package/src/memory/chroma/store.ts +694 -0
  59. package/src/memory/consolidation/archiver.ts +164 -0
  60. package/src/memory/consolidation/merger.ts +186 -0
  61. package/src/memory/consolidation/scorer.ts +138 -0
  62. package/src/memory/context-builder.ts +236 -0
  63. package/src/memory/database.ts +169 -0
  64. package/src/memory/embedding-utils.ts +156 -0
  65. package/src/memory/embeddings.ts +226 -0
  66. package/src/memory/episodic/detector.ts +108 -0
  67. package/src/memory/episodic/manager.ts +334 -0
  68. package/src/memory/episodic/summarizer.ts +179 -0
  69. package/src/memory/episodic/types.ts +52 -0
  70. package/src/memory/index.ts +395 -0
  71. package/src/memory/knowledge-extractor.ts +455 -0
  72. package/src/memory/learning.ts +378 -0
  73. package/src/memory/patterns.ts +396 -0
  74. package/src/memory/schema.ts +56 -0
  75. package/src/memory/search.ts +309 -0
  76. package/src/memory/store.ts +344 -0
  77. package/src/memory/types.ts +121 -0
  78. package/src/optimization/index.ts +10 -0
  79. package/src/optimization/precompute.ts +202 -0
  80. package/src/optimization/semantic-cache.ts +207 -0
  81. package/src/orchestrator/coordinator.ts +272 -0
  82. package/src/orchestrator/decision-logger.ts +228 -0
  83. package/src/orchestrator/event-emitter.ts +198 -0
  84. package/src/orchestrator/event-queue.ts +184 -0
  85. package/src/orchestrator/handlers/base-handler.ts +70 -0
  86. package/src/orchestrator/handlers/context-handler.ts +73 -0
  87. package/src/orchestrator/handlers/decision-handler.ts +204 -0
  88. package/src/orchestrator/handlers/index.ts +10 -0
  89. package/src/orchestrator/handlers/status-handler.ts +131 -0
  90. package/src/orchestrator/handlers/task-handler.ts +171 -0
  91. package/src/orchestrator/index.ts +275 -0
  92. package/src/orchestrator/task-parser.ts +284 -0
  93. package/src/orchestrator/types.ts +98 -0
  94. package/src/phase12/index.ts +456 -0
  95. package/src/prediction/context-anticipator.ts +198 -0
  96. package/src/prediction/decision-predictor.ts +184 -0
  97. package/src/prediction/index.ts +13 -0
  98. package/src/prediction/recommender.ts +268 -0
  99. package/src/reasoning/chain-retrieval.ts +247 -0
  100. package/src/reasoning/counterfactual.ts +248 -0
  101. package/src/reasoning/index.ts +13 -0
  102. package/src/reasoning/synthesizer.ts +169 -0
  103. package/src/retrieval/bm25/index.ts +300 -0
  104. package/src/retrieval/bm25/tokenizer.ts +184 -0
  105. package/src/retrieval/feedback/adaptive.ts +223 -0
  106. package/src/retrieval/feedback/index.ts +16 -0
  107. package/src/retrieval/feedback/metrics.ts +223 -0
  108. package/src/retrieval/feedback/store.ts +283 -0
  109. package/src/retrieval/fusion/index.ts +194 -0
  110. package/src/retrieval/fusion/rrf.ts +163 -0
  111. package/src/retrieval/index.ts +12 -0
  112. package/src/retrieval/pipeline.ts +375 -0
  113. package/src/retrieval/query/expander.ts +198 -0
  114. package/src/retrieval/query/index.ts +27 -0
  115. package/src/retrieval/query/intent-classifier.ts +236 -0
  116. package/src/retrieval/query/temporal-parser.ts +295 -0
  117. package/src/retrieval/reranker/index.ts +188 -0
  118. package/src/retrieval/reranker/model.ts +95 -0
  119. package/src/retrieval/service.ts +125 -0
  120. package/src/retrieval/types.ts +162 -0
  121. package/src/scripts/health-check.ts +118 -0
  122. package/src/scripts/setup.ts +122 -0
  123. package/src/server/handlers/call-tool.ts +194 -0
  124. package/src/server/handlers/index.ts +9 -0
  125. package/src/server/handlers/list-tools.ts +18 -0
  126. package/src/server/handlers/tools/analyze-decision-evolution.ts +71 -0
  127. package/src/server/handlers/tools/auto-remember.ts +200 -0
  128. package/src/server/handlers/tools/create-project.ts +135 -0
  129. package/src/server/handlers/tools/detect-trends.ts +80 -0
  130. package/src/server/handlers/tools/find-cross-project-patterns.ts +73 -0
  131. package/src/server/handlers/tools/get-activity-log.ts +194 -0
  132. package/src/server/handlers/tools/get-code-standards.ts +124 -0
  133. package/src/server/handlers/tools/get-corrections.ts +154 -0
  134. package/src/server/handlers/tools/get-decision-timeline.ts +86 -0
  135. package/src/server/handlers/tools/get-episode.ts +93 -0
  136. package/src/server/handlers/tools/get-patterns.ts +158 -0
  137. package/src/server/handlers/tools/get-phase12-status.ts +63 -0
  138. package/src/server/handlers/tools/get-project-context.ts +75 -0
  139. package/src/server/handlers/tools/get-recommendations.ts +65 -0
  140. package/src/server/handlers/tools/index.ts +33 -0
  141. package/src/server/handlers/tools/init-project.ts +710 -0
  142. package/src/server/handlers/tools/list-episodes.ts +80 -0
  143. package/src/server/handlers/tools/list-projects.ts +125 -0
  144. package/src/server/handlers/tools/rate-memory.ts +95 -0
  145. package/src/server/handlers/tools/recall-similar.ts +87 -0
  146. package/src/server/handlers/tools/recognize-pattern.ts +126 -0
  147. package/src/server/handlers/tools/record-correction.ts +125 -0
  148. package/src/server/handlers/tools/remember-decision.ts +153 -0
  149. package/src/server/handlers/tools/schemas.ts +241 -0
  150. package/src/server/handlers/tools/search-knowledge-graph.ts +89 -0
  151. package/src/server/handlers/tools/smart-context.ts +124 -0
  152. package/src/server/handlers/tools/update-progress.ts +114 -0
  153. package/src/server/handlers/tools/what-if-analysis.ts +73 -0
  154. package/src/server/http-api.ts +474 -0
  155. package/src/server/index.ts +40 -0
  156. package/src/server/mcp-server.ts +283 -0
  157. package/src/server/providers/index.ts +7 -0
  158. package/src/server/providers/prompts.ts +327 -0
  159. package/src/server/providers/resources.ts +427 -0
  160. package/src/server/services.ts +388 -0
  161. package/src/server/types.ts +39 -0
  162. package/src/server/utils/error-handler.ts +155 -0
  163. package/src/server/utils/index.ts +13 -0
  164. package/src/server/utils/memory-indicator.ts +83 -0
  165. package/src/server/utils/request-context.ts +122 -0
  166. package/src/server/utils/response-formatter.ts +124 -0
  167. package/src/server/utils/validators.ts +210 -0
  168. package/src/setup/index.ts +22 -0
  169. package/src/setup/wizard.ts +321 -0
  170. package/src/temporal/evolution.ts +197 -0
  171. package/src/temporal/index.ts +16 -0
  172. package/src/temporal/query-processor.ts +190 -0
  173. package/src/temporal/timeline.ts +259 -0
  174. package/src/temporal/trends.ts +263 -0
  175. package/src/tools/index.ts +24 -0
  176. package/src/tools/registry.ts +106 -0
  177. package/src/tools/schemas.test.ts +30 -0
  178. package/src/tools/schemas.ts +907 -0
  179. package/src/tools/types.ts +412 -0
  180. package/src/utils/circuit-breaker.ts +130 -0
  181. package/src/utils/cleanup.ts +34 -0
  182. package/src/utils/error-handler.ts +132 -0
  183. package/src/utils/error-messages.ts +60 -0
  184. package/src/utils/fallback.ts +45 -0
  185. package/src/utils/index.ts +54 -0
  186. package/src/utils/logger-utils.ts +80 -0
  187. package/src/utils/logger.ts +88 -0
  188. package/src/utils/phase12-helper.ts +56 -0
  189. package/src/utils/retry.ts +94 -0
  190. package/src/utils/transaction.ts +63 -0
  191. package/src/vault/frontmatter.ts +264 -0
  192. package/src/vault/index.ts +318 -0
  193. package/src/vault/paths.ts +106 -0
  194. package/src/vault/query.ts +422 -0
  195. package/src/vault/reader.ts +264 -0
  196. package/src/vault/templates.ts +186 -0
  197. package/src/vault/types.ts +73 -0
  198. package/src/vault/watcher.ts +277 -0
  199. package/src/vault/writer.ts +393 -0
  200. package/tsconfig.json +30 -0
@@ -0,0 +1,60 @@
1
+ export class ErrorMessages {
2
+ static format(error: Error, context?: Record<string, unknown>): string {
3
+ const errorMap: Record<string, string> = {
4
+ 'ENOENT': 'File or directory not found',
5
+ 'EACCES': 'Permission denied',
6
+ 'ECONNREFUSED': 'Connection refused - service may be down',
7
+ 'ETIMEDOUT': 'Operation timed out',
8
+ 'SQLITE_BUSY': 'Database is busy, please try again',
9
+ 'Project not found': 'Project does not exist in vault'
10
+ }
11
+
12
+ for (const [key, message] of Object.entries(errorMap)) {
13
+ if (error.message.includes(key)) {
14
+ return this.formatWithSolution(message, key, context)
15
+ }
16
+ }
17
+
18
+ return `Error: ${error.message}`
19
+ }
20
+
21
+ private static formatWithSolution(
22
+ message: string,
23
+ errorType: string,
24
+ context?: Record<string, unknown>
25
+ ): string {
26
+ let output = `❌ ${message}\n`
27
+
28
+ const solutions: Record<string, string[]> = {
29
+ 'ENOENT': [
30
+ 'Verify the file path is correct',
31
+ 'Check that the project exists in your vault'
32
+ ],
33
+ 'EACCES': [
34
+ 'Check file permissions',
35
+ 'Ensure you have write access to the directory'
36
+ ],
37
+ 'ECONNREFUSED': [
38
+ 'Verify the service is running',
39
+ 'Check your network connection'
40
+ ],
41
+ 'SQLITE_BUSY': [
42
+ 'Close other applications accessing the database',
43
+ 'Try again in a moment'
44
+ ]
45
+ }
46
+
47
+ if (solutions[errorType]) {
48
+ output += '\n💡 Possible solutions:\n'
49
+ solutions[errorType].forEach((solution, i) => {
50
+ output += ` ${i + 1}. ${solution}\n`
51
+ })
52
+ }
53
+
54
+ if (context) {
55
+ output += `\n📋 Context: ${JSON.stringify(context)}\n`
56
+ }
57
+
58
+ return output
59
+ }
60
+ }
@@ -0,0 +1,45 @@
1
+ import type { Logger } from 'pino'
2
+
3
+ export interface FallbackStrategy<T> {
4
+ primary: () => Promise<T>
5
+ fallback: () => Promise<T>
6
+ condition?: (error: Error) => boolean
7
+ }
8
+
9
+ export class FallbackManager {
10
+ private logger: Logger
11
+
12
+ constructor(logger: Logger) {
13
+ this.logger = logger.child({ component: 'fallback-manager' })
14
+ }
15
+
16
+ async execute<T>(strategy: FallbackStrategy<T>): Promise<T> {
17
+ try {
18
+ return await strategy.primary()
19
+
20
+ } catch (error) {
21
+ if (strategy.condition && !strategy.condition(error as Error)) {
22
+ throw error
23
+ }
24
+
25
+ this.logger.warn(
26
+ { error: (error as Error).message },
27
+ 'Primary operation failed, using fallback'
28
+ )
29
+
30
+ try {
31
+ return await strategy.fallback()
32
+
33
+ } catch (fallbackError) {
34
+ this.logger.error(
35
+ {
36
+ primaryError: (error as Error).message,
37
+ fallbackError: (fallbackError as Error).message
38
+ },
39
+ 'Both primary and fallback failed'
40
+ )
41
+ throw fallbackError
42
+ }
43
+ }
44
+ }
45
+ }
@@ -0,0 +1,54 @@
1
+ export {
2
+ createLogger,
3
+ createComponentLogger,
4
+ createRequestLogger,
5
+ generateRequestId
6
+ } from './logger'
7
+
8
+ export {
9
+ logToolCall,
10
+ logToolResult,
11
+ logAgentActivity,
12
+ logPerformance,
13
+ logError,
14
+ logVaultOperation,
15
+ logMemoryOperation,
16
+ createTimer
17
+ } from './logger-utils'
18
+
19
+ export {
20
+ GlobalErrorHandler,
21
+ AppError,
22
+ isOperationalError
23
+ } from './error-handler'
24
+
25
+ export {
26
+ RetryManager
27
+ } from './retry'
28
+
29
+ export {
30
+ CircuitBreaker,
31
+ CircuitState
32
+ } from './circuit-breaker'
33
+
34
+ export {
35
+ FallbackManager
36
+ } from './fallback'
37
+
38
+ export {
39
+ TransactionManager
40
+ } from './transaction'
41
+
42
+ export {
43
+ CleanupManager
44
+ } from './cleanup'
45
+
46
+ export {
47
+ ErrorMessages
48
+ } from './error-messages'
49
+
50
+ export {
51
+ isPhase12Available,
52
+ getPhase12Instance,
53
+ resetPhase12Cache
54
+ } from './phase12-helper'
@@ -0,0 +1,80 @@
1
+ import type { Logger } from 'pino'
2
+
3
+ /** Log an MCP tool call with parameters */
4
+ export function logToolCall(logger: Logger, toolName: string, params: Record<string, unknown>): void {
5
+ logger.info({
6
+ event: 'tool_call',
7
+ tool: toolName,
8
+ params
9
+ }, `Tool called: ${toolName}`)
10
+ }
11
+
12
+ /** Log an MCP tool result */
13
+ export function logToolResult(logger: Logger, toolName: string, success: boolean, duration: number): void {
14
+ logger.info({
15
+ event: 'tool_result',
16
+ tool: toolName,
17
+ success,
18
+ durationMs: duration
19
+ }, `Tool ${success ? 'completed' : 'failed'}: ${toolName}`)
20
+ }
21
+
22
+ /** Log agent activity and decisions */
23
+ export function logAgentActivity(logger: Logger, action: string, context: Record<string, unknown>): void {
24
+ logger.info({
25
+ event: 'agent_activity',
26
+ action,
27
+ ...context
28
+ }, `Agent: ${action}`)
29
+ }
30
+
31
+ /** Log performance metrics */
32
+ export function logPerformance(logger: Logger, operation: string, durationMs: number, metadata?: Record<string, unknown>): void {
33
+ logger.info({
34
+ event: 'performance',
35
+ operation,
36
+ durationMs,
37
+ ...metadata
38
+ }, `Performance: ${operation} took ${durationMs}ms`)
39
+ }
40
+
41
+ /** Log an error with full context */
42
+ export function logError(logger: Logger, error: Error, context?: Record<string, unknown>): void {
43
+ logger.error({
44
+ event: 'error',
45
+ error: {
46
+ name: error.name,
47
+ message: error.message,
48
+ stack: error.stack
49
+ },
50
+ ...context
51
+ }, `Error: ${error.message}`)
52
+ }
53
+
54
+ /** Log vault file operations */
55
+ export function logVaultOperation(logger: Logger, operation: string, filePath: string, metadata?: Record<string, unknown>): void {
56
+ logger.debug({
57
+ event: 'vault_operation',
58
+ operation,
59
+ filePath,
60
+ ...metadata
61
+ }, `Vault: ${operation} - ${filePath}`)
62
+ }
63
+
64
+ /** Log memory/embedding operations */
65
+ export function logMemoryOperation(logger: Logger, operation: string, count: number, metadata?: Record<string, unknown>): void {
66
+ logger.debug({
67
+ event: 'memory_operation',
68
+ operation,
69
+ count,
70
+ ...metadata
71
+ }, `Memory: ${operation} (${count} items)`)
72
+ }
73
+
74
+ /** Create a timer utility for measuring operation duration */
75
+ export function createTimer(): { stop: () => number } {
76
+ const start = performance.now()
77
+ return {
78
+ stop: () => Math.round(performance.now() - start)
79
+ }
80
+ }
@@ -0,0 +1,88 @@
1
+ import pino, { type Logger, type LoggerOptions } from 'pino'
2
+ import { mkdirSync, existsSync } from 'node:fs'
3
+ import { dirname } from 'node:path'
4
+ import type { LogLevel } from '@/config'
5
+
6
+ /** Create the log directory if it doesn't exist */
7
+ function ensureLogDir(logFilePath: string): void {
8
+ const dir = dirname(logFilePath)
9
+ if (!existsSync(dir)) {
10
+ mkdirSync(dir, { recursive: true })
11
+ }
12
+ }
13
+
14
+ /** Create a configured Pino logger instance */
15
+ export function createLogger(level: LogLevel, logFilePath: string): Logger {
16
+ ensureLogDir(logFilePath)
17
+
18
+ const isDev = process.env.NODE_ENV !== 'production'
19
+
20
+ const baseOptions: LoggerOptions = {
21
+ level,
22
+ timestamp: pino.stdTimeFunctions.isoTime,
23
+ base: {
24
+ pid: process.pid,
25
+ hostname: undefined // Remove hostname for cleaner logs
26
+ }
27
+ }
28
+
29
+ if (isDev) {
30
+ // Development: pretty print to console AND write to file
31
+ const targets: pino.TransportTargetOptions[] = [
32
+ {
33
+ target: 'pino-pretty',
34
+ options: {
35
+ colorize: true,
36
+ translateTime: 'SYS:standard',
37
+ ignore: 'pid',
38
+ destination: 2 // stderr for console
39
+ },
40
+ level
41
+ },
42
+ {
43
+ target: 'pino/file',
44
+ options: { destination: logFilePath },
45
+ level
46
+ }
47
+ ]
48
+
49
+ return pino({
50
+ ...baseOptions,
51
+ transport: { targets }
52
+ })
53
+ }
54
+
55
+ // Production: write to file only (MCP uses stdout for JSON-RPC)
56
+ const targets: pino.TransportTargetOptions[] = [
57
+ {
58
+ target: 'pino/file',
59
+ options: { destination: logFilePath },
60
+ level
61
+ },
62
+ {
63
+ target: 'pino/file',
64
+ options: { destination: 2 }, // stderr instead of stdout
65
+ level
66
+ }
67
+ ]
68
+
69
+ return pino({
70
+ ...baseOptions,
71
+ transport: { targets }
72
+ })
73
+ }
74
+
75
+ /** Create a child logger with component context */
76
+ export function createComponentLogger(parentLogger: Logger, component: string): Logger {
77
+ return parentLogger.child({ component })
78
+ }
79
+
80
+ /** Generate a unique request ID */
81
+ export function generateRequestId(): string {
82
+ return `req_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 9)}`
83
+ }
84
+
85
+ /** Create a child logger with request ID context */
86
+ export function createRequestLogger(parentLogger: Logger, requestId: string): Logger {
87
+ return parentLogger.child({ requestId })
88
+ }
@@ -0,0 +1,56 @@
1
+ /**
2
+ * Phase 12 Helper
3
+ * Utilities for checking Phase 12 availability
4
+ * Allows Phase 6.5 to gracefully degrade if Phase 12 isn't fully implemented
5
+ */
6
+
7
+ import type { Phase12Manager } from '@/phase12'
8
+
9
+ // Cache for Phase 12 availability check
10
+ let phase12Available: boolean | null = null
11
+
12
+ /**
13
+ * Check if Phase 12 features are available
14
+ */
15
+ export function isPhase12Available(): boolean {
16
+ if (phase12Available !== null) {
17
+ return phase12Available
18
+ }
19
+
20
+ try {
21
+ // Check if the module can be loaded
22
+ require('@/phase12')
23
+ phase12Available = true
24
+ return true
25
+ } catch {
26
+ phase12Available = false
27
+ return false
28
+ }
29
+ }
30
+
31
+ /**
32
+ * Get Phase 12 instance if available
33
+ * Returns null if Phase 12 is not available or not initialized
34
+ */
35
+ export function getPhase12Instance(): Phase12Manager | null {
36
+ try {
37
+ // Import the service getter
38
+ const { getPhase12Service, isServicesInitialized } = require('@/server/services')
39
+
40
+ if (!isServicesInitialized()) {
41
+ return null
42
+ }
43
+
44
+ return getPhase12Service()
45
+ } catch {
46
+ return null
47
+ }
48
+ }
49
+
50
+ /**
51
+ * Reset the Phase 12 availability cache
52
+ * Useful for testing
53
+ */
54
+ export function resetPhase12Cache(): void {
55
+ phase12Available = null
56
+ }
@@ -0,0 +1,94 @@
1
+ import type { Logger } from 'pino'
2
+
3
+ export interface RetryOptions {
4
+ maxRetries: number
5
+ initialDelay: number
6
+ maxDelay: number
7
+ backoffMultiplier: number
8
+ retryCondition?: (error: Error) => boolean
9
+ onRetry?: (error: Error, attempt: number) => void
10
+ }
11
+
12
+ export class RetryManager {
13
+ private logger: Logger
14
+
15
+ constructor(logger: Logger) {
16
+ this.logger = logger.child({ component: 'retry-manager' })
17
+ }
18
+
19
+ async execute<T>(
20
+ fn: () => Promise<T>,
21
+ options: Partial<RetryOptions> = {}
22
+ ): Promise<T> {
23
+ const opts: RetryOptions = {
24
+ maxRetries: 3,
25
+ initialDelay: 1000,
26
+ maxDelay: 30000,
27
+ backoffMultiplier: 2,
28
+ ...options
29
+ }
30
+
31
+ let lastError: Error | undefined
32
+ let attempt = 0
33
+
34
+ while (attempt <= opts.maxRetries) {
35
+ try {
36
+ return await fn()
37
+
38
+ } catch (error) {
39
+ lastError = error as Error
40
+ attempt++
41
+
42
+ if (attempt > opts.maxRetries) {
43
+ this.logger.error(
44
+ { error: lastError, attempts: attempt },
45
+ 'Max retries exceeded'
46
+ )
47
+ throw lastError
48
+ }
49
+
50
+ if (opts.retryCondition && !opts.retryCondition(lastError)) {
51
+ this.logger.debug(
52
+ { error: lastError },
53
+ 'Retry condition not met, not retrying'
54
+ )
55
+ throw lastError
56
+ }
57
+
58
+ const delay = Math.min(
59
+ opts.initialDelay * Math.pow(opts.backoffMultiplier, attempt - 1),
60
+ opts.maxDelay
61
+ )
62
+
63
+ this.logger.info(
64
+ { error: lastError.message, attempt, delay },
65
+ 'Retrying after delay'
66
+ )
67
+
68
+ if (opts.onRetry) {
69
+ opts.onRetry(lastError, attempt)
70
+ }
71
+
72
+ await this.sleep(delay)
73
+ }
74
+ }
75
+
76
+ throw lastError!
77
+ }
78
+
79
+ static isRetryableError(error: Error): boolean {
80
+ if (error.message.includes('ECONNREFUSED')) return true
81
+ if (error.message.includes('ETIMEDOUT')) return true
82
+ if (error.message.includes('ENOTFOUND')) return true
83
+ if (error.message.includes('SQLITE_BUSY')) return true
84
+ if (error.message.includes('database is locked')) return true
85
+ if (error.message.includes('EBUSY')) return true
86
+ if (error.message.includes('EMFILE')) return true
87
+
88
+ return false
89
+ }
90
+
91
+ private sleep(ms: number): Promise<void> {
92
+ return new Promise(resolve => setTimeout(resolve, ms))
93
+ }
94
+ }
@@ -0,0 +1,63 @@
1
+ import type { Logger } from 'pino'
2
+
3
+ export class TransactionManager {
4
+ private logger: Logger
5
+ private operations: Array<() => Promise<void>> = []
6
+ private rollbacks: Array<() => Promise<void>> = []
7
+
8
+ constructor(logger: Logger) {
9
+ this.logger = logger.child({ component: 'transaction-manager' })
10
+ }
11
+
12
+ add(
13
+ operation: () => Promise<void>,
14
+ rollback: () => Promise<void>
15
+ ): void {
16
+ this.operations.push(operation)
17
+ this.rollbacks.unshift(rollback)
18
+ }
19
+
20
+ async execute(): Promise<void> {
21
+ const executedCount = 0
22
+
23
+ try {
24
+ for (let i = 0; i < this.operations.length; i++) {
25
+ await this.operations[i]()
26
+ }
27
+
28
+ this.logger.info(
29
+ { operationCount: this.operations.length },
30
+ 'Transaction completed successfully'
31
+ )
32
+
33
+ } catch (error) {
34
+ this.logger.error(
35
+ { error, executedCount },
36
+ 'Transaction failed, rolling back'
37
+ )
38
+
39
+ await this.rollback()
40
+ throw error
41
+ }
42
+ }
43
+
44
+ private async rollback(): Promise<void> {
45
+ for (const rollback of this.rollbacks) {
46
+ try {
47
+ await rollback()
48
+ } catch (error) {
49
+ this.logger.error(
50
+ { error },
51
+ 'Rollback operation failed'
52
+ )
53
+ }
54
+ }
55
+
56
+ this.logger.info('Rollback completed')
57
+ }
58
+
59
+ clear(): void {
60
+ this.operations = []
61
+ this.rollbacks = []
62
+ }
63
+ }