fix: returning correct ledger id for localnode (#163)

Signed-off-by: Mariusz Jasuwienas <mariusz.jasuwienas@arianelabs.com>

Author: arianejasuwienas

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/mirror-node-client.js

@@ -48,6 +48,27 @@ class MirrorNodeClient {
         this.mirrorNodeUrl = mirrorNodeUrl;
     }
 
+    /**
+     * Get ledger ID.
+     *
+     * @returns {string}
+     */
+    getLedgerId() {
+        const knownEnvironments = [
+            { name: 'mainnet', id: '0x00' },
+            { name: 'testnet', id: '0x01' },
+            { name: 'previewnet', id: '0x02' },
+            { name: 'localhost', id: '0x03' },
+            { name: '127.0.0.1', id: '0x03' },
+        ];
+        for (let i = 0; i < knownEnvironments.length; i++) {
+            if (this.mirrorNodeUrl.split(knownEnvironments[i].name).length === 2) {
+                return knownEnvironments[i].id;
+            }
+        }
+        return '0x00';
+    }
+
     /**
      * Fetches information about a token by its token ID.
      *

src/index.d.ts

@@ -20,6 +20,17 @@
  *
  */
 interface IMirrorNodeClient {
+    /**
+     * Get ledger ID.
+     *
+     * Returns the guessed ID of the ledger. Since there is only one environment supported at a time,
+     * the ledger ID can be determined using the API URL as input:
+     *  - 0x00 for mainnet,
+     *  - 0x01 for testnet,
+     *  - 0x02 for previewnet.
+     */
+    getLedgerId(): string;
+
     /**
      * Get token by id.
      *

src/index.js

@@ -250,7 +250,7 @@ async function getHtsStorageAt(address, requestedSlot, blockNumber, mirrorNodeCl
     if (unresolvedValues === undefined) {
         const token = await mirrorNodeClient.getTokenById(tokenId, blockNumber);
         if (token === null) return ret(ZERO_HEX_32_BYTE, `Token \`${tokenId}\` not found`);
-        unresolvedValues = slotMapOf(token).load(nrequestedSlot);
+        unresolvedValues = slotMapOf(token, mirrorNodeClient.getLedgerId()).load(nrequestedSlot);
 
         if (unresolvedValues === undefined)
             return ret(ZERO_HEX_32_BYTE, `Requested slot does not match any field slots`);

src/slotmap.js

@@ -216,9 +216,10 @@ function visit(slot, baseSlot, obj, path, map) {
  * Secondly, it builds the actual `SlotMap` starting from the declared fields in the storage layout.
  *
  * @param {Record<string, unknown>} token
+ * @param {string} ledgerId
  * @returns {SlotMap}
  */
-function slotMapOf(token) {
+function slotMapOf(token, ledgerId) {
     token['token_type'] = token['type'];
     token['token_supply_type'] = token['supply_type'] === 'FINITE';
     token['default_kyc_status'] = false;
@@ -227,7 +228,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']}`;