import { OpsEntity, SSM } from '@aws-sdk/client-ssm';

const ssm = new SSM();

async function getOpsItem(title: string): Promise<OpsEntity | undefined> {
  // get opsItem n times with 5s interval
  const maxRetries = 30;
  const delaySeconds = 5;
  for (let i = 1; i <= maxRetries; i++) {
    log({
      message: `Getting opsItem with title ${title} at try ${i}/${maxRetries} with delay ${delaySeconds}s`,
      title: title,
    });

    const res = await ssm.getOpsSummary({
      Filters: [
        {
          Key: 'AWS:OpsItem.Title',
          Values: [title],
          Type: 'Equal',
        },
        {
          Key: 'AWS:OpsItem.Status',
          Values: ['Open'],
          Type: 'Equal',
        },
      ],
    });
    if (res.$metadata.httpStatusCode !== 200) {
      log({
        message: 'Error getOpsSummary',
        title: title,
        err: `httpStatusCode: ${res.$metadata.httpStatusCode}`,
      });

      return undefined;
    }

    if (res.Entities === undefined || res.Entities.length === 0) {
      log({
        message: `No opsItem entity for title. Next try in ${delaySeconds}s`,
        title: title,
        res: res,
      });

      // delay 5s and try again
      await new Promise((resolve) => setTimeout(resolve, delaySeconds * 1000));
      continue;
    }


    if (res.Entities.length > 1) {
      log({
        message: `Too many opsItems: ${res.Entities!.length}`,
        title: title,
        res: res,
      });
      return undefined;
    }

    // 1 ops item found
    return res.Entities[0];
  }

  log({
    message: 'No opsItem entities at all',
    title: title,
  });
  return undefined;
}

export const handler = async (event: any) => {
  const title = event.title || 'test';
  const source = event.source || 'source';
  const description = event.description || 'description';

  log({
    message: 'Closing opsItem',
    event: event,
    title: title,
    source: source,
    description: description,
  });

  try {
    // 1 get opsItem
    const opsEntity = await getOpsItem(title);
    if (opsEntity === undefined) {
      log({
        message: 'opsItem undefined',
      });
      return 'NOK';
    }
    log({
      message: 'Got opsItem',
      title: title,
    });


    const opsItemId = opsEntity!.Id!;
    const opsItemContent = opsEntity!.Data!['AWS:OpsItem'].Content![0];
    const opsItemTitle = opsItemContent.Title;
    const opsItemSource = opsItemContent.Source;
    const opsItemDescription = opsItemContent.Description;

    // as truncate the title and source to 1020 and 60 characters
    if (
      opsItemTitle !== title.substring(0, 1020) + (title.length > 1020 ? ' ...' : '') ||
      opsItemSource !== source.substring(0, 60) + (source.length > 60 ? ' ...' : ''),
      opsItemDescription !== description.substring(0, 1020) + (description.length > 1020 ? ' ...' : '')
    ) {
      log({
        message: 'OpsItem did not match',
        expected: `title: '${title}', source: '${source}', description: '${description}'`,
        got: `title: '${opsItemTitle}', source: '${opsItemSource}', description: '${opsItemDescription}'`,
      });
      return 'NOK';
    }

    // 2 close opsItem
    const resUpdate = await ssm.updateOpsItem({
      OpsItemId: opsItemId,
      Status: 'Resolved',
    });

    if (resUpdate.$metadata.httpStatusCode !== 200) {
      log({
        message: 'Error updateOpsItem',
        title: title,
        err: `httpStatusCode: ${resUpdate.$metadata.httpStatusCode}`,
      });

      return 'NOK';
    }

    log({
      message: 'Updated opsItem',
      title: title,
      id: opsItemId,
      res: resUpdate,
    });

    return 'OK';
  } catch (err) {
    log({
      message: 'Error (catch) getting and closing opsItem',
      err: err,
    });
    return 'NOK';
  }
};

function log(msg: any) {
  console.log(JSON.stringify(msg));
}