@t2000/sdk 0.16.9 → 0.16.11

package/dist/index.js

@@ -3437,14 +3437,21 @@ To access invested funds: t2000 invest sell ${params.amount} ${asset}`,
           investmentCostBasis += pos.costBasis;
         }
       }
+      let strategySuiTotal = 0;
       for (const key of this.portfolio.getAllStrategyKeys()) {
         for (const sp of this.portfolio.getStrategyPositions(key)) {
           if (!(sp.asset in INVESTMENT_ASSETS)) continue;
           const price = assetPrices[sp.asset] ?? 0;
           investmentValue += sp.totalAmount * price;
           investmentCostBasis += sp.costBasis;
+          if (sp.asset === "SUI") strategySuiTotal += sp.totalAmount;
         }
       }
+      if (strategySuiTotal > 0) {
+        const suiPrice2 = assetPrices["SUI"] ?? 0;
+        const gasSui = Math.max(0, bal.gasReserve.sui - strategySuiTotal);
+        bal.gasReserve = { sui: gasSui, usdEquiv: gasSui * suiPrice2 };
+      }
       bal.investment = investmentValue;
       bal.investmentPnL = investmentValue - investmentCostBasis;
     } catch {
@@ -5058,14 +5065,23 @@ To sell investment: t2000 invest sell ${params.amount} ${fromAsset}`,
   // -- Helpers --
   async getFreeBalance(asset) {
     if (!(asset in INVESTMENT_ASSETS)) return Infinity;
+    let walletInvested = 0;
     const pos = this.portfolio.getPosition(asset);
-    const invested = pos?.totalAmount ?? 0;
-    if (invested <= 0) return Infinity;
+    if (pos && pos.totalAmount > 0 && !pos.earning) {
+      walletInvested += pos.totalAmount;
+    }
+    for (const key of this.portfolio.getAllStrategyKeys()) {
+      for (const sp of this.portfolio.getStrategyPositions(key)) {
+        if (sp.asset === asset && sp.totalAmount > 0) {
+          walletInvested += sp.totalAmount;
+        }
+      }
+    }
+    if (walletInvested <= 0 && (!pos || pos.totalAmount <= 0)) return Infinity;
     const assetInfo = SUPPORTED_ASSETS[asset];
     const balance = await this.client.getBalance({ owner: this._address, coinType: assetInfo.type });
     const walletAmount = Number(balance.totalBalance) / 10 ** assetInfo.decimals;
     const gasReserve = asset === "SUI" ? GAS_RESERVE_MIN : 0;
-    const walletInvested = pos?.earning ? 0 : invested;
     return Math.max(0, walletAmount - walletInvested - gasReserve);
   }
   async resolveLending(protocol, asset, capability) {