Merge pull request #109 from GenesysGo/tracy/undefined-compute-units-…

…check Undefined compute units check
GenesysGo · Jun 8, 2024 · 6ad8762 · 6ad8762
2 parents 1b5c54d + 3dff183
commit 6ad8762
Show file tree

Hide file tree

Showing 10 changed files with 34 additions and 14 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@shadow-drive/sdk",
-    "version": "5.1.0",
+    "version": "5.1.1",
     "description": "Interfaces for GenesysGo ShadowDrive",
     "main": "./dist/index.js",
     "module": "./dist/index.js",

diff --git a/src/methods/add-storage.ts b/src/methods/add-storage.ts
@@ -101,11 +101,13 @@ export default async function addStorage(
                         ),
                         this.connection,
                     ]);
-                txn.add(
-                    web3.ComputeBudgetProgram.setComputeUnitLimit({
-                        units: immutableUnits,
-                    })
-                );
+                if (typeof immutableUnits !== "undefined") {
+                    txn.add(
+                        web3.ComputeBudgetProgram.setComputeUnitLimit({
+                            units: immutableUnits,
+                        })
+                    );
+                }
                 txn.add(immutableComputePriceIx);
                 txn.add(increaseImmutableStorageIx);
                 break;
@@ -141,6 +143,8 @@ export default async function addStorage(
                     ),
                     this.connection,
                 ]);
+                if (typeof units !== "undefined") {
+                }
                 txn.add(
                     web3.ComputeBudgetProgram.setComputeUnitLimit({ units })
                 );

diff --git a/src/methods/cancel-storage-account.ts b/src/methods/cancel-storage-account.ts
@@ -39,7 +39,9 @@ export default async function cancelDeleteStorageAccount(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
 
         txn.add(unmarkDeleteAccountIx2);

diff --git a/src/methods/claim-stake.ts b/src/methods/claim-stake.ts
@@ -62,7 +62,9 @@ export default async function claimStake(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
         txn.add(claimStakeIx2);
         let blockInfo = await this.connection.getLatestBlockhash();

diff --git a/src/methods/create-storage-account.ts b/src/methods/create-storage-account.ts
@@ -93,7 +93,9 @@ export default async function createStorageAccount(
         ),
         this.connection,
     ]);
-    txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+    if (typeof units !== undefined) {
+        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+    }
     txn.add(computePriceIx);
     txn.add(initializeAccountIx2);
 

diff --git a/src/methods/delete-storage-account.ts b/src/methods/delete-storage-account.ts
@@ -37,7 +37,9 @@ export default async function deleteStorageAccount(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
 
         txn.add(reqDeleteAccountIx2);

diff --git a/src/methods/make-storage-immutable.ts b/src/methods/make-storage-immutable.ts
@@ -72,7 +72,9 @@ export default async function makeStorageImmutable(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
 
         txn.add(makeImmutableIx2);

diff --git a/src/methods/reduce-storage.ts b/src/methods/reduce-storage.ts
@@ -82,7 +82,9 @@ export default async function reduceStorage(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
         txn.add(decreaseStorageIx2);
 

diff --git a/src/methods/refresh-stake.ts b/src/methods/refresh-stake.ts
@@ -50,7 +50,9 @@ export default async function refreshStake(
             ),
             this.connection,
         ]);
-        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        if (typeof units !== "undefined") {
+            txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+        }
         txn.add(computePriceIx);
 
         txn.add(refreshStakeIx2);

diff --git a/src/methods/top-up.ts b/src/methods/top-up.ts
@@ -45,7 +45,9 @@ export default async function topUp(
         ),
         this.connection,
     ]);
-    txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+    if (typeof units !== "undefined") {
+        txn.add(web3.ComputeBudgetProgram.setComputeUnitLimit({ units }));
+    }
     txn.add(computePriceIx);
 
     txn.add(transferInstruction);