@drift-labs/sdk 0.2.0-master.6 → 0.2.0-master.9

package/src/factory/bigNum.js

@@ -0,0 +1,364 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BigNum = void 0;
+const anchor_1 = require("@project-serum/anchor");
+const assert_1 = require("../assert/assert");
+const numericConstants_1 = require("./../constants/numericConstants");
+class BigNum {
+    constructor(val, precisionVal = new anchor_1.BN(0)) {
+        this.toString = (base, length) => this.val.toString(base, length);
+        this.val = new anchor_1.BN(val);
+        this.precision = new anchor_1.BN(precisionVal);
+    }
+    add(bn) {
+        assert_1.assert(bn.precision.eq(this.precision), 'Adding unequal precisions');
+        return BigNum.from(this.val.add(bn.val), this.precision);
+    }
+    sub(bn) {
+        assert_1.assert(bn.precision.eq(this.precision), 'Subtracting unequal precisions');
+        return BigNum.from(this.val.sub(bn.val), this.precision);
+    }
+    mul(bn) {
+        const mulVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        return BigNum.from(this.val.mul(mulVal.val), this.precision.add(mulVal.precision));
+    }
+    /**
+     * Multiplies by another big number then scales the result down by the big number's precision so that we're in the same precision space
+     * @param bn
+     * @returns
+     */
+    scalarMul(bn) {
+        if (bn instanceof anchor_1.BN)
+            return BigNum.from(this.val.mul(bn), this.precision);
+        return BigNum.from(this.val.mul(bn.val), this.precision.add(bn.precision)).shift(bn.precision.neg());
+    }
+    div(bn) {
+        if (bn instanceof anchor_1.BN)
+            return BigNum.from(this.val.div(bn), this.precision);
+        return BigNum.from(this.val.div(bn.val), this.precision.sub(bn.precision));
+    }
+    /**
+     * Shift precision up or down
+     * @param exponent
+     * @param skipAdjustingPrecision
+     * @returns
+     */
+    shift(exponent, skipAdjustingPrecision = false) {
+        const shiftVal = typeof exponent === 'number' ? new anchor_1.BN(exponent) : exponent;
+        return BigNum.from(shiftVal.isNeg()
+            ? this.val.div(new anchor_1.BN(10).pow(shiftVal))
+            : this.val.mul(new anchor_1.BN(10).pow(shiftVal)), skipAdjustingPrecision ? this.precision : this.precision.add(shiftVal));
+    }
+    /**
+     * Shift to a target precision
+     * @param targetPrecision
+     * @returns
+     */
+    shiftTo(targetPrecision) {
+        return this.shift(targetPrecision.sub(this.precision));
+    }
+    /**
+     * Scale the number by a fraction
+     * @param numerator
+     * @param denominator
+     * @returns
+     */
+    scale(numerator, denominator) {
+        return this.mul(BigNum.from(new anchor_1.BN(numerator))).div(new anchor_1.BN(denominator));
+    }
+    toPercentage(denominator, precision) {
+        return this.shift(precision)
+            .shift(2, true)
+            .div(denominator)
+            .toPrecision(precision);
+    }
+    gt(bn, ignorePrecision) {
+        const comparisonVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        if (!ignorePrecision && !comparisonVal.eq(numericConstants_1.ZERO)) {
+            assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
+        }
+        return this.val.gt(comparisonVal.val);
+    }
+    lt(bn, ignorePrecision) {
+        const comparisonVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
+            assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
+        }
+        return this.val.lt(comparisonVal.val);
+    }
+    gte(bn, ignorePrecision) {
+        const comparisonVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
+            assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
+        }
+        return this.val.gte(comparisonVal.val);
+    }
+    lte(bn, ignorePrecision) {
+        const comparisonVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
+            assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
+        }
+        return this.val.lte(comparisonVal.val);
+    }
+    eq(bn, ignorePrecision) {
+        const comparisonVal = bn instanceof BigNum ? bn : BigNum.from(bn);
+        if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
+            assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
+        }
+        return this.val.eq(comparisonVal.val);
+    }
+    eqZero() {
+        return this.val.eq(numericConstants_1.ZERO);
+    }
+    gtZero() {
+        return this.val.gt(numericConstants_1.ZERO);
+    }
+    ltZero() {
+        return this.val.lt(numericConstants_1.ZERO);
+    }
+    gteZero() {
+        return this.val.gte(numericConstants_1.ZERO);
+    }
+    lteZero() {
+        return this.val.lte(numericConstants_1.ZERO);
+    }
+    abs() {
+        return new BigNum(this.val.abs(), this.precision);
+    }
+    neg() {
+        return new BigNum(this.val.neg(), this.precision);
+    }
+    /**
+     * Pretty print the underlying value in human-readable form. Depends on precision being correct for the output string to be correct
+     * @returns
+     */
+    print() {
+        assert_1.assert(this.precision.gte(numericConstants_1.ZERO), 'Tried to print a BN with precision lower than zero');
+        const plainString = this.toString();
+        const precisionNum = this.precision.toNumber();
+        // make a string with at least the precisionNum number of zeroes
+        let printString = [
+            ...Array(this.precision.toNumber()).fill(0),
+            ...plainString.split(''),
+        ].join('');
+        // inject decimal
+        printString =
+            printString.substring(0, printString.length - precisionNum) +
+                BigNum.delim +
+                printString.substring(printString.length - precisionNum);
+        // remove leading zeroes
+        printString = printString.replace(/^0+/, '');
+        // add zero if leading delim
+        if (printString[0] === BigNum.delim)
+            printString = `0${printString}`;
+        // remove trailing delim
+        if (printString[printString.length - 1] === BigNum.delim)
+            printString = printString.slice(0, printString.length - 1);
+        return printString;
+    }
+    prettyPrint(useTradePrecision, precisionOverride) {
+        const [leftSide, rightSide] = this.printShort(useTradePrecision, precisionOverride).split(BigNum.delim);
+        let formattedLeftSide = leftSide;
+        const isNeg = formattedLeftSide.includes('-');
+        if (isNeg) {
+            formattedLeftSide = formattedLeftSide.replace('-', '');
+        }
+        let index = formattedLeftSide.length - 3;
+        while (index >= 1) {
+            const formattedLeftSideArray = formattedLeftSide.split('');
+            formattedLeftSideArray.splice(index, 0, BigNum.spacer);
+            formattedLeftSide = formattedLeftSideArray.join('');
+            index -= 3;
+        }
+        return `${isNeg ? '-' : ''}${formattedLeftSide}${rightSide ? `${BigNum.delim}${rightSide}` : ''}`;
+    }
+    /**
+     * Print and remove unnecessary trailing zeroes
+     * @returns
+     */
+    printShort(useTradePrecision, precisionOverride) {
+        const printVal = precisionOverride
+            ? this.toPrecision(precisionOverride)
+            : useTradePrecision
+                ? this.toTradePrecision()
+                : this.print();
+        return printVal.replace(/0+$/g, '').replace(/\.$/, '').replace(/,$/, '');
+    }
+    debug() {
+        console.log(`${this.toString()} | ${this.print()} | ${this.precision.toString()}`);
+    }
+    /**
+     * Pretty print with the specified number of decimal places
+     * @param fixedPrecision
+     * @returns
+     */
+    toFixed(fixedPrecision) {
+        const printString = this.print();
+        const [leftSide, rightSide] = printString.split(BigNum.delim);
+        const filledRightSide = [
+            ...(rightSide !== null && rightSide !== void 0 ? rightSide : '').slice(0, fixedPrecision),
+            ...Array(fixedPrecision).fill('0'),
+        ]
+            .slice(0, fixedPrecision)
+            .join('');
+        return `${leftSide}${BigNum.delim}${filledRightSide}`;
+    }
+    /**
+     * Pretty print to the specified number of significant figures
+     * @param fixedPrecision
+     * @returns
+     */
+    toPrecision(fixedPrecision, trailingZeroes = false) {
+        const printString = this.print();
+        let precisionPrintString = printString.slice(0, fixedPrecision + 1);
+        if (!precisionPrintString.includes(BigNum.delim) ||
+            precisionPrintString[precisionPrintString.length - 1] === BigNum.delim) {
+            precisionPrintString = printString.slice(0, fixedPrecision);
+        }
+        const pointsOfPrecision = precisionPrintString.replace(BigNum.delim, '').length;
+        if (pointsOfPrecision < fixedPrecision) {
+            precisionPrintString = [
+                ...precisionPrintString.split(''),
+                ...Array(fixedPrecision - pointsOfPrecision).fill('0'),
+            ].join('');
+        }
+        if (!precisionPrintString.includes(BigNum.delim)) {
+            const delimFullStringLocation = printString.indexOf(BigNum.delim);
+            let skipExponent = false;
+            if (delimFullStringLocation === -1) {
+                // no decimal, not missing any precision
+                skipExponent = true;
+            }
+            if (precisionPrintString[precisionPrintString.length - 1] === BigNum.delim) {
+                // decimal is at end of string, not missing any precision, do nothing
+                skipExponent = true;
+            }
+            if (printString.indexOf(BigNum.delim) === fixedPrecision) {
+                // decimal is at end of string, not missing any precision, do nothing
+                skipExponent = true;
+            }
+            if (!skipExponent) {
+                const exponent = delimFullStringLocation - fixedPrecision;
+                if (trailingZeroes) {
+                    precisionPrintString = `${precisionPrintString}${Array(exponent)
+                        .fill('0')
+                        .join('')}`;
+                }
+                else {
+                    precisionPrintString = `${precisionPrintString}e${exponent}`;
+                }
+            }
+        }
+        return precisionPrintString;
+    }
+    toTradePrecision() {
+        return this.toPrecision(6, true);
+    }
+    /**
+     * Print dollar formatted value. Defaults to fixed decimals two unless a given precision is given.
+     * @param useTradePrecision
+     * @param precisionOverride
+     * @returns
+     */
+    toNotional(useTradePrecision, precisionOverride) {
+        const prefix = `${this.lt(BigNum.zero()) ? `-` : ``}$`;
+        const val = useTradePrecision || precisionOverride
+            ? this.prettyPrint(useTradePrecision, precisionOverride)
+            : BigNum.fromPrint(this.toFixed(2), new anchor_1.BN(2)).prettyPrint();
+        return `${prefix}${val.replace('-', '')}`;
+    }
+    toMillified(precision = 3) {
+        const stringVal = this.print();
+        const [leftSide] = stringVal.split(BigNum.delim);
+        if (!leftSide) {
+            return this.shift(new anchor_1.BN(precision)).toPrecision(precision, true);
+        }
+        if (leftSide.length <= 3) {
+            return this.shift(new anchor_1.BN(precision)).toPrecision(precision, true);
+        }
+        const unitTicks = ['', 'K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y'];
+        const unitNumber = Math.floor((leftSide.length - 1) / 3);
+        const unit = unitTicks[unitNumber];
+        let leadDigits = leftSide.slice(0, precision);
+        if (leadDigits.length < precision) {
+            leadDigits = [
+                ...leadDigits.split(''),
+                ...Array(precision - leadDigits.length).fill('0'),
+            ].join('');
+        }
+        const decimalLocation = leftSide.length - 3 * unitNumber;
+        let leadString = '';
+        if (decimalLocation >= precision) {
+            leadString = `${leadDigits}`;
+        }
+        else {
+            leadString = `${leadDigits.slice(0, decimalLocation)}${BigNum.delim}${leadDigits.slice(decimalLocation)}`;
+        }
+        return `${leadString}${unit}`;
+    }
+    toJSON() {
+        return {
+            val: this.val.toString(),
+            precision: this.precision.toString(),
+        };
+    }
+    isNeg() {
+        return this.lt(numericConstants_1.ZERO, true);
+    }
+    isPos() {
+        return !this.isNeg();
+    }
+    /**
+     * Get the numerical value of the BigNum. This can break if the BigNum is too large.
+     * @returns
+     */
+    toNum() {
+        return parseFloat(this.print());
+    }
+    static fromJSON(json) {
+        return BigNum.from(new anchor_1.BN(json.val), new anchor_1.BN(json.precision));
+    }
+    /**
+     * Create a BigNum instance
+     * @param val
+     * @param precision
+     * @returns
+     */
+    static from(val = numericConstants_1.ZERO, precision) {
+        assert_1.assert(new anchor_1.BN(precision).lt(new anchor_1.BN(100)), 'Tried to create a bignum with precision higher than 10^100');
+        return new BigNum(val, precision);
+    }
+    /**
+     * Create a BigNum instance from a printed BigNum
+     * @param val
+     * @param precisionOverride
+     * @returns
+     */
+    static fromPrint(val, precisionShift) {
+        var _a;
+        // Handle empty number edge cases
+        if (!val)
+            return BigNum.from(numericConstants_1.ZERO, precisionShift);
+        if (!val.replace(BigNum.delim, ''))
+            return BigNum.from(numericConstants_1.ZERO, precisionShift);
+        const [leftSide, rightSide] = val.split(BigNum.delim);
+        const rawBn = new anchor_1.BN(`${leftSide !== null && leftSide !== void 0 ? leftSide : ''}${rightSide !== null && rightSide !== void 0 ? rightSide : ''}`);
+        const rightSideLength = (_a = rightSide === null || rightSide === void 0 ? void 0 : rightSide.length) !== null && _a !== void 0 ? _a : 0;
+        const totalShift = precisionShift
+            ? precisionShift.sub(new anchor_1.BN(rightSideLength))
+            : numericConstants_1.ZERO;
+        return BigNum.from(rawBn, precisionShift).shift(totalShift, true);
+    }
+    static max(a, b) {
+        return a.gt(b) ? a : b;
+    }
+    static min(a, b) {
+        return a.lt(b) ? a : b;
+    }
+    static zero(precision) {
+        return BigNum.from(0, precision);
+    }
+}
+exports.BigNum = BigNum;
+BigNum.delim = '.';
+BigNum.spacer = ',';

package/src/factory/oracleClient.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getOracleClient = void 0;
+const types_1 = require("../types");
+const pythClient_1 = require("../oracles/pythClient");
+const switchboardClient_1 = require("../oracles/switchboardClient");
+const quoteAssetOracleClient_1 = require("../oracles/quoteAssetOracleClient");
+function getOracleClient(oracleSource, connection) {
+    if (types_1.isVariant(oracleSource, 'pyth')) {
+        return new pythClient_1.PythClient(connection);
+    }
+    if (types_1.isVariant(oracleSource, 'switchboard')) {
+        return new switchboardClient_1.SwitchboardClient(connection);
+    }
+    if (types_1.isVariant(oracleSource, 'quoteAsset')) {
+        return new quoteAssetOracleClient_1.QuoteAssetOracleClient();
+    }
+    throw new Error(`Unknown oracle source ${oracleSource}`);
+}
+exports.getOracleClient = getOracleClient;

package/src/index.js

@@ -0,0 +1,69 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pyth = exports.PublicKey = exports.BN = void 0;
+const anchor_1 = require("@project-serum/anchor");
+Object.defineProperty(exports, "BN", { enumerable: true, get: function () { return anchor_1.BN; } });
+const web3_js_1 = require("@solana/web3.js");
+Object.defineProperty(exports, "PublicKey", { enumerable: true, get: function () { return web3_js_1.PublicKey; } });
+const client_1 = __importDefault(require("@pythnetwork/client"));
+exports.pyth = client_1.default;
+__exportStar(require("./mockUSDCFaucet"), exports);
+__exportStar(require("./oracles/types"), exports);
+__exportStar(require("./oracles/pythClient"), exports);
+__exportStar(require("./oracles/switchboardClient"), exports);
+__exportStar(require("./types"), exports);
+__exportStar(require("./constants/markets"), exports);
+__exportStar(require("./accounts/fetch"), exports);
+__exportStar(require("./accounts/webSocketClearingHouseAccountSubscriber"), exports);
+__exportStar(require("./accounts/bulkAccountLoader"), exports);
+__exportStar(require("./accounts/bulkUserSubscription"), exports);
+__exportStar(require("./accounts/pollingClearingHouseAccountSubscriber"), exports);
+__exportStar(require("./accounts/pollingOracleSubscriber"), exports);
+__exportStar(require("./accounts/pollingTokenAccountSubscriber"), exports);
+__exportStar(require("./accounts/types"), exports);
+__exportStar(require("./addresses/pda"), exports);
+__exportStar(require("./admin"), exports);
+__exportStar(require("./clearingHouseUser"), exports);
+__exportStar(require("./clearingHouseUserConfig"), exports);
+__exportStar(require("./clearingHouse"), exports);
+__exportStar(require("./factory/oracleClient"), exports);
+__exportStar(require("./factory/bigNum"), exports);
+__exportStar(require("./events/types"), exports);
+__exportStar(require("./events/eventSubscriber"), exports);
+__exportStar(require("./math/auction"), exports);
+__exportStar(require("./math/conversion"), exports);
+__exportStar(require("./math/funding"), exports);
+__exportStar(require("./math/market"), exports);
+__exportStar(require("./math/position"), exports);
+__exportStar(require("./math/oracles"), exports);
+__exportStar(require("./math/amm"), exports);
+__exportStar(require("./math/trade"), exports);
+__exportStar(require("./math/orders"), exports);
+__exportStar(require("./math/repeg"), exports);
+__exportStar(require("./orders"), exports);
+__exportStar(require("./orderParams"), exports);
+__exportStar(require("./slot/SlotSubscriber"), exports);
+__exportStar(require("./wallet"), exports);
+__exportStar(require("./types"), exports);
+__exportStar(require("./math/utils"), exports);
+__exportStar(require("./config"), exports);
+__exportStar(require("./constants/numericConstants"), exports);
+__exportStar(require("./tx/retryTxSender"), exports);
+__exportStar(require("./util/computeUnits"), exports);
+__exportStar(require("./util/tps"), exports);
+__exportStar(require("./math/bankBalance"), exports);
+__exportStar(require("./constants/banks"), exports);
+__exportStar(require("./clearingHouseConfig"), exports);