@debridge-finance/solana-utils 4.5.2 → 4.5.5

package/dist/browser/index.js

@@ -1,4 +1,4 @@
-import { MessageV0, VersionedTransaction, Transaction, PublicKey, TransactionInstruction, SystemProgram, Keypair } from '@solana/web3.js';
+import { MessageV0, VersionedTransaction, Transaction, PublicKey, TransactionInstruction, SystemProgram, AddressLookupTableAccount, Keypair } from '@solana/web3.js';
 
 var commonjsGlobal = typeof globalThis !== 'undefined' ? globalThis : typeof window !== 'undefined' ? window : typeof global !== 'undefined' ? global : typeof self !== 'undefined' ? self : {};
 
@@ -6943,16 +6943,42 @@ function hashExternalCallBytes(data) {
     }
     return hexToBuffer(solidityKeccak256(["bytes"], [data ? data : buffer.Buffer.from([])]), 32);
 }
+function isSendHashedFlagSet(packedFlags) {
+    const SEND_HASHED_DATA_FLAG_BIT_NUMBER = BigInt(3);
+    const result = (packedFlags >> SEND_HASHED_DATA_FLAG_BIT_NUMBER) & BigInt(1);
+    return result === BigInt(1);
+}
+function getShortcutFromAutoParams(params) {
+    var _a, _b, _c;
+    const shortcut = (_a = params.autoParams) === null || _a === void 0 ? void 0 : _a.shortcut;
+    if (shortcut) {
+        return isBuffer(shortcut) ? shortcut : hexToBuffer(shortcut);
+    }
+    else {
+        const flags = BigInt(new BN(((_b = params.autoParams) === null || _b === void 0 ? void 0 : _b.flags) || 0).toNumber());
+        const data = (_c = params.autoParams) === null || _c === void 0 ? void 0 : _c.data;
+        if (!data)
+            return hashExternalCallBytes(data);
+        if (isSendHashedFlagSet(flags)) {
+            const result = isBuffer(data) ? data : hexToBuffer(data);
+            if (result.length !== 32) {
+                throw new Error(`Incorrect autoParams.data length for SEND_HASHED_DATA_FLAG`);
+            }
+            return result;
+        }
+        return hashExternalCallBytes(data);
+    }
+}
 /**
  * Hashes submission for provided data
  * @param params data to hash
  * @returns submissionId from provided params
  */
 function hashSubmissionIdRaw(params) {
-    var _a, _b, _c, _d;
+    var _a, _b;
     let packParams = [
         "uint256", // submission prefix
-        "uint256", // bridge_id
+        "bytes32", // bridge_id
         "uint256", // source_chain_id
         "uint256", // target_chain_id
         "uint256", // amount
@@ -6964,7 +6990,7 @@ function hashSubmissionIdRaw(params) {
         : typeof params.receiver === "string"
             ? params.receiver.startsWith("0x")
                 ? hexToBuffer(params.receiver)
-                : new PublicKey(params.receiver)
+                : new PublicKey(params.receiver).toBuffer()
             : new PublicKey(params.receiver).toBuffer();
     let packData = [
         SUBMISSION_PREFIX,
@@ -6979,21 +7005,17 @@ function hashSubmissionIdRaw(params) {
         packParams = packParams.concat([
             "uint256", // execution fee
             "uint256", // flag
-            "uint256", // fallback hash
+            "bytes32", // fallback hash
             "uint256", // ext call data hash
             "uint256", // native sender hash
         ]);
-        const shortcut = ((_a = params.autoParams) === null || _a === void 0 ? void 0 : _a.shortcut)
-            ? isBuffer(params.autoParams.shortcut)
-                ? params.autoParams.shortcut
-                : hexToBuffer(params.autoParams.shortcut)
-            : hashExternalCallBytes((_b = params === null || params === void 0 ? void 0 : params.autoParams) === null || _b === void 0 ? void 0 : _b.data);
+        const shortcut = getShortcutFromAutoParams(params);
         const fallback = params.autoParams.fallbackAddress.startsWith("0x")
             ? hexToBuffer(params.autoParams.fallbackAddress)
             : new PublicKey(params.autoParams.fallbackAddress).toBuffer();
         packData = packData.concat([
-            new BN(((_c = params === null || params === void 0 ? void 0 : params.autoParams) === null || _c === void 0 ? void 0 : _c.executionFee) || 0).toArray("be", 32),
-            new BN(((_d = params.autoParams) === null || _d === void 0 ? void 0 : _d.flags) || 0).toArray("be", 32),
+            new BN(((_a = params === null || params === void 0 ? void 0 : params.autoParams) === null || _a === void 0 ? void 0 : _a.executionFee) || 0).toArray("be", 32),
+            new BN(((_b = params.autoParams) === null || _b === void 0 ? void 0 : _b.flags) || 0).toArray("be", 32),
             hashExternalCallBytes(fallback),
             shortcut,
             hashExternalCallBytes(params.autoParams.nativeSender),
@@ -10262,6 +10284,128 @@ function checkIfAssociatedWalletExists(connection, mintAccount, originalWallet)
         return accountData !== null;
     });
 }
+function getTransactionAccountsForSimulationDiff(connection, tx) {
+    var _a, _b;
+    return __awaiter(this, void 0, void 0, function* () {
+        const result = new Set();
+        if ("version" in tx) {
+            const ALTs = tx.message.addressTableLookups.map((alt) => alt.accountKey);
+            const fetchedALTs = yield connection.getMultipleAccountsInfo(ALTs);
+            const parsedALTs = fetchedALTs
+                .map((fetched, i) => fetched
+                ? new AddressLookupTableAccount({
+                    state: AddressLookupTableAccount.deserialize(fetched.data),
+                    key: ALTs[i],
+                })
+                : null)
+                .filter((x) => x != null);
+            const accGetter = tx.message.getAccountKeys({
+                addressLookupTableAccounts: parsedALTs,
+            });
+            for (const acc of [
+                ...accGetter.staticAccountKeys,
+                ...((_b = (_a = accGetter.accountKeysFromLookups) === null || _a === void 0 ? void 0 : _a.writable) !== null && _b !== void 0 ? _b : []),
+            ]) {
+                result.add(acc.toBase58());
+            }
+        }
+        else {
+            for (const ix of tx.instructions) {
+                // we don't need read-only accounts and instruction.programId since they won't change
+                ix.keys
+                    .filter((meta) => meta.isSigner || meta.isWritable)
+                    .map((meta) => result.add(meta.pubkey.toBase58()));
+            }
+        }
+        return Array.from(result);
+    });
+}
+function parseMultipleAccounts(accounts) {
+    return accounts.map((account) => {
+        var _a, _b;
+        let parsedSplAccount;
+        if (account) {
+            const owner = new PublicKey(account.owner);
+            if (owner.equals(TOKEN_PROGRAM_ID)) {
+                if (account.data instanceof buffer.Buffer) {
+                    parsedSplAccount = parseSplAccount(account.data);
+                }
+                else if (account.data instanceof Array) {
+                    parsedSplAccount = parseSplAccount(buffer.Buffer.from(account.data[0], account.data[1]));
+                }
+            }
+        }
+        const parsedAccount = {
+            balance: (_a = account === null || account === void 0 ? void 0 : account.lamports) !== null && _a !== void 0 ? _a : 0,
+        };
+        if (parsedSplAccount) {
+            parsedAccount.spl = {
+                mint: parsedSplAccount.mint,
+                owner: parsedSplAccount.owner,
+                balance: Number((_b = parsedSplAccount.amount) !== null && _b !== void 0 ? _b : 0),
+            };
+        }
+        return parsedAccount;
+    });
+}
+function getSplDiff(pre, post) {
+    var _a, _b, _c, _d, _e, _f;
+    const preBalance = (_a = pre === null || pre === void 0 ? void 0 : pre.balance) !== null && _a !== void 0 ? _a : 0;
+    const postBalance = (_b = post === null || post === void 0 ? void 0 : post.balance) !== null && _b !== void 0 ? _b : 0;
+    return {
+        mint: (_c = pre === null || pre === void 0 ? void 0 : pre.mint) !== null && _c !== void 0 ? _c : post === null || post === void 0 ? void 0 : post.mint,
+        owner: (_d = pre === null || pre === void 0 ? void 0 : pre.owner) !== null && _d !== void 0 ? _d : post === null || post === void 0 ? void 0 : post.owner,
+        preBalance: (_e = pre === null || pre === void 0 ? void 0 : pre.balance) !== null && _e !== void 0 ? _e : 0,
+        postBalance: (_f = post === null || post === void 0 ? void 0 : post.balance) !== null && _f !== void 0 ? _f : 0,
+        diff: postBalance - preBalance,
+    };
+}
+function getTransactionDiff(connection, tx, txAccounts) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const accountsStrings = txAccounts
+            ? txAccounts.map((acc) => acc.toBase58())
+            : yield getTransactionAccountsForSimulationDiff(connection, tx);
+        const accountsPubKeys = txAccounts !== null && txAccounts !== void 0 ? txAccounts : accountsStrings.map((key) => new PublicKey(key));
+        let [accountsInfo, simulatedTx] = yield Promise.all([
+            connection.getMultipleAccountsInfo(accountsPubKeys),
+            connection.simulateTransaction(tx, {
+                replaceRecentBlockhash: true,
+                accounts: {
+                    encoding: "base64",
+                    addresses: accountsStrings,
+                },
+            }),
+        ]);
+        const simulatesInfo = simulatedTx.value.accounts;
+        const preBalances = parseMultipleAccounts(accountsInfo);
+        const postBalances = parseMultipleAccounts(simulatesInfo);
+        const balancesDiff = accountsPubKeys.reduce((map, pubKey, i) => {
+            var _a;
+            if ((_a = accountsInfo[i]) === null || _a === void 0 ? void 0 : _a.executable) {
+                return map;
+            }
+            const preBalance = preBalances[i];
+            const postBalance = postBalances[i];
+            const diff = {
+                pubKey: pubKey,
+                preBalance: preBalance.balance,
+                postBalance: postBalance.balance,
+                diff: postBalance.balance - preBalance.balance,
+            };
+            if (preBalance.spl || postBalance.spl) {
+                const splDiff = getSplDiff(preBalance.spl, postBalance.spl);
+                if (splDiff.diff) {
+                    diff.spl = splDiff;
+                }
+            }
+            if (diff.diff || diff.spl) {
+                map.set(pubKey.toString(), diff);
+            }
+            return map;
+        }, new Map());
+        return balancesDiff;
+    });
+}
 
 var spl = /*#__PURE__*/Object.freeze({
 	__proto__: null,
@@ -10278,7 +10422,11 @@ var spl = /*#__PURE__*/Object.freeze({
 	getTokenAccountsWithBalance: getTokenAccountsWithBalance,
 	buildReplenishWsolBalanceTransaction: buildReplenishWsolBalanceTransaction,
 	updateWsolBalance: updateWsolBalance,
-	checkIfAssociatedWalletExists: checkIfAssociatedWalletExists
+	checkIfAssociatedWalletExists: checkIfAssociatedWalletExists,
+	getTransactionAccountsForSimulationDiff: getTransactionAccountsForSimulationDiff,
+	parseMultipleAccounts: parseMultipleAccounts,
+	getSplDiff: getSplDiff,
+	getTransactionDiff: getTransactionDiff
 });
 
 const deBridge = "DEbrdGj3HsRsAzx6uH4MKyREKxVAfBydijLUF3ygsFfh";