sdk: add referrer info to getArbPerpIx

drift-labs · Nov 13, 2023 · ecc907c · ecc907c
1 parent 1e4d0bd
commit ecc907c
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/ts/sdk/src/jitProxyClient.ts b/ts/sdk/src/jitProxyClient.ts
@@ -197,9 +197,11 @@ export class JitProxyClient {
 	public async getArbPerpIx({
 		makerInfos,
 		marketIndex,
+		referrerInfo,
 	}: {
 		makerInfos: MakerInfo[];
 		marketIndex: number;
+		referrerInfo?: ReferrerInfo;
 	}): Promise<TransactionInstruction> {
 		const userAccounts = [this.driftClient.getUserAccount()];
 		for (const makerInfo of makerInfos) {
@@ -224,6 +226,25 @@ export class JitProxyClient {
 			});
 		}
 
+		if (referrerInfo) {
+			const referrerIsMaker =
+				makerInfos.find((maker) =>
+					maker.maker.equals(referrerInfo.referrer)
+				) !== undefined;
+			if (!referrerIsMaker) {
+				remainingAccounts.push({
+					pubkey: referrerInfo.referrer,
+					isWritable: true,
+					isSigner: false,
+				});
+				remainingAccounts.push({
+					pubkey: referrerInfo.referrerStats,
+					isWritable: true,
+					isSigner: false,
+				});
+			}
+		}
+
 		return this.program.methods
 			.arbPerp(marketIndex)
 			.accounts({