@aurigami/sdk 1.5.9 → 1.5.10

package/dist/sdk.esm.js

@@ -3978,13 +3978,28 @@ var StakingReadWrite = /*#__PURE__*/function (_StakingRead) {
 
   _proto2.unstake = /*#__PURE__*/function () {
     var _unstake = /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/runtime_1.mark(function _callee6(amount) {
+      var msgSender, amountInput, stakedAmount, unstakeAmount;
       return runtime_1.wrap(function _callee6$(_context6) {
         while (1) {
           switch (_context6.prev = _context6.next) {
             case 0:
-              return _context6.abrupt("return", this._auriFairLaunch.connect(this._networkConnection.signer).withdraw(PLYWNEAR_POOL_ID, amount.rawAmount()));
+              _context6.next = 2;
+              return this._networkConnection.signer.getAddress();
 
-            case 1:
+            case 2:
+              msgSender = _context6.sent;
+              amountInput = BigNumber$1.from(amount.rawAmount());
+              _context6.t0 = BigNumber$1;
+              _context6.next = 7;
+              return this.stakeBalanceOf(msgSender);
+
+            case 7:
+              _context6.t1 = _context6.sent.rawAmount();
+              stakedAmount = _context6.t0.from.call(_context6.t0, _context6.t1);
+              unstakeAmount = amountInput.lt(stakedAmount) ? amountInput : stakedAmount.sub(1);
+              return _context6.abrupt("return", this._auriFairLaunch.connect(this._networkConnection.signer).withdraw(PLYWNEAR_POOL_ID, unstakeAmount));
+
+            case 11:
             case "end":
               return _context6.stop();
           }