@chipi-stack/chipi-react 14.1.1 → 14.2.1

package/dist/index.mjs

@@ -438,6 +438,61 @@ function useChipiSession(config) {
       setError(err);
     }
   });
+  const setSpendingPolicyMutation = useMutation({
+    mutationFn: async (policyConfig) => {
+      if (!wallet) throw new Error("Wallet is required");
+      if (!localSession)
+        throw new Error("No session to set policy on. Create and register a session first.");
+      if (!encryptKey) throw new Error("Encryption key (PIN) is required");
+      if (!supportsSession) throw new Error("Wallet does not support sessions");
+      const bearerToken = await getBearerToken();
+      if (!bearerToken) throw new Error("Bearer token is required");
+      const txHash = await chipiSDK.sessions.setSpendingPolicy(
+        {
+          encryptKey,
+          wallet,
+          spendingPolicyConfig: {
+            sessionPublicKey: localSession.publicKey,
+            ...policyConfig
+          }
+        },
+        bearerToken
+      );
+      return txHash;
+    },
+    onSuccess: () => {
+      setError(null);
+    },
+    onError: (err) => {
+      setError(err);
+    }
+  });
+  const removeSpendingPolicyMutation = useMutation({
+    mutationFn: async (token) => {
+      if (!wallet) throw new Error("Wallet is required");
+      if (!localSession) throw new Error("No session to remove policy from");
+      if (!encryptKey) throw new Error("Encryption key (PIN) is required");
+      if (!supportsSession) throw new Error("Wallet does not support sessions");
+      const bearerToken = await getBearerToken();
+      if (!bearerToken) throw new Error("Bearer token is required");
+      const txHash = await chipiSDK.sessions.removeSpendingPolicy(
+        {
+          encryptKey,
+          wallet,
+          sessionPublicKey: localSession.publicKey,
+          token
+        },
+        bearerToken
+      );
+      return txHash;
+    },
+    onSuccess: () => {
+      setError(null);
+    },
+    onError: (err) => {
+      setError(err);
+    }
+  });
   const createSession = useCallback(
     async (params) => {
       return createSessionMutation.mutateAsync(params ?? {});
@@ -466,14 +521,40 @@ function useChipiSession(config) {
   const refetchStatus = useCallback(async () => {
     await sessionStatusQuery.refetch();
   }, [sessionStatusQuery]);
+  const setSpendingPolicy = useCallback(
+    async (policyConfig) => {
+      return setSpendingPolicyMutation.mutateAsync(policyConfig);
+    },
+    [setSpendingPolicyMutation]
+  );
+  const getSpendingPolicy = useCallback(
+    async (token) => {
+      if (!wallet?.publicKey) throw new Error("Wallet is required");
+      if (!localSession?.publicKey) throw new Error("No session to query policy for");
+      return chipiSDK.sessions.getSpendingPolicy({
+        walletAddress: wallet.publicKey,
+        sessionPublicKey: localSession.publicKey,
+        token
+      });
+    },
+    [chipiSDK, wallet, localSession]
+  );
+  const removeSpendingPolicy = useCallback(
+    async (token) => {
+      return removeSpendingPolicyMutation.mutateAsync(token);
+    },
+    [removeSpendingPolicyMutation]
+  );
   const combinedError = useMemo(() => {
-    return error || createSessionMutation.error || registerSessionMutation.error || revokeSessionMutation.error || executeWithSessionMutation.error || sessionStatusQuery.error || null;
+    return error || createSessionMutation.error || registerSessionMutation.error || revokeSessionMutation.error || executeWithSessionMutation.error || setSpendingPolicyMutation.error || removeSpendingPolicyMutation.error || sessionStatusQuery.error || null;
   }, [
     error,
     createSessionMutation.error,
     registerSessionMutation.error,
     revokeSessionMutation.error,
     executeWithSessionMutation.error,
+    setSpendingPolicyMutation.error,
+    removeSpendingPolicyMutation.error,
     sessionStatusQuery.error
   ]);
   return {
@@ -492,18 +573,25 @@ function useChipiSession(config) {
     executeWithSession,
     clearSession,
     refetchStatus,
+    // Spending Policy Actions
+    setSpendingPolicy,
+    getSpendingPolicy,
+    removeSpendingPolicy,
     // Loading states
     isCreating: createSessionMutation.isPending,
     isRegistering: registerSessionMutation.isPending,
     isRevoking: revokeSessionMutation.isPending,
     isExecuting: executeWithSessionMutation.isPending,
     isLoadingStatus: sessionStatusQuery.isLoading,
+    isSettingSpendingPolicy: setSpendingPolicyMutation.isPending,
+    isRemovingSpendingPolicy: removeSpendingPolicyMutation.isPending,
     // Errors
     error: combinedError
   };
 }
 function useMigrateWalletToPasskey() {
   const { chipiSDK } = useChipiContext();
+  const queryClient = useQueryClient();
   const mutation = useMutation({
     mutationFn: async (input) => {
       const { wallet, oldEncryptKey, externalUserId, bearerToken } = input;
@@ -525,10 +613,27 @@ function useMigrateWalletToPasskey() {
           decryptedPrivateKey,
           passkeyResult.encryptKey
         );
+        const updateResult = await chipiSDK.wallets.updateWalletEncryption(
+          {
+            externalUserId,
+            newEncryptedPrivateKey,
+            publicKey: wallet.publicKey
+          },
+          bearerToken
+        );
+        if (updateResult?.success === false) {
+          throw new Error("Backend rejected wallet encryption update");
+        }
         const updatedWallet = {
           ...wallet,
           encryptedPrivateKey: newEncryptedPrivateKey
         };
+        queryClient.invalidateQueries({
+          queryKey: ["chipi-wallet", externalUserId]
+        });
+        queryClient.invalidateQueries({
+          queryKey: ["wallet", externalUserId]
+        });
         return {
           success: true,
           wallet: updatedWallet,