@drift-labs/sdk 0.1.23-master.2 → 0.1.24

package/lib/clearingHouseUser.js

@@ -10,6 +10,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ClearingHouseUser = void 0;
+const types_1 = require("./types");
 const position_1 = require("./math/position");
 const numericConstants_1 = require("./constants/numericConstants");
 const _1 = require(".");
@@ -38,8 +39,8 @@ class ClearingHouseUser {
     static from(clearingHouse, authority) {
         if (clearingHouse.accountSubscriber.type !== 'websocket')
             throw 'This method only works for clearing houses with a websocket account listener. Try using the getClearingHouseUser factory method to initialize a ClearingHouseUser instead';
-        const config = (0, clearingHouseUser_1.getWebSocketClearingHouseUserConfig)(clearingHouse, authority);
-        return (0, clearingHouseUser_1.getClearingHouseUser)(config);
+        const config = clearingHouseUser_1.getWebSocketClearingHouseUserConfig(clearingHouse, authority);
+        return clearingHouseUser_1.getClearingHouseUser(config);
     }
     /**
      * Subscribe to ClearingHouseUser state accounts
@@ -112,7 +113,7 @@ class ClearingHouseUser {
             if (this.userAccountPublicKey) {
                 return this.userAccountPublicKey;
             }
-            this.userAccountPublicKey = yield (0, addresses_1.getUserAccountPublicKey)(this.clearingHouse.program.programId, this.authority);
+            this.userAccountPublicKey = yield addresses_1.getUserAccountPublicKey(this.clearingHouse.program.programId, this.authority);
             return this.userAccountPublicKey;
         });
     }
@@ -121,7 +122,7 @@ class ClearingHouseUser {
             if (this.userOrdersAccountPublicKey) {
                 return this.userOrdersAccountPublicKey;
             }
-            this.userOrdersAccountPublicKey = yield (0, _1.getUserOrdersAccountPublicKey)(this.clearingHouse.program.programId, yield this.getUserAccountPublicKey());
+            this.userOrdersAccountPublicKey = yield _1.getUserOrdersAccountPublicKey(this.clearingHouse.program.programId, yield this.getUserAccountPublicKey());
             return this.userOrdersAccountPublicKey;
         });
     }
@@ -136,19 +137,39 @@ class ClearingHouseUser {
      * calculates Buying Power = FC * MAX_LEVERAGE
      * @returns : Precision QUOTE_PRECISION
      */
-    getBuyingPower() {
+    getBuyingPower(marketIndex) {
         return this.getFreeCollateral()
-            .mul(this.getMaxLeverage('Initial'))
+            .mul(this.getMaxLeverage(marketIndex, 'Initial'))
             .div(numericConstants_1.TEN_THOUSAND);
     }
     /**
-     * calculates Free Collateral = (TC - TPV) * MAX_LEVERAGE
+     * calculates Free Collateral = Total collateral - initial margin requirement
      * @returns : Precision QUOTE_PRECISION
      */
     getFreeCollateral() {
-        return this.getTotalCollateral().sub(this.getTotalPositionValue()
-            .mul(numericConstants_1.TEN_THOUSAND)
-            .div(this.getMaxLeverage('Initial')));
+        const totalCollateral = this.getTotalCollateral();
+        const initialMarginRequirement = this.getInitialMarginRequirement();
+        const freeCollateral = totalCollateral.sub(initialMarginRequirement);
+        return freeCollateral.gte(numericConstants_1.ZERO) ? freeCollateral : numericConstants_1.ZERO;
+    }
+    getInitialMarginRequirement() {
+        return this.getUserPositionsAccount().positions.reduce((marginRequirement, marketPosition) => {
+            const market = this.clearingHouse.getMarket(marketPosition.marketIndex);
+            return marginRequirement.add(_1.calculateBaseAssetValue(market, marketPosition)
+                .mul(new _1.BN(market.marginRatioInitial))
+                .div(numericConstants_1.MARGIN_PRECISION));
+        }, numericConstants_1.ZERO);
+    }
+    /**
+     * @returns The partial margin requirement in USDC. : QUOTE_PRECISION
+     */
+    getPartialMarginRequirement() {
+        return this.getUserPositionsAccount().positions.reduce((marginRequirement, marketPosition) => {
+            const market = this.clearingHouse.getMarket(marketPosition.marketIndex);
+            return marginRequirement.add(_1.calculateBaseAssetValue(market, marketPosition)
+                .mul(new _1.BN(market.marginRatioPartial))
+                .div(numericConstants_1.MARGIN_PRECISION));
+        }, numericConstants_1.ZERO);
     }
     /**
      * calculates unrealized position price pnl
@@ -159,7 +180,7 @@ class ClearingHouseUser {
             .positions.filter((pos) => marketIndex ? pos.marketIndex === marketIndex : true)
             .reduce((pnl, marketPosition) => {
             const market = this.clearingHouse.getMarket(marketPosition.marketIndex);
-            return pnl.add((0, _1.calculatePositionPNL)(market, marketPosition, withFunding));
+            return pnl.add(_1.calculatePositionPNL(market, marketPosition, withFunding));
         }, numericConstants_1.ZERO);
     }
     /**
@@ -171,7 +192,7 @@ class ClearingHouseUser {
             .positions.filter((pos) => marketIndex ? pos.marketIndex === marketIndex : true)
             .reduce((pnl, marketPosition) => {
             const market = this.clearingHouse.getMarket(marketPosition.marketIndex);
-            return pnl.add((0, _1.calculatePositionFundingPNL)(market, marketPosition));
+            return pnl.add(_1.calculatePositionFundingPNL(market, marketPosition));
         }, numericConstants_1.ZERO);
     }
     /**
@@ -189,7 +210,7 @@ class ClearingHouseUser {
     getTotalPositionValue() {
         return this.getUserPositionsAccount().positions.reduce((positionValue, marketPosition) => {
             const market = this.clearingHouse.getMarket(marketPosition.marketIndex);
-            return positionValue.add((0, _1.calculateBaseAssetValue)(market, marketPosition));
+            return positionValue.add(_1.calculateBaseAssetValue(market, marketPosition));
         }, numericConstants_1.ZERO);
     }
     /**
@@ -199,7 +220,7 @@ class ClearingHouseUser {
     getPositionValue(marketIndex) {
         const userPosition = this.getUserPosition(marketIndex) || this.getEmptyPosition(marketIndex);
         const market = this.clearingHouse.getMarket(userPosition.marketIndex);
-        return (0, _1.calculateBaseAssetValue)(market, userPosition);
+        return _1.calculateBaseAssetValue(market, userPosition);
     }
     getPositionSide(currentPosition) {
         if (currentPosition.baseAssetAmount.gt(numericConstants_1.ZERO)) {
@@ -218,10 +239,10 @@ class ClearingHouseUser {
      */
     getPositionEstimatedExitPriceAndPnl(position, amountToClose) {
         const market = this.clearingHouse.getMarket(position.marketIndex);
-        const entryPrice = (0, position_1.calculateEntryPrice)(position);
+        const entryPrice = position_1.calculateEntryPrice(position);
         if (amountToClose) {
             if (amountToClose.eq(numericConstants_1.ZERO)) {
-                return [(0, _1.calculateMarkPrice)(market), numericConstants_1.ZERO];
+                return [_1.calculateMarkPrice(market), numericConstants_1.ZERO];
             }
             position = {
                 baseAssetAmount: amountToClose,
@@ -230,7 +251,7 @@ class ClearingHouseUser {
                 quoteAssetAmount: position.quoteAssetAmount,
             };
         }
-        const baseAssetValue = (0, _1.calculateBaseAssetValue)(market, position);
+        const baseAssetValue = _1.calculateBaseAssetValue(market, position);
         if (position.baseAssetAmount.eq(numericConstants_1.ZERO)) {
             return [numericConstants_1.ZERO, numericConstants_1.ZERO];
         }
@@ -262,24 +283,24 @@ class ClearingHouseUser {
      * @params category {Initial, Partial, Maintenance}
      * @returns : Precision TEN_THOUSAND
      */
-    getMaxLeverage(category) {
-        const chState = this.clearingHouse.getStateAccount();
+    getMaxLeverage(marketIndex, category = 'Initial') {
+        const market = this.clearingHouse.getMarket(marketIndex);
         let marginRatioCategory;
         switch (category) {
             case 'Initial':
-                marginRatioCategory = chState.marginRatioInitial;
+                marginRatioCategory = market.marginRatioInitial;
                 break;
             case 'Maintenance':
-                marginRatioCategory = chState.marginRatioMaintenance;
+                marginRatioCategory = market.marginRatioMaintenance;
                 break;
             case 'Partial':
-                marginRatioCategory = chState.marginRatioPartial;
+                marginRatioCategory = market.marginRatioPartial;
                 break;
             default:
-                marginRatioCategory = chState.marginRatioInitial;
+                marginRatioCategory = market.marginRatioInitial;
                 break;
         }
-        const maxLeverage = numericConstants_1.TEN_THOUSAND.mul(numericConstants_1.TEN_THOUSAND).div(marginRatioCategory);
+        const maxLeverage = numericConstants_1.TEN_THOUSAND.mul(numericConstants_1.TEN_THOUSAND).div(new _1.BN(marginRatioCategory));
         return maxLeverage;
     }
     /**
@@ -294,8 +315,10 @@ class ClearingHouseUser {
         return this.getTotalCollateral().mul(numericConstants_1.TEN_THOUSAND).div(totalPositionValue);
     }
     canBeLiquidated() {
+        const totalCollateral = this.getTotalCollateral();
+        const partialMaintenanceRequirement = this.getPartialMarginRequirement();
         const marginRatio = this.getMarginRatio();
-        const canLiquidate = marginRatio.lte(numericConstants_1.PARTIAL_LIQUIDATION_RATIO);
+        const canLiquidate = totalCollateral.lt(partialMaintenanceRequirement);
         return [canLiquidate, marginRatio];
     }
     /**
@@ -319,97 +342,13 @@ class ClearingHouseUser {
     }
     /**
      * Calculate the liquidation price of a position, with optional parameter to calculate the liquidation price after a trade
-     * @param targetMarket
-     * @param positionBaseSizeChange // change in position size to calculate liquidation price for : Precision 10^13
-     * @param partial
-     * @returns Precision : MARK_PRICE_PRECISION
-     */
-    liquidationPriceOld(targetMarket, positionBaseSizeChange = numericConstants_1.ZERO, partial = false) {
-        // +/-(margin_ratio-liq_ratio) * price_now = price_liq
-        // todo: margin_ratio is not symmetric on price action (both numer and denom change)
-        // margin_ratio = collateral / base_asset_value
-        /* example: assume BTC price is $40k (examine 10% up/down)
-        
-        if 10k deposit and levered 10x short BTC => BTC up $400 means:
-        1. higher base_asset_value (+$4k)
-        2. lower collateral (-$4k)
-        3. (10k - 4k)/(100k + 4k) => 6k/104k => .0576
-
-        for 10x long, BTC down $400:
-        3. (10k - 4k) / (100k - 4k) = 6k/96k => .0625 */
-        const currentPrice = (0, _1.calculateMarkPrice)(this.clearingHouse.getMarket(targetMarket.marketIndex));
-        const totalCollateralUSDC = this.getTotalCollateral();
-        // calculate the total position value ignoring any value from the target market of the trade
-        const totalCurrentPositionValueIgnoringTargetUSDC = this.getTotalPositionValueExcludingMarket(targetMarket.marketIndex);
-        const currentMarketPosition = this.getUserPosition(targetMarket.marketIndex) ||
-            this.getEmptyPosition(targetMarket.marketIndex);
-        const currentMarketPositionBaseSize = currentMarketPosition.baseAssetAmount;
-        // calculate position for current market after trade
-        const proposedMarketPosition = {
-            marketIndex: targetMarket.marketIndex,
-            baseAssetAmount: currentMarketPositionBaseSize.add(positionBaseSizeChange),
-            lastCumulativeFundingRate: new _1.BN(0),
-            quoteAssetAmount: new _1.BN(0),
-            openOrders: new _1.BN(0),
-        };
-        const market = this.clearingHouse.getMarket(proposedMarketPosition.marketIndex);
-        const proposedMarketPositionValueUSDC = (0, _1.calculateBaseAssetValue)(market, proposedMarketPosition);
-        // total position value after trade
-        const targetTotalPositionValueUSDC = totalCurrentPositionValueIgnoringTargetUSDC.add(proposedMarketPositionValueUSDC);
-        let totalFreeCollateralUSDC = this.getTotalCollateral().sub(this.getTotalPositionValue()
-            .mul(numericConstants_1.TEN_THOUSAND)
-            .div(this.getMaxLeverage('Maintenance')));
-        if (partial) {
-            totalFreeCollateralUSDC = this.getTotalCollateral().sub(this.getTotalPositionValue()
-                .mul(numericConstants_1.TEN_THOUSAND)
-                .div(this.getMaxLeverage('Partial')));
-        }
-        // if the position value after the trade is less than total collateral, there is no liq price
-        if (targetTotalPositionValueUSDC.lte(totalFreeCollateralUSDC) &&
-            proposedMarketPosition.baseAssetAmount.gt(numericConstants_1.ZERO)) {
-            return new _1.BN(-1);
-        }
-        // get current margin ratio based on current collateral and proposed total position value
-        let marginRatio;
-        if (proposedMarketPositionValueUSDC.eq(numericConstants_1.ZERO)) {
-            marginRatio = numericConstants_1.BN_MAX;
-        }
-        else {
-            marginRatio = totalCollateralUSDC
-                .mul(numericConstants_1.TEN_THOUSAND)
-                .div(proposedMarketPositionValueUSDC);
-        }
-        let liqRatio = numericConstants_1.FULL_LIQUIDATION_RATIO;
-        if (partial) {
-            liqRatio = numericConstants_1.PARTIAL_LIQUIDATION_RATIO;
-        }
-        // sign of position in current market after the trade
-        const baseAssetSignIsNeg = proposedMarketPosition.baseAssetAmount.isNeg();
-        let pctChange = marginRatio.abs().sub(liqRatio);
-        // if user is short, higher price is liq
-        if (baseAssetSignIsNeg) {
-            pctChange = pctChange.add(numericConstants_1.TEN_THOUSAND);
-        }
-        else {
-            if (numericConstants_1.TEN_THOUSAND.lte(pctChange)) {
-                // no liquidation price, position is a fully/over collateralized long
-                // handle as NaN on UI
-                return new _1.BN(-1);
-            }
-            pctChange = numericConstants_1.TEN_THOUSAND.sub(pctChange);
-        }
-        const liqPrice = currentPrice.mul(pctChange).div(numericConstants_1.TEN_THOUSAND);
-        return liqPrice;
-    }
-    /**
-     * Calculate the liquidation price of a position, with optional parameter to calculate the liquidation price after a trade
-     * @param targetMarket
+     * @param marketPosition
      * @param positionBaseSizeChange // change in position size to calculate liquidation price for : Precision 10^13
      * @param partial
      * @returns Precision : MARK_PRICE_PRECISION
      */
-    liquidationPrice(targetMarket, positionBaseSizeChange = numericConstants_1.ZERO, partial = false) {
-        // solves formula for example calc below
+    liquidationPrice(marketPosition, positionBaseSizeChange = numericConstants_1.ZERO, partial = false) {
+        // solves formula for example canBeLiquidated below
         /* example: assume BTC price is $40k (examine 10% up/down)
 
         if 10k deposit and levered 10x short BTC => BTC up $400 means:
@@ -419,77 +358,86 @@ class ClearingHouseUser {
 
         for 10x long, BTC down $400:
         3. (10k - 4k) / (100k - 4k) = 6k/96k => .0625 */
-        const tc = this.getTotalCollateral();
-        const tpv = this.getTotalPositionValue();
-        const partialLev = 16;
-        const maintLev = 20;
-        const thisLev = partial ? new _1.BN(partialLev) : new _1.BN(maintLev);
+        const totalCollateral = this.getTotalCollateral();
         // calculate the total position value ignoring any value from the target market of the trade
-        const totalCurrentPositionValueIgnoringTargetUSDC = this.getTotalPositionValueExcludingMarket(targetMarket.marketIndex);
-        const currentMarketPosition = this.getUserPosition(targetMarket.marketIndex) ||
-            this.getEmptyPosition(targetMarket.marketIndex);
+        const totalPositionValueExcludingTargetMarket = this.getTotalPositionValueExcludingMarket(marketPosition.marketIndex);
+        const currentMarketPosition = this.getUserPosition(marketPosition.marketIndex) ||
+            this.getEmptyPosition(marketPosition.marketIndex);
         const currentMarketPositionBaseSize = currentMarketPosition.baseAssetAmount;
         const proposedBaseAssetAmount = currentMarketPositionBaseSize.add(positionBaseSizeChange);
         // calculate position for current market after trade
         const proposedMarketPosition = {
-            marketIndex: targetMarket.marketIndex,
+            marketIndex: marketPosition.marketIndex,
             baseAssetAmount: proposedBaseAssetAmount,
             lastCumulativeFundingRate: currentMarketPosition.lastCumulativeFundingRate,
             quoteAssetAmount: new _1.BN(0),
             openOrders: new _1.BN(0),
         };
+        if (proposedBaseAssetAmount.eq(numericConstants_1.ZERO))
+            return new _1.BN(-1);
         const market = this.clearingHouse.getMarket(proposedMarketPosition.marketIndex);
-        const proposedMarketPositionValueUSDC = (0, _1.calculateBaseAssetValue)(market, proposedMarketPosition);
+        const proposedMarketPositionValue = _1.calculateBaseAssetValue(market, proposedMarketPosition);
         // total position value after trade
-        const targetTotalPositionValueUSDC = totalCurrentPositionValueIgnoringTargetUSDC.add(proposedMarketPositionValueUSDC);
-        let totalFreeCollateralUSDC = tc.sub(totalCurrentPositionValueIgnoringTargetUSDC
-            .mul(numericConstants_1.TEN_THOUSAND)
-            .div(this.getMaxLeverage('Maintenance')));
-        if (partial) {
-            totalFreeCollateralUSDC = tc.sub(totalCurrentPositionValueIgnoringTargetUSDC
-                .mul(numericConstants_1.TEN_THOUSAND)
-                .div(this.getMaxLeverage('Partial')));
+        const totalPositionValueAfterTrade = totalPositionValueExcludingTargetMarket.add(proposedMarketPositionValue);
+        const marginRequirementExcludingTargetMarket = this.getUserPositionsAccount().positions.reduce((totalMarginRequirement, position) => {
+            if (!position.marketIndex.eq(marketPosition.marketIndex)) {
+                const market = this.clearingHouse.getMarket(position.marketIndex);
+                const positionValue = _1.calculateBaseAssetValue(market, position);
+                const marketMarginRequirement = positionValue
+                    .mul(partial
+                    ? new _1.BN(market.marginRatioPartial)
+                    : new _1.BN(market.marginRatioMaintenance))
+                    .div(numericConstants_1.MARGIN_PRECISION);
+                totalMarginRequirement = totalMarginRequirement.add(marketMarginRequirement);
+            }
+            return totalMarginRequirement;
+        }, numericConstants_1.ZERO);
+        const freeCollateralExcludingTargetMarket = totalCollateral.sub(marginRequirementExcludingTargetMarket);
+        // if the position value after the trade is less than free collateral, there is no liq price
+        if (totalPositionValueAfterTrade.lte(freeCollateralExcludingTargetMarket) &&
+            proposedMarketPosition.baseAssetAmount.abs().gt(numericConstants_1.ZERO)) {
+            return new _1.BN(-1);
         }
-        let priceDelt;
+        const marginRequirementAfterTrade = marginRequirementExcludingTargetMarket.add(proposedMarketPositionValue
+            .mul(partial
+            ? new _1.BN(market.marginRatioPartial)
+            : new _1.BN(market.marginRatioMaintenance))
+            .div(numericConstants_1.MARGIN_PRECISION));
+        const freeCollateralAfterTrade = totalCollateral.sub(marginRequirementAfterTrade);
+        const marketMaxLeverage = partial
+            ? this.getMaxLeverage(proposedMarketPosition.marketIndex, 'Partial')
+            : this.getMaxLeverage(proposedMarketPosition.marketIndex, 'Maintenance');
+        let priceDelta;
         if (proposedBaseAssetAmount.lt(numericConstants_1.ZERO)) {
-            priceDelt = tc
-                .mul(thisLev)
-                .sub(tpv)
+            priceDelta = freeCollateralAfterTrade
+                .mul(marketMaxLeverage) // precision is TEN_THOUSAND
+                .div(marketMaxLeverage.add(numericConstants_1.TEN_THOUSAND))
                 .mul(numericConstants_1.PRICE_TO_QUOTE_PRECISION)
-                .div(thisLev.add(new _1.BN(1)));
+                .mul(numericConstants_1.AMM_RESERVE_PRECISION)
+                .div(proposedBaseAssetAmount);
         }
         else {
-            priceDelt = tc
-                .mul(thisLev)
-                .sub(tpv)
+            priceDelta = freeCollateralAfterTrade
+                .mul(marketMaxLeverage) // precision is TEN_THOUSAND
+                .div(marketMaxLeverage.sub(numericConstants_1.TEN_THOUSAND))
                 .mul(numericConstants_1.PRICE_TO_QUOTE_PRECISION)
-                .div(thisLev.sub(new _1.BN(1)));
+                .mul(numericConstants_1.AMM_RESERVE_PRECISION)
+                .div(proposedBaseAssetAmount);
         }
-        let currentPrice;
+        let markPriceAfterTrade;
         if (positionBaseSizeChange.eq(numericConstants_1.ZERO)) {
-            currentPrice = (0, _1.calculateMarkPrice)(this.clearingHouse.getMarket(targetMarket.marketIndex));
+            markPriceAfterTrade = _1.calculateMarkPrice(this.clearingHouse.getMarket(marketPosition.marketIndex));
         }
         else {
             const direction = positionBaseSizeChange.gt(numericConstants_1.ZERO)
                 ? _1.PositionDirection.LONG
                 : _1.PositionDirection.SHORT;
-            currentPrice = (0, _1.calculateTradeSlippage)(direction, positionBaseSizeChange.abs(), this.clearingHouse.getMarket(targetMarket.marketIndex), 'base')[3]; // newPrice after swap
-        }
-        // if the position value after the trade is less than total collateral, there is no liq price
-        if (targetTotalPositionValueUSDC.lte(totalFreeCollateralUSDC) &&
-            proposedMarketPosition.baseAssetAmount.gt(numericConstants_1.ZERO)) {
-            return new _1.BN(-1);
+            markPriceAfterTrade = _1.calculateTradeSlippage(direction, positionBaseSizeChange.abs(), this.clearingHouse.getMarket(marketPosition.marketIndex), 'base')[3]; // newPrice after swap
         }
-        if (proposedBaseAssetAmount.eq(numericConstants_1.ZERO))
-            return new _1.BN(-1);
-        const eatMargin2 = priceDelt
-            .mul(numericConstants_1.AMM_RESERVE_PRECISION)
-            .div(proposedBaseAssetAmount);
-        if (eatMargin2.gt(currentPrice)) {
+        if (priceDelta.gt(markPriceAfterTrade)) {
             return new _1.BN(-1);
         }
-        const liqPrice = currentPrice.sub(eatMargin2);
-        return liqPrice;
+        return markPriceAfterTrade.sub(priceDelta);
     }
     /**
      * Calculates the estimated liquidation price for a position after closing a quote amount of the position.
@@ -530,33 +478,24 @@ class ClearingHouseUser {
      *
      * @param targetMarketIndex
      * @param tradeSide
-     * @param userMaxLeverageSetting - leverage : Precision TEN_THOUSAND
      * @returns tradeSizeAllowed : Precision QUOTE_PRECISION
      */
-    getMaxTradeSizeUSDC(targetMarketIndex, tradeSide, userMaxLeverageSetting) {
+    getMaxTradeSizeUSDC(targetMarketIndex, tradeSide) {
         const currentPosition = this.getUserPosition(targetMarketIndex) ||
             this.getEmptyPosition(targetMarketIndex);
-        const targetSide = tradeSide === _1.PositionDirection.SHORT ? 'short' : 'long';
+        const targetSide = types_1.isVariant(tradeSide, 'short') ? 'short' : 'long';
         const currentPositionSide = (currentPosition === null || currentPosition === void 0 ? void 0 : currentPosition.baseAssetAmount.isNeg())
             ? 'short'
             : 'long';
-        const targettingSameSide = !currentPosition
+        const targetingSameSide = !currentPosition
             ? true
             : targetSide === currentPositionSide;
         // add any position we have on the opposite side of the current trade, because we can "flip" the size of this position without taking any extra leverage.
-        const oppositeSizeValueUSDC = targettingSameSide
+        const oppositeSizeValueUSDC = targetingSameSide
             ? numericConstants_1.ZERO
             : this.getPositionValue(targetMarketIndex);
-        // get current leverage
-        const currentLeverage = this.getLeverage();
-        const remainingLeverage = _1.BN.max(userMaxLeverageSetting.sub(currentLeverage), numericConstants_1.ZERO);
-        // get total collateral
-        const totalCollateral = this.getTotalCollateral();
-        // position side allowed based purely on current leverage
-        let maxPositionSize = remainingLeverage
-            .mul(totalCollateral)
-            .div(numericConstants_1.TEN_THOUSAND);
-        if (userMaxLeverageSetting.sub(currentLeverage).gte(numericConstants_1.ZERO)) {
+        let maxPositionSize = this.getBuyingPower(targetMarketIndex);
+        if (maxPositionSize.gte(numericConstants_1.ZERO)) {
             if (oppositeSizeValueUSDC.eq(numericConstants_1.ZERO)) {
                 // case 1 : Regular trade where current total position less than max, and no opposite position to account for
                 // do nothing
@@ -568,25 +507,24 @@ class ClearingHouseUser {
         }
         else {
             // current leverage is greater than max leverage - can only reduce position size
-            if (!targettingSameSide) {
-                const currentPositionQuoteSize = this.getPositionValue(targetMarketIndex);
-                const currentTotalQuoteSize = currentLeverage
-                    .mul(totalCollateral)
-                    .div(numericConstants_1.TEN_THOUSAND);
-                const otherPositionsTotalQuoteSize = currentTotalQuoteSize.sub(currentPositionQuoteSize);
-                const quoteValueOfMaxLeverage = userMaxLeverageSetting
-                    .mul(totalCollateral)
-                    .div(numericConstants_1.TEN_THOUSAND);
-                if (otherPositionsTotalQuoteSize
-                    .sub(currentPositionQuoteSize)
-                    .gte(quoteValueOfMaxLeverage)) {
-                    // case 3: Can only reduce the current position because it will still be greater than max leverage
-                    maxPositionSize = currentPositionQuoteSize;
+            if (!targetingSameSide) {
+                const market = this.clearingHouse.getMarket(targetMarketIndex);
+                const marketPositionValue = this.getPositionValue(targetMarketIndex);
+                const totalCollateral = this.getTotalCollateral();
+                const marginRequirement = this.getInitialMarginRequirement();
+                const marginFreedByClosing = marketPositionValue
+                    .mul(new _1.BN(market.marginRatioInitial))
+                    .div(numericConstants_1.MARGIN_PRECISION);
+                const marginRequirementAfterClosing = marginRequirement.sub(marginFreedByClosing);
+                if (marginRequirementAfterClosing.gt(totalCollateral)) {
+                    maxPositionSize = marketPositionValue;
                 }
                 else {
-                    // case 4: Can reduce the position, and then take extra remaining quote to get to max leverage
-                    const allowedQuoteSizeAfterClosingCurrentPosition = quoteValueOfMaxLeverage.sub(otherPositionsTotalQuoteSize);
-                    maxPositionSize = currentPositionQuoteSize.add(allowedQuoteSizeAfterClosingCurrentPosition);
+                    const freeCollateralAfterClose = totalCollateral.sub(marginRequirementAfterClosing);
+                    const buyingPowerAfterClose = freeCollateralAfterClose
+                        .mul(this.getMaxLeverage(targetMarketIndex))
+                        .div(numericConstants_1.TEN_THOUSAND);
+                    maxPositionSize = marketPositionValue.add(buyingPowerAfterClose);
                 }
             }
             else {
@@ -659,46 +597,5 @@ class ClearingHouseUser {
         }
         return this.getTotalPositionValue().sub(currentMarketPositionValueUSDC);
     }
-    canFillOrder(order) {
-        const userAccount = this.getUserAccount();
-        const userPositionsAccount = this.getUserPositionsAccount();
-        const userPosition = this.getUserPosition(order.marketIndex);
-        const market = this.clearingHouse.getMarket(order.marketIndex);
-        if ((0, position_1.isEmptyPosition)(userPosition)) {
-            return false;
-        }
-        const newState = (0, _1.calculateNewStateAfterOrder)(userAccount, userPosition, market, order);
-        if (newState === null) {
-            return false;
-        }
-        const [userAccountAfter, userPositionAfter, marketAfter] = newState;
-        const totalPositionValue = userPositionsAccount.positions.reduce((positionValue, marketPosition) => {
-            let market = this.clearingHouse.getMarket(marketPosition.marketIndex);
-            if (marketPosition.marketIndex.eq(order.marketIndex)) {
-                market = marketAfter;
-                marketPosition = userPositionAfter;
-            }
-            return positionValue.add((0, _1.calculateBaseAssetValue)(market, marketPosition));
-        }, numericConstants_1.ZERO);
-        if (totalPositionValue.eq(numericConstants_1.ZERO)) {
-            return true;
-        }
-        const unrealizedPnL = userPositionsAccount.positions.reduce((pnl, marketPosition) => {
-            let market = this.clearingHouse.getMarket(marketPosition.marketIndex);
-            pnl = pnl.add((0, _1.calculatePositionFundingPNL)(market, marketPosition).div(numericConstants_1.PRICE_TO_QUOTE_PRECISION));
-            if (marketPosition.marketIndex.eq(order.marketIndex)) {
-                market = marketAfter;
-                marketPosition = userPositionAfter;
-            }
-            // update
-            return pnl.add((0, _1.calculatePositionPNL)(market, marketPosition, false));
-        }, numericConstants_1.ZERO);
-        const totalCollateral = userAccountAfter.collateral.add(unrealizedPnL);
-        const marginRatioAfter = totalCollateral
-            .mul(numericConstants_1.TEN_THOUSAND)
-            .div(totalPositionValue);
-        const marginRatioInitial = this.clearingHouse.getStateAccount().marginRatioInitial;
-        return marginRatioAfter.gte(marginRatioInitial);
-    }
 }
 exports.ClearingHouseUser = ClearingHouseUser;

package/lib/constants/numericConstants.d.ts

@@ -6,8 +6,6 @@ export declare const TWO: BN;
 export declare const TEN_THOUSAND: BN;
 export declare const BN_MAX: BN;
 export declare const MAX_LEVERAGE: BN;
-export declare const FULL_LIQUIDATION_RATIO: BN;
-export declare const PARTIAL_LIQUIDATION_RATIO: BN;
 export declare const QUOTE_PRECISION: BN;
 export declare const MARK_PRICE_PRECISION: BN;
 export declare const FUNDING_PAYMENT_PRECISION: BN;
@@ -17,3 +15,4 @@ export declare const BASE_PRECISION: BN;
 export declare const AMM_TO_QUOTE_PRECISION_RATIO: BN;
 export declare const PRICE_TO_QUOTE_PRECISION: BN;
 export declare const AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO: BN;
+export declare const MARGIN_PRECISION: BN;

package/lib/constants/numericConstants.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO = exports.PRICE_TO_QUOTE_PRECISION = exports.AMM_TO_QUOTE_PRECISION_RATIO = exports.BASE_PRECISION = exports.AMM_RESERVE_PRECISION = exports.PEG_PRECISION = exports.FUNDING_PAYMENT_PRECISION = exports.MARK_PRICE_PRECISION = exports.QUOTE_PRECISION = exports.PARTIAL_LIQUIDATION_RATIO = exports.FULL_LIQUIDATION_RATIO = exports.MAX_LEVERAGE = exports.BN_MAX = exports.TEN_THOUSAND = exports.TWO = exports.ONE = exports.ZERO = void 0;
+exports.MARGIN_PRECISION = exports.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO = exports.PRICE_TO_QUOTE_PRECISION = exports.AMM_TO_QUOTE_PRECISION_RATIO = exports.BASE_PRECISION = exports.AMM_RESERVE_PRECISION = exports.PEG_PRECISION = exports.FUNDING_PAYMENT_PRECISION = exports.MARK_PRICE_PRECISION = exports.QUOTE_PRECISION = exports.MAX_LEVERAGE = exports.BN_MAX = exports.TEN_THOUSAND = exports.TWO = exports.ONE = exports.ZERO = void 0;
 const __1 = require("../");
 exports.ZERO = new __1.BN(0);
 exports.ONE = new __1.BN(1);
@@ -8,8 +8,6 @@ exports.TWO = new __1.BN(2);
 exports.TEN_THOUSAND = new __1.BN(10000);
 exports.BN_MAX = new __1.BN(Number.MAX_SAFE_INTEGER);
 exports.MAX_LEVERAGE = new __1.BN(5);
-exports.FULL_LIQUIDATION_RATIO = new __1.BN(500);
-exports.PARTIAL_LIQUIDATION_RATIO = new __1.BN(625);
 exports.QUOTE_PRECISION = new __1.BN(Math.pow(10, 6));
 exports.MARK_PRICE_PRECISION = new __1.BN(Math.pow(10, 10));
 exports.FUNDING_PAYMENT_PRECISION = new __1.BN(10000);
@@ -19,3 +17,4 @@ exports.BASE_PRECISION = exports.AMM_RESERVE_PRECISION;
 exports.AMM_TO_QUOTE_PRECISION_RATIO = exports.AMM_RESERVE_PRECISION.div(exports.QUOTE_PRECISION); // 10^7
 exports.PRICE_TO_QUOTE_PRECISION = exports.MARK_PRICE_PRECISION.div(exports.QUOTE_PRECISION);
 exports.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO = exports.AMM_RESERVE_PRECISION.mul(exports.PEG_PRECISION).div(exports.QUOTE_PRECISION); // 10^10
+exports.MARGIN_PRECISION = exports.TEN_THOUSAND;

package/lib/examples/makeTradeExample.js

@@ -21,7 +21,7 @@ const getTokenAddress = (mintAddress, userPubKey) => {
 exports.getTokenAddress = getTokenAddress;
 const main = () => __awaiter(void 0, void 0, void 0, function* () {
     // Initialize Drift SDK
-    const sdkConfig = (0, __2.initialize)({ env: 'devnet' });
+    const sdkConfig = __2.initialize({ env: 'devnet' });
     // Set up the Wallet and Provider
     const privateKey = process.env.BOT_PRIVATE_KEY; // stored as an array string
     const keypair = web3_js_1.Keypair.fromSecretKey(Uint8Array.from(JSON.parse(privateKey)));
@@ -35,7 +35,7 @@ const main = () => __awaiter(void 0, void 0, void 0, function* () {
     const lamportsBalance = yield connection.getBalance(wallet.publicKey);
     console.log('SOL balance:', lamportsBalance / Math.pow(10, 9));
     // Misc. other things to set up
-    const usdcTokenAddress = yield (0, exports.getTokenAddress)(sdkConfig.USDC_MINT_ADDRESS, wallet.publicKey.toString());
+    const usdcTokenAddress = yield exports.getTokenAddress(sdkConfig.USDC_MINT_ADDRESS, wallet.publicKey.toString());
     // Set up the Drift Clearing House
     const clearingHousePublicKey = new web3_js_1.PublicKey(sdkConfig.CLEARING_HOUSE_PROGRAM_ID);
     const clearingHouse = __2.ClearingHouse.from(connection, provider.wallet, clearingHousePublicKey);
@@ -47,18 +47,18 @@ const main = () => __awaiter(void 0, void 0, void 0, function* () {
     if (!userAccountExists) {
         //// Create a Clearing House account by Depositing some USDC ($10,000 in this case)
         const depositAmount = new anchor_1.BN(10000).mul(__2.QUOTE_PRECISION);
-        yield clearingHouse.initializeUserAccountAndDepositCollateral(depositAmount, yield (0, exports.getTokenAddress)(usdcTokenAddress.toString(), wallet.publicKey.toString()));
+        yield clearingHouse.initializeUserAccountAndDepositCollateral(depositAmount, yield exports.getTokenAddress(usdcTokenAddress.toString(), wallet.publicKey.toString()));
     }
     yield user.subscribe();
     // Get current price
     const solMarketInfo = __2.Markets.find((market) => market.baseAssetSymbol === 'SOL');
-    const currentMarketPrice = (0, __2.calculateMarkPrice)(clearingHouse.getMarket(solMarketInfo.marketIndex));
-    const formattedPrice = (0, __2.convertToNumber)(currentMarketPrice, __2.MARK_PRICE_PRECISION);
+    const currentMarketPrice = __2.calculateMarkPrice(clearingHouse.getMarket(solMarketInfo.marketIndex));
+    const formattedPrice = __2.convertToNumber(currentMarketPrice, __2.MARK_PRICE_PRECISION);
     console.log(`Current Market Price is $${formattedPrice}`);
     // Estimate the slippage for a $5000 LONG trade
     const solMarketAccount = clearingHouse.getMarket(solMarketInfo.marketIndex);
     const longAmount = new anchor_1.BN(5000).mul(__2.QUOTE_PRECISION);
-    const slippage = (0, __2.convertToNumber)((0, __2.calculateTradeSlippage)(__2.PositionDirection.LONG, longAmount, solMarketAccount)[0], __2.MARK_PRICE_PRECISION);
+    const slippage = __2.convertToNumber(__2.calculateTradeSlippage(__2.PositionDirection.LONG, longAmount, solMarketAccount)[0], __2.MARK_PRICE_PRECISION);
     console.log(`Slippage for a $5000 LONG on the SOL market would be $${slippage}`);
     // Make a $5000 LONG trade
     yield clearingHouse.openPosition(__2.PositionDirection.LONG, longAmount, solMarketInfo.marketIndex);