diff --git a/packages/dapp-kit/src/dapp-kit.ts b/packages/dapp-kit/src/dapp-kit.ts index 2adb89e6..76347df4 100644 --- a/packages/dapp-kit/src/dapp-kit.ts +++ b/packages/dapp-kit/src/dapp-kit.ts @@ -1,3 +1,4 @@ +import type { Net } from '@vechain/connex-driver'; import { DriverNoVendor, SimpleNet, @@ -19,10 +20,12 @@ const cache: Record = {}; * * @param node - The node URL * @param genesis - The genesis block + * @param net - The network */ const createThorDriver = ( node: string, genesis: Connex.Thor.Block, + net: Net, ): DriverNoVendor => { // Stringify the certificate to hash const certificateToHash = JSON.stringify({ @@ -40,7 +43,7 @@ const createThorDriver = ( let driver = cache[key]; if (!driver) { - driver = new DriverNoVendor(new SimpleNet(node), genesis); + driver = new DriverNoVendor(net, genesis); cache[key] = driver; } return driver; @@ -61,7 +64,8 @@ class DAppKit { const genesisBlock = normalizeGenesisBlock(genesis); - const driver = createThorDriver(nodeUrl, genesisBlock); + const net = options.customNet || new SimpleNet(nodeUrl); + const driver = createThorDriver(nodeUrl, genesisBlock, net); const walletManager = new WalletManager(options, driver); diff --git a/packages/dapp-kit/src/types/types.d.ts b/packages/dapp-kit/src/types/types.d.ts index 71eed17d..df6d7e5d 100644 --- a/packages/dapp-kit/src/types/types.d.ts +++ b/packages/dapp-kit/src/types/types.d.ts @@ -51,6 +51,7 @@ interface DAppKitOptions { message?: Connex.Vendor.CertMessage; options?: Connex.Signer.CertOptions; }; + customNet?: Net; } type BaseWallet = ExpandedConnexSigner & {