import { BindOptions, getOptionsSignature } from './bind-options.js'
import { ContractView } from './context.js'
import { BaseProcessor } from './base-processor.js'

import { BaseContract } from 'ethers'
import { MapStateStorage } from '@sentio/runtime'
import { EthChainId } from '@sentio/chain'

export class EthProcessorState extends MapStateStorage<BaseProcessor<any, any>> {
  static INSTANCE = new EthProcessorState()
}

// from abiName_address_chainId => contract wrapper
const contracts = new Map<string, ContractView<BaseContract>>()

function getKey(abiName: string, address: string, network: EthChainId) {
  return [abiName, address.toLowerCase(), network].join('_')
}

// Dedup processor that bind multiple times
export function getProcessor(opts: BindOptions) {
  const sig = getOptionsSignature(opts)
  return EthProcessorState.INSTANCE.getValue(sig)
}

export function addProcessor(opts: BindOptions, processor: BaseProcessor<any, any>) {
  const sig = getOptionsSignature(opts)

  EthProcessorState.INSTANCE.getOrSetValue(sig, processor)
}

export function getContractByABI(abiName: string, address: string, network: EthChainId) {
  const key = getKey(abiName, address, network)
  return contracts.get(key)
}

export function addContractByABI(
  abiName: string,
  address: string,
  network: EthChainId,
  contract: ContractView<BaseContract>
) {
  const key = getKey(abiName, address, network)
  return contracts.set(key, contract)
}