Support both single and multiple url configuration

pimlicolabs · Nov 1, 2024 · 7276ed0 · 7276ed0
1 parent 743cb49
commit 7276ed0
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/packages/wagmi/hooks/useSendTransaction.ts b/packages/wagmi/hooks/useSendTransaction.ts
@@ -31,14 +31,21 @@ const sendTransactionMutationOptions = <config extends Config>(
             if (parameters.capabilities) {
                 const client = config.getClient({ chainId: variables.chainId })
 
+                const paymasterServiceUrl = parameters.capabilities
+                    ?.paymasterService?.url
+                    ? parameters.capabilities?.paymasterService?.url
+                    : parameters.capabilities?.paymasterService[client.chain.id]
+                          ?.url
+
                 return sendCalls(config, {
                     calls: [variables],
                     capabilities: {
                         ...parameters.capabilities,
-                        paymasterService:
-                            parameters.capabilities?.paymasterService[
-                                client.chain.id
-                            ]
+                        paymasterService: paymasterServiceUrl
+                            ? {
+                                  url: paymasterServiceUrl
+                              }
+                            : undefined
                     }
                 })
             }