import { eth_getCode } from "../../rpc/actions/eth_getCode.js";
import { getRpcClient } from "../../rpc/rpc.js";
import type { Hex } from "../../utils/encoding/hex.js";
import type { ThirdwebContract } from "../contract.js";

const BYTECODE_CACHE = new WeakMap<ThirdwebContract, Promise<Hex>>();

/**
 * Retrieves the bytecode of a contract.
 * @param contract - The ThirdwebContract instance.
 * @returns A Promise that resolves to the bytecode of the contract.
 * @example
 * ```ts
 * import { getBytecode } from "thirdweb/contract";
 * const bytecode = await getBytecode(contract);
 * ```
 * @contract
 */
export function getBytecode(contract: ThirdwebContract): Promise<Hex> {
  if (BYTECODE_CACHE.has(contract)) {
    return BYTECODE_CACHE.get(contract) as Promise<Hex>;
  }

  const prom = (async () => {
    const rpcRequest = getRpcClient(contract);
    const result = await eth_getCode(rpcRequest, {
      address: contract.address,
      blockTag: "latest",
    });
    if (result === "0x") {
      BYTECODE_CACHE.delete(contract);
    }
    return result;
  })();
  BYTECODE_CACHE.set(contract, prom);
  return prom;
}