import WorkerClient from './worker-client';
import * as Sentry from '@sentry/node';
import Web3SocketService from './web3-socket';
import Contracts from './contracts';
import { AddressKeys } from '@cardstack/cardpay-sdk';
import LatestEventBlockQueries from './queries/latest-event-block';
import { inject } from '@cardstack/di';

import logger from '@cardstack/logger';
const log = logger('hub/contract-subscription-event-handler');

const CONTRACT_EVENTS = [
  {
    abiName: 'pay-merchant-handler',
    contractName: 'payMerchantHandler' as AddressKeys,
    eventName: 'CustomerPayment',
    taskName: 'notify-customer-payment',
  },
  {
    abiName: 'revenue-pool',
    contractName: 'revenuePool' as AddressKeys,
    eventName: 'MerchantClaim',
    taskName: 'notify-merchant-claim',
  },
];

export class ContractSubscriptionEventHandler {
  contracts: Contracts = inject('contracts', { as: 'contracts' });
  web3: Web3SocketService = inject('web3-socket', { as: 'web3' });
  workerClient: WorkerClient = inject('worker-client', { as: 'workerClient' });
  latestEventBlockQueries: LatestEventBlockQueries = inject('latest-event-block-queries', {
    as: 'latestEventBlockQueries',
  });

  async setupContractEventSubscriptions() {
    let web3Instance = this.web3.getInstance();

    let subscriptionOptions = {};
    let latestBlock = await this.latestEventBlockQueries.read();

    if (latestBlock) {
      subscriptionOptions = { fromBlock: latestBlock };
    }

    for (let contractEvent of CONTRACT_EVENTS) {
      let contract = await this.contracts.getContract(web3Instance, contractEvent.abiName, contractEvent.contractName);

      contract.events[contractEvent.eventName](subscriptionOptions, async (error: Error, event: any) => {
        if (error) {
          Sentry.captureException(error, {
            tags: {
              action: 'contract-subscription-event-handler',
            },
          });
          log.error(`Error in ${contractEvent.contractName} subscription`, error);
        } else {
          log.info(
            `Received ${contractEvent.contractName} event (block number ${event.blockNumber})`,
            event.transactionHash
          );

          await this.latestEventBlockQueries.update(event.blockNumber);
          this.workerClient.addJob(contractEvent.taskName, event.transactionHash);
        }
      });
    }

    log.info('Subscribed to events');
  }
}

declare module '@cardstack/di' {
  interface KnownServices {
    'contract-subscription-event-handler': ContractSubscriptionEventHandler;
  }
}