feat: returning correct ledger id in tokeninfo (#163)

contracts/HtsSystemContractJson.sol

@@ -423,6 +423,7 @@ contract HtsSystemContractJson is HtsSystemContract {
         if (block.chainid == 295) return "0x00"; // Mainnet
         if (block.chainid == 296) return "0x01"; // Testnet
         if (block.chainid == 297) return "0x02"; // Previewnet
+        if (block.chainid == 298) return "0x03"; // Localhost
         return "0x00"; // Default to Mainnet
     }
 

src/forwarder/json-rpc-forwarder.js

@@ -28,6 +28,8 @@ const { getHtsCode, getHtsStorageAt, HTSAddress, LONG_ZERO_PREFIX, getHIP719Code
 /** @type {Partial<import('hardhat/types').HardhatNetworkForkingConfig>} */
 const { url: forkUrl, mirrorNodeUrl, workerPort, localAddresses = [] } = workerData;
 
+require('../slotmap').setLedgerId(workerData.chainId);
+
 assert(mirrorNodeUrl !== undefined, 'json-rpc-forwarder: Missing Mirror Node URL');
 
 debug(

src/slotmap.js

@@ -24,6 +24,19 @@ const {
     storageLayout: { storage, types },
 } = require('../out/HtsSystemContract.sol/HtsSystemContract.json');
 
+let ledgerId = '0x00';
+/**
+ * Sets the `ledgerId` used when retrieving fungible and non-fungible tokens.
+ * The `ledgerId` depends on the `chainId` of the remote network it is forking from.
+ *
+ * @param {number} chainId
+ */
+const setLedgerId = chainId => {
+    const chainIdToLedgerIdMap = { 295: '0x00', 296: '0x01', 297: '0x02', 298: '0x03' };
+    ledgerId =
+        chainIdToLedgerIdMap[/**@type{keyof typeof chainIdToLedgerIdMap}*/ (chainId)] || '0x00';
+};
+
 /**
  * Represents the value in the `SlotMap`.
  * This can be either a `string`, or a function that returns a `string`.
@@ -227,7 +240,7 @@ function slotMapOf(token) {
     // token['inherit_account_key'] = ZERO_HEX_32_BYTE;
     // token['delegatable_contract_id'] = zeroAddress();
     token['treasury'] = token['treasury_account_id'];
-    token['ledger_id'] = '0x00';
+    token['ledger_id'] = ledgerId;
     // Every inner `struct` will be flattened by `visit`,
     // so it uses the last part of the field path, _i.e._, `.second`.
     token['second'] = `${token['expiry_timestamp']}`;
@@ -339,4 +352,4 @@ class PersistentStorageMap {
     }
 }
 
-module.exports = { packValues, slotMapOf, PersistentStorageMap };
+module.exports = { packValues, slotMapOf, PersistentStorageMap, setLedgerId };