@drift-labs/sdk 2.76.0-beta.9 → 2.77.0-beta.0

package/VERSION

@@ -1 +1 @@
-2.76.0-beta.9
+2.77.0-beta.0

package/lib/adminClient.d.ts

@@ -11,6 +11,8 @@ export declare class AdminClient extends DriftClient {
     getInitializeSpotMarketIx(mint: PublicKey, optimalUtilization: number, optimalRate: number, maxRate: number, oracle: PublicKey, oracleSource: OracleSource, initialAssetWeight: number, maintenanceAssetWeight: number, initialLiabilityWeight: number, maintenanceLiabilityWeight: number, imfFactor?: number, liquidatorFee?: number, ifLiquidationFee?: number, activeStatus?: boolean, assetTier?: {
         collateral: {};
     }, scaleInitialAssetWeightStart?: BN, withdrawGuardThreshold?: BN, orderTickSize?: BN, orderStepSize?: BN, ifTotalFactor?: number, name?: string): Promise<TransactionInstruction>;
+    deleteInitializedSpotMarket(marketIndex: number): Promise<TransactionSignature>;
+    getDeleteInitializedSpotMarketIx(marketIndex: number): Promise<TransactionInstruction>;
     initializeSerumFulfillmentConfig(marketIndex: number, serumMarket: PublicKey, serumProgram: PublicKey): Promise<TransactionSignature>;
     getInitializeSerumFulfillmentConfigIx(marketIndex: number, serumMarket: PublicKey, serumProgram: PublicKey): Promise<TransactionInstruction>;
     initializePhoenixFulfillmentConfig(marketIndex: number, phoenixMarket: PublicKey): Promise<TransactionSignature>;

package/lib/adminClient.js

@@ -100,6 +100,30 @@ class AdminClient extends driftClient_1.DriftClient {
         });
         return initializeIx;
     }
+    async deleteInitializedSpotMarket(marketIndex) {
+        const deleteInitializeMarketIx = await this.getDeleteInitializedSpotMarketIx(marketIndex);
+        const tx = await this.buildTransaction(deleteInitializeMarketIx);
+        const { txSig } = await this.sendTransaction(tx, [], this.opts);
+        return txSig;
+    }
+    async getDeleteInitializedSpotMarketIx(marketIndex) {
+        const spotMarketPublicKey = await (0, pda_1.getSpotMarketPublicKey)(this.program.programId, marketIndex);
+        const spotMarketVaultPublicKey = await (0, pda_1.getSpotMarketVaultPublicKey)(this.program.programId, marketIndex);
+        const insuranceFundVaultPublicKey = await (0, pda_1.getInsuranceFundVaultPublicKey)(this.program.programId, marketIndex);
+        return await this.program.instruction.deleteInitializedSpotMarket(marketIndex, {
+            accounts: {
+                state: await this.getStatePublicKey(),
+                admin: this.isSubscribed
+                    ? this.getStateAccount().admin
+                    : this.wallet.publicKey,
+                spotMarket: spotMarketPublicKey,
+                spotMarketVault: spotMarketVaultPublicKey,
+                insuranceFundVault: insuranceFundVaultPublicKey,
+                driftSigner: this.getSignerPublicKey(),
+                tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+            },
+        });
+    }
     async initializeSerumFulfillmentConfig(marketIndex, serumMarket, serumProgram) {
         const initializeIx = await this.getInitializeSerumFulfillmentConfigIx(marketIndex, serumMarket, serumProgram);
         const tx = await this.buildTransaction(initializeIx);

package/lib/constants/perpMarkets.js

@@ -546,6 +546,16 @@ exports.MainnetPerpMarkets = [
         launchTs: 1712240681000,
         oracleSource: __1.OracleSource.Prelaunch,
     },
+    {
+        fullName: 'Tensor',
+        category: ['NFT', 'Solana'],
+        symbol: 'TNSR-PERP',
+        baseAssetSymbol: 'TNSR',
+        marketIndex: 29,
+        oracle: new web3_js_1.PublicKey('7Cfyymx49ipGsgEsCA2XygAB2DUsan4C6Cyb5c8oR5st'),
+        launchTs: 1712593532000,
+        oracleSource: __1.OracleSource.SWITCHBOARD,
+    },
 ];
 exports.PerpMarkets = {
     devnet: exports.DevnetPerpMarkets,

package/lib/constants/spotMarkets.js

@@ -186,6 +186,17 @@ exports.MainnetSpotMarkets = [
         phoenixMarket: new web3_js_1.PublicKey('8dFTCTAbtGuHsdDL8WEPrTU6pXFDrU1QSjBTutw8fwZk'),
         launchTs: 1712149014000,
     },
+    {
+        symbol: 'TNSR',
+        marketIndex: 14,
+        oracle: new web3_js_1.PublicKey('7Cfyymx49ipGsgEsCA2XygAB2DUsan4C6Cyb5c8oR5st'),
+        oracleSource: __1.OracleSource.SWITCHBOARD,
+        mint: new web3_js_1.PublicKey('TNSRxcUxoT9xBG3de7PiJyTDYu7kskLqcpddxnEJAS6'),
+        precision: new __1.BN(10).pow(numericConstants_1.NINE),
+        precisionExp: numericConstants_1.NINE,
+        phoenixMarket: new web3_js_1.PublicKey('AbJCZ9TAJiby5AY3cHcXS2gUdENC6mtsm6m7XpC2ZMvE'),
+        launchTs: 1712593532000,
+    },
 ];
 exports.SpotMarkets = {
     devnet: exports.DevnetSpotMarkets,

package/lib/idl/drift.json

@@ -1,5 +1,5 @@
 {
-    "version": "2.75.0",
+    "version": "2.76.0",
     "name": "drift",
     "instructions": [
         {
@@ -2696,6 +2696,52 @@
                 }
             ]
         },
+        {
+            "name": "deleteInitializedSpotMarket",
+            "accounts": [
+                {
+                    "name": "admin",
+                    "isMut": true,
+                    "isSigner": true
+                },
+                {
+                    "name": "state",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "spotMarket",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "spotMarketVault",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "insuranceFundVault",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "driftSigner",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "tokenProgram",
+                    "isMut": false,
+                    "isSigner": false
+                }
+            ],
+            "args": [
+                {
+                    "name": "marketIndex",
+                    "type": "u16"
+                }
+            ]
+        },
         {
             "name": "initializeSerumFulfillmentConfig",
             "accounts": [

package/lib/priorityFee/driftPriorityFeeMethod.d.ts

@@ -0,0 +1,3 @@
+import { HeliusPriorityFeeLevels } from './heliusPriorityFeeMethod';
+export type DriftPriorityFeeResponse = HeliusPriorityFeeLevels[];
+export declare function fetchDriftPriorityFee(url: string, marketType: string, marketIndex: number): Promise<DriftPriorityFeeResponse>;

package/lib/priorityFee/driftPriorityFeeMethod.js

@@ -0,0 +1,12 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchDriftPriorityFee = void 0;
+const node_fetch_1 = __importDefault(require("node-fetch"));
+async function fetchDriftPriorityFee(url, marketType, marketIndex) {
+    const response = await (0, node_fetch_1.default)(`${url}?marketType=${marketType}&marketIndex=${marketIndex}`);
+    return await response.json();
+}
+exports.fetchDriftPriorityFee = fetchDriftPriorityFee;

package/lib/priorityFee/priorityFeeSubscriber.d.ts

@@ -3,10 +3,16 @@ import { PriorityFeeMethod, PriorityFeeStrategy, PriorityFeeSubscriberConfig } f
 import { AverageOverSlotsStrategy } from './averageOverSlotsStrategy';
 import { MaxOverSlotsStrategy } from './maxOverSlotsStrategy';
 import { HeliusPriorityFeeLevels, HeliusPriorityLevel } from './heliusPriorityFeeMethod';
+import { MarketType } from '..';
+export type DriftMarketInfo = {
+    marketType: string;
+    marketIndex: number;
+};
 export declare class PriorityFeeSubscriber {
     connection: Connection;
     frequencyMs: number;
     addresses: string[];
+    driftMarket?: DriftMarketInfo;
     customStrategy?: PriorityFeeStrategy;
     averageStrategy: AverageOverSlotsStrategy;
     maxStrategy: MaxOverSlotsStrategy;
@@ -14,6 +20,7 @@ export declare class PriorityFeeSubscriber {
     lookbackDistance: number;
     maxFeeMicroLamports?: number;
     priorityFeeMultiplier?: number;
+    driftPriorityFeeEndpoint?: string;
     heliusRpcUrl?: string;
     lastHeliusSample?: HeliusPriorityFeeLevels;
     intervalId?: ReturnType<typeof setTimeout>;
@@ -26,6 +33,7 @@ export declare class PriorityFeeSubscriber {
     subscribe(): Promise<void>;
     private loadForSolana;
     private loadForHelius;
+    private loadForDrift;
     getMaxPriorityFee(): number | undefined;
     updateMaxPriorityFee(newMaxFee: number | undefined): void;
     getPriorityFeeMultiplier(): number;
@@ -38,4 +46,5 @@ export declare class PriorityFeeSubscriber {
     load(): Promise<void>;
     unsubscribe(): Promise<void>;
     updateAddresses(addresses: PublicKey[]): void;
+    updateMarketTypeAndIndex(marketType: MarketType, marketIndex: number): void;
 }

package/lib/priorityFee/priorityFeeSubscriber.js

@@ -6,6 +6,8 @@ const averageOverSlotsStrategy_1 = require("./averageOverSlotsStrategy");
 const maxOverSlotsStrategy_1 = require("./maxOverSlotsStrategy");
 const solanaPriorityFeeMethod_1 = require("./solanaPriorityFeeMethod");
 const heliusPriorityFeeMethod_1 = require("./heliusPriorityFeeMethod");
+const driftPriorityFeeMethod_1 = require("./driftPriorityFeeMethod");
+const __1 = require("..");
 class PriorityFeeSubscriber {
     constructor(config) {
         var _a, _b;
@@ -19,7 +21,10 @@ class PriorityFeeSubscriber {
         this.lastSlotSeen = 0;
         this.connection = config.connection;
         this.frequencyMs = config.frequencyMs;
-        this.addresses = config.addresses.map((address) => address.toBase58());
+        this.addresses = config.addresses
+            ? config.addresses.map((address) => address.toBase58())
+            : [];
+        this.driftMarket = config.driftMarket;
         if (config.customStrategy) {
             this.customStrategy = config.customStrategy;
         }
@@ -42,6 +47,9 @@ class PriorityFeeSubscriber {
                     this.heliusRpcUrl = config.heliusRpcUrl;
                 }
             }
+            else if (this.priorityFeeMethod === types_1.PriorityFeeMethod.DRIFT) {
+                this.driftPriorityFeeEndpoint = config.driftPriorityFeeEndpoint;
+            }
         }
         if (this.priorityFeeMethod === types_1.PriorityFeeMethod.SOLANA) {
             if (this.connection === undefined) {
@@ -74,14 +82,34 @@ class PriorityFeeSubscriber {
         this.lastHeliusSample = (_b = (_a = sample === null || sample === void 0 ? void 0 : sample.result) === null || _a === void 0 ? void 0 : _a.priorityFeeLevels) !== null && _b !== void 0 ? _b : undefined;
         if (this.lastHeliusSample) {
             this.lastAvgStrategyResult =
-                this.heliusRpcUrl[heliusPriorityFeeMethod_1.HeliusPriorityLevel.MEDIUM];
+                this.lastHeliusSample[heliusPriorityFeeMethod_1.HeliusPriorityLevel.MEDIUM];
             this.lastMaxStrategyResult =
-                this.heliusRpcUrl[heliusPriorityFeeMethod_1.HeliusPriorityLevel.UNSAFE_MAX];
+                this.lastHeliusSample[heliusPriorityFeeMethod_1.HeliusPriorityLevel.UNSAFE_MAX];
             if (this.customStrategy) {
                 this.lastCustomStrategyResult = this.customStrategy.calculate(sample);
             }
         }
     }
+    async loadForDrift() {
+        if (!this.driftMarket) {
+            return;
+        }
+        const sample = await (0, driftPriorityFeeMethod_1.fetchDriftPriorityFee)(this.driftPriorityFeeEndpoint, this.driftMarket.marketType, this.driftMarket.marketIndex);
+        if (sample.length > 0) {
+            this.lastAvgStrategyResult = sample[heliusPriorityFeeMethod_1.HeliusPriorityLevel.MEDIUM];
+            this.lastMaxStrategyResult = sample[heliusPriorityFeeMethod_1.HeliusPriorityLevel.UNSAFE_MAX];
+            if (this.customStrategy) {
+                // stay compatible with helius response
+                this.lastCustomStrategyResult = this.customStrategy.calculate({
+                    jsonrpc: '',
+                    result: {
+                        priorityFeeLevels: sample[0],
+                    },
+                    id: '',
+                });
+            }
+        }
+    }
     getMaxPriorityFee() {
         return this.maxFeeMicroLamports;
     }
@@ -136,6 +164,9 @@ class PriorityFeeSubscriber {
             else if (this.priorityFeeMethod === types_1.PriorityFeeMethod.HELIUS) {
                 await this.loadForHelius();
             }
+            else if (this.priorityFeeMethod === types_1.PriorityFeeMethod.DRIFT) {
+                await this.loadForDrift();
+            }
             else {
                 throw new Error(`${this.priorityFeeMethod} load not implemented`);
             }
@@ -155,5 +186,8 @@ class PriorityFeeSubscriber {
     updateAddresses(addresses) {
         this.addresses = addresses.map((k) => k.toBase58());
     }
+    updateMarketTypeAndIndex(marketType, marketIndex) {
+        this.driftMarket = { marketType: (0, __1.getVariant)(marketType), marketIndex };
+    }
 }
 exports.PriorityFeeSubscriber = PriorityFeeSubscriber;

package/lib/priorityFee/types.d.ts

@@ -1,21 +1,25 @@
 import { Connection, PublicKey } from '@solana/web3.js';
 import { SolanaPriorityFeeResponse } from './solanaPriorityFeeMethod';
 import { HeliusPriorityFeeResponse } from './heliusPriorityFeeMethod';
+import { DriftMarketInfo } from './priorityFeeSubscriber';
 export interface PriorityFeeStrategy {
     calculate(samples: SolanaPriorityFeeResponse[] | HeliusPriorityFeeResponse): number;
 }
 export declare enum PriorityFeeMethod {
     SOLANA = "solana",
-    HELIUS = "helius"
+    HELIUS = "helius",
+    DRIFT = "drift"
 }
 export type PriorityFeeSubscriberConfig = {
     connection?: Connection;
     frequencyMs: number;
-    addresses: PublicKey[];
+    addresses?: PublicKey[];
+    driftMarket?: DriftMarketInfo;
     customStrategy?: PriorityFeeStrategy;
     priorityFeeMethod?: PriorityFeeMethod;
     slotsToCheck?: number;
     heliusRpcUrl?: string;
+    driftPriorityFeeEndpoint?: string;
     maxFeeMicroLamports?: number;
     priorityFeeMultiplier?: number;
 };

package/lib/priorityFee/types.js

@@ -5,4 +5,5 @@ var PriorityFeeMethod;
 (function (PriorityFeeMethod) {
     PriorityFeeMethod["SOLANA"] = "solana";
     PriorityFeeMethod["HELIUS"] = "helius";
+    PriorityFeeMethod["DRIFT"] = "drift";
 })(PriorityFeeMethod = exports.PriorityFeeMethod || (exports.PriorityFeeMethod = {}));

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.76.0-beta.9",
+	"version": "2.77.0-beta.0",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/adminClient.ts

@@ -227,6 +227,55 @@ export class AdminClient extends DriftClient {
 		return initializeIx;
 	}
 
+	public async deleteInitializedSpotMarket(
+		marketIndex: number
+	): Promise<TransactionSignature> {
+		const deleteInitializeMarketIx =
+			await this.getDeleteInitializedSpotMarketIx(marketIndex);
+
+		const tx = await this.buildTransaction(deleteInitializeMarketIx);
+
+		const { txSig } = await this.sendTransaction(tx, [], this.opts);
+
+		return txSig;
+	}
+
+	public async getDeleteInitializedSpotMarketIx(
+		marketIndex: number
+	): Promise<TransactionInstruction> {
+		const spotMarketPublicKey = await getSpotMarketPublicKey(
+			this.program.programId,
+			marketIndex
+		);
+
+		const spotMarketVaultPublicKey = await getSpotMarketVaultPublicKey(
+			this.program.programId,
+			marketIndex
+		);
+
+		const insuranceFundVaultPublicKey = await getInsuranceFundVaultPublicKey(
+			this.program.programId,
+			marketIndex
+		);
+
+		return await this.program.instruction.deleteInitializedSpotMarket(
+			marketIndex,
+			{
+				accounts: {
+					state: await this.getStatePublicKey(),
+					admin: this.isSubscribed
+						? this.getStateAccount().admin
+						: this.wallet.publicKey,
+					spotMarket: spotMarketPublicKey,
+					spotMarketVault: spotMarketVaultPublicKey,
+					insuranceFundVault: insuranceFundVaultPublicKey,
+					driftSigner: this.getSignerPublicKey(),
+					tokenProgram: TOKEN_PROGRAM_ID,
+				},
+			}
+		);
+	}
+
 	public async initializeSerumFulfillmentConfig(
 		marketIndex: number,
 		serumMarket: PublicKey,

package/src/constants/perpMarkets.ts

@@ -557,6 +557,16 @@ export const MainnetPerpMarkets: PerpMarketConfig[] = [
 		launchTs: 1712240681000,
 		oracleSource: OracleSource.Prelaunch,
 	},
+	{
+		fullName: 'Tensor',
+		category: ['NFT', 'Solana'],
+		symbol: 'TNSR-PERP',
+		baseAssetSymbol: 'TNSR',
+		marketIndex: 29,
+		oracle: new PublicKey('7Cfyymx49ipGsgEsCA2XygAB2DUsan4C6Cyb5c8oR5st'),
+		launchTs: 1712593532000,
+		oracleSource: OracleSource.SWITCHBOARD,
+	},
 ];
 
 export const PerpMarkets: { [key in DriftEnv]: PerpMarketConfig[] } = {

package/src/constants/spotMarkets.ts

@@ -227,6 +227,19 @@ export const MainnetSpotMarkets: SpotMarketConfig[] = [
 		),
 		launchTs: 1712149014000,
 	},
+	{
+		symbol: 'TNSR',
+		marketIndex: 14,
+		oracle: new PublicKey('7Cfyymx49ipGsgEsCA2XygAB2DUsan4C6Cyb5c8oR5st'),
+		oracleSource: OracleSource.SWITCHBOARD,
+		mint: new PublicKey('TNSRxcUxoT9xBG3de7PiJyTDYu7kskLqcpddxnEJAS6'),
+		precision: new BN(10).pow(NINE),
+		precisionExp: NINE,
+		phoenixMarket: new PublicKey(
+			'AbJCZ9TAJiby5AY3cHcXS2gUdENC6mtsm6m7XpC2ZMvE'
+		),
+		launchTs: 1712593532000,
+	},
 ];
 
 export const SpotMarkets: { [key in DriftEnv]: SpotMarketConfig[] } = {

package/src/idl/drift.json

@@ -1,5 +1,5 @@
 {
-  "version": "2.75.0",
+  "version": "2.76.0",
   "name": "drift",
   "instructions": [
     {
@@ -2696,6 +2696,52 @@
         }
       ]
     },
+    {
+      "name": "deleteInitializedSpotMarket",
+      "accounts": [
+        {
+          "name": "admin",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "state",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "spotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "spotMarketVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftSigner",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "tokenProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        }
+      ]
+    },
     {
       "name": "initializeSerumFulfillmentConfig",
       "accounts": [

package/src/priorityFee/driftPriorityFeeMethod.ts

@@ -0,0 +1,15 @@
+import fetch from 'node-fetch';
+import { HeliusPriorityFeeLevels } from './heliusPriorityFeeMethod';
+
+export type DriftPriorityFeeResponse = HeliusPriorityFeeLevels[];
+
+export async function fetchDriftPriorityFee(
+	url: string,
+	marketType: string,
+	marketIndex: number
+): Promise<DriftPriorityFeeResponse> {
+	const response = await fetch(
+		`${url}?marketType=${marketType}&marketIndex=${marketIndex}`
+	);
+	return await response.json();
+}

package/src/priorityFee/priorityFeeSubscriber.ts

@@ -12,11 +12,19 @@ import {
 	HeliusPriorityLevel,
 	fetchHeliusPriorityFee,
 } from './heliusPriorityFeeMethod';
+import { fetchDriftPriorityFee } from './driftPriorityFeeMethod';
+import { getVariant, MarketType } from '..';
+
+export type DriftMarketInfo = {
+	marketType: string;
+	marketIndex: number;
+};
 
 export class PriorityFeeSubscriber {
 	connection: Connection;
 	frequencyMs: number;
 	addresses: string[];
+	driftMarket?: DriftMarketInfo;
 	customStrategy?: PriorityFeeStrategy;
 	averageStrategy = new AverageOverSlotsStrategy();
 	maxStrategy = new MaxOverSlotsStrategy();
@@ -25,6 +33,7 @@ export class PriorityFeeSubscriber {
 	maxFeeMicroLamports?: number;
 	priorityFeeMultiplier?: number;
 
+	driftPriorityFeeEndpoint?: string;
 	heliusRpcUrl?: string;
 	lastHeliusSample?: HeliusPriorityFeeLevels;
 
@@ -39,13 +48,18 @@ export class PriorityFeeSubscriber {
 	public constructor(config: PriorityFeeSubscriberConfig) {
 		this.connection = config.connection;
 		this.frequencyMs = config.frequencyMs;
-		this.addresses = config.addresses.map((address) => address.toBase58());
+		this.addresses = config.addresses
+			? config.addresses.map((address) => address.toBase58())
+			: [];
+		this.driftMarket = config.driftMarket;
+
 		if (config.customStrategy) {
 			this.customStrategy = config.customStrategy;
 		} else {
 			this.customStrategy = this.averageStrategy;
 		}
 		this.lookbackDistance = config.slotsToCheck ?? 50;
+
 		if (config.priorityFeeMethod) {
 			this.priorityFeeMethod = config.priorityFeeMethod;
 
@@ -61,6 +75,8 @@ export class PriorityFeeSubscriber {
 				} else {
 					this.heliusRpcUrl = config.heliusRpcUrl;
 				}
+			} else if (this.priorityFeeMethod === PriorityFeeMethod.DRIFT) {
+				this.driftPriorityFeeEndpoint = config.driftPriorityFeeEndpoint;
 			}
 		}
 
@@ -111,15 +127,40 @@ export class PriorityFeeSubscriber {
 
 		if (this.lastHeliusSample) {
 			this.lastAvgStrategyResult =
-				this.heliusRpcUrl[HeliusPriorityLevel.MEDIUM];
+				this.lastHeliusSample[HeliusPriorityLevel.MEDIUM];
 			this.lastMaxStrategyResult =
-				this.heliusRpcUrl[HeliusPriorityLevel.UNSAFE_MAX];
+				this.lastHeliusSample[HeliusPriorityLevel.UNSAFE_MAX];
 			if (this.customStrategy) {
 				this.lastCustomStrategyResult = this.customStrategy.calculate(sample!);
 			}
 		}
 	}
 
+	private async loadForDrift(): Promise<void> {
+		if (!this.driftMarket) {
+			return;
+		}
+		const sample = await fetchDriftPriorityFee(
+			this.driftPriorityFeeEndpoint!,
+			this.driftMarket.marketType,
+			this.driftMarket.marketIndex
+		);
+		if (sample.length > 0) {
+			this.lastAvgStrategyResult = sample[HeliusPriorityLevel.MEDIUM];
+			this.lastMaxStrategyResult = sample[HeliusPriorityLevel.UNSAFE_MAX];
+			if (this.customStrategy) {
+				// stay compatible with helius response
+				this.lastCustomStrategyResult = this.customStrategy.calculate({
+					jsonrpc: '',
+					result: {
+						priorityFeeLevels: sample[0],
+					},
+					id: '',
+				});
+			}
+		}
+	}
+
 	public getMaxPriorityFee(): number | undefined {
 		return this.maxFeeMicroLamports;
 	}
@@ -183,6 +224,8 @@ export class PriorityFeeSubscriber {
 				await this.loadForSolana();
 			} else if (this.priorityFeeMethod === PriorityFeeMethod.HELIUS) {
 				await this.loadForHelius();
+			} else if (this.priorityFeeMethod === PriorityFeeMethod.DRIFT) {
+				await this.loadForDrift();
 			} else {
 				throw new Error(`${this.priorityFeeMethod} load not implemented`);
 			}
@@ -207,4 +250,8 @@ export class PriorityFeeSubscriber {
 	public updateAddresses(addresses: PublicKey[]) {
 		this.addresses = addresses.map((k) => k.toBase58());
 	}
+
+	public updateMarketTypeAndIndex(marketType: MarketType, marketIndex: number) {
+		this.driftMarket = { marketType: getVariant(marketType), marketIndex };
+	}
 }

package/src/priorityFee/types.ts

@@ -1,6 +1,7 @@
 import { Connection, PublicKey } from '@solana/web3.js';
 import { SolanaPriorityFeeResponse } from './solanaPriorityFeeMethod';
 import { HeliusPriorityFeeResponse } from './heliusPriorityFeeMethod';
+import { DriftMarketInfo } from './priorityFeeSubscriber';
 
 export interface PriorityFeeStrategy {
 	// calculate the priority fee for a given set of samples.
@@ -13,6 +14,7 @@ export interface PriorityFeeStrategy {
 export enum PriorityFeeMethod {
 	SOLANA = 'solana',
 	HELIUS = 'helius',
+	DRIFT = 'drift',
 }
 
 export type PriorityFeeSubscriberConfig = {
@@ -21,7 +23,9 @@ export type PriorityFeeSubscriberConfig = {
 	/// frequency to make RPC calls to update priority fee samples, in milliseconds
 	frequencyMs: number;
 	/// addresses you plan to write lock, used to determine priority fees
-	addresses: PublicKey[];
+	addresses?: PublicKey[];
+	/// drift market type and index, optionally provide at initialization time if using priorityFeeMethod.DRIFT
+	driftMarket?: DriftMarketInfo;
 	/// custom strategy to calculate priority fees, defaults to AVERAGE
 	customStrategy?: PriorityFeeStrategy;
 	/// method for fetching priority fee samples
@@ -30,6 +34,8 @@ export type PriorityFeeSubscriberConfig = {
 	slotsToCheck?: number;
 	/// url for helius rpc, required if using priorityFeeMethod.HELIUS
 	heliusRpcUrl?: string;
+	/// url for drift cached priority fee endpoint, required if using priorityFeeMethod.DRIFT
+	driftPriorityFeeEndpoint?: string;
 	/// clamp any returned priority fee value to this value.
 	maxFeeMicroLamports?: number;
 	/// multiplier applied to priority fee before maxFeeMicroLamports, defaults to 1.0