import { isLambdaHook } from '@seek/aws-codedeploy-hooks';
import type { SQSEvent } from 'aws-lambda';

import {
  createBatchSQSHandler,
  createHandler,
} from '#src/framework/handler.js';
import { logger } from '#src/framework/logging.js';
import { metricsClient } from '#src/framework/metrics.js';
import { validateJson } from '#src/framework/validation.js';
import {
  scoreJobPublishedEvent,
  scoringService,
} from '#src/services/jobScorer.js';
import { sendPipelineEvent } from '#src/services/pipelineEventSender.js';
import { JobPublishedEventSchema } from '#src/types/pipelineEvents.js';

/**
 * Tests connectivity to ensure appropriate access and network configuration.
 */
const smokeTest = async () => {
  await Promise.all([scoringService.smokeTest(), sendPipelineEvent({}, true)]);
};

export const handler = createHandler<SQSEvent>(async (event, ctx) => {
  // Treat an empty object as our smoke test event.
  if (!Object.entries(event).length) {
    if (process.env.SKIP_SMOKE && isLambdaHook(event, ctx)) {
      // Expedite deployment even if dependencies are unhealthy.
      return;
    }

    // Run dependency checks otherwise.
    logger.debug('Smoke test event received');
    return smokeTest();
  }

  const count = event.Records.length;

  if (!count) {
    throw Error('Received 0 records');
  }
  logger.debug({ count }, 'Received jobs');

  metricsClient.distribution('job.received', count);

  return recordHandler(event, ctx);
});

const recordHandler = createBatchSQSHandler(async (record, _ctx) => {
  const { body } = record;

  // TODO: this throws an error, which will cause the Lambda function to retry
  // the event and eventually send it to your dead-letter queue. If you don't
  // trust your source to provide consistently well-formed input, consider
  // catching and handling this error in code.
  const publishedJob = validateJson(body, JobPublishedEventSchema);

  const scoredJob = await scoreJobPublishedEvent(publishedJob);

  const snsMessageId = await sendPipelineEvent(scoredJob);

  logger.debug({ snsMessageId }, 'Scored job');

  metricsClient.distribution('job.scored', 1);
});