@kamino-finance/klend-sdk 2.10.6

package/dist/utils/oracle.js

@@ -0,0 +1,295 @@
+"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());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cacheOrGetScopePrice = exports.cacheOrGetSwitchboardPrice = exports.cacheOrGetPythPrices = exports.getAllOracleAccounts = exports.getTokenOracleData = exports.CONFIDENCE_FACTOR = exports.MAX_CONFIDENCE_PERCENTAGE = void 0;
+const client_1 = require("@pythnetwork/client");
+const web3_js_1 = require("@solana/web3.js");
+const decimal_js_1 = __importDefault(require("decimal.js"));
+// eslint-disable-next-line import/no-extraneous-dependencies
+const scope_sdk_1 = require("@hubbleprotocol/scope-sdk");
+const pubkey_1 = require("./pubkey");
+const classes_1 = require("../classes");
+const sbv2_lite_1 = __importDefault(require("@switchboard-xyz/sbv2-lite"));
+const kamino_sdk_1 = require("@hubbleprotocol/kamino-sdk");
+const SWITCHBOARD_V2_PROGRAM_ID = new web3_js_1.PublicKey('SW1TCH7qEPTdLsDHRgPuMQjbQxKdH2aBStViMFnt64f');
+// validate price confidence - confidence/price ratio should be less than 2%
+exports.MAX_CONFIDENCE_PERCENTAGE = new decimal_js_1.default('2');
+/// Confidence factor is used to scale the confidence value to a value that can be compared to the price.
+exports.CONFIDENCE_FACTOR = new decimal_js_1.default('100').div(exports.MAX_CONFIDENCE_PERCENTAGE);
+const getScopeAddress = () => {
+    return 'HFn8GnPADiny6XqUoWE8uRPPxb29ikn4yTuPa9MF2fWJ';
+};
+// TODO: Add freshness of the latest price to match sc logic
+function getTokenOracleData(connection, reserves) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const allOracleAccounts = yield getAllOracleAccounts(connection, reserves);
+        const tokenOracleDataForReserves = [];
+        const pythCache = new pubkey_1.PubkeyHashMap();
+        const switchboardCache = new pubkey_1.PubkeyHashMap();
+        const scopeCache = new pubkey_1.PubkeyHashMap();
+        let switchboardV2;
+        for (const reserve of reserves) {
+            let currentBest = undefined;
+            const oracle = {
+                pythAddress: reserve.config.tokenInfo.pythConfiguration.price,
+                switchboardFeedAddress: reserve.config.tokenInfo.switchboardConfiguration.priceAggregator,
+                switchboardTwapAddress: reserve.config.tokenInfo.switchboardConfiguration.twapAggregator,
+                scopeOracleAddress: reserve.config.tokenInfo.scopeConfiguration.priceFeed,
+            };
+            if ((0, pubkey_1.isNotNullPubkey)(oracle.pythAddress)) {
+                const pythPrices = cacheOrGetPythPrices(oracle.pythAddress, pythCache, allOracleAccounts);
+                if (pythPrices && pythPrices.spot) {
+                    currentBest = getBestPrice(currentBest, pythPrices.spot);
+                }
+            }
+            if ((0, pubkey_1.isNotNullPubkey)(oracle.switchboardFeedAddress)) {
+                if (!switchboardV2) {
+                    switchboardV2 = yield sbv2_lite_1.default.loadMainnet(connection);
+                }
+                const switchboardPrice = cacheOrGetSwitchboardPrice(oracle.switchboardFeedAddress, switchboardCache, allOracleAccounts, switchboardV2);
+                if (switchboardPrice) {
+                    currentBest = getBestPrice(currentBest, switchboardPrice);
+                }
+            }
+            if ((0, pubkey_1.isNotNullPubkey)(oracle.scopeOracleAddress)) {
+                const scopePrice = cacheOrGetScopePrice(oracle.scopeOracleAddress, scopeCache, allOracleAccounts, reserve.config.tokenInfo.scopeConfiguration.priceChain);
+                if (scopePrice) {
+                    currentBest = getBestPrice(currentBest, scopePrice);
+                }
+            }
+            if (!currentBest) {
+                console.error(`No price found for reserve: ${(0, classes_1.parseTokenSymbol)(reserve.config.tokenInfo.name)}`);
+                tokenOracleDataForReserves.push([reserve, undefined]);
+                continue;
+            }
+            const tokenOracleData = {
+                mintAddress: reserve.liquidity.mintPubkey,
+                decimals: decimal_js_1.default.pow(10, reserve.liquidity.mintDecimals.toString()),
+                price: new decimal_js_1.default(currentBest.price),
+                timestamp: currentBest.timestamp,
+                valid: currentBest.valid,
+            };
+            tokenOracleDataForReserves.push([reserve, tokenOracleData]);
+        }
+        return tokenOracleDataForReserves;
+    });
+}
+exports.getTokenOracleData = getTokenOracleData;
+function getAllOracleAccounts(connection, reserves) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const allAccounts = [];
+        reserves.forEach((reserve) => {
+            if ((0, pubkey_1.isNotNullPubkey)(reserve.config.tokenInfo.pythConfiguration.price)) {
+                allAccounts.push(reserve.config.tokenInfo.pythConfiguration.price);
+            }
+            if ((0, pubkey_1.isNotNullPubkey)(reserve.config.tokenInfo.switchboardConfiguration.priceAggregator)) {
+                allAccounts.push(reserve.config.tokenInfo.switchboardConfiguration.priceAggregator);
+            }
+            if ((0, pubkey_1.isNotNullPubkey)(reserve.config.tokenInfo.switchboardConfiguration.twapAggregator)) {
+                allAccounts.push(reserve.config.tokenInfo.switchboardConfiguration.twapAggregator);
+            }
+            if ((0, pubkey_1.isNotNullPubkey)(reserve.config.tokenInfo.scopeConfiguration.priceFeed)) {
+                allAccounts.push(reserve.config.tokenInfo.scopeConfiguration.priceFeed);
+            }
+        });
+        const allAccountsDeduped = dedupKeys(allAccounts);
+        const allAccs = yield (0, kamino_sdk_1.batchFetch)(allAccountsDeduped, (chunk) => connection.getMultipleAccountsInfo(chunk));
+        const allAccsMap = new pubkey_1.PubkeyHashMap();
+        allAccs.forEach((acc, i) => {
+            if (acc) {
+                allAccsMap.set(allAccountsDeduped[i], acc);
+            }
+        });
+        return allAccsMap;
+    });
+}
+exports.getAllOracleAccounts = getAllOracleAccounts;
+function dedupKeys(keys) {
+    return new pubkey_1.PublicKeySet(keys).toArray();
+}
+/**
+ * Get pyth price from cache or fetch if not available
+ * @param oracle oracle address
+ * @param cache pyth cache
+ * @param oracleAccounts all oracle accounts
+ */
+function cacheOrGetPythPrices(oracle, cache, oracleAccounts) {
+    const prices = {};
+    const cached = cache.get(oracle);
+    if (cached) {
+        return cached;
+    }
+    else {
+        const result = oracleAccounts.get(oracle);
+        if (result) {
+            try {
+                const { price, timestamp, emaPrice, previousPrice, previousTimestamp, confidence } = (0, client_1.parsePriceData)(result.data);
+                if (price) {
+                    const px = new decimal_js_1.default(price);
+                    prices.spot = {
+                        price: px,
+                        timestamp,
+                        valid: validatePythPx(px, confidence),
+                    };
+                }
+                else {
+                    prices.spot = {
+                        price: new decimal_js_1.default(previousPrice),
+                        timestamp: previousTimestamp,
+                        valid: false,
+                    };
+                }
+                if (emaPrice !== undefined && emaPrice !== null) {
+                    prices.twap = {
+                        price: new decimal_js_1.default(emaPrice.value),
+                        timestamp,
+                        valid: true,
+                    };
+                }
+                if (prices.spot || prices.twap) {
+                    cache.set(oracle, prices);
+                }
+            }
+            catch (error) {
+                console.error(`Error parsing pyth price account ${oracle.toString()} data`, error);
+                return null;
+            }
+        }
+        else {
+            return null;
+        }
+    }
+    return prices;
+}
+exports.cacheOrGetPythPrices = cacheOrGetPythPrices;
+/**
+ * Get switchboard price from cache or fetch if not available
+ * @param oracle oracle address
+ * @param switchboardCache cache for oracle prices
+ * @param oracleAccounts all oracle accounts
+ * @param switchboardV2 loaded switchboard program
+ */
+function cacheOrGetSwitchboardPrice(oracle, switchboardCache, oracleAccounts, switchboardV2) {
+    const cached = switchboardCache.get(oracle);
+    if (cached) {
+        return cached;
+    }
+    else {
+        const info = oracleAccounts.get(oracle);
+        if (info) {
+            if (info.owner.equals(SWITCHBOARD_V2_PROGRAM_ID)) {
+                const agg = switchboardV2.decodeAggregator(info);
+                // @ts-ignore
+                const result = switchboardV2.getLatestAggregatorValue(agg);
+                if (result !== undefined && result !== null) {
+                    const switchboardPx = new decimal_js_1.default(result.toString());
+                    const latestRoundTimestamp = agg.latestConfirmedRound.roundOpenTimestamp;
+                    const ts = BigInt(latestRoundTimestamp.toString());
+                    const valid = validateSwitchboardV2Px(agg);
+                    return {
+                        price: switchboardPx,
+                        timestamp: ts,
+                        valid,
+                    };
+                }
+            }
+            else {
+                console.error('Unrecognized switchboard owner address: ', info.owner.toString());
+                return null;
+            }
+        }
+    }
+    return null;
+}
+exports.cacheOrGetSwitchboardPrice = cacheOrGetSwitchboardPrice;
+/**
+ * Get scope price from cache or fetch if not available
+ * @param oracle oracle address
+ * @param scopeCache cache for oracle prices
+ * @param allOracleAccounts all oracle accounts
+ * @param chain scope chain
+ */
+function cacheOrGetScopePrice(oracle, scopeCache, allOracleAccounts, chain) {
+    if (!(0, pubkey_1.isNotNullPubkey)(oracle) || !chain || !scope_sdk_1.Scope.isScopeChainValid(chain)) {
+        return null;
+    }
+    const scopePrices = scopeCache.get(oracle);
+    if (scopePrices) {
+        return scopeChainToCandidatePrice(chain, scopePrices);
+    }
+    const info = allOracleAccounts.get(oracle);
+    if (info) {
+        const owner = info.owner.toString();
+        if (owner === getScopeAddress()) {
+            const prices = scope_sdk_1.OraclePrices.decode(info.data);
+            scopeCache.set(oracle, prices);
+            return scopeChainToCandidatePrice(chain, prices);
+        }
+        else {
+            console.error('Unrecognized scope owner address: ', owner);
+        }
+    }
+    return null;
+}
+exports.cacheOrGetScopePrice = cacheOrGetScopePrice;
+function getBestPrice(current, next) {
+    if (isBetterPrice(current, next)) {
+        return next;
+    }
+    return current;
+}
+function isBetterPrice(current, next) {
+    if (!current) {
+        return true;
+    }
+    if (current.valid && !next.valid) {
+        return false;
+    }
+    if (!current.valid && next.valid) {
+        return true;
+    }
+    return next.timestamp > current.timestamp;
+}
+function validatePythPx(price, confidence) {
+    const conf50x = new decimal_js_1.default(confidence || 0).mul(exports.CONFIDENCE_FACTOR);
+    return price.gt(conf50x);
+}
+function validateSwitchboardV2Px(agg) {
+    const pxMantissa = new decimal_js_1.default(agg.latestConfirmedRound.result.mantissa.toString());
+    const pxScale = new decimal_js_1.default(agg.latestConfirmedRound.result.scale.toString());
+    const stDevMantissa = new decimal_js_1.default(agg.latestConfirmedRound.stdDeviation.mantissa.toString());
+    const stDevScale = new decimal_js_1.default(agg.latestConfirmedRound.stdDeviation.scale.toString());
+    let conf50xScaled;
+    if (pxScale.gte(stDevScale)) {
+        const scalingFactor = pxScale.sub(stDevScale);
+        const conf50x = stDevMantissa.mul(exports.CONFIDENCE_FACTOR);
+        conf50xScaled = conf50x.mul(scalingFactor);
+    }
+    else {
+        const scalingFactor = stDevScale.sub(pxScale);
+        const conf50x = stDevMantissa.mul(exports.CONFIDENCE_FACTOR);
+        conf50xScaled = conf50x.div(scalingFactor);
+    }
+    return conf50xScaled.gte(pxMantissa);
+}
+function scopeChainToCandidatePrice(chain, prices) {
+    const scopePx = scope_sdk_1.Scope.getPriceFromScopeChain(chain, prices);
+    const valid = scopePx.timestamp.gt('0'); // scope prices are pre-validated
+    return {
+        price: scopePx.price,
+        timestamp: BigInt(scopePx.timestamp.toString()),
+        valid,
+    };
+}
+//# sourceMappingURL=oracle.js.map

package/dist/utils/oracle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oracle.js","sourceRoot":"","sources":["../../src/utils/oracle.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,gDAAqD;AACrD,6CAAqE;AACrE,4DAAiC;AACjC,6DAA6D;AAC7D,yDAAgE;AAChE,qCAAwE;AACxE,wCAA8C;AAC9C,2EAA4D;AAE5D,2DAAwD;AAGxD,MAAM,yBAAyB,GAAG,IAAI,mBAAS,CAAC,6CAA6C,CAAC,CAAC;AAE/F,4EAA4E;AAC/D,QAAA,yBAAyB,GAAY,IAAI,oBAAO,CAAC,GAAG,CAAC,CAAC;AAEnE,yGAAyG;AAC5F,QAAA,iBAAiB,GAAY,IAAI,oBAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,iCAAyB,CAAC,CAAC;AAE5F,MAAM,eAAe,GAAG,GAAG,EAAE;IAC3B,OAAO,8CAA8C,CAAC;AACxD,CAAC,CAAC;AAgBF,4DAA4D;AAC5D,SAAsB,kBAAkB,CACtC,UAAsB,EACtB,QAAmB;;QAEnB,MAAM,iBAAiB,GAAG,MAAM,oBAAoB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC3E,MAAM,0BAA0B,GAAkD,EAAE,CAAC;QACrF,MAAM,SAAS,GAAG,IAAI,sBAAa,EAAyB,CAAC;QAC7D,MAAM,gBAAgB,GAAG,IAAI,sBAAa,EAA6B,CAAC;QACxE,MAAM,UAAU,GAAG,IAAI,sBAAa,EAA2B,CAAC;QAChE,IAAI,aAA6C,CAAC;QAClD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,IAAI,WAAW,GAA+B,SAAS,CAAC;YACxD,MAAM,MAAM,GAAG;gBACb,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK;gBAC7D,sBAAsB,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,eAAe;gBACzF,sBAAsB,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,cAAc;gBACxF,kBAAkB,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,SAAS;aAC1E,CAAC;YACF,IAAI,IAAA,wBAAe,EAAC,MAAM,CAAC,WAAW,CAAC,EAAE;gBACvC,MAAM,UAAU,GAAG,oBAAoB,CAAC,MAAM,CAAC,WAAW,EAAE,SAAS,EAAE,iBAAiB,CAAC,CAAC;gBAC1F,IAAI,UAAU,IAAI,UAAU,CAAC,IAAI,EAAE;oBACjC,WAAW,GAAG,YAAY,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;iBAC1D;aACF;YACD,IAAI,IAAA,wBAAe,EAAC,MAAM,CAAC,sBAAsB,CAAC,EAAE;gBAClD,IAAI,CAAC,aAAa,EAAE;oBAClB,aAAa,GAAG,MAAM,mBAAkB,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;iBAClE;gBACD,MAAM,gBAAgB,GAAG,0BAA0B,CACjD,MAAM,CAAC,sBAAsB,EAC7B,gBAAgB,EAChB,iBAAiB,EACjB,aAAa,CACd,CAAC;gBACF,IAAI,gBAAgB,EAAE;oBACpB,WAAW,GAAG,YAAY,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC;iBAC3D;aACF;YAED,IAAI,IAAA,wBAAe,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE;gBAC9C,MAAM,UAAU,GAAG,oBAAoB,CACrC,MAAM,CAAC,kBAAkB,EACzB,UAAU,EACV,iBAAiB,EACjB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,UAAU,CACvD,CAAC;gBACF,IAAI,UAAU,EAAE;oBACd,WAAW,GAAG,YAAY,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;iBACrD;aACF;YAED,IAAI,CAAC,WAAW,EAAE;gBAChB,OAAO,CAAC,KAAK,CAAC,+BAA+B,IAAA,0BAAgB,EAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAChG,0BAA0B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;gBACtD,SAAS;aACV;YACD,MAAM,eAAe,GAAoB;gBACvC,WAAW,EAAE,OAAO,CAAC,SAAS,CAAC,UAAU;gBACzC,QAAQ,EAAE,oBAAO,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;gBACpE,KAAK,EAAE,IAAI,oBAAO,CAAC,WAAW,CAAC,KAAK,CAAC;gBACrC,SAAS,EAAE,WAAW,CAAC,SAAS;gBAChC,KAAK,EAAE,WAAW,CAAC,KAAK;aACzB,CAAC;YACF,0BAA0B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC;SAC7D;QACD,OAAO,0BAA0B,CAAC;IACpC,CAAC;CAAA;AAlED,gDAkEC;AAID,SAAsB,oBAAoB,CAAC,UAAsB,EAAE,QAAmB;;QACpF,MAAM,WAAW,GAAgB,EAAE,CAAC;QACpC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,IAAA,wBAAe,EAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE;gBACrE,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;aACpE;YACD,IAAI,IAAA,wBAAe,EAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,eAAe,CAAC,EAAE;gBACtF,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,eAAe,CAAC,CAAC;aACrF;YACD,IAAI,IAAA,wBAAe,EAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,cAAc,CAAC,EAAE;gBACrF,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,wBAAwB,CAAC,cAAc,CAAC,CAAC;aACpF;YACD,IAAI,IAAA,wBAAe,EAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,SAAS,CAAC,EAAE;gBAC1E,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;aACzE;QACH,CAAC,CAAC,CAAC;QACH,MAAM,kBAAkB,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,MAAM,IAAA,uBAAU,EAAC,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3G,MAAM,UAAU,GAAG,IAAI,sBAAa,EAAkC,CAAC;QACvE,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YACzB,IAAI,GAAG,EAAE;gBACP,UAAU,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;aAC5C;QACH,CAAC,CAAC,CAAC;QACH,OAAO,UAAU,CAAC;IACpB,CAAC;CAAA;AAzBD,oDAyBC;AAED,SAAS,SAAS,CAAC,IAAiB;IAClC,OAAO,IAAI,qBAAY,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;AAC1C,CAAC;AAOD;;;;;GAKG;AACH,SAAgB,oBAAoB,CAClC,MAAiB,EACjB,KAAiC,EACjC,cAAiC;IAEjC,MAAM,MAAM,GAAe,EAAE,CAAC;IAC9B,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjC,IAAI,MAAM,EAAE;QACV,OAAO,MAAM,CAAC;KACf;SAAM;QACL,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,MAAM,EAAE;YACV,IAAI;gBACF,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,iBAAiB,EAAE,UAAU,EAAE,GAAG,IAAA,uBAAc,EACjG,MAAM,CAAC,IAAI,CACZ,CAAC;gBACF,IAAI,KAAK,EAAE;oBACT,MAAM,EAAE,GAAG,IAAI,oBAAO,CAAC,KAAK,CAAC,CAAC;oBAC9B,MAAM,CAAC,IAAI,GAAG;wBACZ,KAAK,EAAE,EAAE;wBACT,SAAS;wBACT,KAAK,EAAE,cAAc,CAAC,EAAE,EAAE,UAAU,CAAC;qBACtC,CAAC;iBACH;qBAAM;oBACL,MAAM,CAAC,IAAI,GAAG;wBACZ,KAAK,EAAE,IAAI,oBAAO,CAAC,aAAa,CAAC;wBACjC,SAAS,EAAE,iBAAiB;wBAC5B,KAAK,EAAE,KAAK;qBACb,CAAC;iBACH;gBACD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,EAAE;oBAC/C,MAAM,CAAC,IAAI,GAAG;wBACZ,KAAK,EAAE,IAAI,oBAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;wBAClC,SAAS;wBACT,KAAK,EAAE,IAAI;qBACZ,CAAC;iBACH;gBACD,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,EAAE;oBAC9B,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;iBAC3B;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,oCAAoC,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;gBACnF,OAAO,IAAI,CAAC;aACb;SACF;aAAM;YACL,OAAO,IAAI,CAAC;SACb;KACF;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAjDD,oDAiDC;AAED;;;;;;GAMG;AACH,SAAgB,0BAA0B,CACxC,MAAiB,EACjB,gBAAgD,EAChD,cAAiC,EACjC,aAAiC;IAEjC,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,MAAM,EAAE;QACV,OAAO,MAAM,CAAC;KACf;SAAM;QACL,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,IAAI,EAAE;YACR,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,yBAAyB,CAAC,EAAE;gBAChD,MAAM,GAAG,GAAG,aAAa,CAAC,gBAAgB,CAAC,IAAK,CAAC,CAAC;gBAClD,aAAa;gBACb,MAAM,MAAM,GAAmB,aAAa,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;gBAC3E,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI,EAAE;oBAC3C,MAAM,aAAa,GAAG,IAAI,oBAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACrD,MAAM,oBAAoB,GAAO,GAAG,CAAC,oBAAoB,CAAC,kBAAkB,CAAC;oBAC7E,MAAM,EAAE,GAAG,MAAM,CAAC,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACnD,MAAM,KAAK,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC;oBAC3C,OAAO;wBACL,KAAK,EAAE,aAAa;wBACpB,SAAS,EAAE,EAAE;wBACb,KAAK;qBACN,CAAC;iBACH;aACF;iBAAM;gBACL,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACjF,OAAO,IAAI,CAAC;aACb;SACF;KACF;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAlCD,gEAkCC;AAED;;;;;;GAMG;AACH,SAAgB,oBAAoB,CAClC,MAAiB,EACjB,UAAwC,EACxC,iBAAoC,EACpC,KAAe;IAEf,IAAI,CAAC,IAAA,wBAAe,EAAC,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,iBAAK,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE;QACzE,OAAO,IAAI,CAAC;KACb;IAED,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,WAAW,EAAE;QACf,OAAO,0BAA0B,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;KACvD;IACD,MAAM,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,IAAI,EAAE;QACR,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpC,IAAI,KAAK,KAAK,eAAe,EAAE,EAAE;YAC/B,MAAM,MAAM,GAAG,wBAAY,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC/B,OAAO,0BAA0B,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;SAClD;aAAM;YACL,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;SAC5D;KACF;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AA3BD,oDA2BC;AAED,SAAS,YAAY,CAAC,OAAmC,EAAE,IAAoB;IAC7E,IAAI,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE;QAChC,OAAO,IAAI,CAAC;KACb;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,aAAa,CAAC,OAAmC,EAAE,IAAoB;IAC9E,IAAI,CAAC,OAAO,EAAE;QACZ,OAAO,IAAI,CAAC;KACb;IACD,IAAI,OAAO,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;QAChC,OAAO,KAAK,CAAC;KACd;IACD,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,EAAE;QAChC,OAAO,IAAI,CAAC;KACb;IACD,OAAO,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;AAC5C,CAAC;AAED,SAAS,cAAc,CAAC,KAAc,EAAE,UAA8B;IACpE,MAAM,OAAO,GAAG,IAAI,oBAAO,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,yBAAiB,CAAC,CAAC;IACpE,OAAO,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,uBAAuB,CAAC,GAAQ;IACvC,MAAM,UAAU,GAAG,IAAI,oBAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;IACpF,MAAM,OAAO,GAAG,IAAI,oBAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC9E,MAAM,aAAa,GAAG,IAAI,oBAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,MAAM,UAAU,GAAG,IAAI,oBAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACvF,IAAI,aAAsB,CAAC;IAC3B,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;QAC3B,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,yBAAiB,CAAC,CAAC;QACrD,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;KAC5C;SAAM;QACL,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,yBAAiB,CAAC,CAAC;QACrD,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;KAC5C;IACD,OAAO,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;AACvC,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAe,EAAE,MAAoB;IACvE,MAAM,OAAO,GAAG,iBAAK,CAAC,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,iCAAiC;IAC1E,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC/C,KAAK;KACN,CAAC;AACJ,CAAC","sourcesContent":["import { parsePriceData } from '@pythnetwork/client';\nimport { AccountInfo, Connection, PublicKey } from '@solana/web3.js';\nimport Decimal from 'decimal.js';\n// eslint-disable-next-line import/no-extraneous-dependencies\nimport { OraclePrices, Scope } from '@hubbleprotocol/scope-sdk';\nimport { isNotNullPubkey, PubkeyHashMap, PublicKeySet } from './pubkey';\nimport { parseTokenSymbol } from '../classes';\nimport SwitchboardProgram from '@switchboard-xyz/sbv2-lite';\nimport { Reserve } from '../lib';\nimport { batchFetch } from '@hubbleprotocol/kamino-sdk';\nimport BN from 'bn.js';\n\nconst SWITCHBOARD_V2_PROGRAM_ID = new PublicKey('SW1TCH7qEPTdLsDHRgPuMQjbQxKdH2aBStViMFnt64f');\n\n// validate price confidence - confidence/price ratio should be less than 2%\nexport const MAX_CONFIDENCE_PERCENTAGE: Decimal = new Decimal('2');\n\n/// Confidence factor is used to scale the confidence value to a value that can be compared to the price.\nexport const CONFIDENCE_FACTOR: Decimal = new Decimal('100').div(MAX_CONFIDENCE_PERCENTAGE);\n\nconst getScopeAddress = () => {\n  return 'HFn8GnPADiny6XqUoWE8uRPPxb29ikn4yTuPa9MF2fWJ';\n};\n\nexport type TokenOracleData = {\n  mintAddress: PublicKey;\n  decimals: Decimal;\n  price: Decimal;\n  timestamp: bigint;\n  valid: boolean;\n};\n\nexport type CandidatePrice = {\n  price: Decimal;\n  timestamp: bigint;\n  valid: boolean;\n};\n\n// TODO: Add freshness of the latest price to match sc logic\nexport async function getTokenOracleData(\n  connection: Connection,\n  reserves: Reserve[]\n): Promise<Array<[Reserve, TokenOracleData | undefined]>> {\n  const allOracleAccounts = await getAllOracleAccounts(connection, reserves);\n  const tokenOracleDataForReserves: Array<[Reserve, TokenOracleData | undefined]> = [];\n  const pythCache = new PubkeyHashMap<PublicKey, PythPrices>();\n  const switchboardCache = new PubkeyHashMap<PublicKey, CandidatePrice>();\n  const scopeCache = new PubkeyHashMap<PublicKey, OraclePrices>();\n  let switchboardV2: SwitchboardProgram | undefined;\n  for (const reserve of reserves) {\n    let currentBest: CandidatePrice | undefined = undefined;\n    const oracle = {\n      pythAddress: reserve.config.tokenInfo.pythConfiguration.price,\n      switchboardFeedAddress: reserve.config.tokenInfo.switchboardConfiguration.priceAggregator,\n      switchboardTwapAddress: reserve.config.tokenInfo.switchboardConfiguration.twapAggregator,\n      scopeOracleAddress: reserve.config.tokenInfo.scopeConfiguration.priceFeed,\n    };\n    if (isNotNullPubkey(oracle.pythAddress)) {\n      const pythPrices = cacheOrGetPythPrices(oracle.pythAddress, pythCache, allOracleAccounts);\n      if (pythPrices && pythPrices.spot) {\n        currentBest = getBestPrice(currentBest, pythPrices.spot);\n      }\n    }\n    if (isNotNullPubkey(oracle.switchboardFeedAddress)) {\n      if (!switchboardV2) {\n        switchboardV2 = await SwitchboardProgram.loadMainnet(connection);\n      }\n      const switchboardPrice = cacheOrGetSwitchboardPrice(\n        oracle.switchboardFeedAddress,\n        switchboardCache,\n        allOracleAccounts,\n        switchboardV2\n      );\n      if (switchboardPrice) {\n        currentBest = getBestPrice(currentBest, switchboardPrice);\n      }\n    }\n\n    if (isNotNullPubkey(oracle.scopeOracleAddress)) {\n      const scopePrice = cacheOrGetScopePrice(\n        oracle.scopeOracleAddress,\n        scopeCache,\n        allOracleAccounts,\n        reserve.config.tokenInfo.scopeConfiguration.priceChain\n      );\n      if (scopePrice) {\n        currentBest = getBestPrice(currentBest, scopePrice);\n      }\n    }\n\n    if (!currentBest) {\n      console.error(`No price found for reserve: ${parseTokenSymbol(reserve.config.tokenInfo.name)}`);\n      tokenOracleDataForReserves.push([reserve, undefined]);\n      continue;\n    }\n    const tokenOracleData: TokenOracleData = {\n      mintAddress: reserve.liquidity.mintPubkey,\n      decimals: Decimal.pow(10, reserve.liquidity.mintDecimals.toString()),\n      price: new Decimal(currentBest.price),\n      timestamp: currentBest.timestamp,\n      valid: currentBest.valid,\n    };\n    tokenOracleDataForReserves.push([reserve, tokenOracleData]);\n  }\n  return tokenOracleDataForReserves;\n}\n\nexport type AllOracleAccounts = PubkeyHashMap<PublicKey, AccountInfo<Buffer>>;\n\nexport async function getAllOracleAccounts(connection: Connection, reserves: Reserve[]): Promise<AllOracleAccounts> {\n  const allAccounts: PublicKey[] = [];\n  reserves.forEach((reserve) => {\n    if (isNotNullPubkey(reserve.config.tokenInfo.pythConfiguration.price)) {\n      allAccounts.push(reserve.config.tokenInfo.pythConfiguration.price);\n    }\n    if (isNotNullPubkey(reserve.config.tokenInfo.switchboardConfiguration.priceAggregator)) {\n      allAccounts.push(reserve.config.tokenInfo.switchboardConfiguration.priceAggregator);\n    }\n    if (isNotNullPubkey(reserve.config.tokenInfo.switchboardConfiguration.twapAggregator)) {\n      allAccounts.push(reserve.config.tokenInfo.switchboardConfiguration.twapAggregator);\n    }\n    if (isNotNullPubkey(reserve.config.tokenInfo.scopeConfiguration.priceFeed)) {\n      allAccounts.push(reserve.config.tokenInfo.scopeConfiguration.priceFeed);\n    }\n  });\n  const allAccountsDeduped = dedupKeys(allAccounts);\n  const allAccs = await batchFetch(allAccountsDeduped, (chunk) => connection.getMultipleAccountsInfo(chunk));\n  const allAccsMap = new PubkeyHashMap<PublicKey, AccountInfo<Buffer>>();\n  allAccs.forEach((acc, i) => {\n    if (acc) {\n      allAccsMap.set(allAccountsDeduped[i], acc);\n    }\n  });\n  return allAccsMap;\n}\n\nfunction dedupKeys(keys: PublicKey[]): PublicKey[] {\n  return new PublicKeySet(keys).toArray();\n}\n\nexport type PythPrices = {\n  spot?: CandidatePrice;\n  twap?: CandidatePrice;\n};\n\n/**\n * Get pyth price from cache or fetch if not available\n * @param oracle oracle address\n * @param cache pyth cache\n * @param oracleAccounts all oracle accounts\n */\nexport function cacheOrGetPythPrices(\n  oracle: PublicKey,\n  cache: Map<PublicKey, PythPrices>,\n  oracleAccounts: AllOracleAccounts\n): PythPrices | null {\n  const prices: PythPrices = {};\n  const cached = cache.get(oracle);\n  if (cached) {\n    return cached;\n  } else {\n    const result = oracleAccounts.get(oracle);\n    if (result) {\n      try {\n        const { price, timestamp, emaPrice, previousPrice, previousTimestamp, confidence } = parsePriceData(\n          result.data\n        );\n        if (price) {\n          const px = new Decimal(price);\n          prices.spot = {\n            price: px,\n            timestamp,\n            valid: validatePythPx(px, confidence),\n          };\n        } else {\n          prices.spot = {\n            price: new Decimal(previousPrice),\n            timestamp: previousTimestamp,\n            valid: false,\n          };\n        }\n        if (emaPrice !== undefined && emaPrice !== null) {\n          prices.twap = {\n            price: new Decimal(emaPrice.value),\n            timestamp,\n            valid: true,\n          };\n        }\n        if (prices.spot || prices.twap) {\n          cache.set(oracle, prices);\n        }\n      } catch (error) {\n        console.error(`Error parsing pyth price account ${oracle.toString()} data`, error);\n        return null;\n      }\n    } else {\n      return null;\n    }\n  }\n  return prices;\n}\n\n/**\n * Get switchboard price from cache or fetch if not available\n * @param oracle oracle address\n * @param switchboardCache cache for oracle prices\n * @param oracleAccounts all oracle accounts\n * @param switchboardV2 loaded switchboard program\n */\nexport function cacheOrGetSwitchboardPrice(\n  oracle: PublicKey,\n  switchboardCache: Map<PublicKey, CandidatePrice>,\n  oracleAccounts: AllOracleAccounts,\n  switchboardV2: SwitchboardProgram\n): CandidatePrice | null {\n  const cached = switchboardCache.get(oracle);\n  if (cached) {\n    return cached;\n  } else {\n    const info = oracleAccounts.get(oracle);\n    if (info) {\n      if (info.owner.equals(SWITCHBOARD_V2_PROGRAM_ID)) {\n        const agg = switchboardV2.decodeAggregator(info!);\n        // @ts-ignore\n        const result: Big.Big | null = switchboardV2.getLatestAggregatorValue(agg);\n        if (result !== undefined && result !== null) {\n          const switchboardPx = new Decimal(result.toString());\n          const latestRoundTimestamp: BN = agg.latestConfirmedRound.roundOpenTimestamp;\n          const ts = BigInt(latestRoundTimestamp.toString());\n          const valid = validateSwitchboardV2Px(agg);\n          return {\n            price: switchboardPx,\n            timestamp: ts,\n            valid,\n          };\n        }\n      } else {\n        console.error('Unrecognized switchboard owner address: ', info.owner.toString());\n        return null;\n      }\n    }\n  }\n  return null;\n}\n\n/**\n * Get scope price from cache or fetch if not available\n * @param oracle oracle address\n * @param scopeCache cache for oracle prices\n * @param allOracleAccounts all oracle accounts\n * @param chain scope chain\n */\nexport function cacheOrGetScopePrice(\n  oracle: PublicKey,\n  scopeCache: Map<PublicKey, OraclePrices>,\n  allOracleAccounts: AllOracleAccounts,\n  chain: number[]\n): CandidatePrice | null {\n  if (!isNotNullPubkey(oracle) || !chain || !Scope.isScopeChainValid(chain)) {\n    return null;\n  }\n\n  const scopePrices = scopeCache.get(oracle);\n  if (scopePrices) {\n    return scopeChainToCandidatePrice(chain, scopePrices);\n  }\n  const info = allOracleAccounts.get(oracle);\n  if (info) {\n    const owner = info.owner.toString();\n    if (owner === getScopeAddress()) {\n      const prices = OraclePrices.decode(info.data);\n      scopeCache.set(oracle, prices);\n      return scopeChainToCandidatePrice(chain, prices);\n    } else {\n      console.error('Unrecognized scope owner address: ', owner);\n    }\n  }\n\n  return null;\n}\n\nfunction getBestPrice(current: CandidatePrice | undefined, next: CandidatePrice): CandidatePrice | undefined {\n  if (isBetterPrice(current, next)) {\n    return next;\n  }\n  return current;\n}\n\nfunction isBetterPrice(current: CandidatePrice | undefined, next: CandidatePrice): boolean {\n  if (!current) {\n    return true;\n  }\n  if (current.valid && !next.valid) {\n    return false;\n  }\n  if (!current.valid && next.valid) {\n    return true;\n  }\n  return next.timestamp > current.timestamp;\n}\n\nfunction validatePythPx(price: Decimal, confidence: number | undefined): boolean {\n  const conf50x = new Decimal(confidence || 0).mul(CONFIDENCE_FACTOR);\n  return price.gt(conf50x);\n}\n\nfunction validateSwitchboardV2Px(agg: any): boolean {\n  const pxMantissa = new Decimal(agg.latestConfirmedRound.result.mantissa.toString());\n  const pxScale = new Decimal(agg.latestConfirmedRound.result.scale.toString());\n  const stDevMantissa = new Decimal(agg.latestConfirmedRound.stdDeviation.mantissa.toString());\n  const stDevScale = new Decimal(agg.latestConfirmedRound.stdDeviation.scale.toString());\n  let conf50xScaled: Decimal;\n  if (pxScale.gte(stDevScale)) {\n    const scalingFactor = pxScale.sub(stDevScale);\n    const conf50x = stDevMantissa.mul(CONFIDENCE_FACTOR);\n    conf50xScaled = conf50x.mul(scalingFactor);\n  } else {\n    const scalingFactor = stDevScale.sub(pxScale);\n    const conf50x = stDevMantissa.mul(CONFIDENCE_FACTOR);\n    conf50xScaled = conf50x.div(scalingFactor);\n  }\n  return conf50xScaled.gte(pxMantissa);\n}\n\nfunction scopeChainToCandidatePrice(chain: number[], prices: OraclePrices): CandidatePrice {\n  const scopePx = Scope.getPriceFromScopeChain(chain, prices);\n  const valid = scopePx.timestamp.gt('0'); // scope prices are pre-validated\n  return {\n    price: scopePx.price,\n    timestamp: BigInt(scopePx.timestamp.toString()),\n    valid,\n  };\n}\n"]}

package/dist/utils/pubkey.d.ts

@@ -0,0 +1,74 @@
+/// <reference types="node" />
+import { PublicKey } from '@solana/web3.js';
+import BN from 'bn.js';
+export declare const NULL_PUBKEY: PublicKey;
+export declare const COMPUTE_BUDGET_PROGRAM_ID: PublicKey;
+/**
+ * Helper function to check if a configured pubkey is null or default.
+ * @param pubkey
+ * @returns {boolean}
+ */
+export declare function isNotNullPubkey(pubkey: PublicKey): boolean;
+/**
+ * Custom set implementation for PublicKey
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom set implementation where keys are compared with `PublicKey.equals(PublicKey)`
+ */
+export declare class PublicKeySet<T extends PublicKey> {
+    private items;
+    constructor(items: Array<T>);
+    add(item: T): void;
+    contains(item: T): boolean;
+    private equals;
+    isEmpty(): boolean;
+    toArray(): T[];
+}
+/**
+ * Custom map implementation for PublicKey, backed by a list for o(n) lookup speed
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`
+ */
+export declare class PublicKeyMap<K extends PublicKey, V> {
+    private record;
+    set(key: K, value: V): void;
+    get(key: K): V | undefined;
+    has(key: K): boolean;
+    delete(key: K): void;
+    private findIndex;
+    clear(): void;
+    forEach(callback: (value: V, key: K) => void): void;
+    keys(): K[];
+    values(): V[];
+    entries(): [K, V][];
+    isEmpty(): boolean;
+}
+/**
+ * Custom map implementation for PublicKey, backed by a hashmap for o(1) lookup speed
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`, and a hash is derived from the final 32 bits of the public key
+ */
+export declare class PubkeyHashMap<K extends PublicKey, V> implements Map<K, V> {
+    private buckets;
+    size: number;
+    constructor(entries?: readonly (readonly [K, V])[] | null);
+    [Symbol.iterator](): IterableIterator<[K, V]>;
+    [Symbol.toStringTag]: string;
+    set(key: K, value: V): this;
+    get(key: K): V | undefined;
+    has(key: K): boolean;
+    delete(key: K): boolean;
+    clear(): void;
+    isEmpty(): boolean;
+    forEach(callbackfn: (value: V, key: K, map: Map<K, V>) => void, thisArg?: any): void;
+    keys(): IterableIterator<K>;
+    values(): IterableIterator<V>;
+    entries(): IterableIterator<[K, V]>;
+}
+export declare class HashablePublicKey extends PublicKey implements IEquality<HashablePublicKey> {
+    static MASK: BN;
+    constructor(value: PublicKey | string | Buffer | Uint8Array | Array<number>);
+    hashCode(): number;
+    private getBN;
+}
+interface IEquality<T extends IEquality<T>> {
+    equals(other: T): boolean;
+    hashCode(): number;
+}
+export {};

package/dist/utils/pubkey.js

@@ -0,0 +1,229 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _a;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HashablePublicKey = exports.PubkeyHashMap = exports.PublicKeyMap = exports.PublicKeySet = exports.isNotNullPubkey = exports.COMPUTE_BUDGET_PROGRAM_ID = exports.NULL_PUBKEY = void 0;
+const web3_js_1 = require("@solana/web3.js");
+const bn_js_1 = __importDefault(require("bn.js"));
+exports.NULL_PUBKEY = new web3_js_1.PublicKey('nu11111111111111111111111111111111111111111');
+exports.COMPUTE_BUDGET_PROGRAM_ID = new web3_js_1.PublicKey('ComputeBudget111111111111111111111111111111');
+/**
+ * Helper function to check if a configured pubkey is null or default.
+ * @param pubkey
+ * @returns {boolean}
+ */
+function isNotNullPubkey(pubkey) {
+    return pubkey && !pubkey.equals(exports.NULL_PUBKEY) && !pubkey.equals(web3_js_1.PublicKey.default);
+}
+exports.isNotNullPubkey = isNotNullPubkey;
+/**
+ * Custom set implementation for PublicKey
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom set implementation where keys are compared with `PublicKey.equals(PublicKey)`
+ */
+class PublicKeySet {
+    constructor(items) {
+        this.items = [];
+        items.forEach((item) => this.add(item));
+    }
+    add(item) {
+        if (!this.contains(item)) {
+            this.items.push(item);
+        }
+    }
+    contains(item) {
+        return this.items.some((existingItem) => this.equals(existingItem, item));
+    }
+    equals(item1, item2) {
+        return item1.equals(item2);
+    }
+    isEmpty() {
+        return this.items.length === 0;
+    }
+    toArray() {
+        return this.items.slice(); // Return a copy of the array to prevent external modifications
+    }
+}
+exports.PublicKeySet = PublicKeySet;
+/**
+ * Custom map implementation for PublicKey, backed by a list for o(n) lookup speed
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`
+ */
+class PublicKeyMap {
+    constructor() {
+        this.record = [];
+    }
+    set(key, value) {
+        const index = this.findIndex(key);
+        if (index !== -1) {
+            // Update existing entry
+            this.record[index].value = value;
+        }
+        else {
+            // Add new entry
+            this.record.push({ key, value });
+        }
+    }
+    get(key) {
+        const index = this.findIndex(key);
+        return index !== -1 ? this.record[index].value : undefined;
+    }
+    has(key) {
+        return this.findIndex(key) !== -1;
+    }
+    delete(key) {
+        const index = this.findIndex(key);
+        if (index !== -1) {
+            this.record.splice(index, 1);
+        }
+    }
+    findIndex(key) {
+        return this.record.findIndex((entry) => entry.key.equals(key));
+    }
+    clear() {
+        this.record = [];
+    }
+    forEach(callback) {
+        this.record.forEach((entry) => {
+            callback(entry.value, entry.key);
+        });
+    }
+    keys() {
+        return this.record.map((entry) => entry.key);
+    }
+    values() {
+        return this.record.map((entry) => entry.value);
+    }
+    entries() {
+        return this.record.map((entry) => [entry.key, entry.value]);
+    }
+    isEmpty() {
+        return this.record.length === 0;
+    }
+}
+exports.PublicKeyMap = PublicKeyMap;
+/**
+ * Custom map implementation for PublicKey, backed by a hashmap for o(1) lookup speed
+ * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`, and a hash is derived from the final 32 bits of the public key
+ */
+class PubkeyHashMap {
+    constructor(entries) {
+        this[_a] = 'PubkeyHashMap';
+        this.buckets = new Map();
+        this.size = 0;
+        if (entries) {
+            for (const [key, value] of entries) {
+                this.set(key, value);
+            }
+        }
+    }
+    *[Symbol.iterator]() {
+        for (const [, bucket] of this.buckets) {
+            for (const { key, value } of bucket) {
+                yield [key, value];
+            }
+        }
+    }
+    set(key, value) {
+        const hash = new HashablePublicKey(key).hashCode();
+        const bucket = this.buckets.get(hash);
+        if (!bucket) {
+            this.buckets.set(hash, [{ key, value }]);
+            this.size++;
+        }
+        else {
+            const entry = bucket.find((entry) => entry.key.equals(key));
+            if (entry) {
+                entry.value = value;
+            }
+            else {
+                bucket.push({ key, value });
+                this.size++;
+            }
+        }
+        return this;
+    }
+    get(key) {
+        const hash = new HashablePublicKey(key).hashCode();
+        const bucket = this.buckets.get(hash);
+        if (!bucket) {
+            return undefined;
+        }
+        const entry = bucket.find((entry) => entry.key.equals(key));
+        return entry ? entry.value : undefined;
+    }
+    has(key) {
+        const hash = new HashablePublicKey(key).hashCode();
+        const bucket = this.buckets.get(hash);
+        if (!bucket) {
+            return false;
+        }
+        const entry = bucket.find((entry) => entry.key.equals(key));
+        return !!entry;
+    }
+    delete(key) {
+        const hash = new HashablePublicKey(key).hashCode();
+        const bucket = this.buckets.get(hash);
+        if (!bucket) {
+            return false;
+        }
+        const index = bucket.findIndex((entry) => entry.key.equals(key));
+        if (index === -1) {
+            return false;
+        }
+        bucket.splice(index, 1);
+        if (bucket.length === 0) {
+            this.buckets.delete(hash);
+        }
+        this.size--;
+        return true;
+    }
+    clear() {
+        this.buckets = new Map();
+        this.size = 0;
+    }
+    isEmpty() {
+        return this.size === 0;
+    }
+    forEach(callbackfn, thisArg) {
+        this.buckets.forEach((bucket) => {
+            bucket.forEach((entry) => {
+                callbackfn(entry.value, entry.key, this);
+            }, thisArg);
+        }, thisArg);
+    }
+    *keys() {
+        for (const [key] of this) {
+            yield key;
+        }
+    }
+    *values() {
+        for (const [, value] of this) {
+            yield value;
+        }
+    }
+    entries() {
+        return this[Symbol.iterator]();
+    }
+}
+exports.PubkeyHashMap = PubkeyHashMap;
+_a = Symbol.toStringTag;
+class HashablePublicKey extends web3_js_1.PublicKey {
+    constructor(value) {
+        super(value);
+    }
+    hashCode() {
+        let hash = 13;
+        hash = hash * 7 + this.getBN().clone().iuand(HashablePublicKey.MASK).toNumber();
+        return hash;
+    }
+    getBN() {
+        //@ts-ignore
+        return this._bn;
+    }
+}
+exports.HashablePublicKey = HashablePublicKey;
+// We only use the last 32 bits of the public key for hashing
+HashablePublicKey.MASK = new bn_js_1.default(1).shln(32).subn(1);
+//# sourceMappingURL=pubkey.js.map

package/dist/utils/pubkey.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pubkey.js","sourceRoot":"","sources":["../../src/utils/pubkey.ts"],"names":[],"mappings":";;;;;;;AAAA,6CAA4C;AAC5C,kDAAuB;AAEV,QAAA,WAAW,GAAG,IAAI,mBAAS,CAAC,6CAA6C,CAAC,CAAC;AAC3E,QAAA,yBAAyB,GAAG,IAAI,mBAAS,CAAC,6CAA6C,CAAC,CAAC;AAEtG;;;;GAIG;AACH,SAAgB,eAAe,CAAC,MAAiB;IAC/C,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAS,CAAC,OAAO,CAAC,CAAC;AACpF,CAAC;AAFD,0CAEC;AAED;;;GAGG;AACH,MAAa,YAAY;IAGvB,YAAY,KAAe;QAFnB,UAAK,GAAQ,EAAE,CAAC;QAGtB,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,CAAC;IAED,GAAG,CAAC,IAAO;QACT,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SACvB;IACH,CAAC;IAED,QAAQ,CAAC,IAAO;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC;IAC5E,CAAC;IAEO,MAAM,CAAC,KAAQ,EAAE,KAAQ;QAC/B,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,+DAA+D;IAC5F,CAAC;CACF;AA5BD,oCA4BC;AAED;;;GAGG;AACH,MAAa,YAAY;IAAzB;QACU,WAAM,GAA2B,EAAE,CAAC;IA0D9C,CAAC;IAxDC,GAAG,CAAC,GAAM,EAAE,KAAQ;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE;YAChB,wBAAwB;YACxB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;SAClC;aAAM;YACL,gBAAgB;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;SAClC;IACH,CAAC;IAED,GAAG,CAAC,GAAM;QACR,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAClC,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7D,CAAC;IAED,GAAG,CAAC,GAAM;QACR,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,CAAC,GAAM;QACX,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;SAC9B;IACH,CAAC;IAEO,SAAS,CAAC,GAAM;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAED,OAAO,CAAC,QAAoC;QAC1C,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YAC5B,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC;IAClC,CAAC;CACF;AA3DD,oCA2DC;AAED;;;GAGG;AACH,MAAa,aAAa;IAUxB,YAAY,OAA6C;QAkBzD,QAAoB,GAAW,eAAe,CAAC;QAjB7C,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QACd,IAAI,OAAO,EAAE;YACX,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,OAAO,EAAE;gBAClC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;aACtB;SACF;IACH,CAAC;IAED,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QAChB,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;YACrC,KAAK,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,MAAM,EAAE;gBACnC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;aACpB;SACF;IACH,CAAC;IAID,GAAG,CAAC,GAAM,EAAE,KAAQ;QAClB,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YACzC,IAAI,CAAC,IAAI,EAAE,CAAC;SACb;aAAM;YACL,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,IAAI,KAAK,EAAE;gBACT,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC;aACrB;iBAAM;gBACL,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC5B,IAAI,CAAC,IAAI,EAAE,CAAC;aACb;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,GAAG,CAAC,GAAM;QACR,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,SAAS,CAAC;SAClB;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5D,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IACzC,CAAC;IAED,GAAG,CAAC,GAAM;QACR,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,KAAK,CAAC;SACd;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5D,OAAO,CAAC,CAAC,KAAK,CAAC;IACjB,CAAC;IAED,MAAM,CAAC,GAAM;QACX,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,KAAK,CAAC;SACd;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QACjE,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE;YAChB,OAAO,KAAK,CAAC;SACd;QACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;SAC3B;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;IAChB,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,UAAsD,EAAE,OAAa;QAC3E,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvB,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC,EAAE,OAAO,CAAC,CAAC;QACd,CAAC,EAAE,OAAO,CAAC,CAAC;IACd,CAAC;IAED,CAAC,IAAI;QACH,KAAK,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE;YACxB,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAED,CAAC,MAAM;QACL,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;YAC5B,MAAM,KAAK,CAAC;SACb;IACH,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;IACjC,CAAC;CACF;AAtHD,sCAsHC;KA1FE,MAAM,CAAC,WAAW;AA4FrB,MAAa,iBAAkB,SAAQ,mBAAS;IAG9C,YAAY,KAA+D;QACzE,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;IAED,QAAQ;QACN,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,GAAG,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;QAChF,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK;QACX,YAAY;QACZ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;;AAhBH,8CAiBC;AAhBC,6DAA6D;AACtD,sBAAI,GAAG,IAAI,eAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC","sourcesContent":["import { PublicKey } from '@solana/web3.js';\nimport BN from 'bn.js';\n\nexport const NULL_PUBKEY = new PublicKey('nu11111111111111111111111111111111111111111');\nexport const COMPUTE_BUDGET_PROGRAM_ID = new PublicKey('ComputeBudget111111111111111111111111111111');\n\n/**\n * Helper function to check if a configured pubkey is null or default.\n * @param pubkey\n * @returns {boolean}\n */\nexport function isNotNullPubkey(pubkey: PublicKey): boolean {\n  return pubkey && !pubkey.equals(NULL_PUBKEY) && !pubkey.equals(PublicKey.default);\n}\n\n/**\n * Custom set implementation for PublicKey\n * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom set implementation where keys are compared with `PublicKey.equals(PublicKey)`\n */\nexport class PublicKeySet<T extends PublicKey> {\n  private items: T[] = [];\n\n  constructor(items: Array<T>) {\n    items.forEach((item) => this.add(item));\n  }\n\n  add(item: T): void {\n    if (!this.contains(item)) {\n      this.items.push(item);\n    }\n  }\n\n  contains(item: T): boolean {\n    return this.items.some((existingItem) => this.equals(existingItem, item));\n  }\n\n  private equals(item1: T, item2: T): boolean {\n    return item1.equals(item2);\n  }\n\n  isEmpty(): boolean {\n    return this.items.length === 0;\n  }\n\n  toArray(): T[] {\n    return this.items.slice(); // Return a copy of the array to prevent external modifications\n  }\n}\n\n/**\n * Custom map implementation for PublicKey, backed by a list for o(n) lookup speed\n * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`\n */\nexport class PublicKeyMap<K extends PublicKey, V> {\n  private record: { key: K; value: V }[] = [];\n\n  set(key: K, value: V): void {\n    const index = this.findIndex(key);\n    if (index !== -1) {\n      // Update existing entry\n      this.record[index].value = value;\n    } else {\n      // Add new entry\n      this.record.push({ key, value });\n    }\n  }\n\n  get(key: K): V | undefined {\n    const index = this.findIndex(key);\n    return index !== -1 ? this.record[index].value : undefined;\n  }\n\n  has(key: K): boolean {\n    return this.findIndex(key) !== -1;\n  }\n\n  delete(key: K): void {\n    const index = this.findIndex(key);\n    if (index !== -1) {\n      this.record.splice(index, 1);\n    }\n  }\n\n  private findIndex(key: K): number {\n    return this.record.findIndex((entry) => entry.key.equals(key));\n  }\n\n  clear(): void {\n    this.record = [];\n  }\n\n  forEach(callback: (value: V, key: K) => void): void {\n    this.record.forEach((entry) => {\n      callback(entry.value, entry.key);\n    });\n  }\n\n  keys(): K[] {\n    return this.record.map((entry) => entry.key);\n  }\n\n  values(): V[] {\n    return this.record.map((entry) => entry.value);\n  }\n\n  entries(): [K, V][] {\n    return this.record.map((entry) => [entry.key, entry.value]);\n  }\n\n  isEmpty(): boolean {\n    return this.record.length === 0;\n  }\n}\n\n/**\n * Custom map implementation for PublicKey, backed by a hashmap for o(1) lookup speed\n * Calling toString() on a PublicKey is pretty slow and pubkeys do not support `SameValueZero` comparison, so we use a custom map implementation where keys are compared with `PublicKey.equals(PublicKey)`, and a hash is derived from the final 32 bits of the public key\n */\nexport class PubkeyHashMap<K extends PublicKey, V> implements Map<K, V> {\n  private buckets: Map<\n    number,\n    {\n      key: K;\n      value: V;\n    }[]\n  >;\n  size: number;\n\n  constructor(entries?: readonly (readonly [K, V])[] | null) {\n    this.buckets = new Map();\n    this.size = 0;\n    if (entries) {\n      for (const [key, value] of entries) {\n        this.set(key, value);\n      }\n    }\n  }\n\n  *[Symbol.iterator](): IterableIterator<[K, V]> {\n    for (const [, bucket] of this.buckets) {\n      for (const { key, value } of bucket) {\n        yield [key, value];\n      }\n    }\n  }\n\n  [Symbol.toStringTag]: string = 'PubkeyHashMap';\n\n  set(key: K, value: V): this {\n    const hash = new HashablePublicKey(key).hashCode();\n    const bucket = this.buckets.get(hash);\n    if (!bucket) {\n      this.buckets.set(hash, [{ key, value }]);\n      this.size++;\n    } else {\n      const entry = bucket.find((entry) => entry.key.equals(key));\n      if (entry) {\n        entry.value = value;\n      } else {\n        bucket.push({ key, value });\n        this.size++;\n      }\n    }\n    return this;\n  }\n\n  get(key: K): V | undefined {\n    const hash = new HashablePublicKey(key).hashCode();\n    const bucket = this.buckets.get(hash);\n    if (!bucket) {\n      return undefined;\n    }\n    const entry = bucket.find((entry) => entry.key.equals(key));\n    return entry ? entry.value : undefined;\n  }\n\n  has(key: K): boolean {\n    const hash = new HashablePublicKey(key).hashCode();\n    const bucket = this.buckets.get(hash);\n    if (!bucket) {\n      return false;\n    }\n    const entry = bucket.find((entry) => entry.key.equals(key));\n    return !!entry;\n  }\n\n  delete(key: K): boolean {\n    const hash = new HashablePublicKey(key).hashCode();\n    const bucket = this.buckets.get(hash);\n    if (!bucket) {\n      return false;\n    }\n    const index = bucket.findIndex((entry) => entry.key.equals(key));\n    if (index === -1) {\n      return false;\n    }\n    bucket.splice(index, 1);\n    if (bucket.length === 0) {\n      this.buckets.delete(hash);\n    }\n    this.size--;\n    return true;\n  }\n\n  clear(): void {\n    this.buckets = new Map();\n    this.size = 0;\n  }\n\n  isEmpty(): boolean {\n    return this.size === 0;\n  }\n\n  forEach(callbackfn: (value: V, key: K, map: Map<K, V>) => void, thisArg?: any): void {\n    this.buckets.forEach((bucket) => {\n      bucket.forEach((entry) => {\n        callbackfn(entry.value, entry.key, this);\n      }, thisArg);\n    }, thisArg);\n  }\n\n  *keys(): IterableIterator<K> {\n    for (const [key] of this) {\n      yield key;\n    }\n  }\n\n  *values(): IterableIterator<V> {\n    for (const [, value] of this) {\n      yield value;\n    }\n  }\n\n  entries(): IterableIterator<[K, V]> {\n    return this[Symbol.iterator]();\n  }\n}\n\nexport class HashablePublicKey extends PublicKey implements IEquality<HashablePublicKey> {\n  // We only use the last 32 bits of the public key for hashing\n  static MASK = new BN(1).shln(32).subn(1);\n  constructor(value: PublicKey | string | Buffer | Uint8Array | Array<number>) {\n    super(value);\n  }\n\n  hashCode(): number {\n    let hash = 13;\n    hash = hash * 7 + this.getBN().clone().iuand(HashablePublicKey.MASK).toNumber();\n    return hash;\n  }\n\n  private getBN(): BN {\n    //@ts-ignore\n    return this._bn;\n  }\n}\n\ninterface IEquality<T extends IEquality<T>> {\n  equals(other: T): boolean;\n  hashCode(): number;\n}\n"]}