fix: wagmi as optional provider

Author: anshss

packages/auth-browser/src/lib/chains/eth.ts

@@ -94,6 +94,7 @@ interface signAndSaveAuthParams {
   expiration: string;
   uri?: string;
   nonce: string;
+  provider: AuthProvider;
 }
 
 interface IABI {
@@ -124,6 +125,7 @@ interface SignMessageParams {
   body: string;
   web3: Web3Provider;
   account: string;
+  provider: AuthProvider;
 }
 
 interface SignedMessage {
@@ -619,19 +621,28 @@ export const checkAndSignEVMAuthMessage = async ({
   const selectedChain = LIT_CHAINS[chain];
   const expirationString = expiration ?? getDefaultExpiration();
 
-  let web3: Web3Provider;
-  let account: string;
+  let web3: Web3Provider | undefined;
+  let account: string | undefined;
 
   if (provider === AuthProvider.Wagmi) {
     ({ web3, account } = await connectWeb3WithWagmi({
       chainId: selectedChain.chainId,
       walletConnectProjectId,
     }));
-  } else {
+  } else if (provider === AuthProvider.LitConnectModal) {
     ({ web3, account } = await connectWeb3WithLitConnectModal({
       chainId: selectedChain.chainId,
       walletConnectProjectId,
     }));
+  } else {
+    throw new Error('Invalid provider');
+  }
+
+  if (!web3) {
+    throw new Error('Web3Provider is undefined');
+  }
+  if (!account) {
+    throw new Error('Account is undefined');
   }
 
   log(`got web3 and account: ${account}`);
@@ -668,7 +679,7 @@ export const checkAndSignEVMAuthMessage = async ({
 
   // -- 4. case: (current chain id is NOT equal to selected chain) AND is set to switch chain
   if (currentChainIdOrError.result !== selectedChainId && switchChain) {
-    const provider = web3.provider as any;
+    const provider = web3?.provider as any;
 
     // -- (case) if able to switch chain id
     try {
@@ -733,6 +744,7 @@ export const checkAndSignEVMAuthMessage = async ({
         expiration: expirationString,
         uri,
         nonce,
+        provider,
       });
 
       authSigOrError = {
@@ -779,6 +791,7 @@ export const checkAndSignEVMAuthMessage = async ({
       expiration: expirationString,
       uri,
       nonce,
+      provider,
     });
     log('7. authSig:', authSig);
 
@@ -795,6 +808,7 @@ export const checkAndSignEVMAuthMessage = async ({
         expiration: expirationString,
         uri,
         nonce,
+        provider,
       });
     }
     log('8. mustResign:', mustResign);
@@ -819,6 +833,7 @@ export const checkAndSignEVMAuthMessage = async ({
       expiration: expirationString,
       uri,
       nonce,
+      provider,
     });
   }
 
@@ -837,6 +852,7 @@ const _signAndGetAuth = async ({
   expiration,
   uri,
   nonce,
+  provider,
 }: signAndSaveAuthParams): Promise<AuthSig> => {
   await signAndSaveAuthMessage({
     web3,
@@ -846,6 +862,7 @@ const _signAndGetAuth = async ({
     expiration,
     uri,
     nonce,
+    provider,
   });
 
   const authSigOrError = getStorageItem(LOCAL_STORAGE_KEYS.AUTH_SIGNATURE);
@@ -885,6 +902,7 @@ export const signAndSaveAuthMessage = async ({
   expiration,
   uri,
   nonce,
+  provider,
 }: signAndSaveAuthParams): Promise<AuthSig> => {
   // check if it's nodejs
   if (isNode()) {
@@ -925,6 +943,7 @@ export const signAndSaveAuthMessage = async ({
     body,
     web3,
     account: formattedAccount,
+    provider,
   });
 
   // -- 3. prepare auth message
@@ -970,6 +989,7 @@ export const signMessage = async ({
   body,
   web3,
   account,
+  provider,
 }: SignMessageParams): Promise<SignedMessage> => {
   // check if it's nodejs
   if (isNode()) {
@@ -983,7 +1003,12 @@ export const signMessage = async ({
   // -- validate
   if (!web3 || !account) {
     log(`web3: ${web3} OR ${account} not found. Connecting web3..`);
-    const res = await connectWeb3({ chainId: 1 });
+    let res;
+    if (provider === AuthProvider.Wagmi) {
+      res = await connectWeb3WithWagmi({ chainId: 1 });
+    } else {
+      res = await connectWeb3WithLitConnectModal({ chainId: 1 });
+    }
     web3 = res.web3;
     account = res.account;
   }