fix: account for ATA creation in Solana fee queries (#1185)

Signed-off-by: bennett <bennett@umaproject.org>

Author: bmzig

package.json

@@ -1,7 +1,7 @@
 {
   "name": "@across-protocol/sdk",
   "author": "UMA Team",
-  "version": "4.3.42",
+  "version": "4.3.43",
   "license": "AGPL-3.0",
   "homepage": "https://docs.across.to/reference/sdk",
   "files": [

src/relayFeeCalculator/chain-queries/svmQuery.ts

@@ -1,13 +1,20 @@
 import assert from "assert";
-import { getComputeUnitEstimateForTransactionMessageFactory, TransactionSigner } from "@solana/kit";
-import { SVMProvider, SolanaVoidSigner, getFillRelayTx } from "../../arch/svm";
+import {
+  getComputeUnitEstimateForTransactionMessageFactory,
+  TransactionSigner,
+  fetchEncodedAccount,
+  isSome,
+} from "@solana/kit";
+import { SVMProvider, SolanaVoidSigner, getFillRelayTx, toAddress, getAssociatedTokenAddress } from "../../arch/svm";
 import { Coingecko } from "../../coingecko";
 import { CHAIN_IDs } from "../../constants";
 import { getGasPriceEstimate } from "../../gasPriceOracle";
 import { RelayData } from "../../interfaces";
 import { Address, BigNumber, BigNumberish, SvmAddress, TransactionCostEstimate, toBN } from "../../utils";
 import { Logger, QueryInterface, getDefaultRelayer } from "../relayFeeCalculator";
 import { SymbolMappingType } from "./";
+import { TOKEN_PROGRAM_ADDRESS } from "@solana-program/token";
+import { TOKEN_2022_PROGRAM_ADDRESS, getTokenSize, fetchMint, Extension } from "@solana-program/token-2022";
 
 /**
  * A special QueryBase implementation for SVM used for querying gas costs, token prices, and decimals of various tokens
@@ -76,24 +83,49 @@ export class SvmQuery implements QueryInterface {
       repaymentAddress
     );
 
-    const [computeUnitsConsumed, gasPriceEstimate] = await Promise.all([
+    const [computeUnitsConsumed, gasPriceEstimate, tokenAccountInfo] = await Promise.all([
       toBN(await this.computeUnitEstimator(fillRelayTx)),
       getGasPriceEstimate(this.provider, {
         unsignedTx: fillRelayTx,
         baseFeeMultiplier: options.baseFeeMultiplier,
         priorityFeeMultiplier: options.priorityFeeMultiplier,
       }),
+      this.provider.getAccountInfo(toAddress(outputToken), { encoding: "base58" }).send(),
     ]);
 
+    // If the owner of the token account is not the token program, then we can assume that it is the 2022 token program address, in which
+    // case we need to determine the extensions the token has to properly calculate rent exemption.
+    const tokenOwner = tokenAccountInfo!.value!.owner;
+    assert(
+      tokenOwner === TOKEN_2022_PROGRAM_ADDRESS || tokenOwner === TOKEN_PROGRAM_ADDRESS,
+      `${outputToken} has invalid token account owner ${tokenOwner}.`
+    );
+    const recipientAta = await getAssociatedTokenAddress(recipient, outputToken, tokenOwner);
+    const encodedAta = await fetchEncodedAccount(this.provider, recipientAta);
+
     // We can cast the gas price estimate to an SvmGasPriceEstimate here since the oracle should always
     // query the Solana adapter.
     const gasPrice = gasPriceEstimate.baseFee.add(
       gasPriceEstimate.microLamportsPerComputeUnit.mul(computeUnitsConsumed).div(toBN(1_000_000)) // 1_000_000 microLamports/lamport.
     );
+    let tokenGasCost = gasPrice;
+
+    // If the ATA does not exist, we need to factor the rent amount into the token gas cost.
+    if (!encodedAta.exists) {
+      // If the ATA is a non-2022 token, then it will always have a fixed size of 165.
+      let extensions: Extension[] | undefined = undefined;
+      if (tokenOwner === TOKEN_2022_PROGRAM_ADDRESS) {
+        const mint = await fetchMint(this.provider, toAddress(outputToken));
+        extensions = isSome(mint.data.extensions) ? mint.data.extensions.value : undefined;
+      }
+      const tokenAccountSize = getTokenSize(extensions);
+      const rentCostInLamports = await this.provider.getMinimumBalanceForRentExemption(BigInt(tokenAccountSize)).send();
+      tokenGasCost = tokenGasCost.add(toBN(Number(rentCostInLamports)));
+    }
 
     return {
       nativeGasCost: computeUnitsConsumed,
-      tokenGasCost: gasPrice,
+      tokenGasCost,
       gasPrice,
     };
   }