From 8affc4d06b8b203d38786eed524cb9174f9a5b0f Mon Sep 17 00:00:00 2001
From: Mohammad Cheikh <moe@turnkey.io>
Date: Thu, 14 Nov 2024 13:32:44 -0500
Subject: [PATCH] add org id override to addSignature

---
 packages/solana/src/index.ts | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/packages/solana/src/index.ts b/packages/solana/src/index.ts
index 8f95fa529..dd9d166cd 100644
--- a/packages/solana/src/index.ts
+++ b/packages/solana/src/index.ts
@@ -57,13 +57,15 @@ export class TurnkeySigner {
    */
   public async addSignature(
     tx: Transaction | VersionedTransaction,
-    fromAddress: string
+    fromAddress: string,
+    organizationId?: string
   ) {
     const fromKey = new PublicKey(fromAddress);
     const messageToSign: Buffer = this.getMessageToSign(tx);
     const signRawPayloadResult = await this.signRawPayload(
       messageToSign.toString("hex"),
-      fromAddress
+      fromAddress,
+      organizationId!
     );
     const signature = `${signRawPayloadResult?.r}${signRawPayloadResult?.s}`;
 
@@ -161,11 +163,15 @@ export class TurnkeySigner {
     }
   }
 
-  private async signRawPayload(payload: string, signWith: string) {
+  private async signRawPayload(
+    payload: string,
+    signWith: string,
+    organizationId?: string
+  ) {
     if (this.client instanceof TurnkeyClient) {
       const response = await this.client.signRawPayload({
         type: "ACTIVITY_TYPE_SIGN_RAW_PAYLOAD_V2",
-        organizationId: this.organizationId,
+        organizationId: organizationId ?? this.organizationId,
         timestampMs: String(Date.now()),
         parameters: {
           signWith,
@@ -184,6 +190,7 @@ export class TurnkeySigner {
       return assertNonNull(activity?.result?.signRawPayloadResult);
     } else {
       const { activity, r, s, v } = await this.client.signRawPayload({
+        ...(organizationId !== undefined && { organizationId }),
         signWith,
         payload,
         encoding: "PAYLOAD_ENCODING_HEXADECIMAL",