add ata derivation to solana package

tkhq · Dec 11, 2024 · 2ba1bbe · 2ba1bbe
1 parent 0af04b4
commit 2ba1bbe
Show file tree

Hide file tree

Showing 6 changed files with 72 additions and 15 deletions.
diff --git a/examples/with-solana/src/allowSplTransfer.ts b/examples/with-solana/src/allowSplTransfer.ts
diff --git a/examples/with-solana/src/tokenTransfer.ts b/examples/with-solana/src/tokenTransfer.ts
@@ -25,7 +25,7 @@ async function main() {
   const turnkeyWarchest = new PublicKey(TURNKEY_WAR_CHEST);
   const organizationId = process.env.ORGANIZATION_ID!;
   const connection = solanaNetwork.connect();
-
+ 
   const turnkeyClient = new Turnkey({
     apiBaseUrl: process.env.BASE_URL!,
     apiPublicKey: process.env.API_PUBLIC_KEY!,
@@ -38,6 +38,7 @@ async function main() {
     client: turnkeyClient.apiClient(),
   });
 
+
   let solAddress = process.env.SOLANA_ADDRESS!;
   if (!solAddress) {
     solAddress = await createNewSolanaWallet(turnkeyClient.apiClient());

diff --git a/examples/with-solana/src/utils/deriveAta.ts b/examples/with-solana/src/utils/deriveAta.ts
diff --git a/packages/solana/package.json b/packages/solana/package.json
@@ -48,6 +48,7 @@
     "typecheck": "tsc -p tsconfig.typecheck.json"
   },
   "dependencies": {
+    "@solana/spl-token": "^0.4.8",
     "@solana/web3.js": "^1.88.1",
     "@turnkey/http": "workspace:*",
     "@turnkey/sdk-browser": "workspace:*",

diff --git a/packages/solana/src/index.ts b/packages/solana/src/index.ts
@@ -1,4 +1,5 @@
 import { PublicKey, Transaction, VersionedTransaction } from "@solana/web3.js";
+import { getAssociatedTokenAddressSync } from "@solana/spl-token";
 import {
   assertNonNull,
   assertActivityCompleted,
@@ -131,6 +132,24 @@ export class TurnkeySigner {
     return recoveredTransaction;
   }
 
+  /**
+   * This function derives a Solana associated token address for a given Solana wallet address, related to a particular token mint address
+   *
+   * @param walletAddress string representation of the Solana wallet address (base58 encoded)
+   * @param mintAddress string representation of the token mint address (base58 encoded)
+   */
+  public getAssociatedTokenAddress(mintAddress: string, walletAddress:string) {
+    const mintPublicKey = new PublicKey(mintAddress);
+    const walletPublicKey = new PublicKey(walletAddress);
+
+    const associatedTokenAddress = getAssociatedTokenAddressSync(
+        mintPublicKey,
+        walletPublicKey
+    );
+
+    return associatedTokenAddress.toString();
+}
+
   private async signTransactionImpl(
     unsignedTransaction: string,
     signWith: string,

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml