feat: add margin ratio ix to open orders + swift prop

Author: LukasDeco

sdk/src/driftClient.ts

@@ -4018,7 +4018,8 @@ export class DriftClient {
 		bracketOrdersParams = new Array<OptionalOrderParams>(),
 		referrerInfo?: ReferrerInfo,
 		cancelExistingOrders?: boolean,
-		settlePnl?: boolean
+		settlePnl?: boolean,
+		positionMaxLev?: number
 	): Promise<{
 		cancelExistingOrdersTx?: Transaction | VersionedTransaction;
 		settlePnlTx?: Transaction | VersionedTransaction;
@@ -4034,7 +4035,7 @@ export class DriftClient {
 		const marketIndex = orderParams.marketIndex;
 		const orderId = userAccount.nextOrderId;
 
-		const ixPromisesForTxs: Record<TxKeys, Promise<TransactionInstruction>> = {
+		const ixPromisesForTxs: Record<TxKeys, Promise<TransactionInstruction | TransactionInstruction[]>> = {
 			cancelExistingOrdersTx: undefined,
 			settlePnlTx: undefined,
 			fillTx: undefined,
@@ -4043,11 +4044,17 @@ export class DriftClient {
 
 		const txKeys = Object.keys(ixPromisesForTxs);
 
-		ixPromisesForTxs.marketOrderTx = this.getPlaceOrdersIx(
+		const marketOrderTxIxs = positionMaxLev ? this.getPlaceOrdersAndSetPositionMaxLevIx(
+			[orderParams, ...bracketOrdersParams],
+			positionMaxLev,
+			userAccount.subAccountId
+		) : this.getPlaceOrdersIx(
 			[orderParams, ...bracketOrdersParams],
 			userAccount.subAccountId
 		);
 
+		ixPromisesForTxs.marketOrderTx = marketOrderTxIxs;
+
 		/* Cancel open orders in market if requested */
 		if (cancelExistingOrders && isVariant(orderParams.marketType, 'perp')) {
 			ixPromisesForTxs.cancelExistingOrdersTx = this.getCancelOrdersIx(
@@ -4087,7 +4094,7 @@ export class DriftClient {
 		const ixsMap = ixs.reduce((acc, ix, i) => {
 			acc[txKeys[i]] = ix;
 			return acc;
-		}, {}) as MappedRecord<typeof ixPromisesForTxs, TransactionInstruction>;
+		}, {}) as MappedRecord<typeof ixPromisesForTxs, TransactionInstruction | TransactionInstruction[]>;
 
 		const txsMap = (await this.buildTransactionsMap(
 			ixsMap,
@@ -4671,6 +4678,65 @@ export class DriftClient {
 		});
 	}
 
+	public async getPlaceOrdersAndSetPositionMaxLevIx(
+		params: OptionalOrderParams[],
+		positionMaxLev: number,
+		subAccountId?: number
+	): Promise<TransactionInstruction[]> {
+		const user = await this.getUserAccountPublicKey(subAccountId);
+
+		const readablePerpMarketIndex: number[] = [];
+		const readableSpotMarketIndexes: number[] = [];
+		for (const param of params) {
+			if (!param.marketType) {
+				throw new Error('must set param.marketType');
+			}
+			if (isVariant(param.marketType, 'perp')) {
+				readablePerpMarketIndex.push(param.marketIndex);
+			} else {
+				readableSpotMarketIndexes.push(param.marketIndex);
+			}
+		}
+
+		const remainingAccounts = this.getRemainingAccounts({
+			userAccounts: [this.getUserAccount(subAccountId)],
+			readablePerpMarketIndex,
+			readableSpotMarketIndexes,
+			useMarketLastSlotCache: true,
+		});
+
+		for (const param of params) {
+			if (isUpdateHighLeverageMode(param.bitFlags)) {
+				remainingAccounts.push({
+					pubkey: getHighLeverageModeConfigPublicKey(this.program.programId),
+					isWritable: true,
+					isSigner: false,
+				});
+			}
+		}
+
+		const formattedParams = params.map((item) => getOrderParams(item));
+
+		const placeOrdersIxs = await this.program.instruction.placeOrders(formattedParams, {
+			accounts: {
+				state: await this.getStatePublicKey(),
+				user,
+				userStats: this.getUserStatsAccountPublicKey(),
+				authority: this.wallet.publicKey,
+			},
+			remainingAccounts,
+		});
+
+		// TODO: Handle multiple markets?
+		const setPositionMaxLevIxs = await this.getUpdateUserPerpPositionCustomMarginRatioIx(
+			readablePerpMarketIndex[0],
+			1 / positionMaxLev,
+			subAccountId
+		);
+
+		return [placeOrdersIxs, setPositionMaxLevIxs];
+	}
+
 	public async fillPerpOrder(
 		userAccountPublicKey: PublicKey,
 		user: UserAccount,