@tevm/actions/src/LoadState/loadStateProcedure.js

A typesafe library for writing forge scripts in typescript

import { hexToBigInt } from '@tevm/utils'
import { loadStateHandler } from './loadStateHandler.js'

/**
 * Creates a LoadState JSON-RPC Procedure for handling LoadState requests with Ethereumjs EVM
 * @param {import('@tevm/node').TevmNode} client
 * @returns {import('./LoadStateJsonRpcProcedure.js').LoadStateJsonRpcProcedure}
 */
export const loadStateProcedure = (client) => async (request) => {
	const {
		params: [{ state }],
	} = request

	/**
	 * @type {import('@tevm/state').TevmState}
	 */
	const parsedState = {}

	for (const [k, v] of Object.entries(state)) {
		const { nonce, balance, storageRoot, codeHash } = v
		parsedState[/** @type {import('@tevm/utils').Address}*/ (k)] = {
			...v,
			nonce: hexToBigInt(nonce),
			balance: hexToBigInt(balance),
			storageRoot: storageRoot,
			codeHash: codeHash,
		}
	}
	const { errors = [] } = await loadStateHandler(client)({
		state: parsedState,
		throwOnFail: false,
	})

	if (errors.length > 0) {
		const error = /** @type {import('./TevmLoadStateError.js').TevmLoadStateError}*/ (errors[0])
		return {
			jsonrpc: '2.0',
			error: {
				code: error.code,
				message: error.message,
				data: {
					errors: errors.map(({ message }) => message),
				},
			},
			method: 'tevm_loadState',
			...(request.id === undefined ? {} : { id: request.id }),
		}
	}
	return {
		jsonrpc: '2.0',
		result: {},
		method: 'tevm_loadState',
		...(request.id === undefined ? {} : { id: request.id }),
	}
}