@t2000/sdk 0.19.5 → 0.19.6

package/dist/index.d.cts

@@ -261,6 +261,9 @@ declare class PortfolioManager {
     } & StoredPosition>;
     recordEarn(asset: string, protocol: string, apy: number): void;
     recordUnearn(asset: string): void;
+    getStrategyAmountForAsset(asset: string): number;
+    getDirectAmount(asset: string): number;
+    deductFromStrategies(asset: string, amount: number): void;
     closePosition(asset: string): void;
     isEarning(asset: string): boolean;
     getRealizedPnL(): number;

package/dist/index.d.ts

@@ -261,6 +261,9 @@ declare class PortfolioManager {
     } & StoredPosition>;
     recordEarn(asset: string, protocol: string, apy: number): void;
     recordUnearn(asset: string): void;
+    getStrategyAmountForAsset(asset: string): number;
+    getDirectAmount(asset: string): number;
+    deductFromStrategies(asset: string, amount: number): void;
     closePosition(asset: string): void;
     isEarning(asset: string): boolean;
     getRealizedPnL(): number;

package/dist/index.js

@@ -56767,6 +56767,45 @@ var PortfolioManager = class {
     this.data.positions[asset] = pos;
     this.save();
   }
+  getStrategyAmountForAsset(asset) {
+    this.load();
+    let total = 0;
+    for (const bucket of Object.values(this.data.strategies)) {
+      const pos = bucket[asset];
+      if (pos && pos.totalAmount > 0) total += pos.totalAmount;
+    }
+    return total;
+  }
+  getDirectAmount(asset) {
+    this.load();
+    const aggregate = this.data.positions[asset]?.totalAmount ?? 0;
+    const strategyAmount = this.getStrategyAmountForAsset(asset);
+    return Math.max(0, aggregate - strategyAmount);
+  }
+  deductFromStrategies(asset, amount) {
+    this.load();
+    let remaining = amount;
+    for (const [stratKey, bucket] of Object.entries(this.data.strategies)) {
+      if (remaining <= 0) break;
+      const pos = bucket[asset];
+      if (!pos || pos.totalAmount <= 0) continue;
+      const deduct = Math.min(pos.totalAmount, remaining);
+      const costDeduct = pos.avgPrice * deduct;
+      pos.totalAmount -= deduct;
+      pos.costBasis -= costDeduct;
+      if (pos.totalAmount < 1e-6) {
+        pos.totalAmount = 0;
+        pos.costBasis = 0;
+        pos.avgPrice = 0;
+      }
+      remaining -= deduct;
+      const hasPositions = Object.values(bucket).some((p) => p.totalAmount > 0);
+      if (!hasPositions) {
+        delete this.data.strategies[stratKey];
+      }
+    }
+    this.save();
+  }
   closePosition(asset) {
     this.load();
     const pos = this.data.positions[asset];
@@ -58338,6 +58377,7 @@ To sell investment: t2000 invest sell ${params.amount} ${fromAsset}`,
       }
     }
     const price = swapResult.toAmount / sellAmountAsset;
+    const directAmountBeforeSell = this.portfolio.getDirectAmount(params.asset);
     let realizedPnL = 0;
     try {
       realizedPnL = this.portfolio.recordSell({
@@ -58355,6 +58395,12 @@ To sell investment: t2000 invest sell ${params.amount} ${fromAsset}`,
     }
     if (params.usdAmount === "all" && !params._strategyOnly) {
       this.portfolio.closePosition(params.asset);
+      this.portfolio.deductFromStrategies(params.asset, sellAmountAsset);
+    } else if (!params._strategyOnly && sellAmountAsset > 0) {
+      const overflowIntoStrategy = sellAmountAsset - directAmountBeforeSell;
+      if (overflowIntoStrategy > 0) {
+        this.portfolio.deductFromStrategies(params.asset, overflowIntoStrategy);
+      }
     }
     const updatedPos = this.portfolio.getPosition(params.asset);
     const position = {