@cmdoss/memwal-sdk 0.6.2 → 0.8.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 (247) hide show
  1. package/ARCHITECTURE.md +547 -547
  2. package/BENCHMARKS.md +238 -238
  3. package/README.md +310 -181
  4. package/dist/ai-sdk/tools.d.ts +2 -2
  5. package/dist/ai-sdk/tools.js +2 -2
  6. package/dist/client/ClientMemoryManager.js +2 -2
  7. package/dist/client/ClientMemoryManager.js.map +1 -1
  8. package/dist/client/PersonalDataWallet.d.ts.map +1 -1
  9. package/dist/client/SimplePDWClient.d.ts +29 -1
  10. package/dist/client/SimplePDWClient.d.ts.map +1 -1
  11. package/dist/client/SimplePDWClient.js +45 -13
  12. package/dist/client/SimplePDWClient.js.map +1 -1
  13. package/dist/client/namespaces/EmbeddingsNamespace.d.ts +1 -1
  14. package/dist/client/namespaces/EmbeddingsNamespace.js +1 -1
  15. package/dist/client/namespaces/MemoryNamespace.d.ts +31 -0
  16. package/dist/client/namespaces/MemoryNamespace.d.ts.map +1 -1
  17. package/dist/client/namespaces/MemoryNamespace.js +272 -39
  18. package/dist/client/namespaces/MemoryNamespace.js.map +1 -1
  19. package/dist/client/namespaces/consolidated/AINamespace.d.ts +2 -2
  20. package/dist/client/namespaces/consolidated/AINamespace.js +2 -2
  21. package/dist/client/namespaces/consolidated/BlockchainNamespace.d.ts +12 -2
  22. package/dist/client/namespaces/consolidated/BlockchainNamespace.d.ts.map +1 -1
  23. package/dist/client/namespaces/consolidated/BlockchainNamespace.js +62 -4
  24. package/dist/client/namespaces/consolidated/BlockchainNamespace.js.map +1 -1
  25. package/dist/client/namespaces/consolidated/StorageNamespace.d.ts +67 -2
  26. package/dist/client/namespaces/consolidated/StorageNamespace.d.ts.map +1 -1
  27. package/dist/client/namespaces/consolidated/StorageNamespace.js +549 -16
  28. package/dist/client/namespaces/consolidated/StorageNamespace.js.map +1 -1
  29. package/dist/config/ConfigurationHelper.js +61 -61
  30. package/dist/config/defaults.js +2 -2
  31. package/dist/config/defaults.js.map +1 -1
  32. package/dist/graph/GraphService.js +21 -21
  33. package/dist/graph/GraphService.js.map +1 -1
  34. package/dist/index.d.ts +3 -1
  35. package/dist/index.d.ts.map +1 -1
  36. package/dist/index.js +3 -1
  37. package/dist/index.js.map +1 -1
  38. package/dist/infrastructure/seal/EncryptionService.d.ts +9 -5
  39. package/dist/infrastructure/seal/EncryptionService.d.ts.map +1 -1
  40. package/dist/infrastructure/seal/EncryptionService.js +37 -15
  41. package/dist/infrastructure/seal/EncryptionService.js.map +1 -1
  42. package/dist/infrastructure/seal/SealService.d.ts +13 -5
  43. package/dist/infrastructure/seal/SealService.d.ts.map +1 -1
  44. package/dist/infrastructure/seal/SealService.js +36 -34
  45. package/dist/infrastructure/seal/SealService.js.map +1 -1
  46. package/dist/langchain/createPDWRAG.js +30 -30
  47. package/dist/retrieval/MemoryDecryptionPipeline.d.ts.map +1 -1
  48. package/dist/retrieval/MemoryDecryptionPipeline.js +2 -1
  49. package/dist/retrieval/MemoryDecryptionPipeline.js.map +1 -1
  50. package/dist/retrieval/MemoryRetrievalService.d.ts +31 -0
  51. package/dist/retrieval/MemoryRetrievalService.d.ts.map +1 -1
  52. package/dist/retrieval/MemoryRetrievalService.js +44 -4
  53. package/dist/retrieval/MemoryRetrievalService.js.map +1 -1
  54. package/dist/services/CapabilityService.d.ts.map +1 -1
  55. package/dist/services/CapabilityService.js +30 -14
  56. package/dist/services/CapabilityService.js.map +1 -1
  57. package/dist/services/CrossContextPermissionService.d.ts.map +1 -1
  58. package/dist/services/CrossContextPermissionService.js +9 -7
  59. package/dist/services/CrossContextPermissionService.js.map +1 -1
  60. package/dist/services/EmbeddingService.d.ts +28 -1
  61. package/dist/services/EmbeddingService.d.ts.map +1 -1
  62. package/dist/services/EmbeddingService.js +54 -0
  63. package/dist/services/EmbeddingService.js.map +1 -1
  64. package/dist/services/EncryptionService.d.ts.map +1 -1
  65. package/dist/services/EncryptionService.js +6 -5
  66. package/dist/services/EncryptionService.js.map +1 -1
  67. package/dist/services/GeminiAIService.js +309 -309
  68. package/dist/services/IndexManager.d.ts +5 -1
  69. package/dist/services/IndexManager.d.ts.map +1 -1
  70. package/dist/services/IndexManager.js +17 -40
  71. package/dist/services/IndexManager.js.map +1 -1
  72. package/dist/services/QueryService.js +1 -1
  73. package/dist/services/QueryService.js.map +1 -1
  74. package/dist/services/StorageService.d.ts +11 -0
  75. package/dist/services/StorageService.d.ts.map +1 -1
  76. package/dist/services/StorageService.js +73 -10
  77. package/dist/services/StorageService.js.map +1 -1
  78. package/dist/services/TransactionService.d.ts +20 -0
  79. package/dist/services/TransactionService.d.ts.map +1 -1
  80. package/dist/services/TransactionService.js +43 -0
  81. package/dist/services/TransactionService.js.map +1 -1
  82. package/dist/services/ViewService.js +2 -2
  83. package/dist/services/ViewService.js.map +1 -1
  84. package/dist/services/storage/QuiltBatchManager.d.ts +101 -1
  85. package/dist/services/storage/QuiltBatchManager.d.ts.map +1 -1
  86. package/dist/services/storage/QuiltBatchManager.js +410 -20
  87. package/dist/services/storage/QuiltBatchManager.js.map +1 -1
  88. package/dist/services/storage/index.d.ts +1 -1
  89. package/dist/services/storage/index.d.ts.map +1 -1
  90. package/dist/services/storage/index.js.map +1 -1
  91. package/dist/utils/LRUCache.d.ts +106 -0
  92. package/dist/utils/LRUCache.d.ts.map +1 -0
  93. package/dist/utils/LRUCache.js +281 -0
  94. package/dist/utils/LRUCache.js.map +1 -0
  95. package/dist/utils/index.d.ts +1 -0
  96. package/dist/utils/index.d.ts.map +1 -1
  97. package/dist/utils/index.js +2 -0
  98. package/dist/utils/index.js.map +1 -1
  99. package/dist/utils/memoryIndexOnChain.d.ts +212 -0
  100. package/dist/utils/memoryIndexOnChain.d.ts.map +1 -0
  101. package/dist/utils/memoryIndexOnChain.js +312 -0
  102. package/dist/utils/memoryIndexOnChain.js.map +1 -0
  103. package/dist/utils/rebuildIndexNode.d.ts +29 -0
  104. package/dist/utils/rebuildIndexNode.d.ts.map +1 -1
  105. package/dist/utils/rebuildIndexNode.js +366 -98
  106. package/dist/utils/rebuildIndexNode.js.map +1 -1
  107. package/dist/vector/HnswWasmService.d.ts +20 -5
  108. package/dist/vector/HnswWasmService.d.ts.map +1 -1
  109. package/dist/vector/HnswWasmService.js +73 -40
  110. package/dist/vector/HnswWasmService.js.map +1 -1
  111. package/dist/vector/IHnswService.d.ts +10 -1
  112. package/dist/vector/IHnswService.d.ts.map +1 -1
  113. package/dist/vector/IHnswService.js.map +1 -1
  114. package/dist/vector/NodeHnswService.d.ts +16 -0
  115. package/dist/vector/NodeHnswService.d.ts.map +1 -1
  116. package/dist/vector/NodeHnswService.js +84 -5
  117. package/dist/vector/NodeHnswService.js.map +1 -1
  118. package/dist/vector/createHnswService.d.ts +1 -1
  119. package/dist/vector/createHnswService.js +1 -1
  120. package/dist/vector/index.d.ts +1 -1
  121. package/dist/vector/index.js +1 -1
  122. package/package.json +157 -157
  123. package/src/access/PermissionService.ts +635 -635
  124. package/src/aggregation/AggregationService.ts +389 -389
  125. package/src/ai-sdk/PDWVectorStore.ts +715 -715
  126. package/src/ai-sdk/index.ts +65 -65
  127. package/src/ai-sdk/tools.ts +460 -460
  128. package/src/ai-sdk/types.ts +404 -404
  129. package/src/batch/BatchManager.ts +597 -597
  130. package/src/batch/BatchingService.ts +429 -429
  131. package/src/batch/MemoryProcessingCache.ts +492 -492
  132. package/src/batch/index.ts +30 -30
  133. package/src/browser.ts +200 -200
  134. package/src/client/ClientMemoryManager.ts +987 -987
  135. package/src/client/PersonalDataWallet.ts +345 -345
  136. package/src/client/SimplePDWClient.ts +1289 -1222
  137. package/src/client/factory.ts +154 -154
  138. package/src/client/namespaces/AnalyticsNamespace.ts +377 -377
  139. package/src/client/namespaces/BatchNamespace.ts +356 -356
  140. package/src/client/namespaces/CacheNamespace.ts +123 -123
  141. package/src/client/namespaces/CapabilityNamespace.ts +217 -217
  142. package/src/client/namespaces/ClassifyNamespace.ts +169 -169
  143. package/src/client/namespaces/ContextNamespace.ts +297 -297
  144. package/src/client/namespaces/EmbeddingsNamespace.ts +99 -99
  145. package/src/client/namespaces/EncryptionNamespace.ts +221 -221
  146. package/src/client/namespaces/GraphNamespace.ts +468 -468
  147. package/src/client/namespaces/IndexNamespace.ts +361 -361
  148. package/src/client/namespaces/MemoryNamespace.ts +1422 -1135
  149. package/src/client/namespaces/PermissionsNamespace.ts +254 -254
  150. package/src/client/namespaces/PipelineNamespace.ts +220 -220
  151. package/src/client/namespaces/SearchNamespace.ts +1049 -1049
  152. package/src/client/namespaces/StorageNamespace.ts +458 -458
  153. package/src/client/namespaces/TxNamespace.ts +260 -260
  154. package/src/client/namespaces/WalletNamespace.ts +243 -243
  155. package/src/client/namespaces/consolidated/AINamespace.ts +449 -449
  156. package/src/client/namespaces/consolidated/BlockchainNamespace.ts +607 -546
  157. package/src/client/namespaces/consolidated/SecurityNamespace.ts +648 -648
  158. package/src/client/namespaces/consolidated/StorageNamespace.ts +1141 -497
  159. package/src/client/namespaces/consolidated/index.ts +39 -39
  160. package/src/client/signers/KeypairSigner.ts +108 -108
  161. package/src/client/signers/UnifiedSigner.ts +110 -110
  162. package/src/client/signers/WalletAdapterSigner.ts +159 -159
  163. package/src/client/signers/index.ts +26 -26
  164. package/src/config/ConfigurationHelper.ts +412 -412
  165. package/src/config/defaults.ts +51 -51
  166. package/src/config/index.ts +8 -8
  167. package/src/config/validation.ts +70 -70
  168. package/src/core/index.ts +14 -14
  169. package/src/core/interfaces/IService.ts +307 -307
  170. package/src/core/interfaces/index.ts +8 -8
  171. package/src/core/types/capability.ts +297 -297
  172. package/src/core/types/index.ts +870 -870
  173. package/src/core/types/wallet.ts +270 -270
  174. package/src/core/types.ts +9 -9
  175. package/src/core/wallet.ts +222 -222
  176. package/src/embedding/index.ts +19 -19
  177. package/src/embedding/types.ts +357 -357
  178. package/src/errors/index.ts +602 -602
  179. package/src/errors/recovery.ts +461 -461
  180. package/src/errors/validation.ts +567 -567
  181. package/src/generated/pdw/capability.ts +319 -319
  182. package/src/graph/GraphService.ts +887 -887
  183. package/src/graph/KnowledgeGraphManager.ts +728 -728
  184. package/src/graph/index.ts +25 -25
  185. package/src/index.ts +498 -474
  186. package/src/infrastructure/index.ts +22 -22
  187. package/src/infrastructure/seal/EncryptionService.ts +628 -603
  188. package/src/infrastructure/seal/SealService.ts +613 -615
  189. package/src/infrastructure/seal/index.ts +9 -9
  190. package/src/infrastructure/sui/BlockchainManager.ts +627 -627
  191. package/src/infrastructure/sui/SuiService.ts +888 -888
  192. package/src/infrastructure/sui/index.ts +9 -9
  193. package/src/infrastructure/walrus/StorageManager.ts +604 -604
  194. package/src/infrastructure/walrus/WalrusStorageService.ts +612 -612
  195. package/src/infrastructure/walrus/index.ts +9 -9
  196. package/src/langchain/PDWEmbeddings.ts +145 -145
  197. package/src/langchain/PDWVectorStore.ts +456 -456
  198. package/src/langchain/createPDWRAG.ts +303 -303
  199. package/src/langchain/index.ts +47 -47
  200. package/src/permissions/ConsentRepository.browser.ts +249 -249
  201. package/src/permissions/ConsentRepository.ts +364 -364
  202. package/src/pipeline/MemoryPipeline.ts +862 -862
  203. package/src/pipeline/PipelineManager.ts +683 -683
  204. package/src/pipeline/index.ts +26 -26
  205. package/src/retrieval/AdvancedSearchService.ts +629 -629
  206. package/src/retrieval/MemoryAnalyticsService.ts +711 -711
  207. package/src/retrieval/MemoryDecryptionPipeline.ts +825 -824
  208. package/src/retrieval/MemoryRetrievalService.ts +904 -830
  209. package/src/retrieval/index.ts +42 -42
  210. package/src/services/BatchService.ts +352 -352
  211. package/src/services/CapabilityService.ts +464 -448
  212. package/src/services/ClassifierService.ts +465 -465
  213. package/src/services/CrossContextPermissionService.ts +486 -484
  214. package/src/services/EmbeddingService.ts +771 -706
  215. package/src/services/EncryptionService.ts +712 -711
  216. package/src/services/GeminiAIService.ts +753 -753
  217. package/src/services/IndexManager.ts +977 -1004
  218. package/src/services/MemoryIndexService.ts +1003 -1003
  219. package/src/services/MemoryService.ts +369 -369
  220. package/src/services/QueryService.ts +890 -890
  221. package/src/services/StorageService.ts +1182 -1111
  222. package/src/services/TransactionService.ts +838 -790
  223. package/src/services/VectorService.ts +462 -462
  224. package/src/services/ViewService.ts +484 -484
  225. package/src/services/index.ts +25 -25
  226. package/src/services/storage/BlobAttributesManager.ts +333 -333
  227. package/src/services/storage/KnowledgeGraphManager.ts +425 -425
  228. package/src/services/storage/MemorySearchManager.ts +387 -387
  229. package/src/services/storage/QuiltBatchManager.ts +1130 -660
  230. package/src/services/storage/WalrusMetadataManager.ts +268 -268
  231. package/src/services/storage/WalrusStorageManager.ts +287 -287
  232. package/src/services/storage/index.ts +57 -52
  233. package/src/types/index.ts +13 -13
  234. package/src/utils/LRUCache.ts +378 -0
  235. package/src/utils/index.ts +76 -68
  236. package/src/utils/memoryIndexOnChain.ts +507 -0
  237. package/src/utils/rebuildIndex.ts +290 -290
  238. package/src/utils/rebuildIndexNode.ts +771 -424
  239. package/src/vector/BrowserHnswIndexService.ts +758 -758
  240. package/src/vector/HnswWasmService.ts +731 -679
  241. package/src/vector/IHnswService.ts +233 -224
  242. package/src/vector/NodeHnswService.ts +833 -735
  243. package/src/vector/VectorManager.ts +478 -478
  244. package/src/vector/createHnswService.ts +135 -135
  245. package/src/vector/index.ts +56 -56
  246. package/src/wallet/ContextWalletService.ts +656 -656
  247. package/src/wallet/MainWalletService.ts +317 -317
@@ -1,346 +1,346 @@
1
- /**
2
- * PersonalDataWallet Client Extension
3
- *
4
- * Main client extension that follows MystenLabs patterns for Sui ecosystem SDKs.
5
- * Provides a composable API for memory management, chat, storage, and encryption.
6
- */
7
-
8
- import type { ClientWithCoreApi, PDWConfig } from '../types';
9
- import { MemoryService } from '../services/MemoryService';
10
- import { StorageService } from '../services/StorageService';
11
- import { EncryptionService } from '../services/EncryptionService';
12
- import { TransactionService } from '../services/TransactionService';
13
- import { ViewService } from '../services/ViewService';
14
- import { CapabilityService } from '../services/CapabilityService';
15
- import { MainWalletService } from '../wallet/MainWalletService';
16
- import { ContextWalletService } from '../wallet/ContextWalletService';
17
- import { PermissionService } from '../access/PermissionService';
18
- import { AggregationService } from '../aggregation/AggregationService';
19
- import { createDefaultConfig } from '../config/defaults';
20
- import { validateConfig } from '../config/validation';
21
- import type { ConsentRepository } from '../permissions/ConsentRepository';
22
-
23
- // BCS types from generated contracts (ESM static imports)
24
- import * as memoryBcsTypes from '../generated/pdw/memory';
25
- import * as capabilityBcsTypes from '../generated/pdw/capability';
26
- import * as walletBcsTypes from '../generated/pdw/wallet';
27
-
28
- export interface PersonalDataWalletExtension {
29
- // Top-level imperative methods
30
- setConsentRepository: (repository?: ConsentRepository) => void;
31
-
32
- // Storage methods
33
- uploadToStorage: StorageService['upload'];
34
- retrieveFromStorage: StorageService['retrieve'];
35
-
36
- // Organized service methods
37
- tx: {
38
- createMemoryRecord: TransactionService['buildCreateMemoryRecord'];
39
- updateMemoryMetadata: TransactionService['buildUpdateMemoryMetadata'];
40
- deleteMemoryRecord: TransactionService['buildDeleteMemoryRecord'];
41
- grantAccess: TransactionService['buildGrantAccess'];
42
- revokeAccess: TransactionService['buildRevokeAccess'];
43
- registerContent: TransactionService['buildRegisterContent'];
44
- executeBatch: TransactionService['executeBatch'];
45
- };
46
-
47
- call: {
48
- createMemoryRecord: TransactionService['createMemoryRecord'];
49
- updateMemoryMetadata: TransactionService['updateMemoryMetadata'];
50
- deleteMemoryRecord: TransactionService['deleteMemoryRecord'];
51
- grantAccess: TransactionService['grantAccess'];
52
- revokeAccess: TransactionService['revokeAccess'];
53
- executeBatch: TransactionService['executeBatch'];
54
- };
55
-
56
- view: {
57
- getUserMemories: ViewService['getUserMemories'];
58
- getMemoryIndex: ViewService['getMemoryIndex'];
59
- getMemory: ViewService['getMemory'];
60
- getMemoryStats: ViewService['getMemoryStats'];
61
- getAccessPermissions: ViewService['getAccessPermissions'];
62
- getContentRegistry: ViewService['getContentRegistry'];
63
- objectExists: ViewService['objectExists'];
64
- getObjectType: ViewService['getObjectType'];
65
- findMemoryByContentHash: ViewService['findMemoryByContentHash'];
66
- };
67
-
68
- // Wallet architecture services
69
- wallet: {
70
- getMainWallet: MainWalletService['getMainWallet'];
71
- createMainWallet: MainWalletService['createMainWallet'];
72
- deriveContextId: MainWalletService['deriveContextId'];
73
- rotateKeys: MainWalletService['rotateKeys'];
74
- ensureMainWallet: MainWalletService['ensureMainWallet'];
75
- };
76
-
77
- context: {
78
- create: ContextWalletService['create'];
79
- getContext: ContextWalletService['getContext'];
80
- listUserContexts: ContextWalletService['listUserContexts'];
81
- addData: ContextWalletService['addData'];
82
- removeData: ContextWalletService['removeData'];
83
- listData: ContextWalletService['listData'];
84
- ensureContext: ContextWalletService['ensureContext'];
85
- };
86
-
87
- access: {
88
- requestConsent: PermissionService['requestConsent'];
89
- grantPermissions: PermissionService['grantPermissions'];
90
- revokePermissions: PermissionService['revokePermissions'];
91
- checkPermission: PermissionService['checkPermission'];
92
- getGrantsByUser: PermissionService['getGrantsByUser'];
93
- validateOAuthPermission: PermissionService['validateOAuthPermission'];
94
- };
95
-
96
- aggregate: {
97
- query: AggregationService['query'];
98
- queryWithScopes: AggregationService['queryWithScopes'];
99
- search: AggregationService['search'];
100
- getAggregatedStats: AggregationService['getAggregatedStats'];
101
- };
102
-
103
- bcs: {
104
- // Will be populated by generated types from @mysten/codegen
105
- Memory: () => any;
106
- MemoryIndex: () => any;
107
- MemoryMetadata: () => any;
108
- AccessControl: () => any;
109
- };
110
-
111
- // Service instances for advanced usage
112
- memory: MemoryService;
113
- storage: StorageService;
114
- encryption: EncryptionService;
115
- viewService: ViewService;
116
- /** Capability-based access control service (preferred) */
117
- capabilityService: CapabilityService;
118
- /** @deprecated Use capabilityService instead */
119
- mainWalletService: MainWalletService;
120
- /** @deprecated Use ContextNamespace instead */
121
- contextWalletService: ContextWalletService;
122
- permissionService: PermissionService;
123
- aggregationService: AggregationService;
124
-
125
- // Configuration
126
- config: PDWConfig;
127
- }
128
-
129
- export class PersonalDataWallet {
130
- #client: ClientWithCoreApi;
131
- #config: PDWConfig;
132
- #transactions: TransactionService;
133
- #view: ViewService;
134
- #memory: MemoryService;
135
- #storage: StorageService;
136
- #encryption: EncryptionService;
137
- #capability: CapabilityService;
138
- /** @deprecated Use #capability instead */
139
- #mainWallet: MainWalletService;
140
- /** @deprecated Use ContextNamespace instead */
141
- #contextWallet: ContextWalletService;
142
- #permission: PermissionService;
143
- #aggregation: AggregationService;
144
-
145
- constructor(client: ClientWithCoreApi, config?: Partial<PDWConfig>) {
146
- this.#client = client;
147
- this.#config = validateConfig({ ...createDefaultConfig(), ...config });
148
-
149
- // Initialize services
150
- this.#transactions = new TransactionService(client as any, this.#config);
151
- this.#view = new ViewService(client, this.#config);
152
- this.#memory = new MemoryService(client, this.#config);
153
- this.#storage = new StorageService(this.#config);
154
- this.#encryption = new EncryptionService(client, this.#config);
155
-
156
- // Initialize capability service (new architecture)
157
- this.#capability = new CapabilityService({
158
- suiClient: (client as any).client || client,
159
- packageId: this.#config.packageId || ''
160
- });
161
-
162
- // Initialize wallet architecture services (legacy - deprecated)
163
- this.#mainWallet = new MainWalletService({
164
- suiClient: (client as any).client || client,
165
- packageId: this.#config.packageId || ''
166
- });
167
-
168
- this.#contextWallet = new ContextWalletService({
169
- suiClient: (client as any).client || client,
170
- packageId: this.#config.packageId || '',
171
- mainWalletService: this.#mainWallet,
172
- storageService: this.#storage,
173
- encryptionService: this.#encryption
174
- });
175
-
176
- this.#permission = new PermissionService({
177
- suiClient: (client as any).client || client,
178
- packageId: this.#config.packageId || '',
179
- accessRegistryId: this.#config.accessRegistryId || '',
180
- capabilityService: this.#capability,
181
- contextWalletService: this.#contextWallet
182
- });
183
-
184
- this.#aggregation = new AggregationService({
185
- suiClient: (client as any).client || client,
186
- packageId: this.#config.packageId || '',
187
- permissionService: this.#permission,
188
- capabilityService: this.#capability,
189
- contextWalletService: this.#contextWallet
190
- });
191
-
192
- // Bind methods after services are initialized
193
- this.uploadToStorage = this.#storage.upload.bind(this.#storage);
194
- this.retrieveFromStorage = this.#storage.retrieve.bind(this.#storage);
195
- }
196
-
197
- // Top-level imperative methods (declarations)
198
- uploadToStorage: StorageService['upload'];
199
- retrieveFromStorage: StorageService['retrieve'];
200
- setConsentRepository(repository?: ConsentRepository): void {
201
- this.#permission.setConsentRepository(repository);
202
- }
203
-
204
- // Transaction builders
205
- get tx() {
206
- return {
207
- createMemoryRecord: this.#transactions.buildCreateMemoryRecord.bind(this.#transactions),
208
- updateMemoryMetadata: this.#transactions.buildUpdateMemoryMetadata.bind(this.#transactions),
209
- deleteMemoryRecord: this.#transactions.buildDeleteMemoryRecord.bind(this.#transactions),
210
- grantAccess: this.#transactions.buildGrantAccess.bind(this.#transactions),
211
- revokeAccess: this.#transactions.buildRevokeAccess.bind(this.#transactions),
212
- registerContent: this.#transactions.buildRegisterContent.bind(this.#transactions),
213
- executeBatch: this.#transactions.executeBatch.bind(this.#transactions),
214
- };
215
- }
216
-
217
- // Transaction execution (async thunks)
218
- get call() {
219
- return {
220
- createMemoryRecord: this.#transactions.createMemoryRecord.bind(this.#transactions),
221
- updateMemoryMetadata: this.#transactions.updateMemoryMetadata.bind(this.#transactions),
222
- deleteMemoryRecord: this.#transactions.deleteMemoryRecord.bind(this.#transactions),
223
- grantAccess: this.#transactions.grantAccess.bind(this.#transactions),
224
- revokeAccess: this.#transactions.revokeAccess.bind(this.#transactions),
225
- executeBatch: this.#transactions.executeBatch.bind(this.#transactions),
226
- };
227
- }
228
-
229
- // View methods
230
- get view() {
231
- return {
232
- getUserMemories: this.#view.getUserMemories.bind(this.#view),
233
- getMemoryIndex: this.#view.getMemoryIndex.bind(this.#view),
234
- getMemory: this.#view.getMemory.bind(this.#view),
235
- getMemoryStats: this.#view.getMemoryStats.bind(this.#view),
236
- getStorageStats: this.#storage.getStats.bind(this.#storage),
237
- listStoredItems: this.#storage.list.bind(this.#storage),
238
- getAccessPermissions: this.#view.getAccessPermissions.bind(this.#view),
239
- getContentRegistry: this.#view.getContentRegistry.bind(this.#view),
240
- objectExists: this.#view.objectExists.bind(this.#view),
241
- getObjectType: this.#view.getObjectType.bind(this.#view),
242
- findMemoryByContentHash: this.#view.findMemoryByContentHash.bind(this.#view),
243
- };
244
- }
245
-
246
- // BCS types from generated contracts (using static ESM imports)
247
- get bcs() {
248
- return {
249
- // Memory types
250
- Memory: memoryBcsTypes.Memory,
251
- MemoryIndex: memoryBcsTypes.MemoryIndex,
252
- MemoryMetadata: memoryBcsTypes.MemoryMetadata,
253
- MemoryCreated: memoryBcsTypes.MemoryCreated,
254
- MemoryIndexUpdated: memoryBcsTypes.MemoryIndexUpdated,
255
- MemoryMetadataUpdated: memoryBcsTypes.MemoryMetadataUpdated,
256
- MemoryUpdated: memoryBcsTypes.MemoryUpdated,
257
-
258
- // Capability types
259
- MemoryCap: capabilityBcsTypes.MemoryCap,
260
- MemoryCapCreated: capabilityBcsTypes.MemoryCapCreated,
261
-
262
- // Wallet types
263
- ...walletBcsTypes,
264
- };
265
- }
266
-
267
- // Wallet architecture service getters
268
- get wallet() {
269
- return {
270
- getMainWallet: this.#mainWallet.getMainWallet.bind(this.#mainWallet),
271
- createMainWallet: this.#mainWallet.createMainWallet.bind(this.#mainWallet),
272
- deriveContextId: this.#mainWallet.deriveContextId.bind(this.#mainWallet),
273
- rotateKeys: this.#mainWallet.rotateKeys.bind(this.#mainWallet),
274
- ensureMainWallet: this.#mainWallet.ensureMainWallet.bind(this.#mainWallet),
275
- };
276
- }
277
-
278
- get context() {
279
- return {
280
- create: this.#contextWallet.create.bind(this.#contextWallet),
281
- getContext: this.#contextWallet.getContext.bind(this.#contextWallet),
282
- listUserContexts: this.#contextWallet.listUserContexts.bind(this.#contextWallet),
283
- addData: this.#contextWallet.addData.bind(this.#contextWallet),
284
- removeData: this.#contextWallet.removeData.bind(this.#contextWallet),
285
- listData: this.#contextWallet.listData.bind(this.#contextWallet),
286
- ensureContext: this.#contextWallet.ensureContext.bind(this.#contextWallet),
287
- };
288
- }
289
-
290
- get access() {
291
- return {
292
- requestConsent: this.#permission.requestConsent.bind(this.#permission),
293
- grantPermissions: this.#permission.grantPermissions.bind(this.#permission),
294
- revokePermissions: this.#permission.revokePermissions.bind(this.#permission),
295
- checkPermission: this.#permission.checkPermission.bind(this.#permission),
296
- getGrantsByUser: this.#permission.getGrantsByUser.bind(this.#permission),
297
- validateOAuthPermission: this.#permission.validateOAuthPermission.bind(this.#permission),
298
- };
299
- }
300
-
301
- get aggregate() {
302
- return {
303
- query: this.#aggregation.query.bind(this.#aggregation),
304
- queryWithScopes: this.#aggregation.queryWithScopes.bind(this.#aggregation),
305
- search: this.#aggregation.search.bind(this.#aggregation),
306
- getAggregatedStats: this.#aggregation.getAggregatedStats.bind(this.#aggregation),
307
- };
308
- }
309
-
310
- // Service instances
311
- get memory() { return this.#memory; }
312
- get storage() { return this.#storage; }
313
- get encryption() { return this.#encryption; }
314
- get config() { return this.#config; }
315
- get viewService() { return this.#view; }
316
- /** Capability-based access control service (preferred) */
317
- get capabilityService() { return this.#capability; }
318
- /** @deprecated Use capabilityService instead */
319
- get mainWalletService() { return this.#mainWallet; }
320
- /** @deprecated Use ContextNamespace instead */
321
- get contextWalletService() { return this.#contextWallet; }
322
- get permissionService() { return this.#permission; }
323
- get aggregationService() { return this.#aggregation; }
324
-
325
- // Client extension factory
326
- static asClientExtension(config?: Partial<PDWConfig>) {
327
- return {
328
- name: 'pdw' as const,
329
- register: (client: any) => {
330
- // Adapt the client to match our expected interface
331
- const adaptedClient: ClientWithCoreApi = {
332
- core: {
333
- getObject: (objectId: string) => client.getObject({ id: objectId }),
334
- getObjects: (objectIds: string[]) => client.getObjects(objectIds.map(id => ({ id }))),
335
- executeTransaction: (tx: any) => client.executeTransactionBlock({ transactionBlock: tx }),
336
- },
337
- $extend: client.$extend.bind(client),
338
- };
339
- return new PersonalDataWallet(adaptedClient, config);
340
- },
341
- };
342
- }
343
- }
344
-
345
- // Export for easier usage
1
+ /**
2
+ * PersonalDataWallet Client Extension
3
+ *
4
+ * Main client extension that follows MystenLabs patterns for Sui ecosystem SDKs.
5
+ * Provides a composable API for memory management, chat, storage, and encryption.
6
+ */
7
+
8
+ import type { ClientWithCoreApi, PDWConfig } from '../types';
9
+ import { MemoryService } from '../services/MemoryService';
10
+ import { StorageService } from '../services/StorageService';
11
+ import { EncryptionService } from '../services/EncryptionService';
12
+ import { TransactionService } from '../services/TransactionService';
13
+ import { ViewService } from '../services/ViewService';
14
+ import { CapabilityService } from '../services/CapabilityService';
15
+ import { MainWalletService } from '../wallet/MainWalletService';
16
+ import { ContextWalletService } from '../wallet/ContextWalletService';
17
+ import { PermissionService } from '../access/PermissionService';
18
+ import { AggregationService } from '../aggregation/AggregationService';
19
+ import { createDefaultConfig } from '../config/defaults';
20
+ import { validateConfig } from '../config/validation';
21
+ import type { ConsentRepository } from '../permissions/ConsentRepository';
22
+
23
+ // BCS types from generated contracts (ESM static imports)
24
+ import * as memoryBcsTypes from '../generated/pdw/memory';
25
+ import * as capabilityBcsTypes from '../generated/pdw/capability';
26
+ import * as walletBcsTypes from '../generated/pdw/wallet';
27
+
28
+ export interface PersonalDataWalletExtension {
29
+ // Top-level imperative methods
30
+ setConsentRepository: (repository?: ConsentRepository) => void;
31
+
32
+ // Storage methods
33
+ uploadToStorage: StorageService['upload'];
34
+ retrieveFromStorage: StorageService['retrieve'];
35
+
36
+ // Organized service methods
37
+ tx: {
38
+ createMemoryRecord: TransactionService['buildCreateMemoryRecord'];
39
+ updateMemoryMetadata: TransactionService['buildUpdateMemoryMetadata'];
40
+ deleteMemoryRecord: TransactionService['buildDeleteMemoryRecord'];
41
+ grantAccess: TransactionService['buildGrantAccess'];
42
+ revokeAccess: TransactionService['buildRevokeAccess'];
43
+ registerContent: TransactionService['buildRegisterContent'];
44
+ executeBatch: TransactionService['executeBatch'];
45
+ };
46
+
47
+ call: {
48
+ createMemoryRecord: TransactionService['createMemoryRecord'];
49
+ updateMemoryMetadata: TransactionService['updateMemoryMetadata'];
50
+ deleteMemoryRecord: TransactionService['deleteMemoryRecord'];
51
+ grantAccess: TransactionService['grantAccess'];
52
+ revokeAccess: TransactionService['revokeAccess'];
53
+ executeBatch: TransactionService['executeBatch'];
54
+ };
55
+
56
+ view: {
57
+ getUserMemories: ViewService['getUserMemories'];
58
+ getMemoryIndex: ViewService['getMemoryIndex'];
59
+ getMemory: ViewService['getMemory'];
60
+ getMemoryStats: ViewService['getMemoryStats'];
61
+ getAccessPermissions: ViewService['getAccessPermissions'];
62
+ getContentRegistry: ViewService['getContentRegistry'];
63
+ objectExists: ViewService['objectExists'];
64
+ getObjectType: ViewService['getObjectType'];
65
+ findMemoryByContentHash: ViewService['findMemoryByContentHash'];
66
+ };
67
+
68
+ // Wallet architecture services
69
+ wallet: {
70
+ getMainWallet: MainWalletService['getMainWallet'];
71
+ createMainWallet: MainWalletService['createMainWallet'];
72
+ deriveContextId: MainWalletService['deriveContextId'];
73
+ rotateKeys: MainWalletService['rotateKeys'];
74
+ ensureMainWallet: MainWalletService['ensureMainWallet'];
75
+ };
76
+
77
+ context: {
78
+ create: ContextWalletService['create'];
79
+ getContext: ContextWalletService['getContext'];
80
+ listUserContexts: ContextWalletService['listUserContexts'];
81
+ addData: ContextWalletService['addData'];
82
+ removeData: ContextWalletService['removeData'];
83
+ listData: ContextWalletService['listData'];
84
+ ensureContext: ContextWalletService['ensureContext'];
85
+ };
86
+
87
+ access: {
88
+ requestConsent: PermissionService['requestConsent'];
89
+ grantPermissions: PermissionService['grantPermissions'];
90
+ revokePermissions: PermissionService['revokePermissions'];
91
+ checkPermission: PermissionService['checkPermission'];
92
+ getGrantsByUser: PermissionService['getGrantsByUser'];
93
+ validateOAuthPermission: PermissionService['validateOAuthPermission'];
94
+ };
95
+
96
+ aggregate: {
97
+ query: AggregationService['query'];
98
+ queryWithScopes: AggregationService['queryWithScopes'];
99
+ search: AggregationService['search'];
100
+ getAggregatedStats: AggregationService['getAggregatedStats'];
101
+ };
102
+
103
+ bcs: {
104
+ // Will be populated by generated types from @mysten/codegen
105
+ Memory: () => any;
106
+ MemoryIndex: () => any;
107
+ MemoryMetadata: () => any;
108
+ AccessControl: () => any;
109
+ };
110
+
111
+ // Service instances for advanced usage
112
+ memory: MemoryService;
113
+ storage: StorageService;
114
+ encryption: EncryptionService;
115
+ viewService: ViewService;
116
+ /** Capability-based access control service (preferred) */
117
+ capabilityService: CapabilityService;
118
+ /** @deprecated Use capabilityService instead */
119
+ mainWalletService: MainWalletService;
120
+ /** @deprecated Use ContextNamespace instead */
121
+ contextWalletService: ContextWalletService;
122
+ permissionService: PermissionService;
123
+ aggregationService: AggregationService;
124
+
125
+ // Configuration
126
+ config: PDWConfig;
127
+ }
128
+
129
+ export class PersonalDataWallet {
130
+ #client: ClientWithCoreApi;
131
+ #config: PDWConfig;
132
+ #transactions: TransactionService;
133
+ #view: ViewService;
134
+ #memory: MemoryService;
135
+ #storage: StorageService;
136
+ #encryption: EncryptionService;
137
+ #capability: CapabilityService;
138
+ /** @deprecated Use #capability instead */
139
+ #mainWallet: MainWalletService;
140
+ /** @deprecated Use ContextNamespace instead */
141
+ #contextWallet: ContextWalletService;
142
+ #permission: PermissionService;
143
+ #aggregation: AggregationService;
144
+
145
+ constructor(client: ClientWithCoreApi, config?: Partial<PDWConfig>) {
146
+ this.#client = client;
147
+ this.#config = validateConfig({ ...createDefaultConfig(), ...config });
148
+
149
+ // Initialize services
150
+ this.#transactions = new TransactionService(client as any, this.#config);
151
+ this.#view = new ViewService(client, this.#config);
152
+ this.#memory = new MemoryService(client, this.#config);
153
+ this.#storage = new StorageService(this.#config);
154
+ this.#encryption = new EncryptionService(client, this.#config);
155
+
156
+ // Initialize capability service (new architecture)
157
+ this.#capability = new CapabilityService({
158
+ suiClient: (client as any).client || client,
159
+ packageId: this.#config.packageId || ''
160
+ });
161
+
162
+ // Initialize wallet architecture services (legacy - deprecated)
163
+ this.#mainWallet = new MainWalletService({
164
+ suiClient: (client as any).client || client,
165
+ packageId: this.#config.packageId || ''
166
+ });
167
+
168
+ this.#contextWallet = new ContextWalletService({
169
+ suiClient: (client as any).client || client,
170
+ packageId: this.#config.packageId || '',
171
+ mainWalletService: this.#mainWallet,
172
+ storageService: this.#storage,
173
+ encryptionService: this.#encryption
174
+ });
175
+
176
+ this.#permission = new PermissionService({
177
+ suiClient: (client as any).client || client,
178
+ packageId: this.#config.packageId || '',
179
+ accessRegistryId: this.#config.accessRegistryId || '',
180
+ capabilityService: this.#capability,
181
+ contextWalletService: this.#contextWallet
182
+ });
183
+
184
+ this.#aggregation = new AggregationService({
185
+ suiClient: (client as any).client || client,
186
+ packageId: this.#config.packageId || '',
187
+ permissionService: this.#permission,
188
+ capabilityService: this.#capability,
189
+ contextWalletService: this.#contextWallet
190
+ });
191
+
192
+ // Bind methods after services are initialized
193
+ this.uploadToStorage = this.#storage.upload.bind(this.#storage);
194
+ this.retrieveFromStorage = this.#storage.retrieve.bind(this.#storage);
195
+ }
196
+
197
+ // Top-level imperative methods (declarations)
198
+ uploadToStorage: StorageService['upload'];
199
+ retrieveFromStorage: StorageService['retrieve'];
200
+ setConsentRepository(repository?: ConsentRepository): void {
201
+ this.#permission.setConsentRepository(repository);
202
+ }
203
+
204
+ // Transaction builders
205
+ get tx() {
206
+ return {
207
+ createMemoryRecord: this.#transactions.buildCreateMemoryRecord.bind(this.#transactions),
208
+ updateMemoryMetadata: this.#transactions.buildUpdateMemoryMetadata.bind(this.#transactions),
209
+ deleteMemoryRecord: this.#transactions.buildDeleteMemoryRecord.bind(this.#transactions),
210
+ grantAccess: this.#transactions.buildGrantAccess.bind(this.#transactions),
211
+ revokeAccess: this.#transactions.buildRevokeAccess.bind(this.#transactions),
212
+ registerContent: this.#transactions.buildRegisterContent.bind(this.#transactions),
213
+ executeBatch: this.#transactions.executeBatch.bind(this.#transactions),
214
+ };
215
+ }
216
+
217
+ // Transaction execution (async thunks)
218
+ get call() {
219
+ return {
220
+ createMemoryRecord: this.#transactions.createMemoryRecord.bind(this.#transactions),
221
+ updateMemoryMetadata: this.#transactions.updateMemoryMetadata.bind(this.#transactions),
222
+ deleteMemoryRecord: this.#transactions.deleteMemoryRecord.bind(this.#transactions),
223
+ grantAccess: this.#transactions.grantAccess.bind(this.#transactions),
224
+ revokeAccess: this.#transactions.revokeAccess.bind(this.#transactions),
225
+ executeBatch: this.#transactions.executeBatch.bind(this.#transactions),
226
+ };
227
+ }
228
+
229
+ // View methods
230
+ get view() {
231
+ return {
232
+ getUserMemories: this.#view.getUserMemories.bind(this.#view),
233
+ getMemoryIndex: this.#view.getMemoryIndex.bind(this.#view),
234
+ getMemory: this.#view.getMemory.bind(this.#view),
235
+ getMemoryStats: this.#view.getMemoryStats.bind(this.#view),
236
+ getStorageStats: this.#storage.getStats.bind(this.#storage),
237
+ listStoredItems: this.#storage.list.bind(this.#storage),
238
+ getAccessPermissions: this.#view.getAccessPermissions.bind(this.#view),
239
+ getContentRegistry: this.#view.getContentRegistry.bind(this.#view),
240
+ objectExists: this.#view.objectExists.bind(this.#view),
241
+ getObjectType: this.#view.getObjectType.bind(this.#view),
242
+ findMemoryByContentHash: this.#view.findMemoryByContentHash.bind(this.#view),
243
+ };
244
+ }
245
+
246
+ // BCS types from generated contracts (using static ESM imports)
247
+ get bcs() {
248
+ return {
249
+ // Memory types
250
+ Memory: memoryBcsTypes.Memory,
251
+ MemoryIndex: memoryBcsTypes.MemoryIndex,
252
+ MemoryMetadata: memoryBcsTypes.MemoryMetadata,
253
+ MemoryCreated: memoryBcsTypes.MemoryCreated,
254
+ MemoryIndexUpdated: memoryBcsTypes.MemoryIndexUpdated,
255
+ MemoryMetadataUpdated: memoryBcsTypes.MemoryMetadataUpdated,
256
+ MemoryUpdated: memoryBcsTypes.MemoryUpdated,
257
+
258
+ // Capability types
259
+ MemoryCap: capabilityBcsTypes.MemoryCap,
260
+ MemoryCapCreated: capabilityBcsTypes.MemoryCapCreated,
261
+
262
+ // Wallet types
263
+ ...walletBcsTypes,
264
+ };
265
+ }
266
+
267
+ // Wallet architecture service getters
268
+ get wallet() {
269
+ return {
270
+ getMainWallet: this.#mainWallet.getMainWallet.bind(this.#mainWallet),
271
+ createMainWallet: this.#mainWallet.createMainWallet.bind(this.#mainWallet),
272
+ deriveContextId: this.#mainWallet.deriveContextId.bind(this.#mainWallet),
273
+ rotateKeys: this.#mainWallet.rotateKeys.bind(this.#mainWallet),
274
+ ensureMainWallet: this.#mainWallet.ensureMainWallet.bind(this.#mainWallet),
275
+ };
276
+ }
277
+
278
+ get context() {
279
+ return {
280
+ create: this.#contextWallet.create.bind(this.#contextWallet),
281
+ getContext: this.#contextWallet.getContext.bind(this.#contextWallet),
282
+ listUserContexts: this.#contextWallet.listUserContexts.bind(this.#contextWallet),
283
+ addData: this.#contextWallet.addData.bind(this.#contextWallet),
284
+ removeData: this.#contextWallet.removeData.bind(this.#contextWallet),
285
+ listData: this.#contextWallet.listData.bind(this.#contextWallet),
286
+ ensureContext: this.#contextWallet.ensureContext.bind(this.#contextWallet),
287
+ };
288
+ }
289
+
290
+ get access() {
291
+ return {
292
+ requestConsent: this.#permission.requestConsent.bind(this.#permission),
293
+ grantPermissions: this.#permission.grantPermissions.bind(this.#permission),
294
+ revokePermissions: this.#permission.revokePermissions.bind(this.#permission),
295
+ checkPermission: this.#permission.checkPermission.bind(this.#permission),
296
+ getGrantsByUser: this.#permission.getGrantsByUser.bind(this.#permission),
297
+ validateOAuthPermission: this.#permission.validateOAuthPermission.bind(this.#permission),
298
+ };
299
+ }
300
+
301
+ get aggregate() {
302
+ return {
303
+ query: this.#aggregation.query.bind(this.#aggregation),
304
+ queryWithScopes: this.#aggregation.queryWithScopes.bind(this.#aggregation),
305
+ search: this.#aggregation.search.bind(this.#aggregation),
306
+ getAggregatedStats: this.#aggregation.getAggregatedStats.bind(this.#aggregation),
307
+ };
308
+ }
309
+
310
+ // Service instances
311
+ get memory() { return this.#memory; }
312
+ get storage() { return this.#storage; }
313
+ get encryption() { return this.#encryption; }
314
+ get config() { return this.#config; }
315
+ get viewService() { return this.#view; }
316
+ /** Capability-based access control service (preferred) */
317
+ get capabilityService() { return this.#capability; }
318
+ /** @deprecated Use capabilityService instead */
319
+ get mainWalletService() { return this.#mainWallet; }
320
+ /** @deprecated Use ContextNamespace instead */
321
+ get contextWalletService() { return this.#contextWallet; }
322
+ get permissionService() { return this.#permission; }
323
+ get aggregationService() { return this.#aggregation; }
324
+
325
+ // Client extension factory
326
+ static asClientExtension(config?: Partial<PDWConfig>) {
327
+ return {
328
+ name: 'pdw' as const,
329
+ register: (client: any) => {
330
+ // Adapt the client to match our expected interface
331
+ const adaptedClient: ClientWithCoreApi = {
332
+ core: {
333
+ getObject: (objectId: string) => client.getObject({ id: objectId }),
334
+ getObjects: (objectIds: string[]) => client.getObjects(objectIds.map(id => ({ id }))),
335
+ executeTransaction: (tx: any) => client.executeTransactionBlock({ transactionBlock: tx }),
336
+ },
337
+ $extend: client.$extend.bind(client),
338
+ };
339
+ return new PersonalDataWallet(adaptedClient, config);
340
+ },
341
+ };
342
+ }
343
+ }
344
+
345
+ // Export for easier usage
346
346
  export default PersonalDataWallet;