/* eslint-disable */

/**
 * Reference: https://github.com/americanexpress/jest-image-snapshot/blob/main/examples/image-reporter.js
 * 
 * To enable this image reporter, add it to your `jest.config.js` "reporters" definition:
 * "reporters": [ "default", "<rootDir>/image-reporter.js" ]
 * 
 * Image Reporter may not work with jest's --forceExit flag
 */

const fs = require('fs');
const AWS = require('aws-sdk/global');
const S3 = require('aws-sdk/clients/s3'); // this is needed
require('dotenv').config();

const UPLOAD_BUCKET = process.env.S3_BUCKET_NAME;

const s3 = new AWS.S3({ apiVersion: '2006-03-01' });

class ImageReporter {
  constructor(globalConfig, options) {
    this._globalConfig = globalConfig;
    this._options = options;
  }

  onTestResult(test, testResult, aggregateResults) {
    if (
      testResult.numFailingTests &&
      testResult.failureMessage.match(/different from snapshot/)
    ) {
      if (
        !process.env.S3_BUCKET_NAME ||
        !process.env.AWS_SECRET_ACCESS_KEY ||
        !process.env.AWS_ACCESS_KEY_ID
      ) {
        console.log(
          'Missing env variables. Skipping upload of image diff files.'
        );
        return;
      }

      const targetDirectories = [
        './test/__image_snapshots__/',
        './test/__image_snapshots__/__diff_output__/',
      ];
      targetDirectories.forEach(targetDirectory => {
        fs.readdirSync(targetDirectory, { withFileTypes: true }).forEach(
          dirent => {
            if (!dirent.isFile()) return;
            const path = `images/${dirent.name}`;
            const params = {
              Body: fs.readFileSync(`${targetDirectory}/${dirent.name}`),
              Bucket: UPLOAD_BUCKET,
              Key: path,
              ContentType: 'image/png',
            };
            s3.putObject(params, err => {
              if (err) {
                console.log(err, err.stack);
              } else {
                console.log(
                  `Uploaded file to https://${UPLOAD_BUCKET}.s3.amazonaws.com/${path}`
                );
              }
            });
          }
        );
      });
    }
  }
}

module.exports = ImageReporter;