import { type SignerConfig } from "../helpers/config.js";
import { type ContractTransaction, isPolygon } from "../helpers/ethers.js";
import { type TableIdentifier, getContractSetup } from "./contract.js";

export interface SetParams {
  /**
   * Name or tableId and chainId of the token to be transferred.
   */
  tableName: string | TableIdentifier;
  /**
   * Address of the contract to use as a controller.
   */
  controller: string;
}

export async function setController(
  { signer }: SignerConfig,
  params: SetParams
): Promise<ContractTransaction> {
  const { contract, overrides, tableId } = await getContractSetup(
    signer,
    params.tableName
  );
  const caller = await signer.getAddress();
  const controller = params.controller;
  const chainId = await signer.getChainId();
  if (isPolygon(chainId)) {
    const gasLimit = await contract.estimateGas.setController(
      caller,
      tableId,
      controller,
      overrides
    );
    overrides.gasLimit = Math.floor(gasLimit.toNumber() * 1.2);
  }
  return await contract.setController(caller, tableId, controller, overrides);
}

export async function lockController(
  { signer }: SignerConfig,
  tableName: string | TableIdentifier
): Promise<ContractTransaction> {
  const { contract, overrides, tableId } = await getContractSetup(
    signer,
    tableName
  );
  const caller = await signer.getAddress();
  const chainId = await signer.getChainId();
  if (isPolygon(chainId)) {
    const gasLimit = await contract.estimateGas.lockController(
      caller,
      tableId,
      overrides
    );
    overrides.gasLimit = Math.floor(gasLimit.toNumber() * 1.2);
  }
  return await contract.lockController(caller, tableId, overrides);
}

export async function getController(
  { signer }: SignerConfig,
  tableName: string | TableIdentifier
): Promise<string> {
  const { contract, overrides, tableId } = await getContractSetup(
    signer,
    tableName
  );
  return await contract.getController(tableId, overrides);
}