wasabi-solana-ts 1.2.5 → 1.2.6

package/dist/utils/utils.d.ts

@@ -26,7 +26,6 @@ export declare function getPermission(program: Program<WasabiSolana>, admin: Pub
 export declare function uiAmountToAmount(uiAmount: number, decimals: number): BN;
 export declare function amountToUiAmount(amount: BN, decimals: number): number;
 export declare function getTokenProgram(connection: Connection, mint: PublicKey, mintCache?: TokenMintCache): Promise<PublicKey | null>;
-export declare function getTokenProgramAndDecimals(connection: Connection, mint: PublicKey): Promise<[PublicKey, number] | null>;
 export declare const PDA: {
     getLongPool(collateral: PublicKey, currency: PublicKey): PublicKey;
     getShortPool(collateral: PublicKey, currency: PublicKey): PublicKey;

package/dist/utils/utils.js

@@ -1,11 +1,13 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PDA = exports.SEED_PREFIX = exports.METADATA_PROGRAM_ID = exports.WASABI_PROGRAM_ID = exports.SOL_MINT = void 0;
 exports.getPermission = getPermission;
 exports.uiAmountToAmount = uiAmountToAmount;
 exports.amountToUiAmount = amountToUiAmount;
 exports.getTokenProgram = getTokenProgram;
-exports.getTokenProgramAndDecimals = getTokenProgramAndDecimals;
 exports.getMintInfo = getMintInfo;
 exports.getMetaplexMetadata = getMetaplexMetadata;
 exports.getMaxWithdraw = getMaxWithdraw;
@@ -37,6 +39,7 @@ const anchor_1 = require("@coral-xyz/anchor");
 const spl_token_1 = require("@solana/spl-token");
 const js_1 = require("@metaplex-foundation/js");
 const TokenMintCache_1 = require("../cache/TokenMintCache");
+const node_cache_1 = __importDefault(require("node-cache"));
 exports.SOL_MINT = new web3_js_1.PublicKey('So11111111111111111111111111111111111111111');
 exports.WASABI_PROGRAM_ID = new web3_js_1.PublicKey('spicyTHtbmarmUxwFSHYpA8G4uP2nRNq38RReMpoZ9c');
 exports.METADATA_PROGRAM_ID = new web3_js_1.PublicKey('metaqbxxUerdq28cj1RbAWkYQm3ybzjb6a8bt518x1s');
@@ -61,6 +64,18 @@ function findProgramAddress(seeds, programId) {
     const [publicKey] = web3_js_1.PublicKey.findProgramAddressSync(seeds, programId);
     return publicKey;
 }
+const tokenAccountCache = new node_cache_1.default({ stdTTL: 5 * 60 }); // 5 minutes TTL
+const getTokenAccountCached = async (connection, address) => {
+    const cached = tokenAccountCache.get(address.toString());
+    if (cached) {
+        return cached;
+    }
+    const accountInfo = await connection.getAccountInfo(address);
+    if (accountInfo) {
+        tokenAccountCache.set(address.toString(), accountInfo);
+    }
+    return accountInfo;
+};
 async function getPermission(program, admin) {
     let permission;
     const superAdmin = exports.PDA.getSuperAdmin();
@@ -86,19 +101,6 @@ async function getTokenProgram(connection, mint, mintCache) {
     const mintInfo = await mintCache.getAccount(mint);
     return mintInfo.program;
 }
-async function getTokenProgramAndDecimals(connection, mint) {
-    const mintInfo = await connection.getAccountInfo(mint);
-    if (!mintInfo) {
-        return null;
-    }
-    if (mintInfo.owner.equals(spl_token_1.TOKEN_PROGRAM_ID) || mintInfo.owner.equals(spl_token_1.TOKEN_2022_PROGRAM_ID)) {
-        const mintDecimals = spl_token_1.MintLayout.decode(mintInfo.data).decimals;
-        return [mintInfo.owner, mintDecimals];
-    }
-    else {
-        return null;
-    }
-}
 exports.PDA = {
     getLongPool(collateral, currency) {
         return findProgramAddress([
@@ -326,7 +328,7 @@ function isNativeMint(mint) {
 async function createAtaIfNeeded(connection, owner, mint, ata, tokenProgram, payer) {
     if (isNativeMint(mint))
         return null;
-    const account = await connection.getAccountInfo(ata);
+    const account = await getTokenAccountCached(connection, ata);
     if (!account) {
         return (0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(payer ? payer : owner, ata, owner, mint, tokenProgram);
     }
@@ -339,7 +341,7 @@ async function createWrapSolInstruction(connection, owner, amount, includeCleanu
         ? [spl_token_1.NATIVE_MINT_2022, spl_token_1.TOKEN_2022_PROGRAM_ID]
         : [spl_token_1.NATIVE_MINT, spl_token_1.TOKEN_PROGRAM_ID];
     const ownerWrappedSolAta = (0, spl_token_1.getAssociatedTokenAddressSync)(nativeMint, owner, false, tokenProgram);
-    const ownerWrappedSolAccount = await connection.getAccountInfo(ownerWrappedSolAta);
+    const ownerWrappedSolAccount = await getTokenAccountCached(connection, ownerWrappedSolAta);
     if (!ownerWrappedSolAccount) {
         setupIx.push((0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(owner, ownerWrappedSolAta, owner, nativeMint, tokenProgram));
         if (includeCleanupIx) {
@@ -364,7 +366,7 @@ async function createUnwrapSolInstructionWithPayer(connection, payer, owner, use
         ? [spl_token_1.NATIVE_MINT_2022, spl_token_1.TOKEN_2022_PROGRAM_ID]
         : [spl_token_1.NATIVE_MINT, spl_token_1.TOKEN_PROGRAM_ID];
     const ownerWrappedSolAta = (0, spl_token_1.getAssociatedTokenAddressSync)(nativeMint, owner, false, tokenProgram);
-    const ownerWrappedSolAccount = await connection.getAccountInfo(ownerWrappedSolAta);
+    const ownerWrappedSolAccount = await getTokenAccountCached(connection, ownerWrappedSolAta);
     if (!ownerWrappedSolAccount) {
         setupIx.push((0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(payer, ownerWrappedSolAta, owner, nativeMint, tokenProgram));
     }
@@ -401,7 +403,7 @@ async function handleMint(connection, mint, options) {
     }
     if (options.owner) {
         const userAta = (0, spl_token_1.getAssociatedTokenAddressSync)(mint, options.owner, true, tokenProgram);
-        const userTokenAccount = await connection.getAccountInfo(userAta);
+        const userTokenAccount = await getTokenAccountCached(connection, userAta);
         if (!userTokenAccount) {
             instructions.setupIx.push((0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(options.owner, userAta, options.owner, mint, tokenProgram));
         }
@@ -501,7 +503,7 @@ async function handleOpenTokenAccounts({ program, owner, downPayment, fee, mintC
         setupIx.push((0, spl_token_1.createSyncNativeInstruction)(ownerPaymentAta, paymentTokenProgram));
         cleanupIx.push((0, spl_token_1.createCloseAccountInstruction)(ownerPaymentAta, owner, owner, [], paymentTokenProgram));
     }
-    const ownerPaymentAtaInfo = await program.provider.connection.getAccountInfo(ownerPaymentAta);
+    const ownerPaymentAtaInfo = await getTokenAccountCached(program.provider.connection, ownerPaymentAta);
     if (!ownerPaymentAtaInfo) {
         setupIx = [
             (0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(owner, ownerPaymentAta, owner, paymentMint, paymentTokenProgram),
@@ -529,7 +531,7 @@ async function handleCloseTokenAccounts(config, poolAccount) {
     const payoutTokenProgram = poolAccount.isLongPool ? currencyTokenProgram : collateralTokenProgram;
     const payoutIsSol = payoutMint.equals(spl_token_1.NATIVE_MINT);
     const ownerPayoutAta = (0, spl_token_1.getAssociatedTokenAddressSync)(payoutMint, config.owner, false, payoutTokenProgram);
-    const ownerPayoutAtaInfo = await config.program.provider.connection.getAccountInfo(ownerPayoutAta);
+    const ownerPayoutAtaInfo = await getTokenAccountCached(config.program.provider.connection, ownerPayoutAta);
     const setupIx = [];
     const cleanupIx = [];
     if (!ownerPayoutAtaInfo) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wasabi-solana-ts",
-  "version": "1.2.5",
+  "version": "1.2.6",
   "description": "Typescript library for the Wasabi program",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",