From c6c4b94b5c67c6f481c2a3ffda14a01cf2c8243c Mon Sep 17 00:00:00 2001
From: Nesquiko <lukas.castven@gmail.com>
Date: Tue, 21 Jan 2025 11:14:31 +0100
Subject: [PATCH] feat: propagate custom contractNetworks config in
 Safe4337Pack.ts

---
 packages/relay-kit/src/packs/safe-4337/Safe4337Pack.ts | 10 ++++++++--
 packages/relay-kit/src/packs/safe-4337/types.ts        |  4 +++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/packages/relay-kit/src/packs/safe-4337/Safe4337Pack.ts b/packages/relay-kit/src/packs/safe-4337/Safe4337Pack.ts
index 5162c88f2..d217a65d8 100644
--- a/packages/relay-kit/src/packs/safe-4337/Safe4337Pack.ts
+++ b/packages/relay-kit/src/packs/safe-4337/Safe4337Pack.ts
@@ -146,7 +146,8 @@ export class Safe4337Pack extends RelayKitBasePack<{
       bundlerUrl,
       customContracts,
       paymasterOptions,
-      onchainAnalytics
+      onchainAnalytics,
+      contractNetworks
     } = initOptions
 
     let protocolKit: Safe
@@ -272,7 +273,12 @@ export class Safe4337Pack extends RelayKitBasePack<{
         setupTransactions.push(approveToPaymasterTransaction)
       }
 
-      const safeProvider = await SafeProvider.init({ provider, signer, safeVersion })
+      const safeProvider = await SafeProvider.init({
+        provider,
+        signer,
+        safeVersion,
+        contractNetworks
+      })
 
       // third transaction: passkey support via shared signer SafeWebAuthnSharedSigner
       // see: https://github.com/safe-global/safe-modules/blob/main/modules/passkey/contracts/4337/experimental/README.md
diff --git a/packages/relay-kit/src/packs/safe-4337/types.ts b/packages/relay-kit/src/packs/safe-4337/types.ts
index 77ba49ae0..607101512 100644
--- a/packages/relay-kit/src/packs/safe-4337/types.ts
+++ b/packages/relay-kit/src/packs/safe-4337/types.ts
@@ -2,7 +2,8 @@ import { Account, Address, Chain, Hash, Hex, PublicClient, PublicRpcSchema, Tran
 import Safe, {
   DeploymentType,
   SafeProviderConfig,
-  OnchainAnalyticsProps
+  OnchainAnalyticsProps,
+  ContractNetworksConfig
 } from '@safe-global/protocol-kit'
 import {
   EstimateGasData,
@@ -55,6 +56,7 @@ export type Safe4337InitOptions = {
   options: ExistingSafeOptions | PredictedSafeOptions
   paymasterOptions?: PaymasterOptions
   onchainAnalytics?: OnchainAnalyticsProps
+  contractNetworks?: ContractNetworksConfig
 }
 
 export type Safe4337Options = {