From 6523af06380f1d2d90afb9b4759bf81df706d52d Mon Sep 17 00:00:00 2001
From: Lukas deConantsesznak <lukas@lunoho.company>
Date: Wed, 20 Aug 2025 13:32:30 -0600
Subject: [PATCH] feat: optional swift sign message fn

---
 sdk/src/driftClient.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/sdk/src/driftClient.ts b/sdk/src/driftClient.ts
index d1fafa3d05..665a611553 100644
--- a/sdk/src/driftClient.ts
+++ b/sdk/src/driftClient.ts
@@ -6382,7 +6382,8 @@ export class DriftClient {
 		orderParamsMessage:
 			| SignedMsgOrderParamsMessage
 			| SignedMsgOrderParamsDelegateMessage,
-		delegateSigner?: boolean
+		delegateSigner?: boolean,
+		signMessageFunction?: (message: Uint8Array) => Buffer
 	): SignedMsgOrderParams {
 		const borshBuf = this.encodeSignedMsgOrderParamsMessage(
 			orderParamsMessage,
@@ -6391,7 +6392,9 @@ export class DriftClient {
 		const orderParams = Buffer.from(borshBuf.toString('hex'));
 		return {
 			orderParams,
-			signature: this.signMessage(Buffer.from(borshBuf.toString('hex'))),
+			signature: signMessageFunction
+				? signMessageFunction(Buffer.from(borshBuf.toString('hex')))
+				: this.signMessage(Buffer.from(borshBuf.toString('hex'))),
 		};
 	}