@pear-protocol/exchanges-sdk 0.3.0 → 0.5.0

package/dist/tracker/okx/tracker.test.js

@@ -0,0 +1,126 @@
+import { vi, describe, beforeEach, it, expect } from 'vitest';
+import { OkxAccountTracker } from './tracker';
+
+const { fetchAccountConfigMock, fetchBalanceMock, fetchPositionsMock, fetchLeverageInfosMock } = vi.hoisted(() => ({
+  fetchAccountConfigMock: vi.fn(),
+  fetchBalanceMock: vi.fn(),
+  fetchPositionsMock: vi.fn(),
+  fetchLeverageInfosMock: vi.fn()
+}));
+vi.mock("../../client/okx/rest", () => ({
+  fetchAccountConfig: fetchAccountConfigMock,
+  fetchBalance: fetchBalanceMock,
+  fetchPositions: fetchPositionsMock,
+  fetchLeverageInfos: fetchLeverageInfosMock
+}));
+const creds = { api_key: "k", api_secret: "s", api_pass: "p" };
+function buildTracker() {
+  return new OkxAccountTracker({ demo: true, credentials: creds });
+}
+describe("OkxAccountTracker.snapshot", () => {
+  beforeEach(() => {
+    fetchAccountConfigMock.mockReset();
+    fetchBalanceMock.mockReset();
+    fetchPositionsMock.mockReset();
+  });
+  it("returns mapped balance + positions, cross mgnMode only, with no socket opened", async () => {
+    fetchAccountConfigMock.mockResolvedValue({ acctLv: "3" });
+    fetchBalanceMock.mockResolvedValue({
+      totalEq: "1000",
+      adjEq: "900",
+      availEq: "900",
+      mmr: "10",
+      imr: "20",
+      upl: "5",
+      mgnRatio: "0.0500",
+      details: [{ ccy: "USDT", eq: "1000", cashBal: "1000", availEq: "900", availBal: "900", eqUsd: "1000" }]
+    });
+    fetchPositionsMock.mockResolvedValue([
+      {
+        instId: "BTC-USDT-SWAP",
+        posSide: "long",
+        pos: "1",
+        avgPx: "50000",
+        upl: "100",
+        lever: "10",
+        mgnMode: "cross",
+        liqPx: "40000"
+      }
+    ]);
+    const tracker = buildTracker();
+    const snapshot = await tracker.snapshot();
+    expect(snapshot.balance).not.toBeNull();
+    expect(snapshot.balance).toMatchObject({
+      accountType: "OKX_MULTI",
+      totalEquity: "1000",
+      availableToTrade: "900",
+      maintenanceMargin: "10",
+      initialMarginUsed: "20",
+      marginRatio: "0.0500"
+    });
+    expect(snapshot.positions).toHaveLength(1);
+    expect(snapshot.positions[0]).toMatchObject({
+      symbol: "BTC-USDT-SWAP",
+      side: "long",
+      leverage: "10",
+      marginType: "cross",
+      liquidationPrice: "40000"
+    });
+    expect(fetchAccountConfigMock).toHaveBeenCalledTimes(1);
+    expect(fetchBalanceMock).toHaveBeenCalledTimes(1);
+    expect(fetchPositionsMock).toHaveBeenCalledTimes(1);
+    expect(fetchLeverageInfosMock).not.toHaveBeenCalled();
+  });
+  it("returns a null balance when account type cannot be resolved", async () => {
+    fetchAccountConfigMock.mockResolvedValue({ acctLv: "unknown" });
+    fetchBalanceMock.mockResolvedValue({
+      totalEq: "0",
+      adjEq: "0",
+      availEq: "0",
+      mmr: "0",
+      imr: "0",
+      upl: "0",
+      mgnRatio: "0",
+      details: []
+    });
+    fetchPositionsMock.mockResolvedValue([]);
+    const snapshot = await buildTracker().snapshot();
+    expect(snapshot.balance).toBeNull();
+    expect(snapshot.positions).toEqual([]);
+  });
+});
+describe("OkxAccountTracker.snapshotAssets", () => {
+  beforeEach(() => {
+    fetchLeverageInfosMock.mockReset();
+  });
+  it("reads all instIds through one batched leverage-info call and maps rows back by instId", async () => {
+    fetchLeverageInfosMock.mockResolvedValue([
+      { instId: "BTC-USDT-SWAP", lever: "10", mgnMode: "cross" },
+      { instId: "ETH-USDT-SWAP", lever: "3", mgnMode: "cross" }
+    ]);
+    const result = await buildTracker().snapshotAssets(["BTC-USDT-SWAP", "ETH-USDT-SWAP", "DOGE-USDT-SWAP"]);
+    expect(fetchLeverageInfosMock).toHaveBeenCalledTimes(1);
+    expect(fetchLeverageInfosMock).toHaveBeenCalledWith(
+      creds,
+      ["BTC-USDT-SWAP", "ETH-USDT-SWAP", "DOGE-USDT-SWAP"],
+      true
+    );
+    expect(result).toEqual({
+      "BTC-USDT-SWAP": { coin: "BTC-USDT-SWAP", leverage: "10", marginType: "cross" },
+      "ETH-USDT-SWAP": { coin: "ETH-USDT-SWAP", leverage: "3", marginType: "cross" },
+      "DOGE-USDT-SWAP": null
+    });
+  });
+  it("collapses long/short-mode duplicate rows to one entry per instId", async () => {
+    fetchLeverageInfosMock.mockResolvedValue([
+      { instId: "BTC-USDT-SWAP", lever: "10", mgnMode: "cross", posSide: "long" },
+      { instId: "BTC-USDT-SWAP", lever: "10", mgnMode: "cross", posSide: "short" }
+    ]);
+    const result = await buildTracker().snapshotAssets(["BTC-USDT-SWAP"]);
+    expect(result["BTC-USDT-SWAP"]).toEqual({ coin: "BTC-USDT-SWAP", leverage: "10", marginType: "cross" });
+  });
+  it("propagates a failed batched read", async () => {
+    fetchLeverageInfosMock.mockRejectedValue(new Error("50011"));
+    await expect(buildTracker().snapshotAssets(["BTC-USDT-SWAP"])).rejects.toThrow("50011");
+  });
+});

package/dist/tracker/types.js

@@ -15,7 +15,8 @@ const ACCOUNT_TYPE_LABELS = {
   [HyperliquidAccountType.DEX_ABSTRACTION]: "Standard (Shared HIP-3 Balance)",
   [HyperliquidAccountType.PORTFOLIO_MARGIN]: "Portfolio Margin",
   [HyperliquidAccountType.UNIFIED_ACCOUNT]: "Unified Account",
-  [LighterAccountType.BASIC]: "Standard",
+  [LighterAccountType.CLASSIC]: "Classic",
+  [LighterAccountType.UNIFIED]: "Unified",
   [OkxAccountType.SPOT]: "Spot Mode",
   [OkxAccountType.SINGLE_CURRENCY_MARGIN]: "Single-Currency Margin",
   [OkxAccountType.MULTI_CURRENCY_MARGIN]: "Multi-Currency Margin",

package/dist/tracker/utils.test.js

@@ -0,0 +1,8 @@
+import { describe, it, expect } from 'vitest';
+import { positionKey } from './utils';
+
+describe("positionKey", () => {
+  it("joins symbol and side with a colon", () => {
+    expect(positionKey("BTC-PERP", "long")).toBe("BTC-PERP:long");
+  });
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pear-protocol/exchanges-sdk",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "description": "Pear Protocol Exchanges SDK",
   "private": false,
   "type": "module",
@@ -22,13 +22,14 @@
   "scripts": {
     "build": "tsup",
     "clean": "rm -rf dist .turbo",
+    "typecheck": "tsc --noEmit",
     "test": "vitest run",
-    "typecheck": "tsc --noEmit"
+    "test:watch": "vitest"
   },
   "dependencies": {
-    "@pear-protocol/core-sdk": "^1.5.0",
-    "@pear-protocol/types": "^1.6.0",
-    "@pear-protocol/utils": "^0.1.0",
+    "@pear-protocol/core-sdk": "^1.7.0",
+    "@pear-protocol/types": "^1.9.0",
+    "@pear-protocol/utils": "^0.2.0",
     "bignumber.js": "^11.1.1",
     "partysocket": "^1.0.3",
     "viem": "^2.48.8"