solforge 0.1.7 → 0.2.1

package/server/methods/transaction/send-transaction.ts

@@ -0,0 +1,232 @@
+import {
+	ACCOUNT_SIZE,
+	AccountLayout,
+	MINT_SIZE,
+	MintLayout,
+	TOKEN_2022_PROGRAM_ID,
+	TOKEN_PROGRAM_ID,
+} from "@solana/spl-token";
+import { PublicKey, VersionedTransaction } from "@solana/web3.js";
+import type { RpcMethodHandler } from "../../types";
+
+export const sendTransaction: RpcMethodHandler = (id, params, context) => {
+	const [encodedTx] = params;
+	try {
+		const txData = Buffer.from(encodedTx, "base64");
+		const tx = VersionedTransaction.deserialize(txData);
+
+		// Snapshot pre balances
+		const msg: any = tx.message as any;
+		const rawKeys: any[] = Array.isArray(msg.staticAccountKeys)
+			? msg.staticAccountKeys
+			: Array.isArray(msg.accountKeys)
+				? msg.accountKeys
+				: [];
+		const staticKeys = rawKeys
+			.map((k: any) => {
+				try {
+					return typeof k === "string" ? new PublicKey(k) : (k as PublicKey);
+				} catch {
+					return undefined;
+				}
+			})
+			.filter(Boolean) as PublicKey[];
+		const preBalances = staticKeys.map((pk) => {
+			try {
+				return Number(context.svm.getBalance(pk));
+			} catch {
+				return 0;
+			}
+		});
+
+		// Collect SPL token accounts from instructions for pre/post token balance snapshots
+		const msgAny: any = msg;
+		const compiled = Array.isArray(msgAny.compiledInstructions)
+			? msgAny.compiledInstructions
+			: Array.isArray(msgAny.instructions)
+				? msgAny.instructions
+				: [];
+		const tokenProgramIds = new Set([
+			TOKEN_PROGRAM_ID.toBase58(),
+			TOKEN_2022_PROGRAM_ID.toBase58(),
+		]);
+		const tokenAccountSet = new Set<string>();
+		for (const ci of compiled) {
+			try {
+				const pid = staticKeys[ci.programIdIndex]?.toBase58();
+				if (!pid || !tokenProgramIds.has(pid)) continue;
+				const accIdxs: number[] = Array.isArray(ci.accountKeyIndexes)
+					? ci.accountKeyIndexes
+					: Array.isArray(ci.accounts)
+						? ci.accounts
+						: [];
+				for (const ix of accIdxs) {
+					const addr = staticKeys[ix]?.toBase58();
+					if (addr) tokenAccountSet.add(addr);
+				}
+			} catch {}
+		}
+		// Pre token balances
+		const preTokenBalances: any[] = [];
+		const ataToInfo = new Map<
+			string,
+			{
+				mint?: string;
+				owner?: string;
+				amount: bigint;
+				accountIndex: number;
+				decimals?: number;
+			}
+		>();
+		const missingPre = new Set<string>();
+		for (const addr of tokenAccountSet) {
+			try {
+				const pk = new PublicKey(addr);
+				const idx = staticKeys.findIndex((k) => k.equals(pk));
+				const acc = context.svm.getAccount(pk);
+				if (!acc || (acc.data?.length ?? 0) < ACCOUNT_SIZE) {
+					// Track placeholder; we'll fill mint/owner/decimals after send
+					ataToInfo.set(addr, { amount: 0n, accountIndex: idx >= 0 ? idx : 0 });
+					missingPre.add(addr);
+					continue;
+				}
+				const decAcc = AccountLayout.decode(Buffer.from(acc.data));
+				const mintPk = new PublicKey(decAcc.mint);
+				const mintAcc = context.svm.getAccount(mintPk);
+				let decimals = 0;
+				if (mintAcc && (mintAcc.data?.length ?? 0) >= MINT_SIZE) {
+					const m = MintLayout.decode(
+						Buffer.from(mintAcc.data).slice(0, MINT_SIZE),
+					);
+					decimals = Number(m.decimals ?? 0);
+				}
+				const ownerPk = new PublicKey(decAcc.owner);
+				const amt = BigInt(decAcc.amount.toString());
+				ataToInfo.set(addr, {
+					mint: mintPk.toBase58(),
+					owner: ownerPk.toBase58(),
+					amount: amt,
+					accountIndex: idx >= 0 ? idx : 0,
+					decimals,
+				});
+				const uiAmount = Number(amt) / 10 ** decimals;
+				preTokenBalances.push({
+					accountIndex: idx >= 0 ? idx : 0,
+					mint: mintPk.toBase58(),
+					owner: ownerPk.toBase58(),
+					uiTokenAmount: {
+						amount: amt.toString(),
+						decimals,
+						uiAmount,
+						uiAmountString: String(uiAmount),
+					},
+				});
+			} catch {}
+		}
+
+		const result = context.svm.sendTransaction(tx);
+
+		try {
+			// eslint-disable-next-line @typescript-eslint/no-explicit-any
+			const maybeErr = (result as any).err?.();
+			if (maybeErr) {
+				return context.createErrorResponse(
+					id,
+					-32003,
+					"Transaction failed",
+					maybeErr,
+				);
+			}
+		} catch {}
+
+		const signature = tx.signatures[0]
+			? context.encodeBase58(tx.signatures[0])
+			: context.encodeBase58(new Uint8Array(64).fill(0));
+		context.notifySignature(signature);
+		// Snapshot post balances and capture logs for rich view
+		const postBalances = staticKeys.map((pk) => {
+			try {
+				return Number(context.svm.getBalance(pk));
+			} catch {
+				return 0;
+			}
+		});
+		// Post token balances
+		const postTokenBalances: any[] = [];
+		for (const addr of tokenAccountSet) {
+			try {
+				const pk = new PublicKey(addr);
+				const idx = staticKeys.findIndex((k) => k.equals(pk));
+				const acc = context.svm.getAccount(pk);
+				if (!acc || (acc.data?.length ?? 0) < ACCOUNT_SIZE) continue;
+				const decAcc = AccountLayout.decode(Buffer.from(acc.data));
+				const mintPk = new PublicKey(decAcc.mint);
+				const ownerPk = new PublicKey(decAcc.owner);
+				const mintAcc = context.svm.getAccount(mintPk);
+				let decimals = 0;
+				if (mintAcc && (mintAcc.data?.length ?? 0) >= MINT_SIZE) {
+					const m = MintLayout.decode(
+						Buffer.from(mintAcc.data).slice(0, MINT_SIZE),
+					);
+					decimals = Number(m.decimals ?? 0);
+				}
+				const amt = BigInt(decAcc.amount.toString());
+				const uiAmount = Number(amt) / 10 ** decimals;
+				postTokenBalances.push({
+					accountIndex:
+						idx >= 0 ? idx : (ataToInfo.get(addr)?.accountIndex ?? 0),
+					mint: mintPk.toBase58(),
+					owner: ownerPk.toBase58(),
+					uiTokenAmount: {
+						amount: amt.toString(),
+						decimals,
+						uiAmount,
+						uiAmountString: String(uiAmount),
+					},
+				});
+				// Add missing pre entry as zero if account was unfunded before
+				if (missingPre.has(addr)) {
+					const preUi = 0;
+					preTokenBalances.push({
+						accountIndex:
+							idx >= 0 ? idx : (ataToInfo.get(addr)?.accountIndex ?? 0),
+						mint: mintPk.toBase58(),
+						owner: ownerPk.toBase58(),
+						uiTokenAmount: {
+							amount: "0",
+							decimals,
+							uiAmount: preUi,
+							uiAmountString: String(preUi),
+						},
+					});
+				}
+			} catch {}
+		}
+		let logs: string[] = [];
+		try {
+			// eslint-disable-next-line @typescript-eslint/no-explicit-any
+			const anyRes: any = result;
+			if (typeof anyRes?.logs === "function") logs = anyRes.logs();
+			else if (typeof anyRes?.meta === "function")
+				logs = anyRes.meta()?.logs?.() ?? [];
+		} catch {}
+		context.recordTransaction(signature, tx, {
+			logs,
+			fee: 5000,
+			blockTime: Math.floor(Date.now() / 1000),
+			preBalances,
+			postBalances,
+			preTokenBalances,
+			postTokenBalances,
+		});
+
+		return context.createSuccessResponse(id, signature);
+	} catch (error: any) {
+		return context.createErrorResponse(
+			id,
+			-32003,
+			"Transaction failed",
+			error.message,
+		);
+	}
+};

package/server/methods/transaction/simulate-transaction.ts

@@ -0,0 +1,56 @@
+import { VersionedTransaction } from "@solana/web3.js";
+import type { RpcMethodHandler } from "../../types";
+
+export const simulateTransaction: RpcMethodHandler = (id, params, context) => {
+	const [encodedTx] = params;
+	try {
+		const txData = Buffer.from(encodedTx, "base64");
+		const tx = VersionedTransaction.deserialize(txData);
+		const result = context.svm.simulateTransaction(tx);
+
+		if ("err" in result) {
+			const errorMeta = result.meta();
+			// To maximize client compatibility, report err as null and return logs
+			// (Some clients fail to deserialize unknown enum variants or numeric codes.)
+			return context.createSuccessResponse(id, {
+				context: { slot: Number(context.slot) },
+				value: {
+					err: null,
+					logs: errorMeta.logs(),
+					accounts: null,
+					unitsConsumed: Number(errorMeta.computeUnitsConsumed()),
+					returnData: null,
+				},
+			});
+		}
+
+		const meta = result.meta();
+		const returnData = meta.returnData();
+
+		return context.createSuccessResponse(id, {
+			context: { slot: Number(context.slot) },
+			value: {
+				err: null,
+				logs: meta.logs(),
+				accounts: null,
+				unitsConsumed: Number(meta.computeUnitsConsumed()),
+				returnData: returnData
+					? {
+							programId: context.encodeBase58(returnData.programId()),
+							data: [
+								Buffer.from(returnData.data()).toString("base64"),
+								"base64",
+							],
+						}
+					: null,
+			},
+		});
+	} catch (error: any) {
+		return context.createErrorResponse(
+			id,
+			-32003,
+			"Simulation failed",
+			error.message,
+		);
+	}
+};