solforge 0.2.12 → 0.2.13

package/server/rpc-server.ts

@@ -1,521 +0,0 @@
-import {
-	type Keypair,
-	PublicKey,
-	type VersionedTransaction,
-} from "@solana/web3.js";
-import { LiteSVM } from "litesvm";
-import { sqlite } from "../src/db/index";
-import { TxStore } from "../src/db/tx-store";
-import { decodeBase58, encodeBase58 } from "./lib/base58";
-import { fundFaucetIfNeeded, loadOrCreateFaucet } from "./lib/faucet";
-import { rpcMethods } from "./methods";
-import type {
-	JsonRpcRequest,
-	JsonRpcResponse,
-	RpcMethodContext,
-} from "./types";
-
-export class LiteSVMRpcServer {
-	private svm: LiteSVM;
-	private slot: bigint = 1n;
-	private blockHeight: bigint = 1n;
-	private txCount: bigint = 0n;
-	private signatureListeners: Set<(sig: string) => void> = new Set();
-	private knownMints: Set<string> = new Set();
-	private knownPrograms: Set<string> = new Set();
-	private faucet: Keypair;
-	private txRecords: Map<
-		string,
-		{
-			tx: VersionedTransaction;
-			logs: string[];
-			err: unknown;
-			fee: number;
-			slot: number;
-			blockTime?: number;
-			preBalances?: number[];
-			postBalances?: number[];
-			preTokenBalances?: unknown[];
-			postTokenBalances?: unknown[];
-		}
-	> = new Map();
-	private store: TxStore;
-
-	constructor() {
-		this.svm = new LiteSVM()
-			.withSysvars()
-			.withBuiltins()
-			.withDefaultPrograms()
-			// Mint 1,000,000 SOL (1e15 lamports) for local dev
-			.withLamports(1_000_000_000_000_000n)
-			.withBlockhashCheck(true)
-			// keep some tx history so getTransaction/getSignatureStatuses can work
-			.withTransactionHistory(1000n)
-			.withSigverify(false);
-		this.store = new TxStore();
-		// Seed slot/blockHeight/txCount from DB if available for continuity
-		try {
-			const maxRow = sqlite
-				.prepare("SELECT MAX(slot) as m FROM transactions")
-				.get() as { m?: number } | undefined;
-			const cntRow = sqlite
-				.prepare("SELECT COUNT(1) as c FROM transactions")
-				.get() as { c?: number } | undefined;
-			const maxSlot = (maxRow?.m ?? 0) as number;
-			const txc = (cntRow?.c ?? 0) as number;
-			if (maxSlot > 0) {
-				this.slot = BigInt(maxSlot + 1);
-				this.blockHeight = BigInt(maxSlot + 1);
-			}
-			if (txc > 0) {
-				this.txCount = BigInt(txc);
-			}
-		} catch {}
-
-		// Load or create faucet; fund once at startup
-		this.faucet = loadOrCreateFaucet();
-		try {
-			const bal = fundFaucetIfNeeded(this.svm, this.faucet);
-			console.log(
-				`💧 Faucet loaded: ${this.faucet.publicKey.toBase58()} with ${(Number(bal) / 1_000_000_000).toFixed(0)} SOL`,
-			);
-		} catch (e) {
-			console.warn("⚠️ Faucet funding failed:", e);
-			try {
-				const bal =
-					this.svm.getBalance(this.faucet.publicKey as PublicKey) || 0n;
-				console.log(
-					`💧 Faucet balance: ${(Number(bal) / 1_000_000_000).toFixed(9)} SOL`,
-				);
-			} catch {}
-		}
-	}
-
-	// base58 helpers moved to server/lib/base58
-
-	private createSuccessResponse(
-		id: string | number,
-		result: unknown,
-	): JsonRpcResponse {
-		return {
-			jsonrpc: "2.0",
-			id,
-			result,
-		};
-	}
-
-	private createErrorResponse(
-		id: string | number,
-		code: number,
-		message: string,
-		data?: unknown,
-	): JsonRpcResponse {
-		return {
-			jsonrpc: "2.0",
-			id,
-			error: { code, message, data },
-		};
-	}
-
-	private getContext(): RpcMethodContext {
-		return {
-			svm: this.svm,
-			slot: this.slot,
-			blockHeight: this.blockHeight,
-			store: this.store,
-			encodeBase58,
-			decodeBase58,
-			createSuccessResponse: this.createSuccessResponse.bind(this),
-			createErrorResponse: this.createErrorResponse.bind(this),
-			notifySignature: (signature: string) => {
-				for (const cb of this.signatureListeners) cb(signature);
-			},
-			getFaucet: () => this.faucet,
-			getTxCount: () => this.txCount,
-			registerMint: (mint: PublicKey | string) => {
-				try {
-					const pk =
-						typeof mint === "string" ? mint : new PublicKey(mint).toBase58();
-					this.knownMints.add(pk);
-				} catch {}
-			},
-			listMints: () => Array.from(this.knownMints),
-			registerProgram: (program: PublicKey | string) => {
-				try {
-					const pk =
-						typeof program === "string"
-							? program
-							: new PublicKey(program).toBase58();
-					this.knownPrograms.add(pk);
-				} catch {}
-			},
-			listPrograms: () => Array.from(this.knownPrograms),
-			recordTransaction: (signature, tx, meta) => {
-				this.txRecords.set(signature, {
-					tx,
-					logs: meta?.logs || [],
-					err: meta?.err ?? null,
-					fee: meta?.fee ?? 5000,
-					slot: Number(this.slot),
-					blockTime: meta?.blockTime,
-					preBalances: meta?.preBalances,
-					postBalances: meta?.postBalances,
-					preTokenBalances: (
-						meta as { preTokenBalances?: unknown[] } | undefined
-					)?.preTokenBalances,
-					postTokenBalances: (
-						meta as { postTokenBalances?: unknown[] } | undefined
-					)?.postTokenBalances,
-					innerInstructions: meta?.innerInstructions || [],
-					computeUnits:
-						meta?.computeUnits == null ? null : Number(meta.computeUnits),
-					returnData: meta?.returnData ?? null,
-				});
-				try {
-					if (process.env.DEBUG_TX_CAPTURE === "1") {
-						console.debug(
-							`[tx-capture] recordTransaction: sig=${signature} slot=${this.slot} logs=${meta?.logs?.length || 0} inner=${Array.isArray(meta?.innerInstructions) ? meta?.innerInstructions?.length : 0} cu=${meta?.computeUnits ?? null} returnData=${meta?.returnData ? "yes" : "no"}`,
-						);
-					}
-				} catch {}
-
-				// Persist to SQLite for durability and history queries
-				try {
-					const msg = tx.message as unknown as {
-						staticAccountKeys?: unknown[];
-						accountKeys?: unknown[];
-						header?: unknown;
-						isAccountSigner?: (i: number) => boolean;
-						isAccountWritable?: (i: number) => boolean;
-						version?: number;
-					};
-					const rawKeys: unknown[] = Array.isArray(msg.staticAccountKeys)
-						? msg.staticAccountKeys
-						: Array.isArray(msg.accountKeys)
-							? msg.accountKeys
-							: [];
-					const keys: string[] = rawKeys.map((k) => {
-						try {
-							return typeof k === "string" ? k : (k as PublicKey).toBase58();
-						} catch {
-							return String(k);
-						}
-					});
-					const header = msg.header || {
-						numRequiredSignatures: (tx.signatures || []).length,
-						numReadonlySignedAccounts: 0,
-						numReadonlyUnsignedAccounts: 0,
-					};
-					const numReq = Number(header.numRequiredSignatures || 0);
-					const numRoSigned = Number(header.numReadonlySignedAccounts || 0);
-					const numRoUnsigned = Number(header.numReadonlyUnsignedAccounts || 0);
-					const total = keys.length;
-					const unsignedCount = Math.max(0, total - numReq);
-					const writableSignedCutoff = Math.max(0, numReq - numRoSigned);
-					const writableUnsignedCount = Math.max(
-						0,
-						unsignedCount - numRoUnsigned,
-					);
-
-					const accounts = keys.map((addr, i) => {
-						const signer =
-							typeof msg.isAccountSigner === "function"
-								? !!msg.isAccountSigner(i)
-								: i < numReq;
-						let writable = true;
-						if (typeof msg.isAccountWritable === "function")
-							writable = !!msg.isAccountWritable(i);
-						else {
-							if (i < numReq) writable = i < writableSignedCutoff;
-							else writable = i - numReq < writableUnsignedCount;
-						}
-						return { address: addr, index: i, signer, writable };
-					});
-					const version: 0 | "legacy" =
-						typeof msg.version === "number"
-							? msg.version === 0
-								? 0
-								: "legacy"
-							: 0;
-					const rawBase64 = Buffer.from(tx.serialize()).toString("base64");
-					this.store
-						.insertTransactionBundle({
-							signature,
-							slot: Number(this.slot),
-							blockTime: meta?.blockTime,
-							version,
-							fee: Number(meta?.fee ?? 5000),
-							err: meta?.err ?? null,
-							rawBase64,
-							preBalances: Array.isArray(meta?.preBalances)
-								? (meta?.preBalances as number[])
-								: [],
-							postBalances: Array.isArray(meta?.postBalances)
-								? (meta?.postBalances as number[])
-								: [],
-							logs: Array.isArray(meta?.logs) ? (meta?.logs as string[]) : [],
-							preTokenBalances: (() => {
-								const arr = (
-									meta as { preTokenBalances?: unknown[] } | undefined
-								)?.preTokenBalances;
-								return Array.isArray(arr) ? arr : [];
-							})(),
-							postTokenBalances: (() => {
-								const arr = (
-									meta as { postTokenBalances?: unknown[] } | undefined
-								)?.postTokenBalances;
-								return Array.isArray(arr) ? arr : [];
-							})(),
-							innerInstructions: Array.isArray(meta?.innerInstructions)
-								? meta?.innerInstructions
-								: [],
-							computeUnits:
-								meta?.computeUnits == null ? null : Number(meta.computeUnits),
-							returnData: meta?.returnData ?? null,
-							accounts,
-							accountStates: Array.isArray(meta?.accountStates)
-								? meta?.accountStates
-								: [],
-						})
-						.catch(() => {});
-
-					// Upsert account snapshots for static keys
-					const snapshots = keys
-						.map((addr) => {
-							try {
-								const acc = this.svm.getAccount(new PublicKey(addr));
-								if (!acc) return null;
-								const ownerStr = new PublicKey(acc.owner).toBase58();
-								let dataBase64: string | undefined;
-								// Store raw data for SPL Token accounts to reflect balance changes
-								try {
-									if (
-										ownerStr ===
-											"TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA" ||
-										ownerStr === "TokenzQdBNbLqP5VEhdkAS6EPFLC1PHnBqCXEpPxuEb"
-									) {
-										if (acc.data && acc.data.length > 0) {
-											dataBase64 = Buffer.from(acc.data).toString("base64");
-										}
-									}
-								} catch {}
-								return {
-									address: addr,
-									lamports: Number(acc.lamports || 0n),
-									ownerProgram: ownerStr,
-									executable: !!acc.executable,
-									rentEpoch: Number(acc.rentEpoch || 0),
-									dataLen: acc.data?.length ?? 0,
-									dataBase64,
-									lastSlot: Number(this.slot),
-								};
-							} catch {
-								return null;
-							}
-						})
-						.filter(Boolean) as import("../src/db/tx-store").AccountSnapshot[];
-					if (snapshots.length > 0)
-						this.store.upsertAccounts(snapshots).catch(() => {});
-				} catch {}
-			},
-			getRecordedTransaction: (signature) => this.txRecords.get(signature),
-		};
-	}
-
-	onSignatureRecorded(cb: (sig: string) => void) {
-		this.signatureListeners.add(cb);
-		return () => this.signatureListeners.delete(cb);
-	}
-
-	getFaucetAddress(): string {
-		try {
-			return this.faucet.publicKey.toBase58();
-		} catch {
-			return "";
-		}
-	}
-
-	getFaucetBalance(): bigint {
-		try {
-			return this.svm.getBalance(this.faucet.publicKey as PublicKey) ?? 0n;
-		} catch {
-			return 0n;
-		}
-	}
-
-	listKnownMints(): string[] {
-		return Array.from(this.knownMints);
-	}
-
-	listKnownPrograms(): string[] {
-		return Array.from(this.knownPrograms);
-	}
-
-	getSignatureStatus(
-		signature: string,
-	): { slot: number; err: unknown | null } | null {
-		// Prefer local record for reliability
-		const rec = this.txRecords.get(signature);
-		if (rec) {
-			return { slot: rec.slot, err: rec.err ?? null };
-		}
-		try {
-			const sigBytes = decodeBase58(signature);
-			const tx = this.svm.getTransaction(sigBytes);
-			if (!tx) return null;
-			let errVal: unknown = null;
-			try {
-				const raw = (tx as { err?: unknown }).err;
-				errVal = typeof raw === "function" ? (raw as () => unknown)() : raw;
-			} catch {
-				errVal = null;
-			}
-			return { slot: Number(this.slot), err: errVal };
-		} catch {
-			return null;
-		}
-	}
-
-	async handleRequest(request: JsonRpcRequest): Promise<JsonRpcResponse> {
-		const { method, params, id } = request;
-
-		try {
-			const methodHandler = rpcMethods[method];
-
-			if (!methodHandler) {
-				return this.createErrorResponse(
-					id,
-					-32601,
-					`Method not found: ${method}`,
-				);
-			}
-
-			const context = this.getContext();
-			const result = await methodHandler(id, params, context);
-
-			// Update slot and blockHeight for methods that modify state
-			if (["sendTransaction", "requestAirdrop"].includes(method)) {
-				this.slot += 1n;
-				this.blockHeight += 1n;
-				this.txCount += 1n;
-			}
-
-			return result;
-		} catch (error: unknown) {
-			const message = error instanceof Error ? error.message : String(error);
-			return this.createErrorResponse(id, -32603, "Internal error", message);
-		}
-	}
-}
-
-export function createLiteSVMRpcServer(port: number = 8899, host?: string) {
-	const server = new LiteSVMRpcServer();
-
-	const bunServer = Bun.serve({
-		port,
-		hostname: host || process.env.RPC_HOST || "127.0.0.1",
-		async fetch(req) {
-			const DEBUG = process.env.DEBUG_RPC_LOG === "1";
-			const acrh = req.headers.get("Access-Control-Request-Headers");
-			const allowHeaders =
-				acrh && acrh.length > 0
-					? acrh
-					: "Content-Type, Accept, Origin, solana-client";
-			const corsHeaders = {
-				"Access-Control-Allow-Origin": "*",
-				"Access-Control-Allow-Methods": "GET, POST, OPTIONS, HEAD",
-				"Access-Control-Allow-Headers": allowHeaders,
-				"Access-Control-Max-Age": "600",
-				// Help Chrome when accessing local network/localhost from secure context
-				"Access-Control-Allow-Private-Network": "true",
-			} as const;
-
-			if (req.method === "GET") {
-				const url = new URL(req.url);
-				if (url.pathname === "/" || url.pathname === "") {
-					return new Response("ok", {
-						headers: { "Content-Type": "text/plain", ...corsHeaders },
-					});
-				}
-				if (url.pathname === "/health") {
-					return new Response("ok", {
-						headers: { "Content-Type": "text/plain", ...corsHeaders },
-					});
-				}
-				return new Response("Not found", { status: 404, headers: corsHeaders });
-			}
-			if (req.method === "POST") {
-				try {
-					const body = await req.json();
-
-					if (Array.isArray(body)) {
-						if (DEBUG) {
-							try {
-								console.log(
-									"RPC batch:",
-									body.map((b: { method?: string }) => b.method),
-								);
-							} catch {}
-						}
-						const responses = await Promise.all(
-							body.map((request) => server.handleRequest(request)),
-						);
-						return new Response(JSON.stringify(responses), {
-							headers: { "Content-Type": "application/json", ...corsHeaders },
-						});
-					} else {
-						const reqObj = body as JsonRpcRequest;
-						if (DEBUG) {
-							try {
-								console.log(
-									"RPC:",
-									reqObj.method,
-									JSON.stringify(reqObj.params),
-								);
-							} catch {}
-						}
-						const response = await server.handleRequest(reqObj);
-						return new Response(JSON.stringify(response), {
-							headers: { "Content-Type": "application/json", ...corsHeaders },
-						});
-					}
-				} catch (_error) {
-					return new Response(
-						JSON.stringify({
-							jsonrpc: "2.0",
-							id: null,
-							error: {
-								code: -32700,
-								message: "Parse error",
-							},
-						}),
-						{ headers: { "Content-Type": "application/json", ...corsHeaders } },
-					);
-				}
-			}
-
-			if (req.method === "OPTIONS") {
-				return new Response(null, { headers: corsHeaders });
-			}
-			if (req.method === "HEAD") {
-				return new Response(null, { headers: corsHeaders });
-			}
-
-			return new Response("Method not allowed", {
-				status: 405,
-				headers: corsHeaders,
-			});
-		},
-		error(error) {
-			console.error("Server error:", error);
-			return new Response("Internal Server Error", { status: 500 });
-		},
-	});
-
-	const hostname = (host || process.env.RPC_HOST || "127.0.0.1").toString();
-	console.log(`🚀 LiteSVM RPC Server running on http://${hostname}:${port}`);
-	console.log(`   Compatible with Solana RPC API`);
-	console.log(`   Use with: solana config set -u http://${hostname}:${port}`);
-
-	return { httpServer: bunServer, rpcServer: server };
-}

package/server/types.ts

@@ -1,109 +0,0 @@
-import type { Keypair, PublicKey, VersionedTransaction } from "@solana/web3.js";
-import type { LiteSVM } from "litesvm";
-import type { TxStore } from "../src/db/tx-store";
-
-export interface JsonRpcRequest {
-	jsonrpc: "2.0";
-	id: string | number;
-	method: string;
-	params?: unknown;
-}
-
-export interface JsonRpcResponse {
-	jsonrpc: "2.0";
-	id: string | number;
-	result?: unknown;
-	error?: {
-		code: number;
-		message: string;
-		data?: unknown;
-	};
-}
-
-export interface RpcMethodContext {
-	svm: LiteSVM;
-	slot: bigint;
-	blockHeight: bigint;
-	store?: TxStore;
-	encodeBase58: (bytes: Uint8Array) => string;
-	decodeBase58: (str: string) => Uint8Array;
-	createSuccessResponse: (
-		id: string | number,
-		result: unknown,
-	) => JsonRpcResponse;
-	createErrorResponse: (
-		id: string | number,
-		code: number,
-		message: string,
-		data?: unknown,
-	) => JsonRpcResponse;
-	notifySignature: (signature: string) => void;
-	getFaucet: () => Keypair;
-	getTxCount: () => bigint;
-	registerMint?: (mint: PublicKey | string) => void;
-	listMints?: () => string[];
-	registerProgram?: (program: PublicKey | string) => void;
-	listPrograms?: () => string[];
-	recordTransaction: (
-		signature: string,
-		tx: VersionedTransaction,
-		meta?: {
-			logs?: string[];
-			err?: unknown;
-			fee?: number;
-			blockTime?: number;
-			preBalances?: number[];
-			postBalances?: number[];
-			preTokenBalances?: unknown[];
-			postTokenBalances?: unknown[];
-			innerInstructions?: unknown[];
-			computeUnits?: number | bigint | null;
-			returnData?: { programId: string; dataBase64: string } | null;
-			// Optional rich per-account snapshots captured around execution
-			accountStates?: Array<{
-				address: string;
-				pre?: {
-					lamports?: number;
-					ownerProgram?: string;
-					executable?: boolean;
-					rentEpoch?: number;
-					dataLen?: number;
-					dataBase64?: string | null;
-					lastSlot?: number;
-				} | null;
-				post?: {
-					lamports?: number;
-					ownerProgram?: string;
-					executable?: boolean;
-					rentEpoch?: number;
-					dataLen?: number;
-					dataBase64?: string | null;
-					lastSlot?: number;
-				} | null;
-			}>;
-		},
-	) => void;
-	getRecordedTransaction: (signature: string) =>
-		| {
-				tx: VersionedTransaction;
-				logs: string[];
-				err: unknown;
-				fee: number;
-				slot: number;
-				blockTime?: number;
-				preBalances?: number[];
-				postBalances?: number[];
-				preTokenBalances?: unknown[];
-				postTokenBalances?: unknown[];
-				innerInstructions?: unknown[];
-				computeUnits?: number | null;
-				returnData?: { programId: string; dataBase64: string } | null;
-		  }
-		| undefined;
-}
-
-export type RpcMethodHandler = (
-	id: string | number,
-	params: unknown[] | undefined,
-	context: RpcMethodContext,
-) => JsonRpcResponse | Promise<JsonRpcResponse>;