import { evmBasedWeb3 } from '@tatumio/shared-blockchain-evm-based'
import Web3 from 'web3'
import { EvmBasedBlockchain, httpHelper, THIRD_PARTY_API } from '@tatumio/shared-core'

export const polygonWeb3 = (args: { blockchain: EvmBasedBlockchain; client?: Web3 }) => {
  const evmBasedWeb3Result = evmBasedWeb3({
    blockchain: args.blockchain,
  })

  return {
    ...evmBasedWeb3Result,
    getClient(provider?: string, fromPrivateKey?: string): Web3 {
      const web3 = args?.client ?? evmBasedWeb3Result.getClient(provider)

      if (fromPrivateKey) {
        web3.eth.accounts.wallet.add(fromPrivateKey)
        web3.eth.defaultAccount = web3.eth.accounts.wallet[0].address
      }

      return web3
    },
    async getGasPriceInWei(gasStationApiKey?: string): Promise<string> {
      let gasStationUrl = THIRD_PARTY_API.POLYGON_GAS_STATION

      const possiblyGasStationApiKey = gasStationApiKey ?? process.env['TATUM_GAS_STATION_API_KEY']
      if (possiblyGasStationApiKey) {
        gasStationUrl = `${gasStationUrl}?apiKey=${possiblyGasStationApiKey}`
      }

      const data = (await httpHelper.get(gasStationUrl)).data
      const gasPrice = Math.max(30, Math.ceil(data['fast'] ?? 20))

      return Web3.utils.toWei(gasPrice.toString(), 'gwei')
    },
  }
}