import { providers } from 'ethers'
import { Signer } from 'ethers'

// Contract addresses
// eslint-disable-next-line @typescript-eslint/no-var-requires
const DEPLOYED_CONTRACTS = require('@graphprotocol/contracts/addresses.json')

// Contract ABIs
import { Curation } from '@graphprotocol/contracts/dist/typechain/contracts/Curation'
import { DisputeManager } from '@graphprotocol/contracts/dist/typechain/contracts/DisputeManager'
import { EpochManager } from '@graphprotocol/contracts/dist/typechain/contracts/EpochManager'
import { Gns } from '@graphprotocol/contracts/dist/typechain/contracts/Gns'
import { RewardsManager } from '@graphprotocol/contracts/dist/typechain/contracts/RewardsManager'
import { ServiceRegistry } from '@graphprotocol/contracts/dist/typechain/contracts/ServiceRegistry'
import { Staking } from '@graphprotocol/contracts/dist/typechain/contracts/Staking'
import { GraphToken } from '@graphprotocol/contracts/dist/typechain/contracts/GraphToken'
import { GrtAssetHolder } from '@graphprotocol/contracts/dist/typechain/contracts/GrtAssetHolder'
import { AttestationApp } from '@graphprotocol/contracts/dist/typechain/contracts/AttestationApp'
import { Controller } from '@graphprotocol/contracts/dist/typechain/contracts/Controller'

// Contract factories
import { CurationFactory } from '@graphprotocol/contracts/dist/typechain/contracts/CurationFactory'
import { DisputeManagerFactory } from '@graphprotocol/contracts/dist/typechain/contracts/DisputeManagerFactory'
import { EpochManagerFactory } from '@graphprotocol/contracts/dist/typechain/contracts/EpochManagerFactory'
import { GnsFactory } from '@graphprotocol/contracts/dist/typechain/contracts/GnsFactory'
import { RewardsManagerFactory } from '@graphprotocol/contracts/dist/typechain/contracts/RewardsManagerFactory'
import { ServiceRegistryFactory } from '@graphprotocol/contracts/dist/typechain/contracts/ServiceRegistryFactory'
import { StakingFactory } from '@graphprotocol/contracts/dist/typechain/contracts/StakingFactory'
import { GraphTokenFactory } from '@graphprotocol/contracts/dist/typechain/contracts/GraphTokenFactory'
import { GrtAssetHolderFactory } from '@graphprotocol/contracts/dist/typechain/contracts/GrtAssetHolderFactory'
import { AttestationAppFactory } from '@graphprotocol/contracts/dist/typechain/contracts/AttestationAppFactory'
import { ControllerFactory } from '@graphprotocol/contracts/dist/typechain/contracts/ControllerFactory'

export interface NetworkContracts {
  assetHolder: GrtAssetHolder
  attestationApp: AttestationApp
  curation: Curation
  disputeManager: DisputeManager
  epochManager: EpochManager
  gns: Gns
  rewardsManager: RewardsManager
  serviceRegistry: ServiceRegistry
  staking: Staking
  token: GraphToken
  controller: Controller
}

export const connectContracts = async (
  providerOrSigner: providers.Provider | Signer,
  chainId: number,
): Promise<NetworkContracts> => {
  const deployedContracts = DEPLOYED_CONTRACTS[`${chainId}`]

  return {
    assetHolder: GrtAssetHolderFactory.connect(
      deployedContracts.GRTAssetHolder.address,
      providerOrSigner,
    ),
    attestationApp: AttestationAppFactory.connect(
      deployedContracts.AttestationApp.address,
      providerOrSigner,
    ),
    curation: CurationFactory.connect(
      deployedContracts.Curation.address,
      providerOrSigner,
    ),
    disputeManager: DisputeManagerFactory.connect(
      deployedContracts.DisputeManager.address,
      providerOrSigner,
    ),
    epochManager: EpochManagerFactory.connect(
      deployedContracts.EpochManager.address,
      providerOrSigner,
    ),
    gns: GnsFactory.connect(deployedContracts.GNS.address, providerOrSigner),
    rewardsManager: RewardsManagerFactory.connect(
      deployedContracts.RewardsManager.address,
      providerOrSigner,
    ),
    serviceRegistry: ServiceRegistryFactory.connect(
      deployedContracts.ServiceRegistry.address,
      providerOrSigner,
    ),
    staking: StakingFactory.connect(deployedContracts.Staking.address, providerOrSigner),
    token: GraphTokenFactory.connect(
      deployedContracts.GraphToken.address,
      providerOrSigner,
    ),
    controller: ControllerFactory.connect(
      deployedContracts.Controller.address,
      providerOrSigner,
    ),
  }
}