@cmdoss/memwal-sdk 0.8.0 → 1.0.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 (209) hide show
  1. package/README.md +522 -160
  2. package/dist/client/ClientMemoryManager.d.ts.map +1 -1
  3. package/dist/client/ClientMemoryManager.js +25 -8
  4. package/dist/client/ClientMemoryManager.js.map +1 -1
  5. package/dist/client/PersonalDataWallet.d.ts.map +1 -1
  6. package/dist/client/SimplePDWClient.d.ts +62 -2
  7. package/dist/client/SimplePDWClient.d.ts.map +1 -1
  8. package/dist/client/SimplePDWClient.js +96 -11
  9. package/dist/client/SimplePDWClient.js.map +1 -1
  10. package/dist/client/namespaces/IndexNamespace.d.ts +1 -1
  11. package/dist/client/namespaces/IndexNamespace.d.ts.map +1 -1
  12. package/dist/client/namespaces/IndexNamespace.js +7 -4
  13. package/dist/client/namespaces/IndexNamespace.js.map +1 -1
  14. package/dist/client/namespaces/MemoryNamespace.d.ts +47 -0
  15. package/dist/client/namespaces/MemoryNamespace.d.ts.map +1 -1
  16. package/dist/client/namespaces/MemoryNamespace.js +257 -27
  17. package/dist/client/namespaces/MemoryNamespace.js.map +1 -1
  18. package/dist/client/namespaces/consolidated/AdvancedNamespace.d.ts +215 -0
  19. package/dist/client/namespaces/consolidated/AdvancedNamespace.d.ts.map +1 -0
  20. package/dist/client/namespaces/consolidated/AdvancedNamespace.js +214 -0
  21. package/dist/client/namespaces/consolidated/AdvancedNamespace.js.map +1 -0
  22. package/dist/client/namespaces/consolidated/StorageNamespace.d.ts +3 -1
  23. package/dist/client/namespaces/consolidated/StorageNamespace.d.ts.map +1 -1
  24. package/dist/client/namespaces/consolidated/StorageNamespace.js.map +1 -1
  25. package/dist/client/namespaces/consolidated/index.d.ts +1 -0
  26. package/dist/client/namespaces/consolidated/index.d.ts.map +1 -1
  27. package/dist/client/namespaces/consolidated/index.js +1 -0
  28. package/dist/client/namespaces/consolidated/index.js.map +1 -1
  29. package/dist/config/ConfigurationHelper.js +61 -61
  30. package/dist/config/defaults.d.ts.map +1 -1
  31. package/dist/config/defaults.js +9 -4
  32. package/dist/config/defaults.js.map +1 -1
  33. package/dist/config/index.d.ts +1 -0
  34. package/dist/config/index.d.ts.map +1 -1
  35. package/dist/config/index.js +2 -0
  36. package/dist/config/index.js.map +1 -1
  37. package/dist/config/modelDefaults.d.ts +67 -0
  38. package/dist/config/modelDefaults.d.ts.map +1 -0
  39. package/dist/config/modelDefaults.js +91 -0
  40. package/dist/config/modelDefaults.js.map +1 -0
  41. package/dist/core/types/index.d.ts +4 -0
  42. package/dist/core/types/index.d.ts.map +1 -1
  43. package/dist/core/types/index.js.map +1 -1
  44. package/dist/graph/GraphService.d.ts.map +1 -1
  45. package/dist/graph/GraphService.js +22 -21
  46. package/dist/graph/GraphService.js.map +1 -1
  47. package/dist/index.d.ts +1 -1
  48. package/dist/index.d.ts.map +1 -1
  49. package/dist/index.js +1 -1
  50. package/dist/index.js.map +1 -1
  51. package/dist/infrastructure/walrus/WalrusStorageService.d.ts +6 -0
  52. package/dist/infrastructure/walrus/WalrusStorageService.d.ts.map +1 -1
  53. package/dist/infrastructure/walrus/WalrusStorageService.js +23 -4
  54. package/dist/infrastructure/walrus/WalrusStorageService.js.map +1 -1
  55. package/dist/langchain/createPDWRAG.js +30 -30
  56. package/dist/pipeline/MemoryPipeline.d.ts.map +1 -1
  57. package/dist/pipeline/MemoryPipeline.js +2 -1
  58. package/dist/pipeline/MemoryPipeline.js.map +1 -1
  59. package/dist/services/EmbeddingService.d.ts +9 -0
  60. package/dist/services/EmbeddingService.d.ts.map +1 -1
  61. package/dist/services/EmbeddingService.js +31 -10
  62. package/dist/services/EmbeddingService.js.map +1 -1
  63. package/dist/services/GeminiAIService.d.ts.map +1 -1
  64. package/dist/services/GeminiAIService.js +311 -310
  65. package/dist/services/GeminiAIService.js.map +1 -1
  66. package/dist/services/MemoryIndexService.d.ts +2 -0
  67. package/dist/services/MemoryIndexService.d.ts.map +1 -1
  68. package/dist/services/MemoryIndexService.js +11 -4
  69. package/dist/services/MemoryIndexService.js.map +1 -1
  70. package/dist/services/StorageService.d.ts +4 -1
  71. package/dist/services/StorageService.d.ts.map +1 -1
  72. package/dist/services/StorageService.js.map +1 -1
  73. package/dist/services/VectorService.js +1 -1
  74. package/dist/services/VectorService.js.map +1 -1
  75. package/dist/services/storage/QuiltBatchManager.d.ts +7 -0
  76. package/dist/services/storage/QuiltBatchManager.d.ts.map +1 -1
  77. package/dist/services/storage/QuiltBatchManager.js +24 -5
  78. package/dist/services/storage/QuiltBatchManager.js.map +1 -1
  79. package/dist/services/storage/WalrusStorageManager.d.ts +10 -1
  80. package/dist/services/storage/WalrusStorageManager.d.ts.map +1 -1
  81. package/dist/services/storage/WalrusStorageManager.js +53 -12
  82. package/dist/services/storage/WalrusStorageManager.js.map +1 -1
  83. package/dist/vector/BrowserHnswIndexService.js +3 -3
  84. package/dist/vector/BrowserHnswIndexService.js.map +1 -1
  85. package/dist/vector/HnswWasmService.js +1 -1
  86. package/dist/vector/HnswWasmService.js.map +1 -1
  87. package/dist/vector/NodeHnswService.js +5 -5
  88. package/dist/vector/NodeHnswService.js.map +1 -1
  89. package/dist/vector/createHnswService.d.ts +4 -0
  90. package/dist/vector/createHnswService.d.ts.map +1 -1
  91. package/dist/vector/createHnswService.js +15 -3
  92. package/dist/vector/createHnswService.js.map +1 -1
  93. package/package.json +1 -1
  94. package/src/access/PermissionService.ts +635 -635
  95. package/src/aggregation/AggregationService.ts +389 -389
  96. package/src/ai-sdk/PDWVectorStore.ts +715 -715
  97. package/src/ai-sdk/index.ts +65 -65
  98. package/src/ai-sdk/tools.ts +460 -460
  99. package/src/ai-sdk/types.ts +404 -404
  100. package/src/batch/BatchManager.ts +597 -597
  101. package/src/batch/BatchingService.ts +429 -429
  102. package/src/batch/MemoryProcessingCache.ts +492 -492
  103. package/src/batch/index.ts +30 -30
  104. package/src/browser.ts +200 -200
  105. package/src/client/ClientMemoryManager.ts +1004 -987
  106. package/src/client/PersonalDataWallet.ts +345 -345
  107. package/src/client/SimplePDWClient.ts +1387 -1289
  108. package/src/client/factory.ts +154 -154
  109. package/src/client/namespaces/AnalyticsNamespace.ts +377 -377
  110. package/src/client/namespaces/BatchNamespace.ts +356 -356
  111. package/src/client/namespaces/CacheNamespace.ts +123 -123
  112. package/src/client/namespaces/CapabilityNamespace.ts +217 -217
  113. package/src/client/namespaces/ClassifyNamespace.ts +169 -169
  114. package/src/client/namespaces/ContextNamespace.ts +297 -297
  115. package/src/client/namespaces/EncryptionNamespace.ts +221 -221
  116. package/src/client/namespaces/GraphNamespace.ts +468 -468
  117. package/src/client/namespaces/IndexNamespace.ts +364 -361
  118. package/src/client/namespaces/MemoryNamespace.ts +1704 -1422
  119. package/src/client/namespaces/PermissionsNamespace.ts +254 -254
  120. package/src/client/namespaces/PipelineNamespace.ts +220 -220
  121. package/src/client/namespaces/StorageNamespace.ts +458 -458
  122. package/src/client/namespaces/TxNamespace.ts +260 -260
  123. package/src/client/namespaces/WalletNamespace.ts +243 -243
  124. package/src/client/namespaces/consolidated/AdvancedNamespace.ts +264 -0
  125. package/src/client/namespaces/consolidated/BlockchainNamespace.ts +607 -607
  126. package/src/client/namespaces/consolidated/SecurityNamespace.ts +648 -648
  127. package/src/client/namespaces/consolidated/StorageNamespace.ts +1143 -1141
  128. package/src/client/namespaces/consolidated/index.ts +41 -39
  129. package/src/client/signers/KeypairSigner.ts +108 -108
  130. package/src/client/signers/UnifiedSigner.ts +110 -110
  131. package/src/client/signers/WalletAdapterSigner.ts +159 -159
  132. package/src/client/signers/index.ts +26 -26
  133. package/src/config/ConfigurationHelper.ts +412 -412
  134. package/src/config/defaults.ts +56 -51
  135. package/src/config/index.ts +16 -9
  136. package/src/config/modelDefaults.ts +103 -0
  137. package/src/config/validation.ts +70 -70
  138. package/src/core/index.ts +14 -14
  139. package/src/core/interfaces/IService.ts +307 -307
  140. package/src/core/interfaces/index.ts +8 -8
  141. package/src/core/types/capability.ts +297 -297
  142. package/src/core/types/index.ts +874 -870
  143. package/src/core/types/wallet.ts +270 -270
  144. package/src/core/types.ts +9 -9
  145. package/src/core/wallet.ts +222 -222
  146. package/src/embedding/index.ts +19 -19
  147. package/src/embedding/types.ts +357 -357
  148. package/src/errors/index.ts +602 -602
  149. package/src/errors/recovery.ts +461 -461
  150. package/src/errors/validation.ts +567 -567
  151. package/src/generated/pdw/capability.ts +319 -319
  152. package/src/graph/GraphService.ts +888 -887
  153. package/src/graph/KnowledgeGraphManager.ts +728 -728
  154. package/src/graph/index.ts +25 -25
  155. package/src/index.ts +498 -498
  156. package/src/infrastructure/index.ts +22 -22
  157. package/src/infrastructure/seal/EncryptionService.ts +628 -628
  158. package/src/infrastructure/seal/SealService.ts +613 -613
  159. package/src/infrastructure/seal/index.ts +9 -9
  160. package/src/infrastructure/sui/BlockchainManager.ts +627 -627
  161. package/src/infrastructure/sui/SuiService.ts +888 -888
  162. package/src/infrastructure/sui/index.ts +9 -9
  163. package/src/infrastructure/walrus/StorageManager.ts +604 -604
  164. package/src/infrastructure/walrus/WalrusStorageService.ts +637 -612
  165. package/src/infrastructure/walrus/index.ts +9 -9
  166. package/src/langchain/createPDWRAG.ts +303 -303
  167. package/src/langchain/index.ts +47 -47
  168. package/src/permissions/ConsentRepository.browser.ts +249 -249
  169. package/src/permissions/ConsentRepository.ts +364 -364
  170. package/src/pipeline/MemoryPipeline.ts +863 -862
  171. package/src/pipeline/PipelineManager.ts +683 -683
  172. package/src/pipeline/index.ts +26 -26
  173. package/src/retrieval/AdvancedSearchService.ts +629 -629
  174. package/src/retrieval/MemoryAnalyticsService.ts +711 -711
  175. package/src/retrieval/MemoryDecryptionPipeline.ts +825 -825
  176. package/src/retrieval/index.ts +42 -42
  177. package/src/services/BatchService.ts +352 -352
  178. package/src/services/CapabilityService.ts +464 -464
  179. package/src/services/ClassifierService.ts +465 -465
  180. package/src/services/CrossContextPermissionService.ts +486 -486
  181. package/src/services/EmbeddingService.ts +796 -771
  182. package/src/services/EncryptionService.ts +712 -712
  183. package/src/services/GeminiAIService.ts +754 -753
  184. package/src/services/MemoryIndexService.ts +1009 -1003
  185. package/src/services/MemoryService.ts +369 -369
  186. package/src/services/QueryService.ts +890 -890
  187. package/src/services/StorageService.ts +1185 -1182
  188. package/src/services/TransactionService.ts +838 -838
  189. package/src/services/VectorService.ts +462 -462
  190. package/src/services/ViewService.ts +484 -484
  191. package/src/services/index.ts +25 -25
  192. package/src/services/storage/BlobAttributesManager.ts +333 -333
  193. package/src/services/storage/KnowledgeGraphManager.ts +425 -425
  194. package/src/services/storage/MemorySearchManager.ts +387 -387
  195. package/src/services/storage/QuiltBatchManager.ts +1157 -1130
  196. package/src/services/storage/WalrusMetadataManager.ts +268 -268
  197. package/src/services/storage/WalrusStorageManager.ts +333 -287
  198. package/src/services/storage/index.ts +57 -57
  199. package/src/types/index.ts +13 -13
  200. package/src/utils/index.ts +76 -76
  201. package/src/utils/memoryIndexOnChain.ts +507 -507
  202. package/src/vector/BrowserHnswIndexService.ts +758 -758
  203. package/src/vector/HnswWasmService.ts +731 -731
  204. package/src/vector/IHnswService.ts +233 -233
  205. package/src/vector/NodeHnswService.ts +833 -833
  206. package/src/vector/createHnswService.ts +147 -135
  207. package/src/vector/index.ts +56 -56
  208. package/src/wallet/ContextWalletService.ts +656 -656
  209. package/src/wallet/MainWalletService.ts +317 -317
@@ -1,605 +1,605 @@
1
- /**
2
- * StorageManager - Unified Storage Operations Manager
3
- *
4
- * Orchestrates Walrus storage operations with intelligent batching,
5
- * encryption management, and seamless integration with memory processing.
6
- */
7
-
8
- import { WalrusStorageService, WalrusUploadResult, WalrusRetrievalResult, MemoryMetadata } from './WalrusStorageService';
9
- import { ProcessedMemory } from '../../embedding/types';
10
-
11
- export interface StorageManagerConfig {
12
- walrusConfig?: {
13
- network?: 'testnet' | 'mainnet';
14
- enableEncryption?: boolean;
15
- enableBatching?: boolean;
16
- batchSize?: number;
17
- batchDelayMs?: number;
18
- };
19
- enableDistributedStorage?: boolean;
20
- enableCompressionn?: boolean;
21
- retentionPolicyDays?: number;
22
- }
23
-
24
- export interface StorageResult {
25
- success: boolean;
26
- blobId?: string;
27
- metadata?: MemoryMetadata;
28
- error?: string;
29
- processingTimeMs: number;
30
- isEncrypted: boolean;
31
- storageProvider: 'walrus' | 'local' | 'distributed';
32
- }
33
-
34
- export interface RetrievalOptions {
35
- includeMetadata?: boolean;
36
- decryptionKey?: string;
37
- preferCache?: boolean;
38
- timeout?: number;
39
- }
40
-
41
- export interface StorageBatchOperation {
42
- memory: ProcessedMemory;
43
- userId: string;
44
- priority?: 'low' | 'normal' | 'high';
45
- encryptionRequired?: boolean;
46
- }
47
-
48
- export interface StorageBatchResult {
49
- successful: StorageResult[];
50
- failed: Array<{ memory: ProcessedMemory; error: string }>;
51
- totalProcessingTime: number;
52
- averageProcessingTime: number;
53
- }
54
-
55
- export interface StorageStats {
56
- totalOperations: number;
57
- successfulUploads: number;
58
- failedUploads: number;
59
- totalRetrievals: number;
60
- storageProviders: {
61
- walrus: { uploads: number; retrievals: number; success_rate: number };
62
- local: { uploads: number; retrievals: number; success_rate: number };
63
- };
64
- averageUploadTime: number;
65
- averageRetrievalTime: number;
66
- totalStorageUsed: number;
67
- encryptionRate: number;
68
- compressionSavings: number;
69
- }
70
-
71
- /**
72
- * Unified storage manager coordinating multiple storage providers
73
- */
74
- export class StorageManager {
75
- private walrusService: WalrusStorageService;
76
- private pendingOperations = new Map<string, Promise<StorageResult>>();
77
- private batchQueue: StorageBatchOperation[] = [];
78
- private batchTimer?: NodeJS.Timeout;
79
-
80
- private readonly config: Required<StorageManagerConfig>;
81
- private stats: StorageStats = {
82
- totalOperations: 0,
83
- successfulUploads: 0,
84
- failedUploads: 0,
85
- totalRetrievals: 0,
86
- storageProviders: {
87
- walrus: { uploads: 0, retrievals: 0, success_rate: 0 },
88
- local: { uploads: 0, retrievals: 0, success_rate: 0 }
89
- },
90
- averageUploadTime: 0,
91
- averageRetrievalTime: 0,
92
- totalStorageUsed: 0,
93
- encryptionRate: 0,
94
- compressionSavings: 0
95
- };
96
-
97
- constructor(config: StorageManagerConfig = {}) {
98
- this.config = {
99
- walrusConfig: {
100
- network: config.walrusConfig?.network || 'testnet',
101
- enableEncryption: config.walrusConfig?.enableEncryption !== false,
102
- enableBatching: config.walrusConfig?.enableBatching !== false,
103
- batchSize: config.walrusConfig?.batchSize || 10,
104
- batchDelayMs: config.walrusConfig?.batchDelayMs || 5000
105
- },
106
- enableDistributedStorage: config.enableDistributedStorage !== false,
107
- enableCompressionn: config.enableCompressionn !== false,
108
- retentionPolicyDays: config.retentionPolicyDays || 365
109
- };
110
-
111
- // Initialize Walrus service
112
- this.walrusService = new WalrusStorageService({
113
- network: this.config.walrusConfig.network
114
- });
115
- }
116
-
117
- // ==================== MEMORY STORAGE OPERATIONS ====================
118
-
119
- /**
120
- * Store processed memory with comprehensive options
121
- */
122
- async storeMemory(
123
- memory: ProcessedMemory,
124
- userId: string,
125
- options: {
126
- priority?: 'low' | 'normal' | 'high';
127
- enableEncryption?: boolean;
128
- enableBatching?: boolean;
129
- customMetadata?: Record<string, string>;
130
- } = {}
131
- ): Promise<StorageResult> {
132
- const startTime = Date.now();
133
- this.stats.totalOperations++;
134
-
135
- try {
136
- // Check for duplicate operation
137
- const operationKey = `${memory.id}_${userId}`;
138
- const existingOperation = this.pendingOperations.get(operationKey);
139
- if (existingOperation) {
140
- return await existingOperation;
141
- }
142
-
143
- // Create operation promise
144
- const operationPromise = this.executeStoreOperation(memory, userId, options);
145
- this.pendingOperations.set(operationKey, operationPromise);
146
-
147
- try {
148
- const result = await operationPromise;
149
- this.updateStorageStats(result);
150
- return result;
151
- } finally {
152
- this.pendingOperations.delete(operationKey);
153
- }
154
-
155
- } catch (error) {
156
- this.stats.failedUploads++;
157
- return {
158
- success: false,
159
- error: error instanceof Error ? error.message : 'Unknown error',
160
- processingTimeMs: Date.now() - startTime,
161
- isEncrypted: false,
162
- storageProvider: 'local'
163
- };
164
- }
165
- }
166
-
167
- /**
168
- * Store multiple memories in batch
169
- */
170
- async storeMemoryBatch(
171
- operations: StorageBatchOperation[],
172
- options: {
173
- enableProgress?: boolean;
174
- onProgress?: (completed: number, total: number) => void;
175
- maxConcurrent?: number;
176
- } = {}
177
- ): Promise<StorageBatchResult> {
178
- const startTime = Date.now();
179
- const maxConcurrent = options.maxConcurrent || 5;
180
- const successful: StorageResult[] = [];
181
- const failed: Array<{ memory: ProcessedMemory; error: string }> = [];
182
-
183
- // Sort by priority (high -> normal -> low)
184
- const sortedOps = [...operations].sort((a, b) => {
185
- const priorityOrder = { high: 3, normal: 2, low: 1 };
186
- const aPriority = priorityOrder[a.priority || 'normal'];
187
- const bPriority = priorityOrder[b.priority || 'normal'];
188
- return bPriority - aPriority;
189
- });
190
-
191
- // Process in batches to control concurrency
192
- for (let i = 0; i < sortedOps.length; i += maxConcurrent) {
193
- const batch = sortedOps.slice(i, i + maxConcurrent);
194
-
195
- const batchPromises = batch.map(async (op) => {
196
- try {
197
- const result = await this.storeMemory(op.memory, op.userId, {
198
- priority: op.priority,
199
- enableEncryption: op.encryptionRequired
200
- });
201
-
202
- if (result.success) {
203
- successful.push(result);
204
- } else {
205
- failed.push({ memory: op.memory, error: result.error || 'Unknown error' });
206
- }
207
- } catch (error) {
208
- failed.push({ memory: op.memory, error: error instanceof Error ? error.message : 'Unknown error' });
209
- }
210
- });
211
-
212
- await Promise.all(batchPromises);
213
-
214
- // Progress callback
215
- if (options.onProgress) {
216
- options.onProgress(i + batch.length, operations.length);
217
- }
218
- }
219
-
220
- const totalTime = Date.now() - startTime;
221
- const averageTime = operations.length > 0 ? totalTime / operations.length : 0;
222
-
223
- return {
224
- successful,
225
- failed,
226
- totalProcessingTime: totalTime,
227
- averageProcessingTime: averageTime
228
- };
229
- }
230
-
231
- /**
232
- * Add memory to batch queue for deferred processing
233
- */
234
- addToBatchQueue(
235
- memory: ProcessedMemory,
236
- userId: string,
237
- options: {
238
- priority?: 'low' | 'normal' | 'high';
239
- encryptionRequired?: boolean;
240
- } = {}
241
- ): void {
242
- this.batchQueue.push({
243
- memory,
244
- userId,
245
- priority: options.priority || 'normal',
246
- encryptionRequired: options.encryptionRequired
247
- });
248
-
249
- // Schedule batch processing
250
- this.scheduleBatchProcessing();
251
- }
252
-
253
- /**
254
- * Process pending batch queue
255
- */
256
- async processBatchQueue(): Promise<StorageBatchResult> {
257
- if (this.batchQueue.length === 0) {
258
- return {
259
- successful: [],
260
- failed: [],
261
- totalProcessingTime: 0,
262
- averageProcessingTime: 0
263
- };
264
- }
265
-
266
- const operations = [...this.batchQueue];
267
- this.batchQueue.length = 0; // Clear queue
268
-
269
- if (this.batchTimer) {
270
- clearTimeout(this.batchTimer);
271
- this.batchTimer = undefined;
272
- }
273
-
274
- return await this.storeMemoryBatch(operations);
275
- }
276
-
277
- // ==================== RETRIEVAL OPERATIONS ====================
278
-
279
- /**
280
- * Retrieve memory by blob ID
281
- */
282
- async retrieveMemory(
283
- blobId: string,
284
- options: RetrievalOptions = {}
285
- ): Promise<{
286
- memory: ProcessedMemory | null;
287
- metadata: MemoryMetadata | null;
288
- retrievalTimeMs: number;
289
- isFromCache: boolean;
290
- }> {
291
- const startTime = Date.now();
292
- this.stats.totalRetrievals++;
293
-
294
- try {
295
- const result = await this.walrusService.retrieveContent(blobId, options.decryptionKey);
296
-
297
- // Parse content as memory
298
- let memory: ProcessedMemory | null = null;
299
- try {
300
- const parsed = JSON.parse(result.content as string);
301
- if (parsed && typeof parsed === 'object' && parsed.id) {
302
- memory = parsed as ProcessedMemory;
303
- }
304
- } catch {
305
- // Content might not be a memory object
306
- }
307
-
308
- this.updateRetrievalStats(Date.now() - startTime);
309
-
310
- return {
311
- memory,
312
- metadata: options.includeMetadata ? result.metadata : null,
313
- retrievalTimeMs: result.retrievalTimeMs,
314
- isFromCache: result.isFromCache
315
- };
316
-
317
- } catch (error) {
318
- console.error('Memory retrieval failed:', error);
319
- return {
320
- memory: null,
321
- metadata: null,
322
- retrievalTimeMs: Date.now() - startTime,
323
- isFromCache: false
324
- };
325
- }
326
- }
327
-
328
- /**
329
- * Retrieve multiple memories by blob IDs
330
- */
331
- async retrieveMemoryBatch(
332
- blobIds: string[],
333
- options: RetrievalOptions & {
334
- batchSize?: number;
335
- decryptionKeys?: Record<string, string>;
336
- } = {}
337
- ): Promise<Array<{
338
- blobId: string;
339
- memory: ProcessedMemory | null;
340
- metadata: MemoryMetadata | null;
341
- success: boolean;
342
- error?: string;
343
- }>> {
344
- const batchSize = options.batchSize || 10;
345
- const results: Array<{
346
- blobId: string;
347
- memory: ProcessedMemory | null;
348
- metadata: MemoryMetadata | null;
349
- success: boolean;
350
- error?: string;
351
- }> = [];
352
-
353
- // Process in batches
354
- for (let i = 0; i < blobIds.length; i += batchSize) {
355
- const batch = blobIds.slice(i, i + batchSize);
356
-
357
- const batchPromises = batch.map(async (blobId) => {
358
- try {
359
- const decryptionKey = options.decryptionKeys?.[blobId] || options.decryptionKey;
360
- const result = await this.retrieveMemory(blobId, { ...options, decryptionKey });
361
-
362
- return {
363
- blobId,
364
- memory: result.memory,
365
- metadata: result.metadata,
366
- success: result.memory !== null
367
- };
368
- } catch (error) {
369
- return {
370
- blobId,
371
- memory: null,
372
- metadata: null,
373
- success: false,
374
- error: error instanceof Error ? error.message : 'Unknown error'
375
- };
376
- }
377
- });
378
-
379
- const batchResults = await Promise.all(batchPromises);
380
- results.push(...batchResults);
381
- }
382
-
383
- return results;
384
- }
385
-
386
- // ==================== STORAGE MANAGEMENT ====================
387
-
388
- /**
389
- * List user's stored memories
390
- */
391
- async listUserMemories(
392
- userId: string,
393
- options: {
394
- category?: string;
395
- limit?: number;
396
- offset?: number;
397
- sortBy?: 'date' | 'size' | 'importance';
398
- includeMetadata?: boolean;
399
- } = {}
400
- ) {
401
- try {
402
- const blobResult = await this.walrusService.listUserBlobs(userId, {
403
- category: options.category,
404
- limit: options.limit,
405
- offset: options.offset,
406
- sortBy: options.sortBy
407
- });
408
-
409
- const memories = [];
410
-
411
- for (const blob of blobResult.blobs) {
412
- try {
413
- const result = await this.retrieveMemory(blob.blobId, {
414
- includeMetadata: options.includeMetadata
415
- });
416
-
417
- if (result.memory) {
418
- memories.push({
419
- memory: result.memory,
420
- blobId: blob.blobId,
421
- metadata: result.metadata,
422
- blobInfo: blob
423
- });
424
- }
425
- } catch (error) {
426
- console.warn(`Failed to retrieve memory ${blob.blobId}:`, error);
427
- }
428
- }
429
-
430
- return memories;
431
-
432
- } catch (error) {
433
- console.error('Failed to list user memories:', error);
434
- return [];
435
- }
436
- }
437
-
438
- /**
439
- * Delete stored memory
440
- */
441
- async deleteMemory(blobId: string): Promise<boolean> {
442
- try {
443
- return await this.walrusService.deleteBlob(blobId);
444
- } catch (error) {
445
- console.error('Failed to delete memory:', error);
446
- return false;
447
- }
448
- }
449
-
450
- /**
451
- * Get storage analytics
452
- */
453
- getStorageAnalytics() {
454
- const walrusStats = this.walrusService.getStats();
455
- const cacheInfo = this.walrusService.getCacheInfo();
456
-
457
- return {
458
- ...this.stats,
459
- walrusService: walrusStats,
460
- cache: cacheInfo,
461
- batchQueue: {
462
- pending: this.batchQueue.length,
463
- operations: this.pendingOperations.size
464
- }
465
- };
466
- }
467
-
468
- /**
469
- * Cleanup old data based on retention policy
470
- */
471
- async cleanupOldData(): Promise<{
472
- deletedCount: number;
473
- freedSpaceBytes: number;
474
- }> {
475
- // TODO: Implement cleanup based on retention policy
476
- return {
477
- deletedCount: 0,
478
- freedSpaceBytes: 0
479
- };
480
- }
481
-
482
- // ==================== PRIVATE METHODS ====================
483
-
484
- private async executeStoreOperation(
485
- memory: ProcessedMemory,
486
- userId: string,
487
- options: {
488
- priority?: 'low' | 'normal' | 'high';
489
- enableEncryption?: boolean;
490
- customMetadata?: Record<string, string>;
491
- }
492
- ): Promise<StorageResult> {
493
- const startTime = Date.now();
494
-
495
- try {
496
- // Prepare memory data
497
- const memoryData = JSON.stringify(memory);
498
-
499
- // Prepare metadata
500
- const customMetadata = {
501
- owner: userId,
502
- memoryId: memory.id,
503
- ...options.customMetadata
504
- };
505
-
506
- // Upload to Walrus
507
- const uploadResult = await this.walrusService.uploadContentWithMetadata(
508
- memoryData,
509
- userId,
510
- {
511
- category: memory.category,
512
- topic: memory.metadata?.topic,
513
- importance: memory.metadata?.importance || 5,
514
- additionalTags: customMetadata,
515
- enableEncryption: options.enableEncryption
516
- }
517
- );
518
-
519
- this.stats.successfulUploads++;
520
- this.updateUploadStats(Date.now() - startTime);
521
- this.stats.totalStorageUsed += uploadResult.metadata.contentSize;
522
-
523
- if (uploadResult.isEncrypted) {
524
- this.updateEncryptionRate(true);
525
- }
526
-
527
- return {
528
- success: true,
529
- blobId: uploadResult.blobId,
530
- metadata: uploadResult.metadata,
531
- processingTimeMs: uploadResult.uploadTimeMs,
532
- isEncrypted: uploadResult.isEncrypted,
533
- storageProvider: 'walrus'
534
- };
535
-
536
- } catch (error) {
537
- this.stats.failedUploads++;
538
- throw error;
539
- }
540
- }
541
-
542
- private scheduleBatchProcessing(): void {
543
- if (this.batchTimer) {
544
- return; // Already scheduled
545
- }
546
-
547
- // Process when batch is full or after delay
548
- if (this.batchQueue.length >= (this.config.walrusConfig?.batchSize || 10)) {
549
- this.processBatchQueue();
550
- } else {
551
- this.batchTimer = setTimeout(() => {
552
- this.processBatchQueue();
553
- }, this.config.walrusConfig.batchDelayMs);
554
- }
555
- }
556
-
557
- private updateStorageStats(result: StorageResult): void {
558
- if (result.success) {
559
- if (result.storageProvider === 'walrus') {
560
- this.stats.storageProviders.walrus.uploads++;
561
- } else if (result.storageProvider === 'local') {
562
- this.stats.storageProviders.local.uploads++;
563
- }
564
- }
565
-
566
- // Update success rates
567
- this.updateProviderSuccessRates();
568
- }
569
-
570
- private updateUploadStats(uploadTime: number): void {
571
- this.stats.averageUploadTime =
572
- (this.stats.averageUploadTime * (this.stats.successfulUploads - 1) + uploadTime) / this.stats.successfulUploads;
573
- }
574
-
575
- private updateRetrievalStats(retrievalTime: number): void {
576
- this.stats.averageRetrievalTime =
577
- (this.stats.averageRetrievalTime * (this.stats.totalRetrievals - 1) + retrievalTime) / this.stats.totalRetrievals;
578
- }
579
-
580
- private updateEncryptionRate(isEncrypted: boolean): void {
581
- const total = this.stats.successfulUploads;
582
- if (isEncrypted) {
583
- this.stats.encryptionRate = (this.stats.encryptionRate * (total - 1) + 1) / total;
584
- } else {
585
- this.stats.encryptionRate = (this.stats.encryptionRate * (total - 1)) / total;
586
- }
587
- }
588
-
589
- private updateProviderSuccessRates(): void {
590
- const walrusTotal = this.stats.storageProviders.walrus.uploads + this.stats.storageProviders.walrus.retrievals;
591
- const localTotal = this.stats.storageProviders.local.uploads + this.stats.storageProviders.local.retrievals;
592
-
593
- if (walrusTotal > 0) {
594
- this.stats.storageProviders.walrus.success_rate =
595
- this.stats.storageProviders.walrus.uploads / walrusTotal;
596
- }
597
-
598
- if (localTotal > 0) {
599
- this.stats.storageProviders.local.success_rate =
600
- this.stats.storageProviders.local.uploads / localTotal;
601
- }
602
- }
603
- }
604
-
1
+ /**
2
+ * StorageManager - Unified Storage Operations Manager
3
+ *
4
+ * Orchestrates Walrus storage operations with intelligent batching,
5
+ * encryption management, and seamless integration with memory processing.
6
+ */
7
+
8
+ import { WalrusStorageService, WalrusUploadResult, WalrusRetrievalResult, MemoryMetadata } from './WalrusStorageService';
9
+ import { ProcessedMemory } from '../../embedding/types';
10
+
11
+ export interface StorageManagerConfig {
12
+ walrusConfig?: {
13
+ network?: 'testnet' | 'mainnet';
14
+ enableEncryption?: boolean;
15
+ enableBatching?: boolean;
16
+ batchSize?: number;
17
+ batchDelayMs?: number;
18
+ };
19
+ enableDistributedStorage?: boolean;
20
+ enableCompressionn?: boolean;
21
+ retentionPolicyDays?: number;
22
+ }
23
+
24
+ export interface StorageResult {
25
+ success: boolean;
26
+ blobId?: string;
27
+ metadata?: MemoryMetadata;
28
+ error?: string;
29
+ processingTimeMs: number;
30
+ isEncrypted: boolean;
31
+ storageProvider: 'walrus' | 'local' | 'distributed';
32
+ }
33
+
34
+ export interface RetrievalOptions {
35
+ includeMetadata?: boolean;
36
+ decryptionKey?: string;
37
+ preferCache?: boolean;
38
+ timeout?: number;
39
+ }
40
+
41
+ export interface StorageBatchOperation {
42
+ memory: ProcessedMemory;
43
+ userId: string;
44
+ priority?: 'low' | 'normal' | 'high';
45
+ encryptionRequired?: boolean;
46
+ }
47
+
48
+ export interface StorageBatchResult {
49
+ successful: StorageResult[];
50
+ failed: Array<{ memory: ProcessedMemory; error: string }>;
51
+ totalProcessingTime: number;
52
+ averageProcessingTime: number;
53
+ }
54
+
55
+ export interface StorageStats {
56
+ totalOperations: number;
57
+ successfulUploads: number;
58
+ failedUploads: number;
59
+ totalRetrievals: number;
60
+ storageProviders: {
61
+ walrus: { uploads: number; retrievals: number; success_rate: number };
62
+ local: { uploads: number; retrievals: number; success_rate: number };
63
+ };
64
+ averageUploadTime: number;
65
+ averageRetrievalTime: number;
66
+ totalStorageUsed: number;
67
+ encryptionRate: number;
68
+ compressionSavings: number;
69
+ }
70
+
71
+ /**
72
+ * Unified storage manager coordinating multiple storage providers
73
+ */
74
+ export class StorageManager {
75
+ private walrusService: WalrusStorageService;
76
+ private pendingOperations = new Map<string, Promise<StorageResult>>();
77
+ private batchQueue: StorageBatchOperation[] = [];
78
+ private batchTimer?: NodeJS.Timeout;
79
+
80
+ private readonly config: Required<StorageManagerConfig>;
81
+ private stats: StorageStats = {
82
+ totalOperations: 0,
83
+ successfulUploads: 0,
84
+ failedUploads: 0,
85
+ totalRetrievals: 0,
86
+ storageProviders: {
87
+ walrus: { uploads: 0, retrievals: 0, success_rate: 0 },
88
+ local: { uploads: 0, retrievals: 0, success_rate: 0 }
89
+ },
90
+ averageUploadTime: 0,
91
+ averageRetrievalTime: 0,
92
+ totalStorageUsed: 0,
93
+ encryptionRate: 0,
94
+ compressionSavings: 0
95
+ };
96
+
97
+ constructor(config: StorageManagerConfig = {}) {
98
+ this.config = {
99
+ walrusConfig: {
100
+ network: config.walrusConfig?.network || 'testnet',
101
+ enableEncryption: config.walrusConfig?.enableEncryption !== false,
102
+ enableBatching: config.walrusConfig?.enableBatching !== false,
103
+ batchSize: config.walrusConfig?.batchSize || 10,
104
+ batchDelayMs: config.walrusConfig?.batchDelayMs || 5000
105
+ },
106
+ enableDistributedStorage: config.enableDistributedStorage !== false,
107
+ enableCompressionn: config.enableCompressionn !== false,
108
+ retentionPolicyDays: config.retentionPolicyDays || 365
109
+ };
110
+
111
+ // Initialize Walrus service
112
+ this.walrusService = new WalrusStorageService({
113
+ network: this.config.walrusConfig.network
114
+ });
115
+ }
116
+
117
+ // ==================== MEMORY STORAGE OPERATIONS ====================
118
+
119
+ /**
120
+ * Store processed memory with comprehensive options
121
+ */
122
+ async storeMemory(
123
+ memory: ProcessedMemory,
124
+ userId: string,
125
+ options: {
126
+ priority?: 'low' | 'normal' | 'high';
127
+ enableEncryption?: boolean;
128
+ enableBatching?: boolean;
129
+ customMetadata?: Record<string, string>;
130
+ } = {}
131
+ ): Promise<StorageResult> {
132
+ const startTime = Date.now();
133
+ this.stats.totalOperations++;
134
+
135
+ try {
136
+ // Check for duplicate operation
137
+ const operationKey = `${memory.id}_${userId}`;
138
+ const existingOperation = this.pendingOperations.get(operationKey);
139
+ if (existingOperation) {
140
+ return await existingOperation;
141
+ }
142
+
143
+ // Create operation promise
144
+ const operationPromise = this.executeStoreOperation(memory, userId, options);
145
+ this.pendingOperations.set(operationKey, operationPromise);
146
+
147
+ try {
148
+ const result = await operationPromise;
149
+ this.updateStorageStats(result);
150
+ return result;
151
+ } finally {
152
+ this.pendingOperations.delete(operationKey);
153
+ }
154
+
155
+ } catch (error) {
156
+ this.stats.failedUploads++;
157
+ return {
158
+ success: false,
159
+ error: error instanceof Error ? error.message : 'Unknown error',
160
+ processingTimeMs: Date.now() - startTime,
161
+ isEncrypted: false,
162
+ storageProvider: 'local'
163
+ };
164
+ }
165
+ }
166
+
167
+ /**
168
+ * Store multiple memories in batch
169
+ */
170
+ async storeMemoryBatch(
171
+ operations: StorageBatchOperation[],
172
+ options: {
173
+ enableProgress?: boolean;
174
+ onProgress?: (completed: number, total: number) => void;
175
+ maxConcurrent?: number;
176
+ } = {}
177
+ ): Promise<StorageBatchResult> {
178
+ const startTime = Date.now();
179
+ const maxConcurrent = options.maxConcurrent || 5;
180
+ const successful: StorageResult[] = [];
181
+ const failed: Array<{ memory: ProcessedMemory; error: string }> = [];
182
+
183
+ // Sort by priority (high -> normal -> low)
184
+ const sortedOps = [...operations].sort((a, b) => {
185
+ const priorityOrder = { high: 3, normal: 2, low: 1 };
186
+ const aPriority = priorityOrder[a.priority || 'normal'];
187
+ const bPriority = priorityOrder[b.priority || 'normal'];
188
+ return bPriority - aPriority;
189
+ });
190
+
191
+ // Process in batches to control concurrency
192
+ for (let i = 0; i < sortedOps.length; i += maxConcurrent) {
193
+ const batch = sortedOps.slice(i, i + maxConcurrent);
194
+
195
+ const batchPromises = batch.map(async (op) => {
196
+ try {
197
+ const result = await this.storeMemory(op.memory, op.userId, {
198
+ priority: op.priority,
199
+ enableEncryption: op.encryptionRequired
200
+ });
201
+
202
+ if (result.success) {
203
+ successful.push(result);
204
+ } else {
205
+ failed.push({ memory: op.memory, error: result.error || 'Unknown error' });
206
+ }
207
+ } catch (error) {
208
+ failed.push({ memory: op.memory, error: error instanceof Error ? error.message : 'Unknown error' });
209
+ }
210
+ });
211
+
212
+ await Promise.all(batchPromises);
213
+
214
+ // Progress callback
215
+ if (options.onProgress) {
216
+ options.onProgress(i + batch.length, operations.length);
217
+ }
218
+ }
219
+
220
+ const totalTime = Date.now() - startTime;
221
+ const averageTime = operations.length > 0 ? totalTime / operations.length : 0;
222
+
223
+ return {
224
+ successful,
225
+ failed,
226
+ totalProcessingTime: totalTime,
227
+ averageProcessingTime: averageTime
228
+ };
229
+ }
230
+
231
+ /**
232
+ * Add memory to batch queue for deferred processing
233
+ */
234
+ addToBatchQueue(
235
+ memory: ProcessedMemory,
236
+ userId: string,
237
+ options: {
238
+ priority?: 'low' | 'normal' | 'high';
239
+ encryptionRequired?: boolean;
240
+ } = {}
241
+ ): void {
242
+ this.batchQueue.push({
243
+ memory,
244
+ userId,
245
+ priority: options.priority || 'normal',
246
+ encryptionRequired: options.encryptionRequired
247
+ });
248
+
249
+ // Schedule batch processing
250
+ this.scheduleBatchProcessing();
251
+ }
252
+
253
+ /**
254
+ * Process pending batch queue
255
+ */
256
+ async processBatchQueue(): Promise<StorageBatchResult> {
257
+ if (this.batchQueue.length === 0) {
258
+ return {
259
+ successful: [],
260
+ failed: [],
261
+ totalProcessingTime: 0,
262
+ averageProcessingTime: 0
263
+ };
264
+ }
265
+
266
+ const operations = [...this.batchQueue];
267
+ this.batchQueue.length = 0; // Clear queue
268
+
269
+ if (this.batchTimer) {
270
+ clearTimeout(this.batchTimer);
271
+ this.batchTimer = undefined;
272
+ }
273
+
274
+ return await this.storeMemoryBatch(operations);
275
+ }
276
+
277
+ // ==================== RETRIEVAL OPERATIONS ====================
278
+
279
+ /**
280
+ * Retrieve memory by blob ID
281
+ */
282
+ async retrieveMemory(
283
+ blobId: string,
284
+ options: RetrievalOptions = {}
285
+ ): Promise<{
286
+ memory: ProcessedMemory | null;
287
+ metadata: MemoryMetadata | null;
288
+ retrievalTimeMs: number;
289
+ isFromCache: boolean;
290
+ }> {
291
+ const startTime = Date.now();
292
+ this.stats.totalRetrievals++;
293
+
294
+ try {
295
+ const result = await this.walrusService.retrieveContent(blobId, options.decryptionKey);
296
+
297
+ // Parse content as memory
298
+ let memory: ProcessedMemory | null = null;
299
+ try {
300
+ const parsed = JSON.parse(result.content as string);
301
+ if (parsed && typeof parsed === 'object' && parsed.id) {
302
+ memory = parsed as ProcessedMemory;
303
+ }
304
+ } catch {
305
+ // Content might not be a memory object
306
+ }
307
+
308
+ this.updateRetrievalStats(Date.now() - startTime);
309
+
310
+ return {
311
+ memory,
312
+ metadata: options.includeMetadata ? result.metadata : null,
313
+ retrievalTimeMs: result.retrievalTimeMs,
314
+ isFromCache: result.isFromCache
315
+ };
316
+
317
+ } catch (error) {
318
+ console.error('Memory retrieval failed:', error);
319
+ return {
320
+ memory: null,
321
+ metadata: null,
322
+ retrievalTimeMs: Date.now() - startTime,
323
+ isFromCache: false
324
+ };
325
+ }
326
+ }
327
+
328
+ /**
329
+ * Retrieve multiple memories by blob IDs
330
+ */
331
+ async retrieveMemoryBatch(
332
+ blobIds: string[],
333
+ options: RetrievalOptions & {
334
+ batchSize?: number;
335
+ decryptionKeys?: Record<string, string>;
336
+ } = {}
337
+ ): Promise<Array<{
338
+ blobId: string;
339
+ memory: ProcessedMemory | null;
340
+ metadata: MemoryMetadata | null;
341
+ success: boolean;
342
+ error?: string;
343
+ }>> {
344
+ const batchSize = options.batchSize || 10;
345
+ const results: Array<{
346
+ blobId: string;
347
+ memory: ProcessedMemory | null;
348
+ metadata: MemoryMetadata | null;
349
+ success: boolean;
350
+ error?: string;
351
+ }> = [];
352
+
353
+ // Process in batches
354
+ for (let i = 0; i < blobIds.length; i += batchSize) {
355
+ const batch = blobIds.slice(i, i + batchSize);
356
+
357
+ const batchPromises = batch.map(async (blobId) => {
358
+ try {
359
+ const decryptionKey = options.decryptionKeys?.[blobId] || options.decryptionKey;
360
+ const result = await this.retrieveMemory(blobId, { ...options, decryptionKey });
361
+
362
+ return {
363
+ blobId,
364
+ memory: result.memory,
365
+ metadata: result.metadata,
366
+ success: result.memory !== null
367
+ };
368
+ } catch (error) {
369
+ return {
370
+ blobId,
371
+ memory: null,
372
+ metadata: null,
373
+ success: false,
374
+ error: error instanceof Error ? error.message : 'Unknown error'
375
+ };
376
+ }
377
+ });
378
+
379
+ const batchResults = await Promise.all(batchPromises);
380
+ results.push(...batchResults);
381
+ }
382
+
383
+ return results;
384
+ }
385
+
386
+ // ==================== STORAGE MANAGEMENT ====================
387
+
388
+ /**
389
+ * List user's stored memories
390
+ */
391
+ async listUserMemories(
392
+ userId: string,
393
+ options: {
394
+ category?: string;
395
+ limit?: number;
396
+ offset?: number;
397
+ sortBy?: 'date' | 'size' | 'importance';
398
+ includeMetadata?: boolean;
399
+ } = {}
400
+ ) {
401
+ try {
402
+ const blobResult = await this.walrusService.listUserBlobs(userId, {
403
+ category: options.category,
404
+ limit: options.limit,
405
+ offset: options.offset,
406
+ sortBy: options.sortBy
407
+ });
408
+
409
+ const memories = [];
410
+
411
+ for (const blob of blobResult.blobs) {
412
+ try {
413
+ const result = await this.retrieveMemory(blob.blobId, {
414
+ includeMetadata: options.includeMetadata
415
+ });
416
+
417
+ if (result.memory) {
418
+ memories.push({
419
+ memory: result.memory,
420
+ blobId: blob.blobId,
421
+ metadata: result.metadata,
422
+ blobInfo: blob
423
+ });
424
+ }
425
+ } catch (error) {
426
+ console.warn(`Failed to retrieve memory ${blob.blobId}:`, error);
427
+ }
428
+ }
429
+
430
+ return memories;
431
+
432
+ } catch (error) {
433
+ console.error('Failed to list user memories:', error);
434
+ return [];
435
+ }
436
+ }
437
+
438
+ /**
439
+ * Delete stored memory
440
+ */
441
+ async deleteMemory(blobId: string): Promise<boolean> {
442
+ try {
443
+ return await this.walrusService.deleteBlob(blobId);
444
+ } catch (error) {
445
+ console.error('Failed to delete memory:', error);
446
+ return false;
447
+ }
448
+ }
449
+
450
+ /**
451
+ * Get storage analytics
452
+ */
453
+ getStorageAnalytics() {
454
+ const walrusStats = this.walrusService.getStats();
455
+ const cacheInfo = this.walrusService.getCacheInfo();
456
+
457
+ return {
458
+ ...this.stats,
459
+ walrusService: walrusStats,
460
+ cache: cacheInfo,
461
+ batchQueue: {
462
+ pending: this.batchQueue.length,
463
+ operations: this.pendingOperations.size
464
+ }
465
+ };
466
+ }
467
+
468
+ /**
469
+ * Cleanup old data based on retention policy
470
+ */
471
+ async cleanupOldData(): Promise<{
472
+ deletedCount: number;
473
+ freedSpaceBytes: number;
474
+ }> {
475
+ // TODO: Implement cleanup based on retention policy
476
+ return {
477
+ deletedCount: 0,
478
+ freedSpaceBytes: 0
479
+ };
480
+ }
481
+
482
+ // ==================== PRIVATE METHODS ====================
483
+
484
+ private async executeStoreOperation(
485
+ memory: ProcessedMemory,
486
+ userId: string,
487
+ options: {
488
+ priority?: 'low' | 'normal' | 'high';
489
+ enableEncryption?: boolean;
490
+ customMetadata?: Record<string, string>;
491
+ }
492
+ ): Promise<StorageResult> {
493
+ const startTime = Date.now();
494
+
495
+ try {
496
+ // Prepare memory data
497
+ const memoryData = JSON.stringify(memory);
498
+
499
+ // Prepare metadata
500
+ const customMetadata = {
501
+ owner: userId,
502
+ memoryId: memory.id,
503
+ ...options.customMetadata
504
+ };
505
+
506
+ // Upload to Walrus
507
+ const uploadResult = await this.walrusService.uploadContentWithMetadata(
508
+ memoryData,
509
+ userId,
510
+ {
511
+ category: memory.category,
512
+ topic: memory.metadata?.topic,
513
+ importance: memory.metadata?.importance || 5,
514
+ additionalTags: customMetadata,
515
+ enableEncryption: options.enableEncryption
516
+ }
517
+ );
518
+
519
+ this.stats.successfulUploads++;
520
+ this.updateUploadStats(Date.now() - startTime);
521
+ this.stats.totalStorageUsed += uploadResult.metadata.contentSize;
522
+
523
+ if (uploadResult.isEncrypted) {
524
+ this.updateEncryptionRate(true);
525
+ }
526
+
527
+ return {
528
+ success: true,
529
+ blobId: uploadResult.blobId,
530
+ metadata: uploadResult.metadata,
531
+ processingTimeMs: uploadResult.uploadTimeMs,
532
+ isEncrypted: uploadResult.isEncrypted,
533
+ storageProvider: 'walrus'
534
+ };
535
+
536
+ } catch (error) {
537
+ this.stats.failedUploads++;
538
+ throw error;
539
+ }
540
+ }
541
+
542
+ private scheduleBatchProcessing(): void {
543
+ if (this.batchTimer) {
544
+ return; // Already scheduled
545
+ }
546
+
547
+ // Process when batch is full or after delay
548
+ if (this.batchQueue.length >= (this.config.walrusConfig?.batchSize || 10)) {
549
+ this.processBatchQueue();
550
+ } else {
551
+ this.batchTimer = setTimeout(() => {
552
+ this.processBatchQueue();
553
+ }, this.config.walrusConfig.batchDelayMs);
554
+ }
555
+ }
556
+
557
+ private updateStorageStats(result: StorageResult): void {
558
+ if (result.success) {
559
+ if (result.storageProvider === 'walrus') {
560
+ this.stats.storageProviders.walrus.uploads++;
561
+ } else if (result.storageProvider === 'local') {
562
+ this.stats.storageProviders.local.uploads++;
563
+ }
564
+ }
565
+
566
+ // Update success rates
567
+ this.updateProviderSuccessRates();
568
+ }
569
+
570
+ private updateUploadStats(uploadTime: number): void {
571
+ this.stats.averageUploadTime =
572
+ (this.stats.averageUploadTime * (this.stats.successfulUploads - 1) + uploadTime) / this.stats.successfulUploads;
573
+ }
574
+
575
+ private updateRetrievalStats(retrievalTime: number): void {
576
+ this.stats.averageRetrievalTime =
577
+ (this.stats.averageRetrievalTime * (this.stats.totalRetrievals - 1) + retrievalTime) / this.stats.totalRetrievals;
578
+ }
579
+
580
+ private updateEncryptionRate(isEncrypted: boolean): void {
581
+ const total = this.stats.successfulUploads;
582
+ if (isEncrypted) {
583
+ this.stats.encryptionRate = (this.stats.encryptionRate * (total - 1) + 1) / total;
584
+ } else {
585
+ this.stats.encryptionRate = (this.stats.encryptionRate * (total - 1)) / total;
586
+ }
587
+ }
588
+
589
+ private updateProviderSuccessRates(): void {
590
+ const walrusTotal = this.stats.storageProviders.walrus.uploads + this.stats.storageProviders.walrus.retrievals;
591
+ const localTotal = this.stats.storageProviders.local.uploads + this.stats.storageProviders.local.retrievals;
592
+
593
+ if (walrusTotal > 0) {
594
+ this.stats.storageProviders.walrus.success_rate =
595
+ this.stats.storageProviders.walrus.uploads / walrusTotal;
596
+ }
597
+
598
+ if (localTotal > 0) {
599
+ this.stats.storageProviders.local.success_rate =
600
+ this.stats.storageProviders.local.uploads / localTotal;
601
+ }
602
+ }
603
+ }
604
+
605
605
  export default StorageManager;