Update set primary flow for SCWs

Author: arjun-dureja

apps/web/src/hooks/useSetPrimaryBasename.ts

@@ -11,6 +11,9 @@ import useBaseEnsName from 'apps/web/src/hooks/useBaseEnsName';
 import { useErrors } from 'apps/web/contexts/Errors';
 import useWriteContractWithReceipt from 'apps/web/src/hooks/useWriteContractWithReceipt';
 import { useUsernameProfile } from 'apps/web/src/components/Basenames/UsernameProfileContext';
+import useWriteContractsWithLogs from 'apps/web/src/hooks/useWriteContractsWithLogs';
+import useCapabilitiesSafe from 'apps/web/src/hooks/useCapabilitiesSafe';
+import L2ReverseRegistrarAbi from 'apps/web/src/abis/L2ReverseRegistrarAbi';
 
 /*
   A hook to set a name as primary for resolution.
@@ -31,6 +34,9 @@ export default function useSetPrimaryBasename({ secondaryUsername }: UseSetPrima
 
   const { currentWalletIsProfileEditor } = useUsernameProfile();
   const { basenameChain: secondaryUsernameChain } = useBasenameChain(secondaryUsername);
+  const { paymasterService: paymasterServiceEnabled } = useCapabilitiesSafe({
+    chainId: secondaryUsernameChain.id,
+  });
 
   // Get current primary username
   // Note: This is sometimes undefined
@@ -52,6 +58,11 @@ export default function useSetPrimaryBasename({ secondaryUsername }: UseSetPrima
       eventName: 'update_primary_name',
     });
 
+  const { initiateBatchCalls, batchCallsStatus, batchCallsIsLoading } = useWriteContractsWithLogs({
+    chain: secondaryUsernameChain,
+    eventName: 'update_primary_name',
+  });
+
   useEffect(() => {
     if (transactionIsSuccess) {
       refetchPrimaryUsername()
@@ -68,17 +79,43 @@ export default function useSetPrimaryBasename({ secondaryUsername }: UseSetPrima
     if (!address) return undefined;
 
     try {
-      await initiateTransaction({
-        abi: ReverseRegistrarAbi,
-        address: USERNAME_REVERSE_REGISTRAR_ADDRESSES[secondaryUsernameChain.id],
-        args: [
-          address,
-          address,
-          USERNAME_L2_RESOLVER_ADDRESSES[secondaryUsernameChain.id],
-          secondaryUsername,
-        ],
-        functionName: 'setNameForAddr',
-      });
+      if (!paymasterServiceEnabled) {
+        await initiateTransaction({
+          abi: ReverseRegistrarAbi,
+          address: USERNAME_REVERSE_REGISTRAR_ADDRESSES[secondaryUsernameChain.id],
+          args: [
+            address,
+            address,
+            USERNAME_L2_RESOLVER_ADDRESSES[secondaryUsernameChain.id],
+            secondaryUsername,
+          ],
+          functionName: 'setNameForAddr',
+        });
+      } else {
+        await initiateBatchCalls({
+          contracts: [
+            {
+              abi: ReverseRegistrarAbi,
+              address: USERNAME_REVERSE_REGISTRAR_ADDRESSES[secondaryUsernameChain.id],
+              args: [
+                address,
+                address,
+                USERNAME_L2_RESOLVER_ADDRESSES[secondaryUsernameChain.id],
+                secondaryUsername,
+              ],
+              functionName: 'setNameForAddr',
+            },
+            {
+              abi: L2ReverseRegistrarAbi,
+              address: USERNAME_L2_RESOLVER_ADDRESSES[secondaryUsernameChain.id],
+              functionName: 'setName',
+              args: [secondaryUsername],
+            },
+          ],
+          account: address,
+          chain: secondaryUsernameChain,
+        });
+      }
     } catch (error) {
       logError(error, 'Set primary name transaction canceled');
       return undefined;
@@ -89,12 +126,24 @@ export default function useSetPrimaryBasename({ secondaryUsername }: UseSetPrima
     secondaryUsername,
     primaryUsername,
     address,
+    paymasterServiceEnabled,
     initiateTransaction,
-    secondaryUsernameChain.id,
+    secondaryUsernameChain,
+    initiateBatchCalls,
     logError,
   ]);
 
-  const isLoading = transactionIsLoading || primaryUsernameIsLoading || primaryUsernameIsFetching;
+  const isLoading =
+    transactionIsLoading ||
+    primaryUsernameIsLoading ||
+    primaryUsernameIsFetching ||
+    batchCallsIsLoading;
 
-  return { setPrimaryName, canSetUsernameAsPrimary, isLoading, transactionIsSuccess };
+  return {
+    setPrimaryName,
+    canSetUsernameAsPrimary,
+    isLoading,
+    transactionIsSuccess,
+    batchCallsStatus,
+  };
 }