@solana/client 0.0.2 → 0.1.0

package/dist/index.node.cjs

@@ -1358,10 +1358,23 @@ __name(transactionToBase64WithSigners, "transactionToBase64WithSigners");
 
 // src/transactions/prepareTransaction.ts
 var DEFAULT_COMPUTE_UNIT_LIMIT_MULTIPLIER = 1.1;
+var DEFAULT_COMPUTE_UNIT_LIMIT = 2e5;
+var MAX_COMPUTE_UNIT_LIMIT = 14e5;
 function isComputeUnitLimitInstruction(instruction) {
   return instruction.programAddress === computeBudget.COMPUTE_BUDGET_PROGRAM_ADDRESS && instruction.data?.[0] === 2;
 }
 __name(isComputeUnitLimitInstruction, "isComputeUnitLimitInstruction");
+function didExceedComputeBudget(error) {
+  let current = error;
+  while (kit.isSolanaError(current)) {
+    if (kit.isSolanaError(current, kit.SOLANA_ERROR__INSTRUCTION_ERROR__COMPUTATIONAL_BUDGET_EXCEEDED)) {
+      return true;
+    }
+    current = current.cause;
+  }
+  return false;
+}
+__name(didExceedComputeBudget, "didExceedComputeBudget");
 async function estimateComputeUnits(rpc, transaction) {
   let target = transaction;
   const hasLifetime = transaction.lifetimeConstraint !== void 0;
@@ -1373,12 +1386,19 @@ async function estimateComputeUnits(rpc, transaction) {
     );
   }
   const base64Transaction = transactionToBase64(target);
-  const { value } = await rpc.simulateTransaction(base64Transaction, {
-    encoding: "base64",
-    replaceRecentBlockhash: false,
-    sigVerify: false
-  }).send();
-  return Number(value.unitsConsumed ?? 0) || 0;
+  try {
+    const { value } = await rpc.simulateTransaction(base64Transaction, {
+      encoding: "base64",
+      replaceRecentBlockhash: false,
+      sigVerify: false
+    }).send();
+    return Number(value.unitsConsumed ?? 0) || 0;
+  } catch (error) {
+    if (didExceedComputeBudget(error)) {
+      return MAX_COMPUTE_UNIT_LIMIT;
+    }
+    throw error;
+  }
 }
 __name(estimateComputeUnits, "estimateComputeUnits");
 async function prepareTransaction(config) {
@@ -1389,7 +1409,11 @@ async function prepareTransaction(config) {
   const computeLimitIndex = transaction.instructions.findIndex(isComputeUnitLimitInstruction);
   if (computeLimitIndex === -1 || shouldResetComputeUnits) {
     const unitsFromSimulation = await estimateComputeUnits(config.rpc, transaction);
-    const units = Math.max(1, unitsFromSimulation ? Math.ceil(unitsFromSimulation * multiplier) : 2e5);
+    const estimatedUnits = unitsFromSimulation ? Math.ceil(unitsFromSimulation * multiplier) : DEFAULT_COMPUTE_UNIT_LIMIT;
+    const units = Math.min(
+      MAX_COMPUTE_UNIT_LIMIT,
+      Math.max(DEFAULT_COMPUTE_UNIT_LIMIT, Math.max(1, estimatedUnits))
+    );
     const instruction = computeBudget.getSetComputeUnitLimitInstruction({ units });
     if (computeLimitIndex === -1) {
       transaction = kit.appendTransactionMessageInstruction(instruction, transaction);