@ledgerhq/coin-sui 0.9.0 → 0.10.0-nightly.1

package/lib/bridge/getFeesForTransaction.js

@@ -1,9 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+const helpers_1 = require("@ledgerhq/coin-framework/account/helpers");
 const bignumber_js_1 = require("bignumber.js");
 const cryptoassets_1 = require("@ledgerhq/cryptoassets");
 const utils_1 = require("./utils");
 const logic_1 = require("../logic");
+const sdk_1 = require("../network/sdk");
 /**
  * Fetch the transaction fees for a transaction
  *
@@ -25,12 +27,14 @@ async function getEstimatedFees({ account, transaction, }) {
             },
         }), // Remove fees if present since we are fetching fees
     };
+    const subAccount = (0, helpers_1.findSubAccountById)(account, transaction.subAccountId ?? "");
+    const asset = (0, sdk_1.toSuiAsset)(subAccount?.token.contractAddress ?? sdk_1.DEFAULT_COIN_TYPE);
     const fees = await (0, logic_1.estimateFees)({
         recipient: (0, cryptoassets_1.getAbandonSeedAddress)(account.currency.id),
         sender: account.freshAddress,
         amount: BigInt(t.amount.toString()),
         type: "send",
-        asset: { type: "native" },
+        asset,
     });
     return new bignumber_js_1.BigNumber(fees.toString());
 }

package/lib/bridge/getFeesForTransaction.js.map

@@ -1 +1 @@
-{"version":3,"file":"getFeesForTransaction.js","sourceRoot":"","sources":["../../src/bridge/getFeesForTransaction.ts"],"names":[],"mappings":";;AAAA,+CAAyC;AACzC,yDAA+D;AAE/D,mCAA0C;AAC1C,oCAAwC;AAExC;;;;;;;GAOG;AACY,KAAK,UAAU,gBAAgB,CAAC,EAC7C,OAAO,EACP,WAAW,GAIZ;IACC,MAAM,CAAC,GAAG;QACR,GAAG,WAAW;QACd,SAAS,EAAE,IAAA,oCAAqB,EAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,+CAA+C;QAC/C,MAAM,EAAE,IAAA,uBAAe,EAAC;YACtB,OAAO;YACP,WAAW,EAAE;gBACX,GAAG,WAAW;gBACd,IAAI,EAAE,IAAI,wBAAS,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,CAAC;aAC3C;SACF,CAAC,EAAE,oDAAoD;KACzD,CAAC;IAEF,MAAM,IAAI,GAAG,MAAM,IAAA,oBAAY,EAAC;QAC9B,SAAS,EAAE,IAAA,oCAAqB,EAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,MAAM,EAAE,OAAO,CAAC,YAAY;QAC5B,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACnC,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC1B,CAAC,CAAC;IACH,OAAO,IAAI,wBAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;AACxC,CAAC;AA5BD,mCA4BC"}
+{"version":3,"file":"getFeesForTransaction.js","sourceRoot":"","sources":["../../src/bridge/getFeesForTransaction.ts"],"names":[],"mappings":";;AAAA,sEAA8E;AAC9E,+CAAyC;AACzC,yDAA+D;AAE/D,mCAA0C;AAC1C,oCAAwC;AACxC,wCAA+D;AAE/D;;;;;;;GAOG;AACY,KAAK,UAAU,gBAAgB,CAAC,EAC7C,OAAO,EACP,WAAW,GAIZ;IACC,MAAM,CAAC,GAAG;QACR,GAAG,WAAW;QACd,SAAS,EAAE,IAAA,oCAAqB,EAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,+CAA+C;QAC/C,MAAM,EAAE,IAAA,uBAAe,EAAC;YACtB,OAAO;YACP,WAAW,EAAE;gBACX,GAAG,WAAW;gBACd,IAAI,EAAE,IAAI,wBAAS,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,CAAC;aAC3C;SACF,CAAC,EAAE,oDAAoD;KACzD,CAAC;IAEF,MAAM,UAAU,GAAG,IAAA,4BAAkB,EAAC,OAAO,EAAE,WAAW,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;IAC/E,MAAM,KAAK,GAAG,IAAA,gBAAU,EAAC,UAAU,EAAE,KAAK,CAAC,eAAe,IAAI,uBAAiB,CAAC,CAAC;IAEjF,MAAM,IAAI,GAAG,MAAM,IAAA,oBAAY,EAAC;QAC9B,SAAS,EAAE,IAAA,oCAAqB,EAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,MAAM,EAAE,OAAO,CAAC,YAAY;QAC5B,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACnC,IAAI,EAAE,MAAM;QACZ,KAAK;KACN,CAAC,CAAC;IACH,OAAO,IAAI,wBAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;AACxC,CAAC;AA/BD,mCA+BC"}

package/lib/logic/craftTransaction.d.ts

@@ -2,11 +2,11 @@ import BigNumber from "bignumber.js";
 import type { TransactionIntent } from "@ledgerhq/coin-framework/api/index";
 import type { SuiTransactionMode, CoreTransaction } from "../types";
 export type CreateExtrinsicArg = {
-    mode: SuiTransactionMode;
     amount: BigNumber;
     coinType: string;
+    mode: SuiTransactionMode;
     recipient: string;
     useAllAmount?: boolean | undefined;
 };
-export declare function craftTransaction({ sender, amount, recipient, asset, type, }: TransactionIntent): Promise<CoreTransaction>;
+export declare function craftTransaction({ amount, asset, recipient, sender, type, }: TransactionIntent): Promise<CoreTransaction>;
 //# sourceMappingURL=craftTransaction.d.ts.map

package/lib/logic/craftTransaction.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"craftTransaction.d.ts","sourceRoot":"","sources":["../../src/logic/craftTransaction.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,cAAc,CAAC;AACrC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,KAAK,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAIpE,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,kBAAkB,CAAC;IACzB,MAAM,EAAE,SAAS,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACpC,CAAC;AAEF,wBAAsB,gBAAgB,CAAC,EACrC,MAAM,EACN,MAAM,EACN,SAAS,EACT,KAAK,EACL,IAAI,GACL,EAAE,iBAAiB,GAAG,OAAO,CAAC,eAAe,CAAC,CAa9C"}
+{"version":3,"file":"craftTransaction.d.ts","sourceRoot":"","sources":["../../src/logic/craftTransaction.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,cAAc,CAAC;AACrC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,KAAK,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAIpE,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,SAAS,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,kBAAkB,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACpC,CAAC;AAEF,wBAAsB,gBAAgB,CAAC,EACrC,MAAM,EACN,KAAK,EACL,SAAS,EACT,MAAM,EACN,IAAI,GACL,EAAE,iBAAiB,GAAG,OAAO,CAAC,eAAe,CAAC,CAa9C"}

package/lib/logic/craftTransaction.integration.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=craftTransaction.integration.test.d.ts.map

package/lib/logic/craftTransaction.integration.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"craftTransaction.integration.test.d.ts","sourceRoot":"","sources":["../../src/logic/craftTransaction.integration.test.ts"],"names":[],"mappings":""}

package/lib/logic/craftTransaction.integration.test.js

@@ -0,0 +1,56 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const client_1 = require("@mysten/sui/client");
+const config_1 = __importDefault(require("../config"));
+const testUtils_1 = require("../test/testUtils");
+const craftTransaction_1 = require("./craftTransaction");
+const SENDER = "0x33444cf803c690db96527cec67e3c9ab512596f4ba2d4eace43f0b4f716e0164";
+const RECIPIENT = "0x33444cf803c690db96527cec67e3c9ab512596f4ba2d4eace43f0b4f716e0164";
+describe("craftTransaction", () => {
+    beforeAll(() => {
+        config_1.default.setCoinConfig(() => ({
+            status: {
+                type: "active",
+            },
+            node: {
+                url: (0, client_1.getFullnodeUrl)("mainnet"),
+            },
+        }));
+    });
+    it("should craft a native SUI send transaction", async () => {
+        const transactionIntent = {
+            sender: SENDER,
+            recipient: RECIPIENT,
+            amount: BigInt(1000),
+            type: "send",
+            asset: { type: "native" },
+        };
+        const result = await (0, craftTransaction_1.craftTransaction)(transactionIntent);
+        expect(result).toBeDefined();
+        expect(result.unsigned).toBeInstanceOf(Uint8Array);
+        const resultCoinTypes = await (0, testUtils_1.extractCoinTypeFromUnsignedTx)(result.unsigned);
+        expect(resultCoinTypes).toEqual(expect.arrayContaining([expect.stringContaining("sui")]));
+    }, 15000);
+    it("should craft a token send transaction", async () => {
+        const coinType = "0x375f70cf2ae4c00bf37117d0c85a2c71545e6ee05c4a5c7d282cd66a4504b068::usdt::USDT";
+        const transactionIntent = {
+            sender: SENDER,
+            recipient: RECIPIENT,
+            amount: BigInt(1000),
+            type: "send",
+            asset: {
+                type: "token",
+                assetReference: coinType,
+            },
+        };
+        const result = await (0, craftTransaction_1.craftTransaction)(transactionIntent);
+        expect(result).toBeDefined();
+        expect(result.unsigned).toBeInstanceOf(Uint8Array);
+        const resultCoinTypes = await (0, testUtils_1.extractCoinTypeFromUnsignedTx)(result.unsigned);
+        expect(resultCoinTypes).toEqual(expect.arrayContaining([expect.stringContaining("usdt")]));
+    }, 15000);
+});
+//# sourceMappingURL=craftTransaction.integration.test.js.map

package/lib/logic/craftTransaction.integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"craftTransaction.integration.test.js","sourceRoot":"","sources":["../../src/logic/craftTransaction.integration.test.ts"],"names":[],"mappings":";;;;;AACA,+CAAoD;AACpD,uDAAmC;AACnC,iDAAkE;AAClE,yDAAsD;AAEtD,MAAM,MAAM,GAAG,oEAAoE,CAAC;AACpF,MAAM,SAAS,GAAG,oEAAoE,CAAC;AAEvF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,SAAS,CAAC,GAAG,EAAE;QACb,gBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;YAC9B,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;aACf;YACD,IAAI,EAAE;gBACJ,GAAG,EAAE,IAAA,uBAAc,EAAC,SAAS,CAAC;aAC/B;SACF,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,iBAAiB,GAAsB;YAC3C,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAgB,EAAC,iBAAiB,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAEnD,MAAM,eAAe,GAAG,MAAM,IAAA,yCAA6B,EAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5F,CAAC,EAAE,KAAK,CAAC,CAAC;IAEV,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,QAAQ,GACZ,gFAAgF,CAAC;QAEnF,MAAM,iBAAiB,GAAsB;YAC3C,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,cAAc,EAAE,QAAQ;aACzB;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAgB,EAAC,iBAAiB,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAEnD,MAAM,eAAe,GAAG,MAAM,IAAA,yCAA6B,EAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7F,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC"}

package/lib/logic/craftTransaction.js

@@ -7,16 +7,16 @@ exports.craftTransaction = void 0;
 const bignumber_js_1 = __importDefault(require("bignumber.js"));
 const network_1 = __importDefault(require("../network"));
 const sdk_1 = require("../network/sdk");
-async function craftTransaction({ sender, amount, recipient, asset, type, }) {
+async function craftTransaction({ amount, asset, recipient, sender, type, }) {
     let coinType = sdk_1.DEFAULT_COIN_TYPE;
     if (asset.type === "token" && asset.assetReference) {
         coinType = asset.assetReference;
     }
     const unsigned = await network_1.default.createTransaction(sender, {
         amount: (0, bignumber_js_1.default)(amount.toString()),
-        recipient,
         coinType,
         mode: type,
+        recipient,
     });
     return { unsigned };
 }

package/lib/logic/craftTransaction.js.map

@@ -1 +1 @@
-{"version":3,"file":"craftTransaction.js","sourceRoot":"","sources":["../../src/logic/craftTransaction.ts"],"names":[],"mappings":";;;;;;AAAA,gEAAqC;AAGrC,yDAAgC;AAChC,wCAAmD;AAU5C,KAAK,UAAU,gBAAgB,CAAC,EACrC,MAAM,EACN,MAAM,EACN,SAAS,EACT,KAAK,EACL,IAAI,GACc;IAClB,IAAI,QAAQ,GAAG,uBAAiB,CAAC;IACjC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;QACnD,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC;IAClC,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,iBAAM,CAAC,iBAAiB,CAAC,MAAM,EAAE;QACtD,MAAM,EAAE,IAAA,sBAAS,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpC,SAAS;QACT,QAAQ;QACR,IAAI,EAAE,IAA0B;KACjC,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,CAAC;AACtB,CAAC;AAnBD,4CAmBC"}
+{"version":3,"file":"craftTransaction.js","sourceRoot":"","sources":["../../src/logic/craftTransaction.ts"],"names":[],"mappings":";;;;;;AAAA,gEAAqC;AAGrC,yDAAgC;AAChC,wCAAmD;AAU5C,KAAK,UAAU,gBAAgB,CAAC,EACrC,MAAM,EACN,KAAK,EACL,SAAS,EACT,MAAM,EACN,IAAI,GACc;IAClB,IAAI,QAAQ,GAAG,uBAAiB,CAAC;IACjC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;QACnD,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC;IAClC,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,iBAAM,CAAC,iBAAiB,CAAC,MAAM,EAAE;QACtD,MAAM,EAAE,IAAA,sBAAS,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpC,QAAQ;QACR,IAAI,EAAE,IAA0B;QAChC,SAAS;KACV,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,CAAC;AACtB,CAAC;AAnBD,4CAmBC"}

package/lib/logic/estimateFees.d.ts

@@ -1,3 +1,3 @@
 import type { TransactionIntent } from "@ledgerhq/coin-framework/api/index";
-export declare function estimateFees({ recipient, amount, sender, }: TransactionIntent): Promise<bigint>;
+export declare function estimateFees({ recipient, amount, sender, asset, }: TransactionIntent): Promise<bigint>;
 //# sourceMappingURL=estimateFees.d.ts.map

package/lib/logic/estimateFees.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"estimateFees.d.ts","sourceRoot":"","sources":["../../src/logic/estimateFees.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAE5E,wBAAsB,YAAY,CAAC,EACjC,SAAS,EACT,MAAM,EACN,MAAM,GACP,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC,CAUrC"}
+{"version":3,"file":"estimateFees.d.ts","sourceRoot":"","sources":["../../src/logic/estimateFees.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAG5E,wBAAsB,YAAY,CAAC,EACjC,SAAS,EACT,MAAM,EACN,MAAM,EACN,KAAK,GACN,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC,CAcrC"}

package/lib/logic/estimateFees.integration.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=estimateFees.integration.test.d.ts.map

package/lib/logic/estimateFees.integration.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"estimateFees.integration.test.d.ts","sourceRoot":"","sources":["../../src/logic/estimateFees.integration.test.ts"],"names":[],"mappings":""}

package/lib/logic/estimateFees.integration.test.js

@@ -0,0 +1,79 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const client_1 = require("@mysten/sui/client");
+const config_1 = __importDefault(require("../config"));
+const estimateFees_1 = require("./estimateFees");
+const SENDER = "0x6e143fe0a8ca010a86580dafac44298e5b1b7d73efc345356a59a15f0d7824f0";
+const RECIPIENT = "0x33444cf803c690db96527cec67e3c9ab512596f4ba2d4eace43f0b4f716e0164";
+describe("estimateFees", () => {
+    beforeAll(() => {
+        config_1.default.setCoinConfig(() => ({
+            status: {
+                type: "active",
+            },
+            node: {
+                url: (0, client_1.getFullnodeUrl)("mainnet"),
+            },
+        }));
+    });
+    it("should estimate fees for native SUI transaction", async () => {
+        const transactionIntent = {
+            sender: SENDER,
+            recipient: RECIPIENT,
+            amount: BigInt(1000),
+            type: "send",
+            asset: { type: "native" },
+        };
+        const estimatedFees = await (0, estimateFees_1.estimateFees)(transactionIntent);
+        expect(typeof estimatedFees).toBe("bigint");
+        expect(estimatedFees).toBeGreaterThan(1000n);
+        expect(estimatedFees).toBeLessThan(10000000n);
+    }, 25000);
+    it("should estimate fees for token transaction", async () => {
+        const coinType = "0x375f70cf2ae4c00bf37117d0c85a2c71545e6ee05c4a5c7d282cd66a4504b068::usdt::USDT";
+        const transactionIntent = {
+            sender: SENDER,
+            recipient: RECIPIENT,
+            amount: BigInt(1000),
+            type: "send",
+            asset: {
+                type: "token",
+                assetReference: coinType,
+            },
+        };
+        const estimatedFees = await (0, estimateFees_1.estimateFees)(transactionIntent);
+        expect(typeof estimatedFees).toBe("bigint");
+        expect(estimatedFees).toBeGreaterThan(1000n);
+        expect(estimatedFees).toBeLessThan(10000000n);
+    }, 25000);
+    it("should handle concurrent fee estimations", async () => {
+        const transactionIntent = {
+            sender: SENDER,
+            recipient: RECIPIENT,
+            amount: BigInt(1000),
+            type: "send",
+            asset: { type: "native" },
+        };
+        // Run multiple concurrent estimations
+        const promises = Array(5)
+            .fill(0)
+            .map(() => (0, estimateFees_1.estimateFees)(transactionIntent));
+        const results = await Promise.all(promises);
+        // All results should be valid
+        results.forEach(fees => {
+            expect(typeof fees).toBe("bigint");
+            expect(fees).toBeGreaterThan(1000n);
+            expect(fees).toBeLessThan(10000000n);
+        });
+        // Results should be similar (may have slight variations)
+        const values = results.map(r => Number(r));
+        const minValue = Math.min(...values);
+        const maxValue = Math.max(...values);
+        // Should not vary by more than 50% under concurrent load
+        expect((maxValue - minValue) / minValue).toBeLessThan(0.5);
+    }, 25000);
+});
+//# sourceMappingURL=estimateFees.integration.test.js.map

package/lib/logic/estimateFees.integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"estimateFees.integration.test.js","sourceRoot":"","sources":["../../src/logic/estimateFees.integration.test.ts"],"names":[],"mappings":";;;;;AACA,+CAAoD;AACpD,uDAAmC;AACnC,iDAA8C;AAE9C,MAAM,MAAM,GAAG,oEAAoE,CAAC;AACpF,MAAM,SAAS,GAAG,oEAAoE,CAAC;AAEvF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,SAAS,CAAC,GAAG,EAAE;QACb,gBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;YAC9B,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;aACf;YACD,IAAI,EAAE;gBACJ,GAAG,EAAE,IAAA,uBAAc,EAAC,SAAS,CAAC;aAC/B;SACF,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,iBAAiB,GAAsB;YAC3C,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,IAAA,2BAAY,EAAC,iBAAiB,CAAC,CAAC;QAE5D,MAAM,CAAC,OAAO,aAAa,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC,EAAE,KAAK,CAAC,CAAC;IAEV,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,QAAQ,GACZ,gFAAgF,CAAC;QAEnF,MAAM,iBAAiB,GAAsB;YAC3C,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,cAAc,EAAE,QAAQ;aACzB;SACF,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,IAAA,2BAAY,EAAC,iBAAiB,CAAC,CAAC;QAE5D,MAAM,CAAC,OAAO,aAAa,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC,EAAE,KAAK,CAAC,CAAC;IAEV,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,iBAAiB,GAAsB;YAC3C,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,SAAS;YACpB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B,CAAC;QAEF,sCAAsC;QACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC;aACtB,IAAI,CAAC,CAAC,CAAC;aACP,GAAG,CAAC,GAAG,EAAE,CAAC,IAAA,2BAAY,EAAC,iBAAiB,CAAC,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAE5C,8BAA8B;QAC9B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YACrB,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,yDAAyD;QACzD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QAErC,yDAAyD;QACzD,MAAM,CAAC,CAAC,QAAQ,GAAG,QAAQ,CAAC,GAAG,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAC7D,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC"}

package/lib/logic/estimateFees.js

@@ -6,14 +6,19 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.estimateFees = void 0;
 const network_1 = __importDefault(require("../network"));
 const bignumber_js_1 = require("bignumber.js");
-async function estimateFees({ recipient, amount, sender, }) {
+const sdk_1 = require("../network/sdk");
+async function estimateFees({ recipient, amount, sender, asset, }) {
+    let coinType = sdk_1.DEFAULT_COIN_TYPE;
+    if (asset.type === "token" && asset.assetReference) {
+        coinType = asset.assetReference;
+    }
     const { gasBudget } = await network_1.default.paymentInfo(sender, {
         mode: "send",
         family: "sui",
         recipient,
         amount: (0, bignumber_js_1.BigNumber)(amount.toString()),
         errors: {},
-        coinType: "0x2::sui::SUI",
+        coinType,
     });
     return BigInt(gasBudget);
 }

package/lib/logic/estimateFees.js.map

@@ -1 +1 @@
-{"version":3,"file":"estimateFees.js","sourceRoot":"","sources":["../../src/logic/estimateFees.ts"],"names":[],"mappings":";;;;;;AAAA,yDAAgC;AAChC,+CAAyC;AAGlC,KAAK,UAAU,YAAY,CAAC,EACjC,SAAS,EACT,MAAM,EACN,MAAM,GACY;IAClB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,iBAAM,CAAC,WAAW,CAAC,MAAM,EAAE;QACrD,IAAI,EAAE,MAAM;QACZ,MAAM,EAAE,KAAK;QACb,SAAS;QACT,MAAM,EAAE,IAAA,wBAAS,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpC,MAAM,EAAE,EAAE;QACV,QAAQ,EAAE,eAAe;KAC1B,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;AAC3B,CAAC;AAdD,oCAcC"}
+{"version":3,"file":"estimateFees.js","sourceRoot":"","sources":["../../src/logic/estimateFees.ts"],"names":[],"mappings":";;;;;;AAAA,yDAAgC;AAChC,+CAAyC;AAEzC,wCAAmD;AAE5C,KAAK,UAAU,YAAY,CAAC,EACjC,SAAS,EACT,MAAM,EACN,MAAM,EACN,KAAK,GACa;IAClB,IAAI,QAAQ,GAAG,uBAAiB,CAAC;IACjC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;QACnD,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC;IAClC,CAAC;IACD,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,iBAAM,CAAC,WAAW,CAAC,MAAM,EAAE;QACrD,IAAI,EAAE,MAAM;QACZ,MAAM,EAAE,KAAK;QACb,SAAS;QACT,MAAM,EAAE,IAAA,wBAAS,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpC,MAAM,EAAE,EAAE;QACV,QAAQ;KACT,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;AAC3B,CAAC;AAnBD,oCAmBC"}

package/lib/logic/getBalance.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getBalance.d.ts","sourceRoot":"","sources":["../../src/logic/getBalance.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,oCAAoC,CAAC;AAE7D,wBAAsB,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAQpE"}
+{"version":3,"file":"getBalance.d.ts","sourceRoot":"","sources":["../../src/logic/getBalance.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,oCAAoC,CAAC;AAG7D,wBAAsB,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAMpE"}

package/lib/logic/getBalance.integration.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=getBalance.integration.test.d.ts.map

package/lib/logic/getBalance.integration.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"getBalance.integration.test.d.ts","sourceRoot":"","sources":["../../src/logic/getBalance.integration.test.ts"],"names":[],"mappings":""}

package/lib/logic/getBalance.integration.test.js

@@ -0,0 +1,56 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const client_1 = require("@mysten/sui/client");
+const config_1 = __importDefault(require("../config"));
+const getBalance_1 = require("./getBalance");
+const SENDER = "0x33444cf803c690db96527cec67e3c9ab512596f4ba2d4eace43f0b4f716e0164";
+describe("getBalance", () => {
+    beforeAll(() => {
+        config_1.default.setCoinConfig(() => ({
+            status: {
+                type: "active",
+            },
+            node: {
+                url: (0, client_1.getFullnodeUrl)("testnet"),
+            },
+        }));
+    });
+    it("should fetch native SUI balance", async () => {
+        const balances = await (0, getBalance_1.getBalance)(SENDER);
+        expect(balances.length).toBeGreaterThanOrEqual(1);
+        expect(balances[0]).toMatchObject({
+            asset: { type: "native" },
+        });
+        expect(typeof balances[0].value).toBe("bigint");
+        expect(balances[0].value).toBeGreaterThanOrEqual(0n);
+    }, 10000);
+    it("should fetch token balances", async () => {
+        const balances = await (0, getBalance_1.getBalance)(SENDER);
+        expect(balances.length).toBeGreaterThanOrEqual(1);
+        const tokenBalances = balances.filter(balance => balance.asset.type === "token");
+        tokenBalances.forEach(balance => {
+            expect(balance.asset.type).toBe("token");
+            if (balance.asset.type === "token") {
+                expect(balance.asset.assetReference).toMatch(/^0x[a-fA-F0-9]+::[a-zA-Z0-9_]+::[a-zA-Z0-9_]+$/);
+            }
+            expect(typeof balance.value).toBe("bigint");
+            expect(balance.value).toBeGreaterThanOrEqual(0n);
+        });
+    }, 15000);
+    it("should properly parse token asset reference", async () => {
+        const balances = await (0, getBalance_1.getBalance)(SENDER);
+        const usdTokens = balances.filter(balance => balance.asset.type === "token" &&
+            balance.asset.assetReference?.toLowerCase().includes("usd"));
+        usdTokens.forEach(balance => {
+            expect(balance.asset.type).toBe("token");
+            if (balance.asset.type === "token") {
+                expect(balance.asset.assetReference).toBeTruthy();
+            }
+            expect(typeof balance.value).toBe("bigint");
+        });
+    }, 15000);
+});
+//# sourceMappingURL=getBalance.integration.test.js.map

package/lib/logic/getBalance.integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getBalance.integration.test.js","sourceRoot":"","sources":["../../src/logic/getBalance.integration.test.ts"],"names":[],"mappings":";;;;;AAAA,+CAAoD;AACpD,uDAAmC;AACnC,6CAA0C;AAE1C,MAAM,MAAM,GAAG,oEAAoE,CAAC;AAEpF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,SAAS,CAAC,GAAG,EAAE;QACb,gBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;YAC9B,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;aACf;YACD,IAAI,EAAE;gBACJ,GAAG,EAAE,IAAA,uBAAc,EAAC,SAAS,CAAC;aAC/B;SACF,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAU,EAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;IACvD,CAAC,EAAE,KAAK,CAAC,CAAC;IAEV,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAU,EAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAElD,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QACjF,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC9B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzC,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACnC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,OAAO,CAC1C,gDAAgD,CACjD,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,KAAK,CAAC,CAAC;IAEV,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAU,EAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAC/B,OAAO,CAAC,EAAE,CACR,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO;YAC9B,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9D,CAAC;QAEF,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC1B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzC,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACnC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,UAAU,EAAE,CAAC;YACpD,CAAC;YACD,MAAM,CAAC,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC"}

package/lib/logic/getBalance.js

@@ -2,14 +2,25 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getBalance = void 0;
 const network_1 = require("../network");
+const sdk_1 = require("../network/sdk");
 async function getBalance(address) {
-    const { balance } = await (0, network_1.getAccount)(address);
-    return [
-        {
-            value: BigInt(balance.toString()),
-            asset: { type: "native" },
-        },
-    ];
+    const [native, staking] = await Promise.all([
+        getNativeBalance(address),
+        getStakingBalances(address),
+    ]);
+    return [...native, ...staking];
 }
 exports.getBalance = getBalance;
+const getNativeBalance = async (address) => {
+    const balances = await (0, network_1.getAllBalancesCached)(address);
+    return balances.map(({ coinType, totalBalance }) => ({
+        value: BigInt(totalBalance),
+        asset: (0, sdk_1.toSuiAsset)(coinType),
+    }));
+};
+const getStakingBalances = (address) => (0, network_1.getStakes)(address).then(stakes => stakes.map(stake => ({
+    value: stake.amount,
+    asset: stake.asset,
+    stake: stake,
+})));
 //# sourceMappingURL=getBalance.js.map

package/lib/logic/getBalance.js.map

@@ -1 +1 @@
-{"version":3,"file":"getBalance.js","sourceRoot":"","sources":["../../src/logic/getBalance.ts"],"names":[],"mappings":";;;AAAA,wCAAwC;AAGjC,KAAK,UAAU,UAAU,CAAC,OAAe;IAC9C,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,IAAA,oBAAU,EAAC,OAAO,CAAC,CAAC;IAC9C,OAAO;QACL;YACE,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B;KACF,CAAC;AACJ,CAAC;AARD,gCAQC"}
+{"version":3,"file":"getBalance.js","sourceRoot":"","sources":["../../src/logic/getBalance.ts"],"names":[],"mappings":";;;AAAA,wCAA6D;AAE7D,wCAA4C;AAErC,KAAK,UAAU,UAAU,CAAC,OAAe;IAC9C,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QAC1C,gBAAgB,CAAC,OAAO,CAAC;QACzB,kBAAkB,CAAC,OAAO,CAAC;KAC5B,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC;AACjC,CAAC;AAND,gCAMC;AAED,MAAM,gBAAgB,GAAG,KAAK,EAAE,OAAe,EAAsB,EAAE;IACrE,MAAM,QAAQ,GAAG,MAAM,IAAA,8BAAoB,EAAC,OAAO,CAAC,CAAC;IACrD,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QACnD,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC;QAC3B,KAAK,EAAE,IAAA,gBAAU,EAAC,QAAQ,CAAC;KAC5B,CAAC,CAAC,CAAC;AACN,CAAC,CAAC;AAEF,MAAM,kBAAkB,GAAG,CAAC,OAAe,EAAsB,EAAE,CACjE,IAAA,mBAAS,EAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAC/B,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACnB,KAAK,EAAE,KAAK,CAAC,MAAM;IACnB,KAAK,EAAE,KAAK,CAAC,KAAK;IAClB,KAAK,EAAE,KAAK;CACb,CAAC,CAAC,CACJ,CAAC"}

package/lib/logic/getBalance.test.js

@@ -2,21 +2,63 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const getBalance_1 = require("./getBalance");
 const network_1 = require("../network");
-// Mock the getAccount function
 jest.mock("../network", () => ({
-    getAccount: jest.fn().mockResolvedValue({ balance: 1000000000 }),
+    getAllBalancesCached: jest.fn().mockResolvedValue([
+        { coinType: "0x2::sui::SUI", totalBalance: 1000000000 },
+        { coinType: "0x3::usdt::USDT", totalBalance: 500000000 },
+    ]),
+    getStakes: jest.fn().mockResolvedValue([]),
 }));
+const mockedGetStakes = jest.mocked(network_1.getStakes);
 describe("getBalance", () => {
     beforeEach(() => {
         jest.clearAllMocks();
     });
-    it("should return the correct balance as bigint", async () => {
-        // Mock the getAccount response
-        const mockBalance = { balance: 1000000000 };
+    it("should return the correct native SUI balance", async () => {
         const address = "0x123";
         const result = await (0, getBalance_1.getBalance)(address);
-        expect(network_1.getAccount).toHaveBeenCalledWith(address);
-        expect(result[0].value).toBe(BigInt(mockBalance.balance.toString()));
+        expect(result[0]).toMatchObject({
+            value: BigInt(1000000000),
+            asset: { type: "native" },
+        });
+    });
+    it("should return the correct USDT token balance", async () => {
+        const address = "0x123";
+        const result = await (0, getBalance_1.getBalance)(address);
+        expect(result[1]).toMatchObject({
+            value: BigInt(500000000),
+            asset: { type: "token", assetReference: "0x3::usdt::USDT" },
+        });
+    });
+    it("should return staking balances when stakes are available", async () => {
+        const mockStakes = [
+            {
+                uid: "stake_1",
+                address: "0x123",
+                delegate: "0xvalidator1",
+                state: "active",
+                asset: { type: "native" },
+                amount: BigInt(2000000000),
+            },
+        ];
+        mockedGetStakes.mockResolvedValueOnce(mockStakes);
+        const address = "0x123";
+        const result = await (0, getBalance_1.getBalance)(address);
+        expect(network_1.getAllBalancesCached).toHaveBeenCalledWith(address);
+        expect(network_1.getStakes).toHaveBeenCalledWith(address);
+        expect(result).toHaveLength(3);
+        expect(result[2]).toMatchObject({
+            value: BigInt(2000000000),
+            asset: { type: "native" },
+            stake: {
+                uid: "stake_1",
+                address: "0x123",
+                delegate: "0xvalidator1",
+                state: "active",
+                asset: { type: "native" },
+                amount: BigInt(2000000000),
+            },
+        });
     });
 });
 //# sourceMappingURL=getBalance.test.js.map

package/lib/logic/getBalance.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"getBalance.test.js","sourceRoot":"","sources":["../../src/logic/getBalance.test.ts"],"names":[],"mappings":";;AAAA,6CAA0C;AAC1C,wCAAwC;AAExC,+BAA+B;AAC/B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7B,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;CACjE,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,+BAA+B;QAC/B,MAAM,WAAW,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;QAE5C,MAAM,OAAO,GAAG,OAAO,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAA,uBAAU,EAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,CAAC,oBAAU,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"getBalance.test.js","sourceRoot":"","sources":["../../src/logic/getBalance.test.ts"],"names":[],"mappings":";;AAAA,6CAA0C;AAC1C,wCAA6D;AAG7D,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;QAChD,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,EAAE;QACvD,EAAE,QAAQ,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,EAAE;KACzD,CAAC;IACF,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;CAC3C,CAAC,CAAC,CAAC;AAEJ,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAS,CAAC,CAAC;AAE/C,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,OAAO,GAAG,OAAO,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAA,uBAAU,EAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9B,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC;YACzB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,OAAO,GAAG,OAAO,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAA,uBAAU,EAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9B,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC;YACxB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE;SAC5D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,UAAU,GAAY;YAC1B;gBACE,GAAG,EAAE,SAAS;gBACd,OAAO,EAAE,OAAO;gBAChB,QAAQ,EAAE,cAAc;gBACxB,KAAK,EAAE,QAAQ;gBACf,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC;aAC3B;SACF,CAAC;QAEF,eAAe,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAElD,MAAM,OAAO,GAAG,OAAO,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAA,uBAAU,EAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,CAAC,8BAAoB,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAM,CAAC,mBAAS,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9B,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC;YACzB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YACzB,KAAK,EAAE;gBACL,GAAG,EAAE,SAAS;gBACd,OAAO,EAAE,OAAO;gBAChB,QAAQ,EAAE,cAAc;gBACxB,KAAK,EAAE,QAAQ;gBACf,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC;aAC3B;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/lib/logic/index.d.ts

@@ -6,4 +6,5 @@ export { getBalance } from "./getBalance";
 export { lastBlock } from "./lastBlock";
 export { getBlock, getBlockInfo } from "./getBlock";
 export { listOperations } from "./listOperations";
+export { getStakes, getRewards } from "./staking";
 //# sourceMappingURL=index.d.ts.map

package/lib/logic/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/logic/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,KAAK,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/logic/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,KAAK,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC"}

package/lib/logic/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.listOperations = exports.getBlockInfo = exports.getBlock = exports.lastBlock = exports.getBalance = exports.combine = exports.broadcast = exports.estimateFees = exports.craftTransaction = void 0;
+exports.getRewards = exports.getStakes = exports.listOperations = exports.getBlockInfo = exports.getBlock = exports.lastBlock = exports.getBalance = exports.combine = exports.broadcast = exports.estimateFees = exports.craftTransaction = void 0;
 var craftTransaction_1 = require("./craftTransaction");
 Object.defineProperty(exports, "craftTransaction", { enumerable: true, get: function () { return craftTransaction_1.craftTransaction; } });
 var estimateFees_1 = require("./estimateFees");
@@ -18,4 +18,7 @@ Object.defineProperty(exports, "getBlock", { enumerable: true, get: function ()
 Object.defineProperty(exports, "getBlockInfo", { enumerable: true, get: function () { return getBlock_1.getBlockInfo; } });
 var listOperations_1 = require("./listOperations");
 Object.defineProperty(exports, "listOperations", { enumerable: true, get: function () { return listOperations_1.listOperations; } });
+var staking_1 = require("./staking");
+Object.defineProperty(exports, "getStakes", { enumerable: true, get: function () { return staking_1.getStakes; } });
+Object.defineProperty(exports, "getRewards", { enumerable: true, get: function () { return staking_1.getRewards; } });
 //# sourceMappingURL=index.js.map

package/lib/logic/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/logic/index.ts"],"names":[],"mappings":";;;AAAA,uDAA+E;AAAtE,oHAAA,gBAAgB,OAAA;AACzB,+CAA8C;AAArC,4GAAA,YAAY,OAAA;AACrB,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA;AAChB,2CAA0C;AAAjC,wGAAA,UAAU,OAAA;AACnB,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,uCAAoD;AAA3C,oGAAA,QAAQ,OAAA;AAAE,wGAAA,YAAY,OAAA;AAC/B,mDAAkD;AAAzC,gHAAA,cAAc,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/logic/index.ts"],"names":[],"mappings":";;;AAAA,uDAA+E;AAAtE,oHAAA,gBAAgB,OAAA;AACzB,+CAA8C;AAArC,4GAAA,YAAY,OAAA;AACrB,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA;AAChB,2CAA0C;AAAjC,wGAAA,UAAU,OAAA;AACnB,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,uCAAoD;AAA3C,oGAAA,QAAQ,OAAA;AAAE,wGAAA,YAAY,OAAA;AAC/B,mDAAkD;AAAzC,gHAAA,cAAc,OAAA;AACvB,qCAAkD;AAAzC,oGAAA,SAAS,OAAA;AAAE,qGAAA,UAAU,OAAA"}

package/lib/logic/staking.d.ts

@@ -0,0 +1,4 @@
+import { Cursor, Page, Stake, Reward } from "@ledgerhq/coin-framework/api/types";
+export declare const getStakes: (address: string, _cursor?: Cursor) => Promise<Page<Stake>>;
+export declare const getRewards: (_address: string, _cursor?: Cursor) => Promise<Page<Reward>>;
+//# sourceMappingURL=staking.d.ts.map

package/lib/logic/staking.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"staking.d.ts","sourceRoot":"","sources":["../../src/logic/staking.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAGjF,eAAO,MAAM,SAAS,YAAa,MAAM,YAAY,MAAM,KAAG,QAAQ,KAAK,KAAK,CAAC,CAEhF,CAAC;AAEF,eAAO,MAAM,UAAU,aAAc,MAAM,YAAY,MAAM,KAAG,QAAQ,KAAK,MAAM,CAAC,CAEnF,CAAC"}

package/lib/logic/staking.js

@@ -0,0 +1,36 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRewards = exports.getStakes = void 0;
+const sdk = __importStar(require("../network"));
+const getStakes = (address, _cursor) => {
+    return sdk.getStakes(address).then(stakes => ({ items: stakes }));
+};
+exports.getStakes = getStakes;
+const getRewards = (_address, _cursor) => {
+    throw new Error("getRewards is not supported");
+};
+exports.getRewards = getRewards;
+//# sourceMappingURL=staking.js.map

package/lib/logic/staking.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"staking.js","sourceRoot":"","sources":["../../src/logic/staking.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,gDAAkC;AAE3B,MAAM,SAAS,GAAG,CAAC,OAAe,EAAE,OAAgB,EAAwB,EAAE;IACnF,OAAO,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;AACpE,CAAC,CAAC;AAFW,QAAA,SAAS,aAEpB;AAEK,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAE,OAAgB,EAAyB,EAAE;IACtF,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;AACjD,CAAC,CAAC;AAFW,QAAA,UAAU,cAErB"}

package/lib/network/index.d.ts

@@ -1,16 +1,16 @@
-import { getAccount, getAccountBalances, getOperations, paymentInfo, createTransaction, executeTransactionBlock } from "./sdk";
-export { getAccount, getAccountBalances, getOperations, paymentInfo, createTransaction, executeTransactionBlock, };
+import { getAccountBalances, getAllBalancesCached, getOperations, getBlock, getBlockInfo, getStakes, paymentInfo, createTransaction, executeTransactionBlock } from "./sdk";
+export { getAccountBalances, getAllBalancesCached, getOperations, getBlock, getBlockInfo, getStakes, paymentInfo, createTransaction, executeTransactionBlock, };
 declare const _default: {
-    getAccount: (addr: string) => Promise<{
-        blockHeight: number;
-        balance: import("bignumber.js").BigNumber;
-    }>;
     getAccountBalances: (addr: string) => Promise<{
         coinType: string;
         blockHeight: number;
         balance: import("bignumber.js").BigNumber;
     }[]>;
+    getAllBalancesCached: import("@ledgerhq/live-network/lib/cache").CacheRes<[owner: string], import("@mysten/sui/dist/cjs/client").CoinBalance[]>;
     getOperations: (accountId: string, addr: string, cursor?: string | null | undefined) => Promise<import("@ledgerhq/types-live").Operation[]>;
+    getBlock: (id: string) => Promise<import("@ledgerhq/coin-framework/lib/api").Block>;
+    getBlockInfo: (id: string) => Promise<import("@ledgerhq/coin-framework/lib/api").BlockInfo>;
+    getStakes: (address: string) => Promise<import("@ledgerhq/coin-framework/lib/api").Stake[]>;
     paymentInfo: (sender: string, fakeTransaction: import("..").Transaction) => Promise<{
         gasBudget: string;
         totalGasUsed: bigint;

package/lib/network/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/network/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,kBAAkB,EAClB,aAAa,EACb,WAAW,EACX,iBAAiB,EACjB,uBAAuB,EACxB,MAAM,OAAO,CAAC;AAEf,OAAO,EACL,UAAU,EACV,kBAAkB,EAClB,aAAa,EACb,WAAW,EACX,iBAAiB,EACjB,uBAAuB,GACxB,CAAC;;;;;;;;;;;;;;;;;;;;AAEF,wBAOE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/network/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,SAAS,EACT,WAAW,EACX,iBAAiB,EACjB,uBAAuB,EACxB,MAAM,OAAO,CAAC;AAEf,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,SAAS,EACT,WAAW,EACX,iBAAiB,EACjB,uBAAuB,GACxB,CAAC;;;;;;;;;;;;;;;;;;;;AAEF,wBAUE"}