import { BigNumber, BigNumberish } from "ethers";
import { arrayify, Interface } from "ethers/lib/utils";

import { TracerDependencies } from "../types";
import { compareBytecode } from "../utils";
import { colorContract, colorFunctionSuccess, colorKey } from "../utils/colors";

import { formatParam } from "./param";
import { formatResult } from "./result";

export async function formatContract(
  code: string,
  value: BigNumberish,
  salt: BigNumberish | null,
  deployedAddress: string | null,
  dependencies: TracerDependencies
) {
  value = BigNumber.from(value);
  if (salt !== null) {
    salt = BigNumber.from(salt);
  }
  const names = await dependencies.artifacts.getAllFullyQualifiedNames();

  for (const name of names) {
    const artifact = await dependencies.artifacts.readArtifact(name);
    const iface = new Interface(artifact.abi);

    if (
      artifact.bytecode.length > 2 &&
      artifact.bytecode.length <= code.length &&
      compareBytecode(artifact.bytecode, code) > 0.5
    ) {
      // we found the artifact with matching bytecode
      try {
        const constructorParamsDecoded = iface._decodeParams(
          iface.deploy.inputs,
          "0x" + code.slice(artifact.bytecode.length)
        );
        const inputArgs = formatResult(
          constructorParamsDecoded,
          iface.deploy.inputs,
          { decimals: -1, shorten: false },
          dependencies
        );
        const extra = [];
        if (value.gt(0)) {
          extra.push(`value: ${formatParam(value, dependencies)}`);
        }
        if (salt !== null) {
          extra.push(
            `salt: ${formatParam(
              salt.gt(2 ** 32) ? salt.toHexString() : salt,
              dependencies
            )}`
          );
        }
        return `${colorContract(artifact.contractName)}.${colorFunctionSuccess(
          "constructor"
        )}${extra.length !== 0 ? `{${extra.join(",")}}` : ""}(${inputArgs})${
          deployedAddress
            ? ` → (${formatParam(deployedAddress, dependencies)})`
            : ""
        }`;
      } catch {}
    }
  }

  return `${colorContract("UnknownContract")}(${colorKey("deployCodeSize=")}${
    arrayify(code).length
  })`;
}