anvil.ts

import { spawnSync } from "child_process";
import { useHandler } from "../../static/scripts/rewards/web3/use-rpc-handler";
import { RPCHandler } from "@ubiquity-dao/rpc-handler";

class Anvil {
  rpcs: string[] = [];
  rpcHandler: RPCHandler | null = null;
  networkIdToFork = 100;
  networkIdForAnvilToForkAs = 31337;

  async init() {
    this.rpcHandler = useHandler(this.networkIdToFork);
    console.log(`[RPCHandler] Fetching RPCs...`);
    await this.rpcHandler.testRpcPerformance();
    const latencies: Record<string, number> = this.rpcHandler.getLatencies();
    const sorted = Object.entries(latencies).sort(([, a], [, b]) => a - b);
    console.log(
      `Fetched ${sorted.length} RPCs.\nFastest: ${sorted[0][0]} (${sorted[0][1]}ms)\nSlowest: ${sorted[sorted.length - 1][0]} (${sorted[sorted.length - 1][1]}ms)`
    );

    this.rpcs = sorted.map(([rpc]) => rpc.split("__")[1]);
  }

  async run() {
    await this.init();
    console.log(`Starting Anvil...`);
    const isSuccess = await this.spawner(this.rpcs.shift());

    if (!isSuccess) {
      throw new Error(`Anvil failed to start`);
    }
  }

  async spawner(rpc?: string): Promise<boolean> {
    if (!rpc) {
      console.log(`No RPCs left to try`);
      return false;
    }

    console.log(`Forking with RPC: ${rpc}`);

    const anvil = spawnSync("anvil", ["--chain-id", `${this.networkIdForAnvilToForkAs}`, "--fork-url", rpc, "--host", "127.0.0.1", "--port", "8545"], {
      stdio: "inherit",
    });

    if (anvil.status !== 0) {
      console.log(`Anvil failed to start with RPC: ${rpc}`);
      console.log(`Retrying with next RPC...`);
      return this.spawner(this.rpcs.shift());
    }

    return true;
  }
}

async function main() {
  const anvil = new Anvil();
  await anvil.run();
}

main().catch((error) => {
  console.error(error);
  process.exit(1);
});