Merge pull request #7 from avianlabs/programid-in-publickey-extension

allow to specify program ID for associatedTokenAddress()
avianlabs · Feb 13, 2024 · f0bd4f3 · f0bd4f3
2 parents a35b04a + 1e3a2e2
commit f0bd4f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...otlin/src/commonMain/kotlin/net/avianlabs/solana/domain/program/AssociatedTokenProgram.kt b/...otlin/src/commonMain/kotlin/net/avianlabs/solana/domain/program/AssociatedTokenProgram.kt
@@ -74,10 +74,11 @@ public object AssociatedTokenProgram : Program(
 
 public fun PublicKey.associatedTokenAddress(
   tokenMintAddress: PublicKey,
+  programId: PublicKey = TokenProgram.programId,
 ): ProgramDerivedAddress = Program.findProgramAddress(
   listOf(
     bytes.copyOf(),
-    TokenProgram.programId.toByteArray(),
+    programId.toByteArray(),
     tokenMintAddress.toByteArray()
   ),
   AssociatedTokenProgram.programId,