@debridge-finance/solana-utils 4.5.2 → 4.5.3

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 : {};
 
@@ -10262,6 +10262,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 +10400,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";