import { BrowserInfo } from "@common-module/app";
import { Confirm } from "@common-module/app-components";
import { MaterialIcon } from "@common-module/material-icons";
import {
  BaseContract,
  ContractTransactionResponse,
  Contract as EthersContract,
  Interface,
  InterfaceAbi,
  JsonRpcProvider,
  TransactionReceipt,
} from "ethers";
import ChainInfo from "./ChainInfo.js";
import WalletService from "./WalletService.js";

export default abstract class Contract<CT extends BaseContract> {
  private chain!: ChainInfo;
  private address!: string;

  protected provider!: JsonRpcProvider;
  protected viewContract!: CT;

  constructor(private abi: Interface | InterfaceAbi) {}

  public init(chain: ChainInfo, address: string) {
    this.chain = chain;
    this.address = address;
    this.viewContract = new EthersContract(
      address,
      this.abi,
      this.provider = new JsonRpcProvider(chain.rpc),
    ) as any;
  }

  protected async wait(
    run: (contract: CT) => Promise<ContractTransactionResponse>,
  ): Promise<{
    contract: CT;
    receipt: TransactionReceipt | undefined;
  }> {
    try {
      // fix for MetaMask bug on mobile
      if (
        BrowserInfo.isMobileDevice &&
        WalletService.loggedInWallet === "metamask"
      ) {
        new Confirm({
          icon: new MaterialIcon("warning"),
          title: "Transaction Request",
          message: "Please confirm the transaction in MetaMask",
          onConfirm: () => WalletService.openWallet(),
        });
      }

      const contract = new EthersContract(
        this.address,
        this.abi,
        await WalletService.getSigner(this.chain.id),
      ) as any;
      const tx = await run(contract);
      const checkReceipt = async () => {
        while (true) {
          const receipt = await this.provider.getTransactionReceipt(tx.hash);
          if (receipt?.blockNumber && receipt.status === 1) return receipt;
          await new Promise((resolve) => setTimeout(resolve, 3000));
        }
      };
      const receipt = await Promise.race([checkReceipt(), tx.wait()]);
      return {
        contract,
        receipt: receipt ?? undefined,
      };
    } catch (e: any) {
      console.error(e);

      // fix for MetaMask bug on mobile
      if (e.message?.includes("MetaMask is not connected/installed,")) {
        await WalletService.disconnect();
        return await this.wait(run); // retry
      } else throw e;
    }
  }
}