@drift-labs/sdk 2.85.0-beta.9 → 2.86.0-beta.0

package/lib/idl/drift.json

@@ -1,5 +1,5 @@
 {
-    "version": "2.84.0",
+    "version": "2.85.0",
     "name": "drift",
     "instructions": [
         {
@@ -2547,6 +2547,86 @@
                 }
             ]
         },
+        {
+            "name": "updatePythPullOracle",
+            "accounts": [
+                {
+                    "name": "keeper",
+                    "isMut": true,
+                    "isSigner": true
+                },
+                {
+                    "name": "pythSolanaReceiver",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "encodedVaa",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "priceFeed",
+                    "isMut": true,
+                    "isSigner": false
+                }
+            ],
+            "args": [
+                {
+                    "name": "feedId",
+                    "type": {
+                        "array": [
+                            "u8",
+                            32
+                        ]
+                    }
+                },
+                {
+                    "name": "params",
+                    "type": "bytes"
+                }
+            ]
+        },
+        {
+            "name": "postPythPullOracleUpdateAtomic",
+            "accounts": [
+                {
+                    "name": "keeper",
+                    "isMut": true,
+                    "isSigner": true
+                },
+                {
+                    "name": "pythSolanaReceiver",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "guardianSet",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "priceFeed",
+                    "isMut": true,
+                    "isSigner": false
+                }
+            ],
+            "args": [
+                {
+                    "name": "feedId",
+                    "type": {
+                        "array": [
+                            "u8",
+                            32
+                        ]
+                    }
+                },
+                {
+                    "name": "params",
+                    "type": "bytes"
+                }
+            ]
+        },
         {
             "name": "initialize",
             "accounts": [
@@ -5296,6 +5376,47 @@
                     "type": "u16"
                 }
             ]
+        },
+        {
+            "name": "initializePythPullOracle",
+            "accounts": [
+                {
+                    "name": "admin",
+                    "isMut": true,
+                    "isSigner": true
+                },
+                {
+                    "name": "pythSolanaReceiver",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "priceFeed",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "systemProgram",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "state",
+                    "isMut": false,
+                    "isSigner": false
+                }
+            ],
+            "args": [
+                {
+                    "name": "feedId",
+                    "type": {
+                        "array": [
+                            "u8",
+                            32
+                        ]
+                    }
+                }
+            ]
         }
     ],
     "accounts": [
@@ -12029,6 +12150,44 @@
             "code": 6268,
             "name": "MaxBorrows",
             "msg": "Can not borow more than max borrows"
+        },
+        {
+            "code": 6269,
+            "name": "OracleUpdatesNotMonotonic",
+            "msg": "Updates must be monotonically increasing"
+        },
+        {
+            "code": 6270,
+            "name": "OraclePriceFeedMessageMismatch",
+            "msg": "Trying to update price feed with the wrong feed id"
+        },
+        {
+            "code": 6271,
+            "name": "OracleUnsupportedMessageType",
+            "msg": "The message in the update must be a PriceFeedMessage"
+        },
+        {
+            "code": 6272,
+            "name": "OracleDeserializeMessageFailed",
+            "msg": "Could not deserialize the message in the update"
+        },
+        {
+            "code": 6273,
+            "name": "OracleWrongGuardianSetOwner",
+            "msg": "Wrong guardian set owner in update price atomic"
+        },
+        {
+            "code": 6274,
+            "name": "OracleWrongWriteAuthority",
+            "msg": "Oracle post update atomic price feed account must be drift program"
+        },
+        {
+            "code": 6275,
+            "name": "OracleWrongVaaOwner",
+            "msg": "Oracle vaa owner must be wormhole program"
         }
-    ]
+    ],
+    "metadata": {
+        "address": "dRiftyHA39MWEi3m9aunc5MzRF1JYuBsbn6VPcn33UH"
+    }
 }

package/lib/math/oracles.d.ts

@@ -1,4 +1,5 @@
 /// <reference types="bn.js" />
+/// <reference types="node" />
 import { AMM, OracleGuardRails } from '../types';
 import { OraclePriceData } from '../oracles/types';
 import { BN, HistoricalOracleData, PerpMarketAccount } from '../index';
@@ -9,3 +10,4 @@ export declare function isOracleTooDivergent(amm: AMM, oraclePriceData: OraclePr
 export declare function calculateLiveOracleTwap(histOracleData: HistoricalOracleData, oraclePriceData: OraclePriceData, now: BN, period: BN): BN;
 export declare function calculateLiveOracleStd(amm: AMM, oraclePriceData: OraclePriceData, now: BN): BN;
 export declare function getNewOracleConfPct(amm: AMM, oraclePriceData: OraclePriceData, reservePrice: BN, now: BN): BN;
+export declare function trimVaaSignatures(vaa: Buffer, n?: number): Buffer;

package/lib/math/oracles.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getNewOracleConfPct = exports.calculateLiveOracleStd = exports.calculateLiveOracleTwap = exports.isOracleTooDivergent = exports.isOracleValid = exports.getMaxConfidenceIntervalMultiplier = exports.oraclePriceBands = void 0;
+exports.trimVaaSignatures = exports.getNewOracleConfPct = exports.calculateLiveOracleStd = exports.calculateLiveOracleTwap = exports.isOracleTooDivergent = exports.isOracleValid = exports.getMaxConfidenceIntervalMultiplier = exports.oraclePriceBands = void 0;
 const types_1 = require("../types");
 const numericConstants_1 = require("../constants/numericConstants");
 const index_1 = require("../index");
@@ -119,3 +119,16 @@ function getNewOracleConfPct(amm, oraclePriceData, reservePrice, now) {
     return confIntervalPctResult;
 }
 exports.getNewOracleConfPct = getNewOracleConfPct;
+function trimVaaSignatures(vaa, n = 3) {
+    const currentNumSignatures = vaa[5];
+    if (n > currentNumSignatures) {
+        throw new Error("Resulting VAA can't have more signatures than the original VAA");
+    }
+    const trimmedVaa = Buffer.concat([
+        vaa.subarray(0, 6 + n * 66),
+        vaa.subarray(6 + currentNumSignatures * 66),
+    ]);
+    trimmedVaa[5] = n;
+    return trimmedVaa;
+}
+exports.trimVaaSignatures = trimVaaSignatures;

package/lib/oracles/pythPullClient.js

@@ -16,7 +16,7 @@ class PythPullClient {
         new __1.Wallet(new web3_js_1.Keypair()), {
             commitment: connection.commitment,
         });
-        this.receiver = new anchor_1.Program(pyth_solana_receiver_1.pythSolanaReceiverIdl, pyth_solana_receiver_1.DEFAULT_RECEIVER_PROGRAM_ID, provider);
+        this.receiver = new anchor_1.Program(pyth_solana_receiver_1.pythSolanaReceiverIdl, __1.DRIFT_ORACLE_RECEIVER_ID, provider);
         this.decodeFunc =
             this.receiver.account.priceUpdateV2.coder.accounts.decodeUnchecked.bind(this.receiver.account.priceUpdateV2.coder.accounts);
     }

package/lib/tx/baseTxSender.js

@@ -238,6 +238,7 @@ class BaseTxSender {
         const logs = transactionResult.meta.logMessages;
         const lastLog = logs[logs.length - 1];
         const friendlyMessage = (_b = lastLog === null || lastLog === void 0 ? void 0 : lastLog.match(/(failed:) (.+)/)) === null || _b === void 0 ? void 0 : _b[2];
+        // @ts-ignore
         throw new web3_js_1.SendTransactionError({
             action: 'send',
             signature: txSig,

package/lib/tx/blockhashFetcher/baseBlockhashFetcher.d.ts

@@ -0,0 +1,8 @@
+import { BlockhashWithExpiryBlockHeight, Commitment, Connection } from '@solana/web3.js';
+import { BlockhashFetcher } from './types';
+export declare class BaseBlockhashFetcher implements BlockhashFetcher {
+    private connection;
+    private blockhashCommitment;
+    constructor(connection: Connection, blockhashCommitment: Commitment);
+    getLatestBlockhash(): Promise<BlockhashWithExpiryBlockHeight | undefined>;
+}

package/lib/tx/blockhashFetcher/baseBlockhashFetcher.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseBlockhashFetcher = void 0;
+class BaseBlockhashFetcher {
+    constructor(connection, blockhashCommitment) {
+        this.connection = connection;
+        this.blockhashCommitment = blockhashCommitment;
+    }
+    async getLatestBlockhash() {
+        return this.connection.getLatestBlockhash(this.blockhashCommitment);
+    }
+}
+exports.BaseBlockhashFetcher = BaseBlockhashFetcher;

package/lib/tx/blockhashFetcher/cachedBlockhashFetcher.d.ts

@@ -0,0 +1,28 @@
+import { BlockhashWithExpiryBlockHeight, Commitment, Connection } from '@solana/web3.js';
+import { BlockhashFetcher } from './types';
+/**
+ * Fetches the latest blockhash and caches it for a configurable amount of time.
+ *
+ * - Prevents RPC spam by reusing cached values
+ * - Retries on failure with exponential backoff
+ * - Prevents concurrent requests for the same blockhash
+ */
+export declare class CachedBlockhashFetcher implements BlockhashFetcher {
+    private connection;
+    private blockhashCommitment;
+    private retryCount;
+    private retrySleepTimeMs;
+    private staleCacheTimeMs;
+    private recentBlockhashCache;
+    private blockhashFetchingPromise;
+    constructor(connection: Connection, blockhashCommitment: Commitment, retryCount: number, retrySleepTimeMs: number, staleCacheTimeMs: number);
+    private fetchBlockhashWithRetry;
+    private sleep;
+    private updateBlockhashCache;
+    getLatestBlockhash(): Promise<BlockhashWithExpiryBlockHeight | undefined>;
+    private isCacheStale;
+    /**
+     * Refresh the blockhash cache, await a pending refresh if it exists
+     */
+    private refreshBlockhash;
+}

package/lib/tx/blockhashFetcher/cachedBlockhashFetcher.js

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CachedBlockhashFetcher = void 0;
+/**
+ * Fetches the latest blockhash and caches it for a configurable amount of time.
+ *
+ * - Prevents RPC spam by reusing cached values
+ * - Retries on failure with exponential backoff
+ * - Prevents concurrent requests for the same blockhash
+ */
+class CachedBlockhashFetcher {
+    constructor(connection, blockhashCommitment, retryCount, retrySleepTimeMs, staleCacheTimeMs) {
+        this.connection = connection;
+        this.blockhashCommitment = blockhashCommitment;
+        this.retryCount = retryCount;
+        this.retrySleepTimeMs = retrySleepTimeMs;
+        this.staleCacheTimeMs = staleCacheTimeMs;
+        this.recentBlockhashCache = { value: undefined, lastUpdated: 0 };
+        this.blockhashFetchingPromise = null;
+    }
+    async fetchBlockhashWithRetry() {
+        for (let i = 0; i < this.retryCount; i++) {
+            try {
+                return await this.connection.getLatestBlockhash(this.blockhashCommitment);
+            }
+            catch (err) {
+                if (i === this.retryCount - 1) {
+                    throw new Error('Failed to fetch blockhash after maximum retries');
+                }
+                await this.sleep(this.retrySleepTimeMs * 2 ** i);
+            }
+        }
+        throw new Error('Failed to fetch blockhash after maximum retries');
+    }
+    sleep(ms) {
+        return new Promise((resolve) => setTimeout(resolve, ms));
+    }
+    async updateBlockhashCache() {
+        const result = await this.fetchBlockhashWithRetry();
+        this.recentBlockhashCache = {
+            value: result,
+            lastUpdated: Date.now(),
+        };
+    }
+    async getLatestBlockhash() {
+        if (this.isCacheStale()) {
+            await this.refreshBlockhash();
+        }
+        return this.recentBlockhashCache.value;
+    }
+    isCacheStale() {
+        const lastUpdateTime = this.recentBlockhashCache.lastUpdated;
+        return (!lastUpdateTime || Date.now() > lastUpdateTime + this.staleCacheTimeMs);
+    }
+    /**
+     * Refresh the blockhash cache, await a pending refresh if it exists
+     */
+    async refreshBlockhash() {
+        if (!this.blockhashFetchingPromise) {
+            this.blockhashFetchingPromise = this.updateBlockhashCache();
+            try {
+                await this.blockhashFetchingPromise;
+            }
+            finally {
+                this.blockhashFetchingPromise = null;
+            }
+        }
+        else {
+            await this.blockhashFetchingPromise;
+        }
+    }
+}
+exports.CachedBlockhashFetcher = CachedBlockhashFetcher;

package/lib/tx/blockhashFetcher/types.d.ts

@@ -0,0 +1,4 @@
+import { BlockhashWithExpiryBlockHeight } from '@solana/web3.js';
+export interface BlockhashFetcher {
+    getLatestBlockhash(): Promise<BlockhashWithExpiryBlockHeight | undefined>;
+}

package/lib/tx/blockhashFetcher/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/tx/txHandler.d.ts

@@ -13,6 +13,14 @@ export type TxBuildingProps = {
     recentBlockhash?: BlockhashWithExpiryBlockHeight;
     wallet?: IWallet;
 };
+export type TxHandlerConfig = {
+    blockhashCachingEnabled?: boolean;
+    blockhashCachingConfig?: {
+        retryCount: number;
+        retrySleepTimeMs: number;
+        staleCacheTimeMs: number;
+    };
+};
 /**
  * This class is responsible for creating and signing transactions.
  */
@@ -25,6 +33,7 @@ export declare class TxHandler {
     private preSignedCb?;
     private onSignedCb?;
     private blockhashCommitment;
+    private blockHashFetcher;
     constructor(props: {
         connection: Connection;
         wallet: IWallet;
@@ -34,6 +43,7 @@ export declare class TxHandler {
             onSignedCb?: (txSigs: DriftClientMetricsEvents['txSigned']) => void;
             preSignedCb?: () => void;
         };
+        config?: TxHandlerConfig;
     });
     private addHashAndExpiryToLookup;
     private getProps;

package/lib/tx/txHandler.js

@@ -8,6 +8,9 @@ const web3_js_1 = require("@solana/web3.js");
 const txParamProcessor_1 = require("./txParamProcessor");
 const bs58_1 = __importDefault(require("bs58"));
 const computeUnits_1 = require("../util/computeUnits");
+const cachedBlockhashFetcher_1 = require("./blockhashFetcher/cachedBlockhashFetcher");
+const baseBlockhashFetcher_1 = require("./blockhashFetcher/baseBlockhashFetcher");
+const utils_1 = require("./utils");
 /**
  * Explanation for SIGNATURE_BLOCK_AND_EXPIRY:
  *
@@ -15,12 +18,15 @@ const computeUnits_1 = require("../util/computeUnits");
  */
 const DEV_TRY_FORCE_TX_TIMEOUTS = process.env.DEV_TRY_FORCE_TX_TIMEOUTS === 'true' || false;
 exports.COMPUTE_UNITS_DEFAULT = 200000;
+const BLOCKHASH_FETCH_RETRY_COUNT = 3;
+const BLOCKHASH_FETCH_RETRY_SLEEP = 200;
+const RECENT_BLOCKHASH_STALE_TIME_MS = 2000; // Reuse blockhashes within this timeframe during bursts of tx contruction
 /**
  * This class is responsible for creating and signing transactions.
  */
 class TxHandler {
     constructor(props) {
-        var _a, _b, _c, _d;
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p;
         this.blockHashToLastValidBlockHeightLookup = {};
         this.returnBlockHeightsWithSignedTxCallbackData = false;
         this.blockhashCommitment = 'finalized';
@@ -28,11 +34,14 @@ class TxHandler {
         this.connection = props.connection;
         this.wallet = props.wallet;
         this.confirmationOptions = props.confirmationOptions;
+        this.blockHashFetcher = ((_a = props === null || props === void 0 ? void 0 : props.config) === null || _a === void 0 ? void 0 : _a.blockhashCachingEnabled)
+            ? new cachedBlockhashFetcher_1.CachedBlockhashFetcher(this.connection, this.blockhashCommitment, (_d = (_c = (_b = props === null || props === void 0 ? void 0 : props.config) === null || _b === void 0 ? void 0 : _b.blockhashCachingConfig) === null || _c === void 0 ? void 0 : _c.retryCount) !== null && _d !== void 0 ? _d : BLOCKHASH_FETCH_RETRY_COUNT, (_g = (_f = (_e = props === null || props === void 0 ? void 0 : props.config) === null || _e === void 0 ? void 0 : _e.blockhashCachingConfig) === null || _f === void 0 ? void 0 : _f.retrySleepTimeMs) !== null && _g !== void 0 ? _g : BLOCKHASH_FETCH_RETRY_SLEEP, (_k = (_j = (_h = props === null || props === void 0 ? void 0 : props.config) === null || _h === void 0 ? void 0 : _h.blockhashCachingConfig) === null || _j === void 0 ? void 0 : _j.staleCacheTimeMs) !== null && _k !== void 0 ? _k : RECENT_BLOCKHASH_STALE_TIME_MS)
+            : new baseBlockhashFetcher_1.BaseBlockhashFetcher(this.connection, this.blockhashCommitment);
         // #Optionals
         this.returnBlockHeightsWithSignedTxCallbackData =
-            (_b = (_a = props.opts) === null || _a === void 0 ? void 0 : _a.returnBlockHeightsWithSignedTxCallbackData) !== null && _b !== void 0 ? _b : false;
-        this.onSignedCb = (_c = props.opts) === null || _c === void 0 ? void 0 : _c.onSignedCb;
-        this.preSignedCb = (_d = props.opts) === null || _d === void 0 ? void 0 : _d.preSignedCb;
+            (_m = (_l = props.opts) === null || _l === void 0 ? void 0 : _l.returnBlockHeightsWithSignedTxCallbackData) !== null && _m !== void 0 ? _m : false;
+        this.onSignedCb = (_o = props.opts) === null || _o === void 0 ? void 0 : _o.onSignedCb;
+        this.preSignedCb = (_p = props.opts) === null || _p === void 0 ? void 0 : _p.preSignedCb;
     }
     addHashAndExpiryToLookup(hashAndExpiry) {
         if (!this.returnBlockHeightsWithSignedTxCallbackData)
@@ -50,8 +59,8 @@ class TxHandler {
      *
      * @returns
      */
-    getLatestBlockhashForTransaction() {
-        return this.connection.getLatestBlockhash(this.blockhashCommitment);
+    async getLatestBlockhashForTransaction() {
+        return this.blockHashFetcher.getLatestBlockhash();
     }
     /**
      * Applies recent blockhash and signs a given transaction
@@ -80,7 +89,7 @@ class TxHandler {
         return signedTx;
     }
     isVersionedTransaction(tx) {
-        return (tx === null || tx === void 0 ? void 0 : tx.message) && true;
+        return (0, utils_1.isVersionedTransaction)(tx);
     }
     isLegacyTransaction(tx) {
         return !this.isVersionedTransaction(tx);

package/lib/tx/utils.d.ts

@@ -0,0 +1,2 @@
+import { Transaction, VersionedTransaction } from '@solana/web3.js';
+export declare const isVersionedTransaction: (tx: Transaction | VersionedTransaction) => boolean;

package/lib/tx/utils.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isVersionedTransaction = void 0;
+const web3_js_1 = require("@solana/web3.js");
+const isVersionedTransaction = (tx) => {
+    const version = tx === null || tx === void 0 ? void 0 : tx.version;
+    const isVersionedTx = tx instanceof web3_js_1.VersionedTransaction || version !== undefined;
+    return isVersionedTx;
+};
+exports.isVersionedTransaction = isVersionedTransaction;

package/lib/util/pythPullOracleUtils.d.ts

@@ -0,0 +1,2 @@
+export declare function trimFeedId(feedId: string): string;
+export declare function getFeedIdUint8Array(feedId: string): Uint8Array;

package/lib/util/pythPullOracleUtils.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getFeedIdUint8Array = exports.trimFeedId = void 0;
+function trimFeedId(feedId) {
+    if (feedId.startsWith('0x')) {
+        return feedId.slice(2);
+    }
+    return feedId;
+}
+exports.trimFeedId = trimFeedId;
+function getFeedIdUint8Array(feedId) {
+    const trimmedFeedId = trimFeedId(feedId);
+    return Uint8Array.from(Buffer.from(trimmedFeedId, 'hex'));
+}
+exports.getFeedIdUint8Array = getFeedIdUint8Array;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.85.0-beta.9",
+	"version": "2.86.0-beta.0",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",
@@ -38,6 +38,7 @@
 		"@ellipsis-labs/phoenix-sdk": "^1.4.2",
 		"@project-serum/serum": "^0.13.38",
 		"@pythnetwork/client": "2.5.3",
+		"@pythnetwork/price-service-sdk": "^1.7.1",
 		"@pythnetwork/pyth-solana-receiver": "^0.7.0",
 		"@solana/spl-token": "0.3.7",
 		"@solana/web3.js": "1.92.3",
@@ -65,6 +66,7 @@
 		"mocha": "^10.0.0",
 		"object-sizeof": "^2.6.3",
 		"prettier": "3.0.1",
+		"sinon": "^18.0.0",
 		"ts-node": "^10.8.0",
 		"typescript": "^4.9.5"
 	},

package/src/addresses/pda.ts

@@ -236,3 +236,16 @@ export function getPrelaunchOraclePublicKey(
 		programId
 	)[0];
 }
+
+export function getPythPullOraclePublicKey(
+	progarmId: PublicKey,
+	feedId: Uint8Array
+): PublicKey {
+	return PublicKey.findProgramAddressSync(
+		[
+			Buffer.from(anchor.utils.bytes.utf8.encode('pyth_pull')),
+			Buffer.from(feedId),
+		],
+		progarmId
+	)[0];
+}

package/src/adminClient.ts

@@ -1,5 +1,6 @@
 import {
 	PublicKey,
+	SystemProgram,
 	SYSVAR_RENT_PUBKEY,
 	TransactionInstruction,
 	TransactionSignature,
@@ -28,6 +29,7 @@ import {
 	getPhoenixFulfillmentConfigPublicKey,
 	getProtocolIfSharesTransferConfigPublicKey,
 	getPrelaunchOraclePublicKey,
+	getPythPullOraclePublicKey,
 } from './addresses/pda';
 import { squareRootBN } from './math/utils';
 import { TOKEN_PROGRAM_ID } from '@solana/spl-token';
@@ -43,6 +45,8 @@ import {
 import { calculateTargetPriceTrade } from './math/trade';
 import { calculateAmmReservesAfterSwap, getSwapDirection } from './math/amm';
 import { PROGRAM_ID as PHOENIX_PROGRAM_ID } from '@ellipsis-labs/phoenix-sdk';
+import { DRIFT_ORACLE_RECEIVER_ID } from './config';
+import { getFeedIdUint8Array } from './util/pythPullOracleUtils';
 
 export class AdminClient extends DriftClient {
 	public async initialize(
@@ -3528,4 +3532,39 @@ export class AdminClient extends DriftClient {
 			},
 		});
 	}
+
+	public async initializePythPullOracle(
+		feedId: string
+	): Promise<TransactionSignature> {
+		const initializePythPullOracleIx = await this.getInitializePythPullOracleIx(
+			feedId
+		);
+		const tx = await this.buildTransaction(initializePythPullOracleIx);
+		const { txSig } = await this.sendTransaction(tx, [], this.opts);
+
+		return txSig;
+	}
+
+	public async getInitializePythPullOracleIx(
+		feedId: string
+	): Promise<TransactionInstruction> {
+		const feedIdBuffer = getFeedIdUint8Array(feedId);
+		return await this.program.instruction.initializePythPullOracle(
+			feedIdBuffer,
+			{
+				accounts: {
+					admin: this.isSubscribed
+						? this.getStateAccount().admin
+						: this.wallet.publicKey,
+					state: await this.getStatePublicKey(),
+					systemProgram: SystemProgram.programId,
+					priceFeed: getPythPullOraclePublicKey(
+						this.program.programId,
+						feedIdBuffer
+					),
+					pythSolanaReceiver: DRIFT_ORACLE_RECEIVER_ID,
+				},
+			}
+		);
+	}
 }

package/src/config.ts

@@ -19,6 +19,7 @@ type DriftConfig = {
 	PYTH_ORACLE_MAPPING_ADDRESS: string;
 	DRIFT_PROGRAM_ID: string;
 	JIT_PROXY_PROGRAM_ID?: string;
+	DRIFT_ORACLE_RECEIVER_ID: string;
 	USDC_MINT_ADDRESS: string;
 	SERUM_V3: string;
 	PHOENIX: string;
@@ -27,11 +28,14 @@ type DriftConfig = {
 	SPOT_MARKETS: SpotMarketConfig[];
 	MARKET_LOOKUP_TABLE: string;
 	SERUM_LOOKUP_TABLE?: string;
+	PYTH_PULL_ORACLE_LOOKUP_TABLE?: string;
 };
 
 export type DriftEnv = 'devnet' | 'mainnet-beta';
 
 export const DRIFT_PROGRAM_ID = 'dRiftyHA39MWEi3m9aunc5MzRF1JYuBsbn6VPcn33UH';
+export const DRIFT_ORACLE_RECEIVER_ID =
+	'G6EoTTTgpkNBtVXo96EQp2m6uwwVh2Kt6YidjkmQqoha';
 
 export const configs: { [key in DriftEnv]: DriftConfig } = {
 	devnet: {
@@ -47,6 +51,7 @@ export const configs: { [key in DriftEnv]: DriftConfig } = {
 		PERP_MARKETS: DevnetPerpMarkets,
 		SPOT_MARKETS: DevnetSpotMarkets,
 		MARKET_LOOKUP_TABLE: 'FaMS3U4uBojvGn5FSDEPimddcXsCfwkKsFgMVVnDdxGb',
+		DRIFT_ORACLE_RECEIVER_ID,
 	},
 	'mainnet-beta': {
 		ENV: 'mainnet-beta',
@@ -62,6 +67,7 @@ export const configs: { [key in DriftEnv]: DriftConfig } = {
 		SPOT_MARKETS: MainnetSpotMarkets,
 		MARKET_LOOKUP_TABLE: 'D9cnvzswDikQDf53k4HpQ3KJ9y1Fv3HGGDFYMXnK5T6c',
 		SERUM_LOOKUP_TABLE: 'GPZkp76cJtNL2mphCvT6FXkJCVPpouidnacckR6rzKDN',
+		DRIFT_ORACLE_RECEIVER_ID,
 	},
 };