@chipi-stack/chipi-react 14.1.1 → 14.2.0

package/dist/hooks.mjs

@@ -400,6 +400,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 ?? {});
@@ -428,14 +483,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 {
@@ -454,12 +535,18 @@ 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
   };