@curvefi/llamalend-api 1.1.10 → 2.0.1

package/lib/lendMarkets/modules/common/loanBase.js

@@ -0,0 +1,801 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import memoize from "memoizee";
+import { _getAddress, parseUnits, BN, ensureAllowance, hasAllowance, ensureAllowanceEstimateGas, formatUnits, smartNumber, _mulBy1_3, DIGas, calculateFutureLeverage, MAX_ACTIVE_BAND, fromBN, } from "../../../utils";
+import { _getUserCollateral } from "../../../external-api";
+export class LoanBaseModule {
+    constructor(market) {
+        this.createLoanMaxRecvAllRanges = memoize((collateral) => __awaiter(this, void 0, void 0, function* () {
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const calls = [];
+            for (let N = this.market.minBands; N <= this.market.maxBands; N++) {
+                // TODO: currentDebt is no longer passed - now we pass user address or zero address, currentDebt is calculated under the hood
+                calls.push(this.llamalend.contracts[this.market.addresses.controller].multicallContract.max_borrowable(_collateral, N, 0));
+            }
+            const _amounts = yield this.llamalend.multicallProvider.all(calls);
+            const res = {};
+            for (let N = this.market.minBands; N <= this.market.maxBands; N++) {
+                res[N] = formatUnits(_amounts[N - this.market.minBands], this.market.borrowed_token.decimals);
+            }
+            return res;
+        }), {
+            promise: true,
+            maxAge: 5 * 60 * 1000, // 5m
+        });
+        this.estimateGas = {
+            createLoanApprove: this.createLoanApproveEstimateGas.bind(this),
+            createLoan: this.createLoanEstimateGas.bind(this),
+            borrowMoreApprove: this.borrowMoreApproveEstimateGas.bind(this),
+            borrowMore: this.borrowMoreEstimateGas.bind(this),
+            addCollateralApprove: this.addCollateralApproveEstimateGas.bind(this),
+            addCollateral: this.addCollateralEstimateGas.bind(this),
+            removeCollateral: this.removeCollateralEstimateGas.bind(this),
+            repayApprove: this.repayApproveEstimateGas.bind(this),
+            repay: this.repayEstimateGas.bind(this),
+            fullRepayApprove: this.fullRepayApproveEstimateGas.bind(this),
+            fullRepay: this.fullRepayEstimateGas.bind(this),
+            liquidateApprove: this.liquidateApproveEstimateGas.bind(this),
+            liquidate: this.liquidateEstimateGas.bind(this),
+            selfLiquidateApprove: this.selfLiquidateApproveEstimateGas.bind(this),
+            selfLiquidate: this.selfLiquidateEstimateGas.bind(this),
+            partialSelfLiquidateApprove: this.partialSelfLiquidateApproveEstimateGas.bind(this),
+            partialSelfLiquidate: this.partialSelfLiquidateEstimateGas.bind(this),
+        };
+        this.market = market;
+        this.llamalend = market.getLlamalend();
+    }
+    _checkRange(range) {
+        if (range < this.market.minBands)
+            throw Error(`range must be >= ${this.market.minBands}`);
+        if (range > this.market.maxBands)
+            throw Error(`range must be <= ${this.market.maxBands}`);
+    }
+    createLoanMaxRecv(collateral, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._checkRange(range);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            // TODO: currentDebt is no longer passed - now we pass user address or zero address, currentDebt is calculated under the hood
+            return formatUnits(yield contract.max_borrowable(_collateral, range, 0, this.llamalend.constantOptions), this.market.borrowed_token.decimals);
+        });
+    }
+    getMaxRange(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const maxRecv = yield this.createLoanMaxRecvAllRanges(collateral);
+            for (let N = this.market.minBands; N <= this.market.maxBands; N++) {
+                if (BN(debt).gt(BN(maxRecv[N])))
+                    return N - 1;
+            }
+            return this.market.maxBands;
+        });
+    }
+    _calcN1(_collateral, _debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._checkRange(range);
+            // TODO: add 4th parameter - user address in calculate_debt_n1
+            return yield this.llamalend.contracts[this.market.addresses.controller].contract.calculate_debt_n1(_collateral, _debt, range, this.llamalend.constantOptions);
+        });
+    }
+    _calcN1AllRanges(_collateral, _debt, maxN) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const calls = [];
+            // TODO: add 4th parameter - user address in calculate_debt_n1
+            for (let N = this.market.minBands; N <= maxN; N++) {
+                calls.push(this.llamalend.contracts[this.market.addresses.controller].multicallContract.calculate_debt_n1(_collateral, _debt, N));
+            }
+            return yield this.llamalend.multicallProvider.all(calls);
+        });
+    }
+    _createLoanBands(collateral, debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _n1 = yield this._calcN1(parseUnits(collateral, this.market.collateral_token.decimals), parseUnits(debt, this.market.borrowed_token.decimals), range);
+            const _n2 = _n1 + BigInt(range - 1);
+            return [_n2, _n1];
+        });
+    }
+    _createLoanBandsAllRanges(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const maxN = yield this.getMaxRange(collateral, debt);
+            const _n1_arr = yield this._calcN1AllRanges(parseUnits(collateral, this.market.collateral_token.decimals), parseUnits(debt, this.market.borrowed_token.decimals), maxN);
+            const _n2_arr = [];
+            for (let N = this.market.minBands; N <= maxN; N++) {
+                _n2_arr.push(_n1_arr[N - this.market.minBands] + BigInt(N - 1));
+            }
+            const res = {};
+            for (let N = this.market.minBands; N <= maxN; N++) {
+                res[N] = [_n2_arr[N - this.market.minBands], _n1_arr[N - this.market.minBands]];
+            }
+            return res;
+        });
+    }
+    createLoanBands(collateral, debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._createLoanBands(collateral, debt, range);
+            return [Number(_n2), Number(_n1)];
+        });
+    }
+    createLoanBandsAllRanges(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _bandsAllRanges = yield this._createLoanBandsAllRanges(collateral, debt);
+            const bandsAllRanges = {};
+            for (let N = this.market.minBands; N <= this.market.maxBands; N++) {
+                if (_bandsAllRanges[N]) {
+                    bandsAllRanges[N] = _bandsAllRanges[N].map(Number);
+                }
+                else {
+                    bandsAllRanges[N] = null;
+                }
+            }
+            return bandsAllRanges;
+        });
+    }
+    createLoanPrices(collateral, debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._createLoanBands(collateral, debt, range);
+            return yield this.market.prices.getPrices(_n2, _n1);
+        });
+    }
+    createLoanPricesAllRanges(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _bandsAllRanges = yield this._createLoanBandsAllRanges(collateral, debt);
+            const pricesAllRanges = {};
+            for (let N = this.market.minBands; N <= this.market.maxBands; N++) {
+                if (_bandsAllRanges[N]) {
+                    pricesAllRanges[N] = yield this.market.prices.calcPrices(..._bandsAllRanges[N]);
+                }
+                else {
+                    pricesAllRanges[N] = null;
+                }
+            }
+            return pricesAllRanges;
+        });
+    }
+    createLoanHealth(collateral_1, debt_1, range_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, debt, range, full = true) {
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const _debt = parseUnits(debt, this.market.borrowed_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            // TODO: verify parameters
+            let _health = yield contract.health_calculator(this.llamalend.constants.ZERO_ADDRESS, _collateral, _debt, full, range, this.llamalend.constantOptions);
+            _health = _health * BigInt(100);
+            return formatUnits(_health);
+        });
+    }
+    createLoanIsApproved(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield hasAllowance.call(this.llamalend, [this.market.collateral_token.address], [collateral], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    createLoanApproveEstimateGas(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.collateral_token.address], [collateral], this.market.addresses.controller);
+        });
+    }
+    createLoanApprove(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowance.call(this.llamalend, [this.market.collateral_token.address], [collateral], this.market.addresses.controller);
+        });
+    }
+    _createLoan(collateral, debt, range, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (yield this.market.userPosition.userLoanExists())
+                throw Error("Loan already created");
+            this._checkRange(range);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const _debt = parseUnits(debt, this.market.borrowed_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = yield contract.create_loan.estimateGas(_collateral, _debt, range, Object.assign({}, this.llamalend.constantOptions));
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            // TODO: ABI has changed (see docs)
+            return (yield contract.create_loan(_collateral, _debt, range, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    createLoanEstimateGas(collateral, debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!(yield this.createLoanIsApproved(collateral)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._createLoan(collateral, debt, range, true);
+        });
+    }
+    createLoan(collateral, debt, range) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.createLoanApprove(collateral);
+            return yield this._createLoan(collateral, debt, range, false);
+        });
+    }
+    // ---------------- BORROW MORE ----------------
+    borrowMoreMaxRecv(collateralAmount) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _collateral: _currentCollateral, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt();
+            const _collateral = _currentCollateral + parseUnits(collateralAmount, this.market.collateral_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const _debt = yield contract.max_borrowable(_collateral, _N, _currentDebt, this.llamalend.constantOptions);
+            // TODO: max_borrowable(deltaCollateral, _N, user) - debt is now calculated under the hood, return as is
+            return formatUnits(_debt - _currentDebt, this.market.borrowed_token.decimals);
+        });
+    }
+    _borrowMoreBands(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _collateral: _currentCollateral, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt();
+            if (_currentDebt === BigInt(0))
+                throw Error(`Loan for ${this.llamalend.signerAddress} does not exist`);
+            const _collateral = _currentCollateral + parseUnits(collateral, this.market.collateral_token.decimals);
+            const _debt = _currentDebt + parseUnits(debt, this.market.borrowed_token.decimals);
+            const _n1 = yield this._calcN1(_collateral, _debt, Number(_N));
+            const _n2 = _n1 + _N - BigInt(1);
+            return [_n2, _n1];
+        });
+    }
+    borrowMoreBands(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._borrowMoreBands(collateral, debt);
+            return [Number(_n2), Number(_n1)];
+        });
+    }
+    borrowMorePrices(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._borrowMoreBands(collateral, debt);
+            return yield this.market.prices.getPrices(_n2, _n1);
+        });
+    }
+    borrowMoreHealth(collateral_1, debt_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, debt, full = true, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const _debt = parseUnits(debt, this.market.borrowed_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            let _health = yield contract.health_calculator(address, _collateral, _debt, full, 0, this.llamalend.constantOptions);
+            _health = _health * BigInt(100);
+            return formatUnits(_health);
+        });
+    }
+    borrowMoreIsApproved(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield hasAllowance.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    borrowMoreApproveEstimateGas(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.market.addresses.controller);
+        });
+    }
+    borrowMoreApprove(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowance.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.market.addresses.controller);
+        });
+    }
+    _borrowMore(collateral, debt, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { borrowed, debt: currentDebt } = yield this.market.userPosition.userState();
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${this.llamalend.signerAddress} does not exist`);
+            if (Number(borrowed) > 0)
+                throw Error(`User ${this.llamalend.signerAddress} is already in liquidation mode`);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const _debt = parseUnits(debt, this.market.borrowed_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = yield contract.borrow_more.estimateGas(_collateral, _debt, Object.assign({}, this.llamalend.constantOptions));
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            return (yield contract.borrow_more(_collateral, _debt, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    borrowMoreEstimateGas(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!(yield this.borrowMoreIsApproved(collateral)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._borrowMore(collateral, debt, true);
+        });
+    }
+    borrowMore(collateral, debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.borrowMoreApprove(collateral);
+            return yield this._borrowMore(collateral, debt, false);
+        });
+    }
+    borrowMoreFutureLeverage(collateral_1, debt_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, debt, userAddress = '') {
+            userAddress = _getAddress.call(this.llamalend, userAddress);
+            const { stateCollateral, totalDepositFromUser } = yield this.market.userPosition.getCurrentLeverageParams(userAddress);
+            const collateralFromDebt = yield this.market.amm.swapExpected(0, 1, debt);
+            const futureCollateralState = BN(stateCollateral).plus(collateralFromDebt);
+            const futureTotalDepositFromUserPrecise = BN(totalDepositFromUser).plus(collateral);
+            return futureCollateralState.div(futureTotalDepositFromUserPrecise).toString();
+        });
+    }
+    // ---------------- ADD COLLATERAL ----------------
+    _addCollateralBands(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const { _collateral: _currentCollateral, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt(address);
+            if (_currentDebt === BigInt(0))
+                throw Error(`Loan for ${address} does not exist`);
+            const _collateral = _currentCollateral + parseUnits(collateral, this.market.collateral_token.decimals);
+            const _n1 = yield this._calcN1(_collateral, _currentDebt, Number(_N));
+            const _n2 = _n1 + _N - BigInt(1);
+            return [_n2, _n1];
+        });
+    }
+    addCollateralBands(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, address = "") {
+            const [_n2, _n1] = yield this._addCollateralBands(collateral, address);
+            return [Number(_n2), Number(_n1)];
+        });
+    }
+    addCollateralPrices(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, address = "") {
+            const [_n2, _n1] = yield this._addCollateralBands(collateral, address);
+            return yield this.market.prices.getPrices(_n2, _n1);
+        });
+    }
+    addCollateralHealth(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, full = true, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            let _health = yield contract.health_calculator(address, _collateral, 0, full, 0, this.llamalend.constantOptions);
+            _health = _health * BigInt(100);
+            return formatUnits(_health);
+        });
+    }
+    addCollateralIsApproved(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield hasAllowance.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    addCollateralApproveEstimateGas(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.market.addresses.controller);
+        });
+    }
+    addCollateralApprove(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowance.call(this.llamalend, [this.market.addresses.collateral_token], [collateral], this.market.addresses.controller);
+        });
+    }
+    _addCollateral(collateral, address, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { borrowed, debt: currentDebt } = yield this.market.userPosition.userState(address);
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${address} does not exist`);
+            if (Number(borrowed) > 0)
+                throw Error(`User ${address} is already in liquidation mode`);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = yield contract.add_collateral.estimateGas(_collateral, address, Object.assign({}, this.llamalend.constantOptions));
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            return (yield contract.add_collateral(_collateral, address, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    addCollateralEstimateGas(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            if (!(yield this.addCollateralIsApproved(collateral)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._addCollateral(collateral, address, true);
+        });
+    }
+    addCollateral(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            yield this.addCollateralApprove(collateral);
+            return yield this._addCollateral(collateral, address, false);
+        });
+    }
+    addCollateralFutureLeverage(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, userAddress = '') {
+            userAddress = _getAddress.call(this.llamalend, userAddress);
+            const [userCollateral, { collateral: currentCollateral }] = yield Promise.all([
+                _getUserCollateral(this.llamalend.constants.NETWORK_NAME, this.market.addresses.controller, userAddress),
+                this.market.userPosition.userState(userAddress),
+            ]);
+            const total_deposit_from_user = userCollateral.total_deposit_from_user_precise;
+            return calculateFutureLeverage(currentCollateral, total_deposit_from_user, collateral, 'add');
+        });
+    }
+    // ---------------- REMOVE COLLATERAL ----------------
+    maxRemovable() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _collateral: _currentCollateral, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt();
+            const _requiredCollateral = yield this.llamalend.contracts[this.market.addresses.controller].contract.min_collateral(_currentDebt, _N, this.llamalend.constantOptions);
+            return formatUnits(_currentCollateral - _requiredCollateral, this.market.collateral_token.decimals);
+        });
+    }
+    _removeCollateralBands(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _collateral: _currentCollateral, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt();
+            if (_currentDebt === BigInt(0))
+                throw Error(`Loan for ${this.llamalend.signerAddress} does not exist`);
+            const _collateral = _currentCollateral - parseUnits(collateral, this.market.collateral_token.decimals);
+            const _n1 = yield this._calcN1(_collateral, _currentDebt, Number(_N));
+            const _n2 = _n1 + _N - BigInt(1);
+            return [_n2, _n1];
+        });
+    }
+    removeCollateralBands(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._removeCollateralBands(collateral);
+            return [Number(_n2), Number(_n1)];
+        });
+    }
+    removeCollateralPrices(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [_n2, _n1] = yield this._removeCollateralBands(collateral);
+            return yield this.market.prices.getPrices(_n2, _n1);
+        });
+    }
+    removeCollateralHealth(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, full = true, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals) * BigInt(-1);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            let _health = yield contract.health_calculator(address, _collateral, 0, full, 0, this.llamalend.constantOptions);
+            _health = _health * BigInt(100);
+            return formatUnits(_health);
+        });
+    }
+    _removeCollateral(collateral, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { borrowed, debt: currentDebt } = yield this.market.userPosition.userState();
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${this.llamalend.signerAddress} does not exist`);
+            if (Number(borrowed) > 0)
+                throw Error(`User ${this.llamalend.signerAddress} is already in liquidation mode`);
+            const _collateral = parseUnits(collateral, this.market.collateral_token.decimals);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = yield contract.remove_collateral.estimateGas(_collateral, this.llamalend.constantOptions);
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            return (yield contract.remove_collateral(_collateral, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    removeCollateralEstimateGas(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this._removeCollateral(collateral, true);
+        });
+    }
+    removeCollateral(collateral) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this._removeCollateral(collateral, false);
+        });
+    }
+    removeCollateralFutureLeverage(collateral_1) {
+        return __awaiter(this, arguments, void 0, function* (collateral, userAddress = '') {
+            userAddress = _getAddress.call(this.llamalend, userAddress);
+            const [userCollateral, { collateral: currentCollateral }] = yield Promise.all([
+                _getUserCollateral(this.llamalend.constants.NETWORK_NAME, this.market.addresses.controller, userAddress),
+                this.market.userPosition.userState(userAddress),
+            ]);
+            const total_deposit_from_user = userCollateral.total_deposit_from_user_precise;
+            return calculateFutureLeverage(currentCollateral, total_deposit_from_user, collateral, 'remove');
+        });
+    }
+    // ---------------- REPAY ----------------
+    // TODO: add shrink: bool parameter
+    // If bool is true, _N = n2 - n1 + 1 -> _N = n2 - activeBand
+    _repayBands(debt, address) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _collateral: _currentCollateral, _borrowed, _debt: _currentDebt, _N } = yield this.market.userPosition.userStateBigInt(address);
+            if (_currentDebt === BigInt(0))
+                throw Error(`Loan for ${address} does not exist`);
+            // && shrink === false
+            if (_borrowed > BigInt(0))
+                return yield this.market.userPosition.userBandsBigInt(address);
+            // If shrink == true, then _debt = _currentDebt - parseUnits(debt, this.market.borrowed_token.decimals) - _borrowed
+            const _debt = _currentDebt - parseUnits(debt, this.market.borrowed_token.decimals);
+            const _n1 = yield this._calcN1(_currentCollateral, _debt, Number(_N));
+            const _n2 = _n1 + _N - BigInt(1);
+            return [_n2, _n1];
+        });
+    }
+    repayBands(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, address = "") {
+            const [_n2, _n1] = yield this._repayBands(debt, address);
+            return [Number(_n2), Number(_n1)];
+        });
+    }
+    repayPrices(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, address = "") {
+            const [_n2, _n1] = yield this._repayBands(debt, address);
+            return yield this.market.prices.getPrices(_n2, _n1);
+        });
+    }
+    repayIsApproved(debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield hasAllowance.call(this.llamalend, [this.market.borrowed_token.address], [debt], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    repayApproveEstimateGas(debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.borrowed_token.address], [debt], this.market.addresses.controller);
+        });
+    }
+    repayApprove(debt) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowance.call(this.llamalend, [this.market.borrowed_token.address], [debt], this.market.addresses.controller);
+        });
+    }
+    repayHealth(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, full = true, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const _debt = parseUnits(debt) * BigInt(-1);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            let _health = yield contract.health_calculator(address, 0, _debt, full, 0, this.llamalend.constantOptions);
+            _health = _health * BigInt(100);
+            return formatUnits(_health);
+        });
+    }
+    _repay(debt, address, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            address = _getAddress.call(this.llamalend, address);
+            const { debt: currentDebt } = yield this.market.userPosition.userState(address);
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${address} does not exist`);
+            const _debt = parseUnits(debt);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const [, n1] = yield this.market.userPosition.userBands(address);
+            const { borrowed } = yield this.market.userPosition.userState(address);
+            const n = (BN(borrowed).gt(0)) ? MAX_ACTIVE_BAND : n1 - 1; // In liquidation mode it doesn't matter if active band moves
+            const gas = yield contract.repay.estimateGas(_debt, address, n, this.llamalend.constantOptions);
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            // TODO: shrink parameter is added, which changes bands calculation
+            return (yield contract.repay(_debt, address, n, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    repayEstimateGas(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, address = "") {
+            if (!(yield this.repayIsApproved(debt)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._repay(debt, address, true);
+        });
+    }
+    repay(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, address = "") {
+            yield this.repayApprove(debt);
+            return yield this._repay(debt, address, false);
+        });
+    }
+    repayFutureLeverage(debt_1) {
+        return __awaiter(this, arguments, void 0, function* (debt, userAddress = '') {
+            userAddress = _getAddress.call(this.llamalend, userAddress);
+            const { stateCollateral, totalDepositFromUser } = yield this.market.userPosition.getCurrentLeverageParams(userAddress);
+            const collateralFromDebt = yield this.market.amm.swapExpected(0, 1, debt);
+            const futureCollateralState = BN(stateCollateral);
+            const futureTotalDepositFromUserPrecise = BN(totalDepositFromUser).plus(collateralFromDebt);
+            return futureCollateralState.div(futureTotalDepositFromUserPrecise).toString();
+        });
+    }
+    // ---------------- FULL REPAY ----------------
+    _fullRepayAmount() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            // TODO: now debt is _borrowed
+            const { debt } = yield this.market.userPosition.userState(address);
+            return BN(debt).times(1.0001).toString();
+        });
+    }
+    fullRepayIsApproved() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const fullRepayAmount = yield this._fullRepayAmount(address);
+            return yield this.repayIsApproved(fullRepayAmount);
+        });
+    }
+    fullRepayApproveEstimateGas() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const fullRepayAmount = yield this._fullRepayAmount(address);
+            return yield this.repayApproveEstimateGas(fullRepayAmount);
+        });
+    }
+    fullRepayApprove() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const fullRepayAmount = yield this._fullRepayAmount(address);
+            return yield this.repayApprove(fullRepayAmount);
+        });
+    }
+    fullRepayEstimateGas() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const fullRepayAmount = yield this._fullRepayAmount(address);
+            if (!(yield this.repayIsApproved(fullRepayAmount)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._repay(fullRepayAmount, address, true);
+        });
+    }
+    fullRepay() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const fullRepayAmount = yield this._fullRepayAmount(address);
+            yield this.repayApprove(fullRepayAmount);
+            return yield this._repay(fullRepayAmount, address, false);
+        });
+    }
+    // ---------------- LIQUIDATE ----------------
+    tokensToLiquidate() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const _tokens = yield this.llamalend.contracts[this.market.addresses.controller].contract.tokens_to_liquidate(address, this.llamalend.constantOptions);
+            return formatUnits(_tokens, this.market.borrowed_token.decimals);
+        });
+    }
+    calcPartialFrac(amount_1) {
+        return __awaiter(this, arguments, void 0, function* (amount, address = "") {
+            address = _getAddress.call(this.llamalend, address);
+            const tokensToLiquidate = yield this.tokensToLiquidate(address);
+            const amountBN = BN(amount);
+            const tokensToLiquidateBN = BN(tokensToLiquidate);
+            if (amountBN.gt(tokensToLiquidateBN))
+                throw Error("Amount cannot be greater than total tokens to liquidate");
+            if (amountBN.lte(0))
+                throw Error("Amount must be greater than 0");
+            // Calculate frac = amount / tokensToLiquidate * 10**18
+            // 100% = 10**18
+            const fracDecimalBN = amountBN.div(tokensToLiquidateBN);
+            const frac = fromBN(fracDecimalBN);
+            return {
+                frac: frac.toString(),
+                fracDecimal: fracDecimalBN.toString(),
+                amount: amountBN.toString(),
+            };
+        });
+    }
+    liquidateIsApproved() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            const tokensToLiquidate = yield this.tokensToLiquidate(address);
+            return yield hasAllowance.call(this.llamalend, [this.market.addresses.borrowed_token], [tokensToLiquidate], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    liquidateApproveEstimateGas() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            const tokensToLiquidate = yield this.tokensToLiquidate(address);
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.addresses.borrowed_token], [tokensToLiquidate], this.market.addresses.controller);
+        });
+    }
+    liquidateApprove() {
+        return __awaiter(this, arguments, void 0, function* (address = "") {
+            const tokensToLiquidate = yield this.tokensToLiquidate(address);
+            return yield ensureAllowance.call(this.llamalend, [this.market.addresses.borrowed_token], [tokensToLiquidate], this.market.addresses.controller);
+        });
+    }
+    _liquidate(address, slippage, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { borrowed, debt: currentDebt } = yield this.market.userPosition.userState(address);
+            if (slippage <= 0)
+                throw Error("Slippage must be > 0");
+            if (slippage > 100)
+                throw Error("Slippage must be <= 100");
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${address} does not exist`);
+            if (Number(borrowed) === 0)
+                throw Error(`User ${address} is not in liquidation mode`);
+            const minAmountBN = BN(borrowed).times(100 - slippage).div(100);
+            const _minAmount = fromBN(minAmountBN);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = (yield contract.liquidate.estimateGas(address, _minAmount, this.llamalend.constantOptions));
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            return (yield contract.liquidate(address, _minAmount, Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    _partialLiquidate(address, partialFrac, slippage, estimateGas) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { borrowed, debt: currentDebt } = yield this.market.userPosition.userState(address);
+            if (slippage <= 0)
+                throw Error("Slippage must be > 0");
+            if (slippage > 100)
+                throw Error("Slippage must be <= 100");
+            if (Number(currentDebt) === 0)
+                throw Error(`Loan for ${address} does not exist`);
+            if (Number(borrowed) === 0)
+                throw Error(`User ${address} is not in liquidation mode`);
+            const frac = partialFrac.frac;
+            const fracBN = BN(partialFrac.fracDecimal);
+            const borrowedBN = BN(borrowed);
+            const expectedBorrowedBN = borrowedBN.times(fracBN);
+            const minAmountBN = expectedBorrowedBN.times(100 - slippage).div(100);
+            const _minAmount = fromBN(minAmountBN);
+            const contract = this.llamalend.contracts[this.market.addresses.controller].contract;
+            const gas = (yield contract.liquidate_extended.estimateGas(address, _minAmount, frac, this.llamalend.constants.ZERO_ADDRESS, [], this.llamalend.constantOptions));
+            if (estimateGas)
+                return smartNumber(gas);
+            yield this.llamalend.updateFeeData();
+            const gasLimit = _mulBy1_3(DIGas(gas));
+            return (yield contract.liquidate_extended(address, _minAmount, frac, this.llamalend.constants.ZERO_ADDRESS, [], Object.assign(Object.assign({}, this.llamalend.options), { gasLimit }))).hash;
+        });
+    }
+    liquidateEstimateGas(address_1) {
+        return __awaiter(this, arguments, void 0, function* (address, slippage = 0.1) {
+            if (!(yield this.liquidateIsApproved(address)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._liquidate(address, slippage, true);
+        });
+    }
+    liquidate(address_1) {
+        return __awaiter(this, arguments, void 0, function* (address, slippage = 0.1) {
+            yield this.liquidateApprove(address);
+            return yield this._liquidate(address, slippage, false);
+        });
+    }
+    // ---------------- SELF-LIQUIDATE ----------------
+    selfLiquidateIsApproved() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this.liquidateIsApproved();
+        });
+    }
+    selfLiquidateApproveEstimateGas() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.liquidateApproveEstimateGas();
+        });
+    }
+    selfLiquidateApprove() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this.liquidateApprove();
+        });
+    }
+    selfLiquidateEstimateGas() {
+        return __awaiter(this, arguments, void 0, function* (slippage = 0.1) {
+            if (!(yield this.selfLiquidateIsApproved()))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._liquidate(this.llamalend.signerAddress, slippage, true);
+        });
+    }
+    selfLiquidate() {
+        return __awaiter(this, arguments, void 0, function* (slippage = 0.1) {
+            yield this.selfLiquidateApprove();
+            return yield this._liquidate(this.llamalend.signerAddress, slippage, false);
+        });
+    }
+    // ---------------- PARTIAL SELF-LIQUIDATE ----------------
+    partialSelfLiquidateIsApproved(partialFrac) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield hasAllowance.call(this.llamalend, [this.market.addresses.borrowed_token], [partialFrac.amount], this.llamalend.signerAddress, this.market.addresses.controller);
+        });
+    }
+    partialSelfLiquidateApproveEstimateGas(partialFrac) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowanceEstimateGas.call(this.llamalend, [this.market.addresses.borrowed_token], [partialFrac.amount], this.market.addresses.controller);
+        });
+    }
+    partialSelfLiquidateApprove(partialFrac) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield ensureAllowance.call(this.llamalend, [this.market.addresses.borrowed_token], [partialFrac.amount], this.market.addresses.controller);
+        });
+    }
+    partialSelfLiquidateEstimateGas(partialFrac_1) {
+        return __awaiter(this, arguments, void 0, function* (partialFrac, slippage = 0.1) {
+            if (!(yield this.partialSelfLiquidateIsApproved(partialFrac)))
+                throw Error("Approval is needed for gas estimation");
+            return yield this._partialLiquidate(this.llamalend.signerAddress, partialFrac, slippage, true);
+        });
+    }
+    partialSelfLiquidate(partialFrac_1) {
+        return __awaiter(this, arguments, void 0, function* (partialFrac, slippage = 0.1) {
+            yield this.partialSelfLiquidateApprove(partialFrac);
+            return yield this._partialLiquidate(this.llamalend.signerAddress, partialFrac, slippage, false);
+        });
+    }
+}