ethereumClient.ts

import {
  WalletClient,
  createWalletClient,
  createPublicClient,
  http,
  PrivateKeyAccount,
  PublicClient,
} from "viem";
import { privateKeyToAccount } from "viem/accounts";
import { optimism } from "viem/chains";
import { CONTRACT_ABI, OPTIMISTIC_ETHERSCAN_TX } from "./utils/constants";
import { Submission } from "./model/submission";
import { getUserById } from "./controller/userController";
import { generateSHA256Hash } from "./utils/shared";
import { sendMessage } from "./discordBot";
import { getLevelById } from "./controller/levelController";

enum SubmissionTypeConverter {
  "solidity" = 0,
  "yul" = 1,
  "vyper" = 2,
  "huff" = 3,
  "bytecode" = 4,
}

enum SubmissionOptimizedForConverter {
  "gas" = 0,
  "size" = 1,
}

let ethereumClient: PublicClient;
let account: PrivateKeyAccount;
let walletClient: WalletClient;
let nonce: number;
const transport = http(process.env.API_URL);

const connect = async () => {
  try {
    account = privateKeyToAccount(`0x${process.env.PRIVATE_KEY}`);
    //<any> seems to fix a viem error when working with optimism/base, etc..
    ethereumClient = createPublicClient<any>({
      batch: {
        multicall: true,
      },
      chain: optimism,
      transport,
    });

    walletClient = createWalletClient({
      chain: optimism,
      transport,
    });

    nonce = await ethereumClient.getTransactionCount({
      address: account.address,
    });
  } catch (error) {
    console.log("## ETHEREUM CLIENT CONNECT ERROR: ", error);
  }
};

connect();

const storeSubmissionOnChain = async (
  user_id: number,
  level_id: number,
  submissions: Submission[]
) => {
  const user = await getUserById(user_id);
  const level = await getLevelById(level_id);
  for (const s of submissions) {
    const { request } = await ethereumClient.simulateContract({
      account: account,
      nonce: nonce,
      address: process.env.CONTRACT_ADDRESS as `0x${string}`,
      abi: CONTRACT_ABI,
      functionName: "submit",
      args: [
        user?.wallet_address,
        {
          id: s?.id,
          level_id: s?.level_id,
          gas: s?.gas,
          size: s?.size,
          solutionType:
            SubmissionTypeConverter[
              s?.type as keyof typeof SubmissionTypeConverter
            ], //just to avoid warnings
          optimized_for:
            SubmissionOptimizedForConverter[
              s?.optimized_for as keyof typeof SubmissionOptimizedForConverter
            ], //just to avoid warnings
          submitted_at: Date.parse(s?.submitted_at.toString()),
          bytecode_hash: `0x${generateSHA256Hash(s?.bytecode)}`,
          user_name: user?.name,
        },
      ],
    });

    nonce++;
    const txHash = await walletClient.writeContract(request);
    sendMessage(
      `${user?.name} submitted a new solution for level ${level?.name}. [See the transaction here.](${OPTIMISTIC_ETHERSCAN_TX}${txHash})`
    );
  }
};

export { storeSubmissionOnChain };