@aurigami/sdk 1.5.8 → 1.5.10

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.5.8",
+  "version": "1.5.10",
   "license": "MIT",
   "main": "dist/index.js",
   "typings": "dist/index.d.ts",

package/src/SDK.ts

@@ -15,6 +15,7 @@ import {
   ALL_STAKING_POOLS,
   BORROW_LIMIT_BUFFER,
   DECIMAL_PRECISION,
+  INF,
   networkAddresses,
 } from './constants';
 import { decimalFactor, isSameAddress } from './helpers';
@@ -327,6 +328,12 @@ export class SdkReadWrite extends SdkRead {
       );
   }
 
+  public async claimLpRewards(): Promise<TransactionResponse> {
+    return this._auriFairLaunch
+      .connect(this._networkConnection.signer!)
+      .harvest(await this._networkConnection.signer!.getAddress(), 0, INF);
+  }
+
   public async stake(amount: TokenAmount): Promise<TransactionResponse> {
     return new StakingReadWrite(this._networkConnection).stake(amount);
   }

package/src/Staking.ts

@@ -154,9 +154,18 @@ export class StakingReadWrite extends StakingRead {
   }
 
   public async unstake(amount: TokenAmount): Promise<TransactionResponse> {
+    const msgSender = await this._networkConnection.signer!.getAddress();
+    const amountInput = BN.from(amount.rawAmount());
+    const stakedAmount = BN.from(
+      (await this.stakeBalanceOf(msgSender)).rawAmount()
+    );
+    // unstakeAmount = min(amountInput, stakedAmount - 1)
+    const unstakeAmount = amountInput.lt(stakedAmount)
+      ? amountInput
+      : stakedAmount.sub(1);
     return this._auriFairLaunch
       .connect(this._networkConnection.signer!)
-      .withdraw(PLYWNEAR_POOL_ID, amount.rawAmount());
+      .withdraw(PLYWNEAR_POOL_ID, unstakeAmount);
   }
 }