@sonarwatch/portfolio-plugins 0.12.36 → 0.12.38

package/src/plugins/drift/perpHelpers/position.js

@@ -0,0 +1,184 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.hasOpenOrders = exports.isEmptyPosition = exports.positionCurrentDirection = exports.findDirectionToClose = exports.calculateCostBasis = exports.calculateEntryPrice = exports.calculateBreakEvenPrice = exports.positionIsAvailable = exports.calculatePositionFundingPNL = exports.calculatePositionPNL = exports.calculateBaseAssetValue = void 0;
+const bn_js_1 = __importDefault(require("bn.js"));
+const constants_1 = require("./constants");
+const types_1 = require("./types");
+const amm_1 = require("./amm");
+const margin_1 = require("./margin");
+function calculateBaseAssetValue(market, userPosition, oraclePriceData, useSpread = true, skipUpdate = false) {
+    if (userPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return constants_1.ZERO;
+    }
+    const directionToClose = findDirectionToClose(userPosition);
+    let prepegAmm;
+    if (!skipUpdate) {
+        if (market.amm.baseSpread > 0 && useSpread) {
+            const { baseAssetReserve, quoteAssetReserve, sqrtK, newPeg } = (0, amm_1.calculateUpdatedAMMSpreadReserves)(market.amm, directionToClose, oraclePriceData);
+            prepegAmm = {
+                baseAssetReserve,
+                quoteAssetReserve,
+                sqrtK,
+                pegMultiplier: newPeg,
+            };
+        }
+        else {
+            prepegAmm = (0, amm_1.calculateUpdatedAMM)(market.amm, oraclePriceData);
+        }
+    }
+    else {
+        prepegAmm = market.amm;
+    }
+    const [newQuoteAssetReserve] = (0, amm_1.calculateAmmReservesAfterSwap)(prepegAmm, 'base', userPosition.baseAssetAmount.abs(), (0, amm_1.getSwapDirection)('base', directionToClose));
+    switch (directionToClose) {
+        case types_1.PositionDirection.SHORT:
+            return prepegAmm.quoteAssetReserve
+                .sub(newQuoteAssetReserve)
+                .mul(prepegAmm.pegMultiplier)
+                .div(constants_1.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO);
+        case types_1.PositionDirection.LONG:
+            return newQuoteAssetReserve
+                .sub(prepegAmm.quoteAssetReserve)
+                .mul(prepegAmm.pegMultiplier)
+                .div(constants_1.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO)
+                .add(constants_1.ONE);
+        default:
+            throw new Error('Failed to calculateBaseAssetValue');
+    }
+}
+exports.calculateBaseAssetValue = calculateBaseAssetValue;
+/**
+ * calculatePositionPNL
+ * = BaseAssetAmount * (Avg Exit Price - Avg Entry Price)
+ * @param market
+ * @param PerpPosition
+ * @param withFunding (adds unrealized funding payment pnl to result)
+ * @param oraclePriceData
+ * @returns BaseAssetAmount : Precision QUOTE_PRECISION
+ */
+function calculatePositionPNL(market, perpPosition, oraclePriceData, withFunding = false) {
+    if (perpPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return perpPosition.quoteAssetAmount;
+    }
+    const baseAssetValue = (0, margin_1.calculateBaseAssetValueWithOracle)(market, perpPosition, oraclePriceData);
+    const baseAssetValueSign = perpPosition.baseAssetAmount.isNeg()
+        ? new bn_js_1.default(-1)
+        : new bn_js_1.default(1);
+    let pnl = baseAssetValue
+        .mul(baseAssetValueSign)
+        .add(perpPosition.quoteAssetAmount);
+    if (withFunding) {
+        const fundingRatePnL = calculatePositionFundingPNL(market, perpPosition);
+        pnl = pnl.add(fundingRatePnL);
+    }
+    return pnl;
+}
+exports.calculatePositionPNL = calculatePositionPNL;
+/**
+ *
+ * @param market
+ * @param PerpPosition
+ * @returns // QUOTE_PRECISION
+ */
+function calculatePositionFundingPNL(market, perpPosition) {
+    if (perpPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return constants_1.ZERO;
+    }
+    let ammCumulativeFundingRate;
+    if (perpPosition.baseAssetAmount.gt(constants_1.ZERO)) {
+        ammCumulativeFundingRate = market.amm.cumulativeFundingRateLong;
+    }
+    else {
+        ammCumulativeFundingRate = market.amm.cumulativeFundingRateShort;
+    }
+    const perPositionFundingRate = ammCumulativeFundingRate
+        .sub(perpPosition.lastCumulativeFundingRate)
+        .mul(perpPosition.baseAssetAmount)
+        .div(constants_1.AMM_RESERVE_PRECISION)
+        .div(constants_1.FUNDING_RATE_BUFFER_PRECISION)
+        .mul(new bn_js_1.default(-1));
+    return perPositionFundingRate;
+}
+exports.calculatePositionFundingPNL = calculatePositionFundingPNL;
+function positionIsAvailable(position) {
+    return (position.baseAssetAmount.eq(constants_1.ZERO) &&
+        position.openOrders === 0 &&
+        position.quoteAssetAmount.eq(constants_1.ZERO) &&
+        position.lpShares.eq(constants_1.ZERO));
+}
+exports.positionIsAvailable = positionIsAvailable;
+/**
+ *
+ * @param userPosition
+ * @returns Precision: PRICE_PRECISION (10^6)
+ */
+function calculateBreakEvenPrice(userPosition) {
+    if (userPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return constants_1.ZERO;
+    }
+    return userPosition.quoteBreakEvenAmount
+        .mul(constants_1.PRICE_PRECISION)
+        .mul(constants_1.AMM_TO_QUOTE_PRECISION_RATIO)
+        .div(userPosition.baseAssetAmount)
+        .abs();
+}
+exports.calculateBreakEvenPrice = calculateBreakEvenPrice;
+/**
+ *
+ * @param userPosition
+ * @returns Precision: PRICE_PRECISION (10^6)
+ */
+function calculateEntryPrice(userPosition) {
+    if (userPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return constants_1.ZERO;
+    }
+    return userPosition.quoteEntryAmount
+        .mul(constants_1.PRICE_PRECISION)
+        .mul(constants_1.AMM_TO_QUOTE_PRECISION_RATIO)
+        .div(userPosition.baseAssetAmount)
+        .abs();
+}
+exports.calculateEntryPrice = calculateEntryPrice;
+/**
+ *
+ * @param userPosition
+ * @returns Precision: PRICE_PRECISION (10^10)
+ */
+function calculateCostBasis(userPosition, includeSettledPnl = false) {
+    if (userPosition.baseAssetAmount.eq(constants_1.ZERO)) {
+        return constants_1.ZERO;
+    }
+    return userPosition.quoteAssetAmount
+        .add(includeSettledPnl ? userPosition.settledPnl : constants_1.ZERO)
+        .mul(constants_1.PRICE_PRECISION)
+        .mul(constants_1.AMM_TO_QUOTE_PRECISION_RATIO)
+        .div(userPosition.baseAssetAmount)
+        .abs();
+}
+exports.calculateCostBasis = calculateCostBasis;
+function findDirectionToClose(userPosition) {
+    return userPosition.baseAssetAmount.gt(constants_1.ZERO)
+        ? types_1.PositionDirection.SHORT
+        : types_1.PositionDirection.LONG;
+}
+exports.findDirectionToClose = findDirectionToClose;
+function positionCurrentDirection(userPosition) {
+    return userPosition.baseAssetAmount.gte(constants_1.ZERO)
+        ? types_1.PositionDirection.LONG
+        : types_1.PositionDirection.SHORT;
+}
+exports.positionCurrentDirection = positionCurrentDirection;
+function isEmptyPosition(userPosition) {
+    return userPosition.baseAssetAmount.eq(constants_1.ZERO) && userPosition.openOrders === 0;
+}
+exports.isEmptyPosition = isEmptyPosition;
+function hasOpenOrders(position) {
+    return (position.openOrders !== 0 ||
+        !position.openBids.eq(constants_1.ZERO) ||
+        !position.openAsks.eq(constants_1.ZERO));
+}
+exports.hasOpenOrders = hasOpenOrders;
+//# sourceMappingURL=position.js.map

package/src/plugins/drift/perpHelpers/position.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"position.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/position.ts"],"names":[],"mappings":";;;;;;AAAA,kDAAuB;AAEvB,2CAQqB;AACrB,mCAA6D;AAC7D,+BAKe;AACf,qCAA6D;AAE7D,SAAgB,uBAAuB,CACrC,MAAkB,EAClB,YAA0B,EAC1B,eAAgC,EAChC,SAAS,GAAG,IAAI,EAChB,UAAU,GAAG,KAAK;IAElB,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,gBAAI,CAAC;IACd,CAAC;IAED,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAC5D,IAAI,SAA8D,CAAC;IAEnE,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,IAAI,MAAM,CAAC,GAAG,CAAC,UAAU,GAAG,CAAC,IAAI,SAAS,EAAE,CAAC;YAC3C,MAAM,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,GAC1D,IAAA,uCAAiC,EAC/B,MAAM,CAAC,GAAG,EACV,gBAAgB,EAChB,eAAe,CAChB,CAAC;YACJ,SAAS,GAAG;gBACV,gBAAgB;gBAChB,iBAAiB;gBACjB,KAAK;gBACL,aAAa,EAAE,MAAM;aACtB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,IAAA,yBAAmB,EAAC,MAAM,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,oBAAoB,CAAC,GAAG,IAAA,mCAA6B,EAC1D,SAAS,EACT,MAAM,EACN,YAAY,CAAC,eAAe,CAAC,GAAG,EAAE,EAClC,IAAA,sBAAgB,EAAC,MAAM,EAAE,gBAAgB,CAAC,CAC3C,CAAC;IAEF,QAAQ,gBAAgB,EAAE,CAAC;QACzB,KAAK,yBAAiB,CAAC,KAAK;YAC1B,OAAO,SAAS,CAAC,iBAAiB;iBAC/B,GAAG,CAAC,oBAAoB,CAAC;iBACzB,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;iBAC5B,GAAG,CAAC,kDAAsC,CAAC,CAAC;QAEjD,KAAK,yBAAiB,CAAC,IAAI;YACzB,OAAO,oBAAoB;iBACxB,GAAG,CAAC,SAAS,CAAC,iBAAiB,CAAC;iBAChC,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;iBAC5B,GAAG,CAAC,kDAAsC,CAAC;iBAC3C,GAAG,CAAC,eAAG,CAAC,CAAC;QAEd;YACE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AA3DD,0DA2DC;AAED;;;;;;;;GAQG;AACH,SAAgB,oBAAoB,CAClC,MAAkB,EAClB,YAA0B,EAC1B,eAAgC,EAChC,WAAW,GAAG,KAAK;IAEnB,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,YAAY,CAAC,gBAAgB,CAAC;IACvC,CAAC;IAED,MAAM,cAAc,GAAG,IAAA,0CAAiC,EACtD,MAAM,EACN,YAAY,EACZ,eAAe,CAChB,CAAC;IAEF,MAAM,kBAAkB,GAAG,YAAY,CAAC,eAAe,CAAC,KAAK,EAAE;QAC7D,CAAC,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC;QACZ,CAAC,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC;IACd,IAAI,GAAG,GAAG,cAAc;SACrB,GAAG,CAAC,kBAAkB,CAAC;SACvB,GAAG,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;IAEtC,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,cAAc,GAAG,2BAA2B,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAEzE,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AA9BD,oDA8BC;AAED;;;;;GAKG;AACH,SAAgB,2BAA2B,CACzC,MAAkB,EAClB,YAA0B;IAE1B,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,gBAAI,CAAC;IACd,CAAC;IAED,IAAI,wBAA4B,CAAC;IACjC,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,wBAAwB,GAAG,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC;IAClE,CAAC;SAAM,CAAC;QACN,wBAAwB,GAAG,MAAM,CAAC,GAAG,CAAC,0BAA0B,CAAC;IACnE,CAAC;IAED,MAAM,sBAAsB,GAAG,wBAAwB;SACpD,GAAG,CAAC,YAAY,CAAC,yBAAyB,CAAC;SAC3C,GAAG,CAAC,YAAY,CAAC,eAAe,CAAC;SACjC,GAAG,CAAC,iCAAqB,CAAC;SAC1B,GAAG,CAAC,yCAA6B,CAAC;SAClC,GAAG,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEnB,OAAO,sBAAsB,CAAC;AAChC,CAAC;AAvBD,kEAuBC;AAED,SAAgB,mBAAmB,CAAC,QAAsB;IACxD,OAAO,CACL,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC;QACjC,QAAQ,CAAC,UAAU,KAAK,CAAC;QACzB,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC,gBAAI,CAAC;QAClC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAI,CAAC,CAC3B,CAAC;AACJ,CAAC;AAPD,kDAOC;AAED;;;;GAIG;AACH,SAAgB,uBAAuB,CAAC,YAA0B;IAChE,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,gBAAI,CAAC;IACd,CAAC;IAED,OAAO,YAAY,CAAC,oBAAoB;SACrC,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,wCAA4B,CAAC;SACjC,GAAG,CAAC,YAAY,CAAC,eAAe,CAAC;SACjC,GAAG,EAAE,CAAC;AACX,CAAC;AAVD,0DAUC;AAED;;;;GAIG;AACH,SAAgB,mBAAmB,CAAC,YAA0B;IAC5D,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,gBAAI,CAAC;IACd,CAAC;IAED,OAAO,YAAY,CAAC,gBAAgB;SACjC,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,wCAA4B,CAAC;SACjC,GAAG,CAAC,YAAY,CAAC,eAAe,CAAC;SACjC,GAAG,EAAE,CAAC;AACX,CAAC;AAVD,kDAUC;AAED;;;;GAIG;AACH,SAAgB,kBAAkB,CAChC,YAA0B,EAC1B,iBAAiB,GAAG,KAAK;IAEzB,IAAI,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC1C,OAAO,gBAAI,CAAC;IACd,CAAC;IAED,OAAO,YAAY,CAAC,gBAAgB;SACjC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,gBAAI,CAAC;SACvD,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,wCAA4B,CAAC;SACjC,GAAG,CAAC,YAAY,CAAC,eAAe,CAAC;SACjC,GAAG,EAAE,CAAC;AACX,CAAC;AAdD,gDAcC;AAED,SAAgB,oBAAoB,CAClC,YAA0B;IAE1B,OAAO,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC;QAC1C,CAAC,CAAC,yBAAiB,CAAC,KAAK;QACzB,CAAC,CAAC,yBAAiB,CAAC,IAAI,CAAC;AAC7B,CAAC;AAND,oDAMC;AAED,SAAgB,wBAAwB,CACtC,YAA0B;IAE1B,OAAO,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,gBAAI,CAAC;QAC3C,CAAC,CAAC,yBAAiB,CAAC,IAAI;QACxB,CAAC,CAAC,yBAAiB,CAAC,KAAK,CAAC;AAC9B,CAAC;AAND,4DAMC;AAED,SAAgB,eAAe,CAAC,YAA0B;IACxD,OAAO,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC,gBAAI,CAAC,IAAI,YAAY,CAAC,UAAU,KAAK,CAAC,CAAC;AAChF,CAAC;AAFD,0CAEC;AAED,SAAgB,aAAa,CAAC,QAAsB;IAClD,OAAO,CACL,QAAQ,CAAC,UAAU,KAAK,CAAC;QACzB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAI,CAAC;QAC3B,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAI,CAAC,CAC5B,CAAC;AACJ,CAAC;AAND,sCAMC"}

package/src/plugins/drift/perpHelpers/prelaunchOracle.d.ts

@@ -0,0 +1,3 @@
+/// <reference types="node" />
+import { OraclePriceData } from './types';
+export declare function getPreLaunchOraclePriceDataFromBuffer(buffer: Buffer): OraclePriceData;

package/src/plugins/drift/perpHelpers/prelaunchOracle.js

@@ -0,0 +1,20 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getPreLaunchOraclePriceDataFromBuffer = void 0;
+const bn_js_1 = __importDefault(require("bn.js"));
+const struct_1 = require("../struct");
+function getPreLaunchOraclePriceDataFromBuffer(buffer) {
+    const [prelaunchOracle] = struct_1.preLaunchOracleStruct.deserialize(buffer);
+    return {
+        price: new bn_js_1.default(prelaunchOracle.price.toString(10), 10),
+        slot: new bn_js_1.default(prelaunchOracle.ammLastUpdateSlot.toString(10), 10),
+        confidence: new bn_js_1.default(prelaunchOracle.confidence.toString(10), 10),
+        hasSufficientNumberOfDataPoints: true,
+        maxPrice: new bn_js_1.default(prelaunchOracle.maxPrice.toString(10), 10),
+    };
+}
+exports.getPreLaunchOraclePriceDataFromBuffer = getPreLaunchOraclePriceDataFromBuffer;
+//# sourceMappingURL=prelaunchOracle.js.map

package/src/plugins/drift/perpHelpers/prelaunchOracle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prelaunchOracle.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/prelaunchOracle.ts"],"names":[],"mappings":";;;;;;AAAA,kDAAuB;AAEvB,sCAAkD;AAElD,SAAgB,qCAAqC,CACnD,MAAc;IAEd,MAAM,CAAC,eAAe,CAAC,GAAG,8BAAqB,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAEpE,OAAO;QACL,KAAK,EAAE,IAAI,eAAE,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACrD,IAAI,EAAE,IAAI,eAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAChE,UAAU,EAAE,IAAI,eAAE,CAAC,eAAe,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/D,+BAA+B,EAAE,IAAI;QACrC,QAAQ,EAAE,IAAI,eAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;KAC5D,CAAC;AACJ,CAAC;AAZD,sFAYC"}

package/src/plugins/drift/perpHelpers/pyth.d.ts

@@ -0,0 +1,5 @@
+/// <reference types="node" />
+import BN from 'bn.js';
+import { OraclePriceData } from './types';
+export declare function getPythOraclePriceDataFromBuffer(buffer: Buffer, multiple: BN, stableCoin: boolean): OraclePriceData;
+export declare function convertPythPrice(price: number, exponent: number, multiple: BN): BN;

package/src/plugins/drift/perpHelpers/pyth.js

@@ -0,0 +1,42 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.convertPythPrice = exports.getPythOraclePriceDataFromBuffer = void 0;
+/* eslint-disable no-param-reassign */
+const bn_js_1 = __importDefault(require("bn.js"));
+const helpers_1 = require("../../../utils/solana/pyth/helpers");
+const constants_1 = require("./constants");
+function getPythOraclePriceDataFromBuffer(buffer, multiple, stableCoin) {
+    const priceData = (0, helpers_1.parsePriceData)(buffer);
+    const confidence = convertPythPrice(priceData.confidence || 0, priceData.exponent, multiple);
+    const minPublishers = Math.min(priceData.numComponentPrices, 3);
+    let price = convertPythPrice(priceData.aggregate.price, priceData.exponent, multiple);
+    if (stableCoin) {
+        price = getStableCoinPrice(price, confidence);
+    }
+    return {
+        price,
+        slot: new bn_js_1.default(priceData.lastSlot.toString()),
+        confidence,
+        twap: convertPythPrice(priceData.emaPrice.value, priceData.exponent, multiple),
+        twapConfidence: convertPythPrice(priceData.emaConfidence.value, priceData.exponent, multiple),
+        hasSufficientNumberOfDataPoints: priceData.numQuoters >= minPublishers,
+    };
+}
+exports.getPythOraclePriceDataFromBuffer = getPythOraclePriceDataFromBuffer;
+function convertPythPrice(price, exponent, multiple) {
+    exponent = Math.abs(exponent);
+    const pythPrecision = constants_1.TEN.pow(new bn_js_1.default(exponent).abs()).div(multiple);
+    return new bn_js_1.default(price * Math.pow(10, exponent)).mul(constants_1.PRICE_PRECISION).div(pythPrecision);
+}
+exports.convertPythPrice = convertPythPrice;
+const fiveBPS = new bn_js_1.default(500);
+function getStableCoinPrice(price, confidence) {
+    if (price.sub(constants_1.QUOTE_PRECISION).abs().lt(bn_js_1.default.min(confidence, fiveBPS))) {
+        return constants_1.QUOTE_PRECISION;
+    }
+    return price;
+}
+//# sourceMappingURL=pyth.js.map

package/src/plugins/drift/perpHelpers/pyth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pyth.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/pyth.ts"],"names":[],"mappings":";;;;;;AAAA,sCAAsC;AACtC,kDAAuB;AACvB,gEAAoE;AACpE,2CAAoE;AAGpE,SAAgB,gCAAgC,CAC9C,MAAc,EACd,QAAY,EACZ,UAAmB;IAEnB,MAAM,SAAS,GAAG,IAAA,wBAAc,EAAC,MAAM,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,gBAAgB,CACjC,SAAS,CAAC,UAAU,IAAI,CAAC,EACzB,SAAS,CAAC,QAAQ,EAClB,QAAQ,CACT,CAAC;IACF,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;IAChE,IAAI,KAAK,GAAG,gBAAgB,CAC1B,SAAS,CAAC,SAAS,CAAC,KAAK,EACzB,SAAS,CAAC,QAAQ,EAClB,QAAQ,CACT,CAAC;IACF,IAAI,UAAU,EAAE,CAAC;QACf,KAAK,GAAG,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC;IAED,OAAO;QACL,KAAK;QACL,IAAI,EAAE,IAAI,eAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAC3C,UAAU;QACV,IAAI,EAAE,gBAAgB,CACpB,SAAS,CAAC,QAAQ,CAAC,KAAK,EACxB,SAAS,CAAC,QAAQ,EAClB,QAAQ,CACT;QACD,cAAc,EAAE,gBAAgB,CAC9B,SAAS,CAAC,aAAa,CAAC,KAAK,EAC7B,SAAS,CAAC,QAAQ,EAClB,QAAQ,CACT;QACD,+BAA+B,EAAE,SAAS,CAAC,UAAU,IAAI,aAAa;KACvE,CAAC;AACJ,CAAC;AArCD,4EAqCC;AAED,SAAgB,gBAAgB,CAC9B,KAAa,EACb,QAAgB,EAChB,QAAY;IAEZ,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9B,MAAM,aAAa,GAAG,eAAG,CAAC,GAAG,CAAC,IAAI,eAAE,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpE,OAAO,IAAI,eAAE,CAAC,KAAK,GAAG,SAAA,EAAE,EAAI,QAAQ,CAAA,CAAC,CAAC,GAAG,CAAC,2BAAe,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;AAChF,CAAC;AARD,4CAQC;AAED,MAAM,OAAO,GAAG,IAAI,eAAE,CAAC,GAAG,CAAC,CAAC;AAC5B,SAAS,kBAAkB,CAAC,KAAS,EAAE,UAAc;IACnD,IAAI,KAAK,CAAC,GAAG,CAAC,2BAAe,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,eAAE,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,EAAE,CAAC;QACrE,OAAO,2BAAe,CAAC;IACzB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/src/plugins/drift/perpHelpers/repeg.d.ts

@@ -0,0 +1,14 @@
+import BN from 'bn.js';
+import { AMM } from '../struct';
+export declare function calculateAdjustKCost(amm: AMM, numerator: BN, denomenator: BN): BN;
+/**
+ * Helper function calculating adjust pegMultiplier (repeg) cost
+ *
+ * @param amm
+ * @param newPeg
+ * @returns cost : Precision QUOTE_ASSET_PRECISION
+ */
+export declare function calculateRepegCost(amm: AMM, newPeg: BN): BN;
+export declare function calculateBudgetedKBN(x: BN, y: BN, budget: BN, Q: BN, d: BN): [BN, BN];
+export declare function calculateBudgetedK(amm: AMM, cost: BN): [BN, BN];
+export declare function calculateBudgetedPeg(amm: AMM, budget: BN, targetPrice: BN): BN;

package/src/plugins/drift/perpHelpers/repeg.js

@@ -0,0 +1,162 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.calculateBudgetedPeg = exports.calculateBudgetedK = exports.calculateBudgetedKBN = exports.calculateRepegCost = exports.calculateAdjustKCost = void 0;
+/* eslint-disable @typescript-eslint/naming-convention */
+const bn_js_1 = __importDefault(require("bn.js"));
+const constants_1 = require("./constants");
+const assert_1 = require("./assert");
+function calculateAdjustKCost(amm, numerator, denomenator) {
+    // const k = market.amm.sqrtK.mul(market.amm.sqrtK);
+    const x = amm.baseAssetReserve;
+    const y = amm.quoteAssetReserve;
+    const d = amm.baseAssetAmountWithAmm;
+    const Q = amm.pegMultiplier;
+    const quoteScale = y.mul(d).mul(Q); // .div(AMM_RESERVE_PRECISION);
+    const p = numerator.mul(constants_1.PRICE_PRECISION).div(denomenator);
+    const cost = quoteScale
+        .mul(constants_1.PERCENTAGE_PRECISION)
+        .mul(constants_1.PERCENTAGE_PRECISION)
+        .div(x.add(d))
+        .sub(quoteScale
+        .mul(p)
+        .mul(constants_1.PERCENTAGE_PRECISION)
+        .mul(constants_1.PERCENTAGE_PRECISION)
+        .div(constants_1.PRICE_PRECISION)
+        .div(x.mul(p).div(constants_1.PRICE_PRECISION).add(d)))
+        .div(constants_1.PERCENTAGE_PRECISION)
+        .div(constants_1.PERCENTAGE_PRECISION)
+        .div(constants_1.AMM_TO_QUOTE_PRECISION_RATIO)
+        .div(constants_1.PEG_PRECISION);
+    return cost.mul(new bn_js_1.default(-1));
+}
+exports.calculateAdjustKCost = calculateAdjustKCost;
+// /**
+//  * Helper function calculating adjust k cost
+//  * @param amm
+//  * @param numerator
+//  * @param denomenator
+//  * @returns cost : Precision QUOTE_ASSET_PRECISION
+//  */
+// export function calculateAdjustKCost2(
+// 	amm: AMM,
+// 	numerator: BN,
+// 	denomenator: BN
+// ): BN {
+// 	// const k = market.amm.sqrtK.mul(market.amm.sqrtK);
+// 	const directionToClose = amm.baseAssetAmountWithAmm.gt(ZERO)
+// 		? PositionDirection.SHORT
+// 		: PositionDirection.LONG;
+// 	const [newQuoteAssetReserve, _newBaseAssetReserve] =
+// 		calculateAmmReservesAfterSwap(
+// 			amm,
+// 			'base',
+// 			amm.baseAssetAmountWithAmm.abs(),
+// 			getSwapDirection('base', directionToClose)
+// 		);
+// }
+/**
+ * Helper function calculating adjust pegMultiplier (repeg) cost
+ *
+ * @param amm
+ * @param newPeg
+ * @returns cost : Precision QUOTE_ASSET_PRECISION
+ */
+function calculateRepegCost(amm, newPeg) {
+    const dqar = amm.quoteAssetReserve.sub(amm.terminalQuoteAssetReserve);
+    const cost = dqar
+        .mul(newPeg.sub(amm.pegMultiplier))
+        .div(constants_1.AMM_TO_QUOTE_PRECISION_RATIO)
+        .div(constants_1.PEG_PRECISION);
+    return cost;
+}
+exports.calculateRepegCost = calculateRepegCost;
+function calculateBudgetedKBN(x, y, budget, Q, d) {
+    (0, assert_1.assert)(Q.gt(new bn_js_1.default(0)));
+    const C = budget.mul(new bn_js_1.default(-1));
+    let dSign = new bn_js_1.default(1);
+    if (d.lt(new bn_js_1.default(0))) {
+        dSign = new bn_js_1.default(-1);
+    }
+    const pegged_y_d_d = y
+        .mul(d)
+        .mul(d)
+        .mul(Q)
+        .div(constants_1.AMM_RESERVE_PRECISION)
+        .div(constants_1.AMM_RESERVE_PRECISION)
+        .div(constants_1.PEG_PRECISION);
+    const numer1 = pegged_y_d_d;
+    const numer2 = C.mul(d)
+        .div(constants_1.QUOTE_PRECISION)
+        .mul(x.add(d))
+        .div(constants_1.AMM_RESERVE_PRECISION)
+        .mul(dSign);
+    const denom1 = C.mul(x)
+        .mul(x.add(d))
+        .div(constants_1.AMM_RESERVE_PRECISION)
+        .div(constants_1.QUOTE_PRECISION);
+    const denom2 = pegged_y_d_d;
+    // protocol is spending to increase k
+    if (C.lt(constants_1.ZERO)) {
+        // thus denom1 is negative and solution is unstable
+        if (denom1.abs().gt(denom2.abs())) {
+            // console.log('denom1 > denom2', denom1.toString(), denom2.toString());
+            // console.log('budget cost exceeds stable K solution');
+            return [new bn_js_1.default(10000), new bn_js_1.default(1)];
+        }
+    }
+    const numerator = numer1.sub(numer2).div(constants_1.AMM_TO_QUOTE_PRECISION_RATIO);
+    const denominator = denom1.add(denom2).div(constants_1.AMM_TO_QUOTE_PRECISION_RATIO);
+    return [numerator, denominator];
+}
+exports.calculateBudgetedKBN = calculateBudgetedKBN;
+function calculateBudgetedK(amm, cost) {
+    // wolframalpha.com
+    // (1/(x+d) - p/(x*p+d))*y*d*Q = C solve for p
+    // p = (d(y*d*Q - C(x+d))) / (C*x(x+d) + y*d*d*Q)
+    // numer
+    //   =  y*d*d*Q - Cxd - Cdd
+    //   =  y/x*Q*d*d - Cd - Cd/x
+    //   = mark      - C/d - C/(x)
+    //   =  mark/C    - 1/d - 1/x
+    // denom
+    // = C*x*x + C*x*d + y*d*d*Q
+    // = x/d**2 + 1 / d + mark/C
+    // todo: assumes k = x * y
+    // otherwise use: (y(1-p) + (kp^2/(x*p+d)) - k/(x+d)) * Q = C solve for p
+    const x = amm.baseAssetReserve;
+    const y = amm.quoteAssetReserve;
+    const d = amm.baseAssetAmountWithAmm;
+    const Q = amm.pegMultiplier;
+    const [numerator, denominator] = calculateBudgetedKBN(x, y, cost, Q, d);
+    return [numerator, denominator];
+}
+exports.calculateBudgetedK = calculateBudgetedK;
+function calculateBudgetedPeg(amm, budget, targetPrice) {
+    let perPegCost = amm.quoteAssetReserve
+        .sub(amm.terminalQuoteAssetReserve)
+        .div(constants_1.AMM_RESERVE_PRECISION.div(constants_1.PRICE_PRECISION));
+    if (perPegCost.gt(constants_1.ZERO)) {
+        perPegCost = perPegCost.add(constants_1.ONE);
+    }
+    else if (perPegCost.lt(constants_1.ZERO)) {
+        perPegCost = perPegCost.sub(constants_1.ONE);
+    }
+    const targetPeg = targetPrice
+        .mul(amm.baseAssetReserve)
+        .div(amm.quoteAssetReserve)
+        .div(constants_1.PRICE_DIV_PEG);
+    const pegChangeDirection = targetPeg.sub(amm.pegMultiplier);
+    const useTargetPeg = (perPegCost.lt(constants_1.ZERO) && pegChangeDirection.gt(constants_1.ZERO)) ||
+        (perPegCost.gt(constants_1.ZERO) && pegChangeDirection.lt(constants_1.ZERO));
+    if (perPegCost.eq(constants_1.ZERO) || useTargetPeg) {
+        return targetPeg;
+    }
+    const budgetDeltaPeg = budget.mul(constants_1.PEG_PRECISION).div(perPegCost);
+    const newPeg = bn_js_1.default.max(constants_1.ONE, amm.pegMultiplier.add(budgetDeltaPeg));
+    return newPeg;
+}
+exports.calculateBudgetedPeg = calculateBudgetedPeg;
+//# sourceMappingURL=repeg.js.map

package/src/plugins/drift/perpHelpers/repeg.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"repeg.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/repeg.ts"],"names":[],"mappings":";;;;;;AAAA,yDAAyD;AACzD,kDAAuB;AACvB,2CAUqB;AAErB,qCAAkC;AAElC,SAAgB,oBAAoB,CAClC,GAAQ,EACR,SAAa,EACb,WAAe;IAEf,oDAAoD;IACpD,MAAM,CAAC,GAAG,GAAG,CAAC,gBAAgB,CAAC;IAC/B,MAAM,CAAC,GAAG,GAAG,CAAC,iBAAiB,CAAC;IAEhC,MAAM,CAAC,GAAG,GAAG,CAAC,sBAAsB,CAAC;IACrC,MAAM,CAAC,GAAG,GAAG,CAAC,aAAa,CAAC;IAE5B,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,+BAA+B;IAEnE,MAAM,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,2BAAe,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAE1D,MAAM,IAAI,GAAG,UAAU;SACpB,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACb,GAAG,CACF,UAAU;SACP,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,2BAAe,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAC7C;SACA,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,gCAAoB,CAAC;SACzB,GAAG,CAAC,wCAA4B,CAAC;SACjC,GAAG,CAAC,yBAAa,CAAC,CAAC;IAEtB,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9B,CAAC;AAlCD,oDAkCC;AAED,MAAM;AACN,+CAA+C;AAC/C,gBAAgB;AAChB,sBAAsB;AACtB,wBAAwB;AACxB,qDAAqD;AACrD,MAAM;AACN,yCAAyC;AACzC,aAAa;AACb,kBAAkB;AAClB,mBAAmB;AACnB,UAAU;AACV,wDAAwD;AACxD,gEAAgE;AAChE,8BAA8B;AAC9B,8BAA8B;AAE9B,wDAAwD;AACxD,mCAAmC;AACnC,UAAU;AACV,aAAa;AACb,uCAAuC;AACvC,gDAAgD;AAChD,OAAO;AACP,IAAI;AAEJ;;;;;;GAMG;AACH,SAAgB,kBAAkB,CAAC,GAAQ,EAAE,MAAU;IACrD,MAAM,IAAI,GAAG,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACtE,MAAM,IAAI,GAAG,IAAI;SACd,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;SAClC,GAAG,CAAC,wCAA4B,CAAC;SACjC,GAAG,CAAC,yBAAa,CAAC,CAAC;IACtB,OAAO,IAAI,CAAC;AACd,CAAC;AAPD,gDAOC;AAED,SAAgB,oBAAoB,CAClC,CAAK,EACL,CAAK,EACL,MAAU,EACV,CAAK,EACL,CAAK;IAEL,IAAA,eAAM,EAAC,CAAC,CAAC,EAAE,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxB,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjC,IAAI,KAAK,GAAG,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC;IACtB,IAAI,CAAC,CAAC,EAAE,CAAC,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACpB,KAAK,GAAG,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,YAAY,GAAG,CAAC;SACnB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,iCAAqB,CAAC;SAC1B,GAAG,CAAC,iCAAqB,CAAC;SAC1B,GAAG,CAAC,yBAAa,CAAC,CAAC;IAEtB,MAAM,MAAM,GAAG,YAAY,CAAC;IAC5B,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;SACpB,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACb,GAAG,CAAC,iCAAqB,CAAC;SAC1B,GAAG,CAAC,KAAK,CAAC,CAAC;IAEd,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;SACpB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACb,GAAG,CAAC,iCAAqB,CAAC;SAC1B,GAAG,CAAC,2BAAe,CAAC,CAAC;IACxB,MAAM,MAAM,GAAG,YAAY,CAAC;IAE5B,qCAAqC;IACrC,IAAI,CAAC,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QACf,mDAAmD;QACnD,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAClC,wEAAwE;YACxE,wDAAwD;YACxD,OAAO,CAAC,IAAI,eAAE,CAAC,KAAK,CAAC,EAAE,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,wCAA4B,CAAC,CAAC;IACvE,MAAM,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,wCAA4B,CAAC,CAAC;IAEzE,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;AAClC,CAAC;AAjDD,oDAiDC;AAED,SAAgB,kBAAkB,CAAC,GAAQ,EAAE,IAAQ;IACnD,mBAAmB;IACnB,8CAA8C;IAC9C,iDAAiD;IAEjD,QAAQ;IACR,2BAA2B;IAC3B,6BAA6B;IAC7B,8BAA8B;IAC9B,6BAA6B;IAE7B,QAAQ;IACR,4BAA4B;IAC5B,4BAA4B;IAE5B,0BAA0B;IAC1B,yEAAyE;IAEzE,MAAM,CAAC,GAAG,GAAG,CAAC,gBAAgB,CAAC;IAC/B,MAAM,CAAC,GAAG,GAAG,CAAC,iBAAiB,CAAC;IAEhC,MAAM,CAAC,GAAG,GAAG,CAAC,sBAAsB,CAAC;IACrC,MAAM,CAAC,GAAG,GAAG,CAAC,aAAa,CAAC;IAE5B,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,GAAG,oBAAoB,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAExE,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;AAClC,CAAC;AA3BD,gDA2BC;AAED,SAAgB,oBAAoB,CAClC,GAAQ,EACR,MAAU,EACV,WAAe;IAEf,IAAI,UAAU,GAAG,GAAG,CAAC,iBAAiB;SACnC,GAAG,CAAC,GAAG,CAAC,yBAAyB,CAAC;SAClC,GAAG,CAAC,iCAAqB,CAAC,GAAG,CAAC,2BAAe,CAAC,CAAC,CAAC;IAEnD,IAAI,UAAU,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QACxB,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,eAAG,CAAC,CAAC;IACnC,CAAC;SAAM,IAAI,UAAU,CAAC,EAAE,CAAC,gBAAI,CAAC,EAAE,CAAC;QAC/B,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,eAAG,CAAC,CAAC;IACnC,CAAC;IAED,MAAM,SAAS,GAAG,WAAW;SAC1B,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC;SACzB,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC;SAC1B,GAAG,CAAC,yBAAa,CAAC,CAAC;IAEtB,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAE5D,MAAM,YAAY,GAChB,CAAC,UAAU,CAAC,EAAE,CAAC,gBAAI,CAAC,IAAI,kBAAkB,CAAC,EAAE,CAAC,gBAAI,CAAC,CAAC;QACpD,CAAC,UAAU,CAAC,EAAE,CAAC,gBAAI,CAAC,IAAI,kBAAkB,CAAC,EAAE,CAAC,gBAAI,CAAC,CAAC,CAAC;IAEvD,IAAI,UAAU,CAAC,EAAE,CAAC,gBAAI,CAAC,IAAI,YAAY,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,yBAAa,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACjE,MAAM,MAAM,GAAG,eAAE,CAAC,GAAG,CAAC,eAAG,EAAE,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;IAElE,OAAO,MAAM,CAAC;AAChB,CAAC;AAlCD,oDAkCC"}

package/src/plugins/drift/perpHelpers/switchboard.d.ts

@@ -0,0 +1,3 @@
+/// <reference types="node" />
+import { OraclePriceData } from './types';
+export declare function getSwitchboardOraclePriceDataFromBuffer(buffer: Buffer): OraclePriceData;

package/src/plugins/drift/perpHelpers/switchboard.js

@@ -0,0 +1,31 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getSwitchboardOraclePriceDataFromBuffer = void 0;
+const bn_js_1 = __importDefault(require("bn.js"));
+const constants_1 = require("./constants");
+const structs_1 = require("../../switchboard/structs");
+function getSwitchboardOraclePriceDataFromBuffer(buffer) {
+    const [aggregatorAccountData] = structs_1.aggregatorAccountStruct.deserialize(buffer);
+    const price = convertSwitchboardDecimal(aggregatorAccountData.latestConfirmedRound.result);
+    const confidence = bn_js_1.default.max(convertSwitchboardDecimal(aggregatorAccountData.latestConfirmedRound.stdDeviation), price.divn(1000));
+    const hasSufficientNumberOfDataPoints = aggregatorAccountData.latestConfirmedRound.numSuccess >=
+        aggregatorAccountData.minOracleResults;
+    const slot = new bn_js_1.default(aggregatorAccountData.latestConfirmedRound.roundOpenSlot.toString());
+    return {
+        price,
+        slot,
+        confidence,
+        hasSufficientNumberOfDataPoints,
+    };
+}
+exports.getSwitchboardOraclePriceDataFromBuffer = getSwitchboardOraclePriceDataFromBuffer;
+function convertSwitchboardDecimal(switchboardDecimal) {
+    const switchboardPrecision = constants_1.TEN.pow(new bn_js_1.default(switchboardDecimal.scale));
+    return new bn_js_1.default(switchboardDecimal.mantissa.toString(10), 10)
+        .mul(constants_1.PRICE_PRECISION)
+        .div(switchboardPrecision);
+}
+//# sourceMappingURL=switchboard.js.map

package/src/plugins/drift/perpHelpers/switchboard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"switchboard.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/switchboard.ts"],"names":[],"mappings":";;;;;;AAAA,kDAAuB;AAGvB,2CAAmD;AACnD,uDAAoE;AAEpE,SAAgB,uCAAuC,CACrD,MAAc;IAEd,MAAM,CAAC,qBAAqB,CAAC,GAAG,iCAAuB,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE5E,MAAM,KAAK,GAAG,yBAAyB,CACrC,qBAAqB,CAAC,oBAAoB,CAAC,MAAM,CAClD,CAAC;IAEF,MAAM,UAAU,GAAG,eAAE,CAAC,GAAG,CACvB,yBAAyB,CACvB,qBAAqB,CAAC,oBAAoB,CAAC,YAAY,CACxD,EACD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CACjB,CAAC;IAEF,MAAM,+BAA+B,GACnC,qBAAqB,CAAC,oBAAoB,CAAC,UAAU;QACrD,qBAAqB,CAAC,gBAAgB,CAAC;IAEzC,MAAM,IAAI,GAAG,IAAI,eAAE,CACjB,qBAAqB,CAAC,oBAAoB,CAAC,aAAa,CAAC,QAAQ,EAAE,CACpE,CAAC;IACF,OAAO;QACL,KAAK;QACL,IAAI;QACJ,UAAU;QACV,+BAA+B;KAChC,CAAC;AACJ,CAAC;AA7BD,0FA6BC;AAED,SAAS,yBAAyB,CAAC,kBAGlC;IACC,MAAM,oBAAoB,GAAG,eAAG,CAAC,GAAG,CAAC,IAAI,eAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;IACvE,OAAO,IAAI,eAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;SACxD,GAAG,CAAC,2BAAe,CAAC;SACpB,GAAG,CAAC,oBAAoB,CAAC,CAAC;AAC/B,CAAC"}

package/src/plugins/drift/perpHelpers/types.d.ts

@@ -0,0 +1,28 @@
+import BN from 'bn.js';
+export type OraclePriceData = {
+    price: BN;
+    slot: BN;
+    confidence: BN;
+    hasSufficientNumberOfDataPoints: boolean;
+    twap?: BN;
+    twapConfidence?: BN;
+    maxPrice?: BN;
+};
+export declare class PositionDirection {
+    static readonly LONG: {
+        long: {};
+    };
+    static readonly SHORT: {
+        short: {};
+    };
+}
+export declare class SwapDirection {
+    static readonly ADD: {
+        add: {};
+    };
+    static readonly REMOVE: {
+        remove: {};
+    };
+}
+export type AssetType = 'quote' | 'base';
+export declare function isVariant(object: object, type: string): boolean;

package/src/plugins/drift/perpHelpers/types.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isVariant = exports.SwapDirection = exports.PositionDirection = void 0;
+class PositionDirection {
+}
+exports.PositionDirection = PositionDirection;
+PositionDirection.LONG = { long: {} };
+PositionDirection.SHORT = { short: {} };
+class SwapDirection {
+}
+exports.SwapDirection = SwapDirection;
+SwapDirection.ADD = { add: {} };
+SwapDirection.REMOVE = { remove: {} };
+function isVariant(object, type) {
+    // eslint-disable-next-line no-prototype-builtins
+    return object.hasOwnProperty(type);
+}
+exports.isVariant = isVariant;
+//# sourceMappingURL=types.js.map

package/src/plugins/drift/perpHelpers/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../../../packages/plugins/src/plugins/drift/perpHelpers/types.ts"],"names":[],"mappings":";;;AAaA,MAAa,iBAAiB;;AAA9B,8CAGC;AAFiB,sBAAI,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;AACpB,uBAAK,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;AAGxC,MAAa,aAAa;;AAA1B,sCAGC;AAFiB,iBAAG,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;AAClB,oBAAM,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;AAK1C,SAAgB,SAAS,CAAC,MAAc,EAAE,IAAY;IACpD,iDAAiD;IACjD,OAAO,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC;AAHD,8BAGC"}

package/src/plugins/drift/prepMarketsJob.d.ts

@@ -0,0 +1,3 @@
+import { Job } from '../../Job';
+declare const job: Job;
+export default job;

package/src/plugins/drift/prepMarketsJob.js

@@ -0,0 +1,38 @@
+"use strict";
+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());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const portfolio_core_1 = require("@sonarwatch/portfolio-core");
+const constants_1 = require("./constants");
+const struct_1 = require("./struct");
+const clients_1 = require("../../utils/clients");
+const solana_1 = require("../../utils/solana");
+const filters_1 = require("./filters");
+const getOraclePrice_1 = require("./perpHelpers/getOraclePrice");
+const executor = (cache) => __awaiter(void 0, void 0, void 0, function* () {
+    const client = (0, clients_1.getClientSolana)();
+    const marketAccounts = yield (0, solana_1.getParsedProgramAccounts)(client, struct_1.perpMarketStruct, constants_1.driftProgram, filters_1.perpMarketsFilters);
+    const indexes = marketAccounts.map((m) => [m.marketIndex, m.pubkey.toString()]);
+    yield cache.setItem(constants_1.perpMarketsIndexesKey, indexes, {
+        prefix: constants_1.platformId,
+        networkId: portfolio_core_1.NetworkId.solana,
+    });
+    for (let i = 0; i < marketAccounts.length; i++) {
+        const acc = marketAccounts[i];
+        yield (0, getOraclePrice_1.getOraclePrice)(acc.amm.oracle.toString(), acc.amm.oracleSource, client);
+    }
+});
+const job = {
+    id: `${constants_1.platformId}-perp-markets`,
+    executor,
+    label: 'normal',
+};
+exports.default = job;
+//# sourceMappingURL=prepMarketsJob.js.map

package/src/plugins/drift/prepMarketsJob.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepMarketsJob.js","sourceRoot":"","sources":["../../../../../../packages/plugins/src/plugins/drift/prepMarketsJob.ts"],"names":[],"mappings":";;;;;;;;;;;AAAA,+DAAuD;AAGvD,2CAA8E;AAC9E,qCAA4C;AAC5C,iDAAsD;AACtD,+CAA8D;AAC9D,uCAA+C;AAC/C,iEAA8D;AAE9D,MAAM,QAAQ,GAAgB,CAAO,KAAY,EAAE,EAAE;IACnD,MAAM,MAAM,GAAG,IAAA,yBAAe,GAAE,CAAC;IAEjC,MAAM,cAAc,GAAG,MAAM,IAAA,iCAAwB,EACnD,MAAM,EACN,yBAAgB,EAChB,wBAAY,EACZ,4BAAkB,CACnB,CAAC;IACF,MAAM,OAAO,GAA4B,cAAc,CAAC,GAAG,CACzD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAU,CACrD,CAAC;IACF,MAAM,KAAK,CAAC,OAAO,CAAC,iCAAqB,EAAE,OAAO,EAAE;QAClD,MAAM,EAAE,sBAAU;QAClB,SAAS,EAAE,0BAAS,CAAC,MAAM;KAC5B,CAAC,CAAC;IAEH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAA,+BAAc,EAClB,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,EACzB,GAAG,CAAC,GAAG,CAAC,YAAY,EACpB,MAAM,CACP,CAAC;IACJ,CAAC;AACH,CAAC,CAAA,CAAC;AAEF,MAAM,GAAG,GAAQ;IACf,EAAE,EAAE,GAAG,sBAAU,eAAe;IAChC,QAAQ;IACR,KAAK,EAAE,QAAQ;CAChB,CAAC;AACF,kBAAe,GAAG,CAAC"}