@gelatonetwork/smartwallet-react-privy 0.0.2-alpha.1 → 0.0.2-alpha.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gelatonetwork/smartwallet-react-privy",
-  "version": "0.0.2-alpha.1",
+  "version": "0.0.2-alpha.4",
   "author": "Gelato",
   "type": "module",
   "description": "Context provider supporting Privy WaaS",
@@ -10,7 +10,9 @@
   "peerDependencies": {
     "typescript": "^5.3",
     "viem": "^2.28",
-    "wagmi": "^2.15"
+    "wagmi": "^2.15",
+    "@gelatonetwork/smartwallet-react-types": "0.0.2-alpha.4",
+    "@gelatonetwork/smartwallet": "0.0.2-alpha.4"
   },
   "peerDependenciesMeta": {
     "typescript": {
@@ -30,7 +32,8 @@
     "viem": "^2.28",
     "wagmi": "^2.15",
     "zod": "^3.24.3",
-    "@gelatonetwork/smartwallet-react-types": "0.0.2-alpha.1"
+    "@gelatonetwork/smartwallet-react-types": "0.0.2-alpha.4",
+    "@gelatonetwork/smartwallet": "0.0.2-alpha.4"
   },
   "devDependencies": {
     "@types/react": "^18",

package/src/provider.tsx

@@ -1,22 +1,24 @@
+import {
+  type GelatoSmartWalletClient,
+  createGelatoSmartWalletClient
+} from "@gelatonetwork/smartwallet";
+import type { wallet } from "@gelatonetwork/smartwallet-react-types";
 import { PrivyProvider, usePrivy, useSignAuthorization, useWallets } from "@privy-io/react-auth";
 import { WagmiProvider, createConfig } from "@privy-io/wagmi";
 import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
 import { ChainId } from "caip";
 import { createContext, useContext, useEffect, useState } from "react";
+import type { FC, ReactNode } from "react";
 import {
   type Account,
   type Chain,
   type Hex,
   type Transport,
-  type WalletClient,
   createWalletClient,
   custom
 } from "viem";
 import * as chains from "viem/chains";
 import { extractChain } from "viem/utils";
-
-import type { wallet } from "@gelatonetwork/smartwallet-react-types";
-import type { FC, ReactNode } from "react";
 import type { Config as WagmiConfig } from "wagmi";
 
 type GelatoSmartWalletPrivyContextType = wallet.ProviderContext;
@@ -40,31 +42,36 @@ type GelatoSmartWalletPrivyContextProps = wallet.ProviderProps;
 const GelatoSmartWalletPrivyInternal: FC<{
   children: ReactNode;
   wagmi: { config: WagmiConfig | undefined };
-}> = ({ children, wagmi }) => {
+  apiKey?: string | undefined;
+}> = ({ children, wagmi, apiKey }) => {
   const { ready, authenticated, logout } = usePrivy();
   const { wallets, ready: walletsReady } = useWallets();
   const { signAuthorization } = useSignAuthorization();
 
-  const [walletClient, setWalletClient] = useState<WalletClient | null>(null);
+  const [smartWalletClient, setSmartWalletClient] = useState<GelatoSmartWalletClient<
+    Transport,
+    Chain,
+    Account
+  > | null>(null);
 
   const logoutWrapper = async () => {
-    if (!walletClient) {
+    if (!smartWalletClient) {
       return;
     }
 
-    setWalletClient(null);
+    setSmartWalletClient(null);
     await logout();
   };
 
   const switchNetwork = async (chain: Chain) => {
-    if (!walletClient) {
+    if (!smartWalletClient) {
       return;
     }
 
     const primaryWallet = wallets[0];
 
     await primaryWallet.switchChain(chain.id);
-    walletClient.switchChain({ id: chain.id });
+    smartWalletClient.switchChain({ id: chain.id });
   };
 
   useEffect(() => {
@@ -73,7 +80,7 @@ const GelatoSmartWalletPrivyInternal: FC<{
     }
 
     if (!authenticated || !wallets || wallets.length === 0) {
-      setWalletClient(null);
+      setSmartWalletClient(null);
       return;
     }
 
@@ -91,13 +98,13 @@ const GelatoSmartWalletPrivyInternal: FC<{
         }
 
         const provider = await primaryWallet.getEthereumProvider();
-        const walletClient = createWalletClient({
+        const client = createWalletClient({
           account: primaryWallet.address as Hex,
           chain,
           transport: custom(provider)
         });
 
-        walletClient.signAuthorization = async (parameters) => {
+        client.signAuthorization = async (parameters) => {
           const { chainId, nonce } = parameters;
           const contractAddress = parameters.contractAddress ?? parameters.address;
 
@@ -114,21 +121,27 @@ const GelatoSmartWalletPrivyInternal: FC<{
           return signedAuthorization;
         };
 
-        setWalletClient(walletClient);
+        const walletClientGelato = createGelatoSmartWalletClient<Transport, Chain, Account>(
+          client,
+          apiKey
+        );
+        setSmartWalletClient(walletClientGelato);
       } catch (error) {
         console.error("Failed to get wallet client:", error);
       }
     };
 
     fetchWalletClient();
-  }, [ready, wallets, walletsReady, authenticated, signAuthorization]);
+  }, [ready, wallets, walletsReady, authenticated, signAuthorization, apiKey]);
 
   return (
     <GelatoSmartWalletPrivyProviderContext.Provider
       value={{
+        gelato: {
+          client: smartWalletClient as GelatoSmartWalletClient<Transport, Chain, Account>
+        },
         wagmi: {
-          config: wagmi.config,
-          client: walletClient as WalletClient<Transport, Chain, Account>
+          config: wagmi.config
         },
         logout: logoutWrapper,
         switchNetwork