@stacks/rendezvous 0.5.0 → 0.6.0

package/dist/citizen.js

@@ -44,13 +44,25 @@ const issueFirstClassCitizenship = (manifestDir, manifestPath, remoteDataSetting
     }));
     const sortedContractsByEpoch = (0, exports.groupContractsByEpochFromSimnetPlan)(simnetPlan);
     const simnetAddresses = [...simnet.getAccounts().values()];
-    const balancesMap = new Map(Array.from(simnetAddresses, (address) => {
+    const stxBalancesMap = new Map(simnetAddresses.map((address) => {
         const balanceHex = simnet.runSnippet(`(stx-get-balance '${address})`);
         return [address, (0, transactions_1.cvToValue)((0, transactions_1.hexToCV)(balanceHex))];
     }));
+    const sbtcBalancesMap = new Map(simnetAddresses.map((address) => {
+        try {
+            const { result: getBalanceResult } = simnet.callReadOnlyFn("SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token", "get-balance", [transactions_1.Cl.principal(address)], address);
+            // If the previous read-only call works, the user is working with
+            // sBTC. This means we can proceed with restoring sBTC balances.
+            const sbtcBalance = (0, transactions_1.cvToJSON)(getBalanceResult).value.value;
+            return [address, sbtcBalance];
+        }
+        catch (e) {
+            return [address, 0];
+        }
+    }));
     yield simnet.initEmptySession(remoteDataSettings);
     simnetAddresses.forEach((address) => {
-        simnet.mintSTX(address, balancesMap.get(address));
+        simnet.mintSTX(address, stxBalancesMap.get(address));
     });
     // Combine the target contract with its tests into a single contract. The
     // resulting contract will replace the target contract in the simnet.
@@ -63,6 +75,14 @@ const issueFirstClassCitizenship = (manifestDir, manifestPath, remoteDataSetting
     ]));
     // Deploy the contracts to the simnet in the correct order.
     yield deployContracts(simnet, sortedContractsByEpoch, (name, props) => (0, exports.getContractSource)([sutContractName], rendezvousSources, name, props, manifestDir));
+    // Filter out addresses with zero balance. They do not need to be restored.
+    const sbtcBalancesToRestore = new Map([...sbtcBalancesMap.entries()].filter(([_, balance]) => balance !== 0));
+    // After all the contracts and requirements are deployed, if the test wallets
+    // had sBTC balances previously, restore them. If no test wallet previously
+    // owned sBTC, skip this step.
+    if ([...sbtcBalancesToRestore.keys()].length > 0) {
+        restoreSbtcBalances(simnet, sbtcBalancesToRestore);
+    }
     return simnet;
 });
 exports.issueFirstClassCitizenship = issueFirstClassCitizenship;
@@ -240,3 +260,85 @@ function scheduleRendezvous(targetContractSource, tests) {
     (ok (map-set context function-name {called: called})))`;
     return `${targetContractSource}\n\n${context}\n\n${tests}`;
 }
+/**
+ * Utility function that restores the test wallets' initial sBTC balances in
+ * the re-initialized first-class citizenship simnet.
+ *
+ * @param simnet The simnet instance.
+ * @param sbtcBalancesMap A map containing the test wallets' balances to be
+ * restored.
+ */
+const restoreSbtcBalances = (simnet, sbtcBalancesMap) => {
+    // For each address present in the balances map, restore the balance.
+    [...sbtcBalancesMap.entries()]
+        // Re-assure the map does not contain nil balances.
+        .filter(([_, balance]) => balance !== 0)
+        .forEach(([address, balance]) => {
+        // To deposit sBTC, one needs a txId and a sweep txId. A deposit transaction
+        // must have a unique txId and sweep txId.
+        const txId = getUniqueHex();
+        const sweepTxId = getUniqueHex();
+        mintSbtc(simnet, balance, address, txId, sweepTxId);
+    });
+};
+/**
+ * Utility function to deposit an amount of sBTC to a Stacks address.
+ *
+ * @param simnet The simnet instance.
+ * @param amountSats The amount to mint in sats.
+ * @param recipient The Stacks address to mint sBTC to.
+ * @param txId A unique hex to use for the deposit.
+ * @param sweepTxId A unique hex to use for the deposit.
+ */
+const mintSbtc = (simnet, amountSats, recipient, txId, sweepTxId) => {
+    // Calling `get-burn-header` only works for past block heights. We mine one
+    // empty Stacks block if the initial height is 0.
+    if (simnet.blockHeight === 0) {
+        simnet.mineEmptyBlock();
+    }
+    const previousStacksHeight = simnet.blockHeight - 1;
+    const burnHash = simnet.callReadOnlyFn("SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-deposit", "get-burn-header", [
+        // (height uint)
+        transactions_1.Cl.uint(previousStacksHeight),
+    ], simnet.deployer).result;
+    if (burnHash === null || burnHash.type === transactions_1.ClarityType.OptionalNone) {
+        throw new Error("Something went wrong trying to retrieve the burn header.");
+    }
+    const completeDepositTx = (0, transactions_1.cvToJSON)(simnet.callPublicFn("SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-deposit", "complete-deposit-wrapper", [
+        // (txid (buff 32))
+        transactions_1.Cl.bufferFromHex(txId),
+        // (vout-index uint)
+        transactions_1.Cl.uint(1),
+        // (amount uint)
+        transactions_1.Cl.uint(amountSats),
+        // (recipient principal)
+        transactions_1.Cl.principal(recipient),
+        // (burn-hash (buff 32))
+        burnHash.value,
+        // (burn-height uint)
+        transactions_1.Cl.uint(previousStacksHeight),
+        // (sweep-txid (buff 32))
+        transactions_1.Cl.bufferFromHex(sweepTxId),
+    ], "SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4").result);
+    // If the deposit transaction fails, an unexpected outcome can happen. Throw
+    // an error if the transaction is not successful.
+    if (!completeDepositTx.success) {
+        throw new Error("Something went wrong trying to restore sBTC balances.");
+    }
+};
+/**
+ * Utility function that generates a random, unique hex to be used as txId in
+ * `mintSbtc`.
+ *
+ * @returns A random hex string.
+ */
+const getUniqueHex = () => {
+    let hex;
+    // Generate a 32-byte (64 character) random hex string.
+    const bytes = new Uint8Array(32);
+    crypto.getRandomValues(bytes);
+    hex = Array.from(bytes)
+        .map((byte) => byte.toString(16).padStart(2, "0"))
+        .join("");
+    return hex;
+};

package/dist/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@stacks/rendezvous",
-    "version": "0.5.0",
+    "version": "0.6.0",
     "description": "Meet your contract's vulnerabilities head-on.",
     "main": "app.js",
     "bin": {
@@ -30,7 +30,7 @@
         "url": "https://github.com/stacks-network/rendezvous.git"
     },
     "dependencies": {
-        "@hirosystems/clarinet-sdk": "2.14.0",
+        "@hirosystems/clarinet-sdk": "2.15.2",
         "@stacks/transactions": "^6.16.1",
         "ansicolor": "^2.0.3",
         "fast-check": "^3.20.0",
@@ -38,7 +38,7 @@
         "yaml": "^2.6.1"
     },
     "devDependencies": {
-        "@hirosystems/clarinet-sdk-wasm": "2.14.0",
+        "@hirosystems/clarinet-sdk-wasm": "2.15.2",
         "@types/jest": "^29.5.12",
         "jest": "^29.7.0",
         "ts-jest": "^29.2.5",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stacks/rendezvous",
-  "version": "0.5.0",
+  "version": "0.6.0",
   "description": "Meet your contract's vulnerabilities head-on.",
   "main": "app.js",
   "bin": {
@@ -30,7 +30,7 @@
     "url": "https://github.com/stacks-network/rendezvous.git"
   },
   "dependencies": {
-    "@hirosystems/clarinet-sdk": "2.14.0",
+    "@hirosystems/clarinet-sdk": "2.15.2",
     "@stacks/transactions": "^6.16.1",
     "ansicolor": "^2.0.3",
     "fast-check": "^3.20.0",
@@ -38,7 +38,7 @@
     "yaml": "^2.6.1"
   },
   "devDependencies": {
-    "@hirosystems/clarinet-sdk-wasm": "2.14.0",
+    "@hirosystems/clarinet-sdk-wasm": "2.15.2",
     "@types/jest": "^29.5.12",
     "jest": "^29.7.0",
     "ts-jest": "^29.2.5",