import { BigNumber, BigNumberish } from "ethers";
import { formatEther, Fragment, FunctionFragment } from "ethers/lib/utils";

import { SEPARATOR } from "../constants";
import { Obj, TracerDependencies } from "../types";
import { fetchContractDecimals, getBetterContractName } from "../utils";
import {
  colorContract,
  colorExtra,
  colorFunctioFail,
  colorFunctionSuccess,
  colorKey,
  colorValue,
} from "../utils/colors";

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

export async function formatCall(
  to: string | undefined,
  input: string,
  ret: string,
  value: BigNumberish,
  gasUsed: BigNumberish,
  gasLimit: BigNumberish,
  success: boolean,
  dependencies: TracerDependencies
) {
  let contractName: string | undefined;
  let contractDecimals: number | undefined;
  let strictlyHideAddress: boolean = false;
  let functionName: string | undefined;
  let inputResult: Obj<any> | undefined;
  let returnResult: Obj<any> | undefined;
  let fragment: Fragment | undefined;

  const precompileResult = formatPrecompile(to, input, ret);
  if (precompileResult) {
    contractName = "Precompiles";
    strictlyHideAddress = true;
    ({ name: functionName, inputResult, returnResult } = precompileResult);
  } else {
    try {
      ({
        fragment,
        contractName,
        inputResult,
        returnResult,
      } = await dependencies.tracerEnv.decoder!.decodeFunction(input, ret));

      // use just contract name
      contractName = contractName.split(":")[1];
    } catch {}

    // find a better contract name
    if (to) {
      const betterContractName = await getBetterContractName(to, dependencies);
      if (betterContractName) {
        contractName = betterContractName;
      } else if (contractName) {
        dependencies.tracerEnv.nameTags[to] = contractName;
      }
    }

    // if ERC20 method found then fetch decimals
    if (
      to &&
      (input.slice(0, 10) === "0x70a08231" || // balanceOf
      input.slice(0, 10) === "0xa9059cbb" || // transfer
        input.slice(0, 10) === "0x23b872dd") // transferFrom
    ) {
      // see if we already know the decimals
      const { cache } = dependencies.tracerEnv._internal;
      const decimals = cache.tokenDecimals.get(to);
      if (decimals) {
        // if we know decimals then use it
        contractDecimals = decimals !== -1 ? decimals : undefined;
      } else {
        // otherwise fetch it
        contractDecimals = await fetchContractDecimals(to, dependencies);
        // and cache it
        if (contractDecimals !== undefined) {
          cache.tokenDecimals.set(to, contractDecimals);
        } else {
          cache.tokenDecimals.set(to, -1);
        }
        cache.save();
      }
    }
  }

  const extra = [];
  if ((value = BigNumber.from(value)).gt(0)) {
    extra.push(
      `value${SEPARATOR}${formatParam(value, dependencies, { isEther: true })}`
    );
  }
  if (
    (gasLimit = BigNumber.from(gasLimit)).gt(0) &&
    (gasUsed = BigNumber.from(gasUsed)).gt(0) &&
    dependencies.tracerEnv.gasCost
  ) {
    extra.push(`gasLimit${SEPARATOR}${formatParam(gasLimit, dependencies)}`);
    extra.push(`gasUsed${SEPARATOR}${formatParam(gasUsed, dependencies)}`);
  }

  const colorFunction = success ? colorFunctionSuccess : colorFunctioFail;

  if (inputResult && (functionName || fragment)) {
    // format known stuff
    const inputArgs = formatResult(
      inputResult,
      fragment?.inputs,
      { decimals: contractDecimals, shorten: false },
      dependencies
    );
    const outputArgs = returnResult
      ? formatResult(
          returnResult,
          (fragment as FunctionFragment)?.outputs,
          { decimals: contractDecimals, shorten: true },
          dependencies
        )
      : // if return data is not decoded, then show return data only if call was success
      ret !== "0x" && success !== false // success can be undefined
      ? ret
      : "";

    const nameToPrint = contractName ?? "UnknownContract";

    return `${
      to &&
      !strictlyHideAddress &&
      (dependencies.tracerEnv.showAddresses ||
        nameToPrint === "UnknownContract")
        ? `${colorContract(nameToPrint)}(${to})`
        : colorContract(nameToPrint)
    }.${colorFunction(functionName ?? fragment?.name)}${
      extra.length !== 0 ? colorExtra(`{${extra.join(", ")}}`) : ""
    }(${inputArgs})${outputArgs ? ` → (${outputArgs})` : ""}`;
  }

  // format unknown stuff
  if (contractName) {
    return `${
      to && dependencies.tracerEnv.showAddresses
        ? `${colorContract(contractName)}(${to})`
        : colorContract(contractName)
    }.<${colorFunction("UnknownFunction")}>${
      extra.length !== 0 ? colorExtra(`{${extra.join(", ")}}`) : ""
    }(${colorKey("input" + SEPARATOR)}${colorValue(input)}, ${colorKey(
      "ret" + SEPARATOR
    )}${colorValue(ret)})`;
  } else if (input === "0x" && ret === "0x" && value.gt(0)) {
    return `${colorFunction("EtherTransfer")}${
      extra.length !== 0 ? colorExtra(`{${extra.join(", ")}}`) : ""
    }(${colorKey("to" + SEPARATOR)}${colorValue(to)})`;
  } else {
    return `${colorFunction("UnknownContractAndFunction")}${
      extra.length !== 0 ? colorExtra(`{${extra.join(", ")}}`) : ""
    }(${colorKey("to" + SEPARATOR)}${colorValue(to)}, ${colorKey(
      "input" + SEPARATOR
    )}${input}, ${colorKey("ret" + SEPARATOR)}${colorValue(ret || "0x")})`;
  }
}