import Web3 from 'web3';
import { AbiItem } from 'web3-utils';
import { ETHTxProvider } from '../eth';
import { Contract } from 'web3-eth-contract';
import { ERC20Abi, MULTISENDAbi } from './abi';
const { toBN } = Web3.utils;

export class ERC20TxProvider extends ETHTxProvider {
  getERC20Contract(tokenContractAddress: string): Contract {
    const web3 = new Web3();
    const contract = new web3.eth.Contract(ERC20Abi as AbiItem[], tokenContractAddress);
    return contract;
  }

  create(params: {
    recipients: Array<{ address: string; amount: string }>;
    nonce: number;
    gasPrice?: number;
    data: string;
    gasLimit: number;
    tokenAddress: string;
    network: string;
    chainId?: number;
    contractAddress?: string;
    maxGasFee?: number;
    priorityGasFee?: number;
  }) {
    const { tokenAddress, contractAddress } = params;
    const data = this.encodeData(params);
    const recipients = [{ address: contractAddress || tokenAddress, amount: '0' }];
    const newParams = { ...params, recipients, data };
    return super.create(newParams);
  }

  encodeData(params: {
    recipients: Array<{ address: string; amount: string }>;
    tokenAddress: string;
    contractAddress?: string;
  }) {
    const { tokenAddress, recipients, contractAddress } = params;
    if (recipients.length > 1) {
      const addresses = [];
      const amounts = [];
      for (let recipient of recipients) {
        addresses.push(recipient.address);
        amounts.push(toBN(BigInt(recipient.amount).toString()));
      }
      const multisendContract = this.getMultiSendContract(contractAddress);
      return multisendContract.methods.sendErc20(tokenAddress, addresses, amounts).encodeABI();
    } else {
      const [{ address, amount }] = params.recipients;
      const amountBN = toBN(BigInt(amount).toString());
      const data = this.getERC20Contract(tokenAddress)
        .methods.transfer(address, amountBN)
        .encodeABI();
      return data;
    }
  }
}