@drift-labs/jit-proxy 0.2.0 → 0.4.0

package/lib/jitProxyClient.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="bn.js" />
-import { BN, DriftClient, PostOnlyParams, TxParams, UserAccount } from '@drift-labs/sdk';
+import { BN, DriftClient, PostOnlyParams, ReferrerInfo, TxParams, UserAccount } from '@drift-labs/sdk';
 import { PublicKey, TransactionInstruction } from '@solana/web3.js';
 import { TxSigAndSlot } from '@drift-labs/sdk/lib/tx/types';
 export declare type JitIxParams = {
@@ -13,6 +13,7 @@ export declare type JitIxParams = {
     ask: BN;
     postOnly: PostOnlyParams | null;
     priceType?: PriceType;
+    referrerInfo?: ReferrerInfo;
 };
 export declare class PriceType {
     static readonly LIMIT: {
@@ -30,5 +31,5 @@ export declare class JitProxyClient {
         programId: PublicKey;
     });
     jit(params: JitIxParams, txParams?: TxParams): Promise<TxSigAndSlot>;
-    getJitIx({ takerKey, takerStatsKey, taker, takerOrderId, maxPosition, minPosition, bid, ask, postOnly, priceType, }: JitIxParams): Promise<TransactionInstruction>;
+    getJitIx({ takerKey, takerStatsKey, taker, takerOrderId, maxPosition, minPosition, bid, ask, postOnly, priceType, referrerInfo, }: JitIxParams): Promise<TransactionInstruction>;
 }

package/lib/jitProxyClient.js

@@ -19,7 +19,7 @@ class JitProxyClient {
         const tx = await this.driftClient.buildTransaction([ix], txParams);
         return await this.driftClient.sendTransaction(tx);
     }
-    async getJitIx({ takerKey, takerStatsKey, taker, takerOrderId, maxPosition, minPosition, bid, ask, postOnly = null, priceType = PriceType.LIMIT, }) {
+    async getJitIx({ takerKey, takerStatsKey, taker, takerOrderId, maxPosition, minPosition, bid, ask, postOnly = null, priceType = PriceType.LIMIT, referrerInfo, }) {
         const order = taker.orders.find((order) => order.orderId === takerOrderId);
         const remainingAccounts = this.driftClient.getRemainingAccounts({
             userAccounts: [taker, this.driftClient.getUserAccount()],
@@ -30,6 +30,18 @@ class JitProxyClient {
                 ? [order.marketIndex]
                 : [],
         });
+        if (referrerInfo) {
+            remainingAccounts.push({
+                pubkey: referrerInfo.referrer,
+                isWritable: true,
+                isSigner: false,
+            });
+            remainingAccounts.push({
+                pubkey: referrerInfo.referrerStats,
+                isWritable: true,
+                isSigner: false,
+            });
+        }
         if (sdk_1.isVariant(order.marketType, 'spot')) {
             remainingAccounts.push({
                 pubkey: this.driftClient.getSpotMarketAccount(order.marketIndex).vault,

package/lib/jitter.d.ts

@@ -1,7 +1,7 @@
 /// <reference types="bn.js" />
 import { JitProxyClient, PriceType } from './jitProxyClient';
 import { PublicKey } from '@solana/web3.js';
-import { AuctionSubscriber, BN, DriftClient, Order, UserAccount } from '@drift-labs/sdk';
+import { AuctionSubscriber, BN, DriftClient, Order, UserAccount, UserStatsMap } from '@drift-labs/sdk';
 export declare type UserFilter = (userAccount: UserAccount, userKey: string, order: Order) => boolean;
 export declare type JitParams = {
     bid: BN;
@@ -14,14 +14,16 @@ export declare class Jitter {
     auctionSubscriber: AuctionSubscriber;
     driftClient: DriftClient;
     jitProxyClient: JitProxyClient;
+    userStatsMap: UserStatsMap;
     perpParams: Map<number, JitParams>;
     spotParams: Map<number, JitParams>;
     onGoingAuctions: Map<string, Promise<void>>;
     userFilter: UserFilter;
-    constructor({ auctionSubscriber, jitProxyClient, driftClient, }: {
+    constructor({ auctionSubscriber, jitProxyClient, driftClient, userStatsMap, }: {
         driftClient: DriftClient;
         auctionSubscriber: AuctionSubscriber;
         jitProxyClient: JitProxyClient;
+        userStatsMap?: UserStatsMap;
     });
     subscribe(): Promise<void>;
     createTryFill(taker: UserAccount, takerKey: PublicKey, takerStatsKey: PublicKey, order: Order, orderSignature: string): () => Promise<void>;

package/lib/jitter.js

@@ -3,15 +3,21 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Jitter = void 0;
 const sdk_1 = require("@drift-labs/sdk");
 class Jitter {
-    constructor({ auctionSubscriber, jitProxyClient, driftClient, }) {
+    constructor({ auctionSubscriber, jitProxyClient, driftClient, userStatsMap, }) {
         this.perpParams = new Map();
         this.spotParams = new Map();
         this.onGoingAuctions = new Map();
         this.auctionSubscriber = auctionSubscriber;
         this.driftClient = driftClient;
         this.jitProxyClient = jitProxyClient;
+        this.userStatsMap = userStatsMap || new sdk_1.UserStatsMap(this.driftClient, {
+            type: 'polling',
+            accountLoader: new sdk_1.BulkAccountLoader(this.driftClient.connection, 'confirmed', 0),
+        });
     }
     async subscribe() {
+        await this.driftClient.subscribe();
+        await this.userStatsMap.subscribe();
         await this.auctionSubscriber.subscribe();
         this.auctionSubscriber.eventEmitter.on('onAccountUpdate', async (taker, takerKey, slot) => {
             const takerKeyString = takerKey.toBase58();
@@ -58,6 +64,8 @@ class Jitter {
                     this.onGoingAuctions.delete(orderSignature);
                     return;
                 }
+                const takerStats = await this.userStatsMap.mustGet(taker.authority.toString());
+                const referrerInfo = takerStats.getReferrerInfo();
                 console.log(`Trying to fill ${orderSignature}`);
                 try {
                     const { txSig } = await this.jitProxyClient.jit({
@@ -71,6 +79,7 @@ class Jitter {
                         ask: params.ask,
                         postOnly: null,
                         priceType: params.priceType,
+                        referrerInfo,
                     });
                     console.log(`Filled ${orderSignature} txSig ${txSig}`);
                     await sleep(10000);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@drift-labs/jit-proxy",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "dependencies": {
     "@coral-xyz/anchor": "^0.26.0",
     "@solana/web3.js": "1.73.2",

package/src/jitProxyClient.ts

@@ -4,6 +4,7 @@ import {
 	isVariant,
 	PostOnlyParams,
 	QUOTE_SPOT_MARKET_INDEX,
+	ReferrerInfo,
 	TxParams,
 	UserAccount,
 } from '@drift-labs/sdk';
@@ -23,6 +24,7 @@ export type JitIxParams = {
 	ask: BN;
 	postOnly: PostOnlyParams | null;
 	priceType?: PriceType;
+	referrerInfo?: ReferrerInfo;
 };
 
 export class PriceType {
@@ -65,6 +67,7 @@ export class JitProxyClient {
 		ask,
 		postOnly = null,
 		priceType = PriceType.LIMIT,
+		referrerInfo,
 	}: JitIxParams): Promise<TransactionInstruction> {
 		const order = taker.orders.find((order) => order.orderId === takerOrderId);
 		const remainingAccounts = this.driftClient.getRemainingAccounts({
@@ -77,6 +80,19 @@ export class JitProxyClient {
 				: [],
 		});
 
+		if (referrerInfo) {
+			remainingAccounts.push({
+				pubkey: referrerInfo.referrer,
+				isWritable: true,
+				isSigner: false,
+			});
+			remainingAccounts.push({
+				pubkey: referrerInfo.referrerStats,
+				isWritable: true,
+				isSigner: false,
+			});
+		}
+
 		if (isVariant(order.marketType, 'spot')) {
 			remainingAccounts.push({
 				pubkey: this.driftClient.getSpotMarketAccount(order.marketIndex).vault,

package/src/jitter.ts

@@ -2,13 +2,13 @@ import { JitProxyClient, PriceType } from './jitProxyClient';
 import { PublicKey } from '@solana/web3.js';
 import {
 	AuctionSubscriber,
-	BN,
+	BN, BulkAccountLoader,
 	DriftClient,
 	getUserStatsAccountPublicKey,
 	hasAuctionPrice,
 	isVariant,
 	Order,
-	UserAccount,
+	UserAccount, UserMap, UserStatsMap,
 } from '@drift-labs/sdk';
 
 export type UserFilter = (
@@ -28,6 +28,7 @@ export class Jitter {
 	auctionSubscriber: AuctionSubscriber;
 	driftClient: DriftClient;
 	jitProxyClient: JitProxyClient;
+	userStatsMap: UserStatsMap;
 
 	perpParams = new Map<number, JitParams>();
 	spotParams = new Map<number, JitParams>();
@@ -40,17 +41,26 @@ export class Jitter {
 		auctionSubscriber,
 		jitProxyClient,
 		driftClient,
+		userStatsMap,
 	}: {
 		driftClient: DriftClient;
 		auctionSubscriber: AuctionSubscriber;
 		jitProxyClient: JitProxyClient;
+		userStatsMap?: UserStatsMap;
 	}) {
 		this.auctionSubscriber = auctionSubscriber;
 		this.driftClient = driftClient;
 		this.jitProxyClient = jitProxyClient;
+		this.userStatsMap = userStatsMap || new UserStatsMap(this.driftClient, {
+			type: 'polling',
+			accountLoader: new BulkAccountLoader(this.driftClient.connection, 'confirmed', 0),
+		});
 	}
 
 	async subscribe(): Promise<void> {
+		await this.driftClient.subscribe();
+		await this.userStatsMap.subscribe();
+
 		await this.auctionSubscriber.subscribe();
 		this.auctionSubscriber.eventEmitter.on(
 			'onAccountUpdate',
@@ -132,6 +142,9 @@ export class Jitter {
 					return;
 				}
 
+				const takerStats = await this.userStatsMap.mustGet(taker.authority.toString());
+				const referrerInfo = takerStats.getReferrerInfo();
+
 				console.log(`Trying to fill ${orderSignature}`);
 				try {
 					const { txSig } = await this.jitProxyClient.jit({
@@ -145,6 +158,7 @@ export class Jitter {
 						ask: params.ask,
 						postOnly: null,
 						priceType: params.priceType,
+						referrerInfo,
 					});
 
 					console.log(`Filled ${orderSignature} txSig ${txSig}`);