feat: allow custom net injection

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<string, DriverNoVendor | undefined> = {};
  *
  * @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);
 

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 & {