fix: allow custom chain ID in JS scripts

hirosystems · Oct 14, 2024 · e240b15 · e240b15
1 parent fdb340a
commit e240b15
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/stacking/common.ts b/stacking/common.ts
@@ -15,6 +15,7 @@ import {
   AccountsApi,
 } from '@stacks/blockchain-api-client';
 import pino, { Logger } from 'pino';
+import { ChainID } from '@stacks/common';
 
 const serviceName = process.env.SERVICE_NAME || 'JS';
 export let logger: Logger;
@@ -37,8 +38,11 @@ if (process.env.STACKS_LOG_JSON === '1') {
   });
 }
 
+export const CHAIN_ID = parseEnvInt('CHAIN_ID', false) ?? ChainID.Testnet;
+
 export const nodeUrl = `http://${process.env.STACKS_CORE_RPC_HOST}:${process.env.STACKS_CORE_RPC_PORT}`;
 export const network = new StacksTestnet({ url: nodeUrl });
+network.chainId = CHAIN_ID;
 const apiConfig = new Configuration({
   basePath: nodeUrl,
 });

diff --git a/stacking/stacking.ts b/stacking/stacking.ts
@@ -112,7 +112,7 @@ async function stackStx(poxInfo: PoxInfo, account: Account, balance: bigint) {
   if (typeof stackAmount === 'number') {
     amountToStx = BigInt(stackAmount) * 1_000_000n;
   }
-  
+
   if (amountToStx > balance) {
     throw new Error(
       `Insufficient balance to stack-stx (amount=${amountToStx}, balance=${balance})`