@walletmesh/aztec-rpc-wallet 0.3.0 → 0.3.1

package/src/aztecRemoteWallet.ts

@@ -186,10 +186,10 @@ export class AztecRemoteWallet implements Wallet {
       .execute() as Promise<AztecWalletMethodMap['aztec_getContractClassMetadata']['result']>;
   }
 
-  async addCapsule(capsule: Fr[]): Promise<void> {
+  async addCapsule(contract: AztecAddress, storageSlot: Fr, capsule: Fr[]): Promise<void> {
     const result = (await this._provider
       .chain(this._providerChainId)
-      .call('aztec_addCapsule', { capsule })
+      .call('aztec_addCapsule', { contract, storageSlot, capsule })
       .execute()) as AztecWalletMethodMap['aztec_addCapsule']['result'];
     if (!result) throw new Error('Failed to add capsule');
   }

package/src/handlers/aztecAccountWallet.test.ts

@@ -473,9 +473,10 @@ describe('Aztec Account Wallet Handler', () => {
       const mockCapsule = [await Fr.random(), await Fr.random(), await Fr.random()];
       wallet.addCapsule = vi.fn().mockResolvedValue(undefined);
 
-      const result = await aztecWalletHandler(context, 'aztec_addCapsule', { capsule: mockCapsule });
-      expect(result).toBe(true);
-      expect(wallet.addCapsule).toHaveBeenCalledWith(mockCapsule);
+      const contract = await AztecAddress.random();
+      const storageSlot = await Fr.random();
+      await aztecWalletHandler(context, 'aztec_addCapsule', { contract, storageSlot, capsule: mockCapsule });
+      expect(wallet.addCapsule).toHaveBeenCalledWith(contract, storageSlot, mockCapsule);
     });
   });
 

package/src/handlers/aztecAccountWallet.ts

@@ -105,8 +105,7 @@ async function handleAddCapsule(
   wallet: AccountWallet,
   params: AztecWalletMethodMap['aztec_addCapsule']['params'],
 ): Promise<AztecWalletMethodMap['aztec_addCapsule']['result']> {
-  await wallet.addCapsule(params.capsule);
-  return true;
+  await wallet.addCapsule(params.contract, params.storageSlot, params.capsule);
 }
 
 async function handleGetAddress(

package/src/provider.test.ts

@@ -65,6 +65,9 @@ describe('AztecProvider', () => {
       // Create a spy on the on method before creating provider
       const onSpy = vi.spyOn(WalletRouterProvider.prototype, 'on');
 
+      // Create a new provider
+      new AztecProvider(mockTransport);
+
       expect(onSpy).toHaveBeenCalledWith('wm_walletStateChanged', expect.any(Function));
       expect(onSpy).toHaveBeenCalledWith('wm_sessionTerminated', expect.any(Function));
 

package/src/types.ts

@@ -181,7 +181,7 @@ export interface AztecWalletMethodMap extends AztecWalletBaseMethodMap {
   };
 
   /* Capsules */
-  aztec_addCapsule: { params: { capsule: Fr[] }; result: boolean };
+  aztec_addCapsule: { params: { contract: AztecAddress; storageSlot: Fr; capsule: Fr[] }; result: undefined };
 
   /* Accounts */
   aztec_getAddress: { result: AztecAddress };