import { Abstract, Type } from '@nestjs/common'
import { SQSRecord } from 'aws-lambda'
import { last } from 'lodash'

import { logNotification } from '../middlewares'

export async function runEvent<TInput = any>(
  record: SQSRecord,
  controller: Type<TInput> | Abstract<TInput> | string | symbol,
): Promise<TInput> {
  logNotification(record)

  const name = record?.eventSourceARN
    ? last(record.eventSourceARN.split(':')).replace(`-${process.env.STAGE}`, '')
    : null

  if (!controller[`process${name}`]) throw new Error(`Processor for queue "${name}" not found`)

  return controller[`process${name}`](record)
}

export async function runNotificationHook<TInput = any>(
  controller: Type<TInput> | Abstract<TInput> | string | symbol,
  records: SQSRecord[],
): Promise<void> {
  await Promise.all(records.map(record => runEvent(record, controller)))
}