index.js

'use strict';

const Notifier = require('@runnerty/module-core').Notifier;
const interpreter = require('@runnerty/interpreter-core');
const nodemailer = require('nodemailer');
const aws = require('@aws-sdk/client-ses');
const { defaultProvider } = require('@aws-sdk/credential-provider-node');
const path = require('path');
const fs = require('fs');

class mailNotifier extends Notifier {
  constructor(notification) {
    super(notification);
  }

  send(notification) {
    const endOptions = {};
    notification.to = notification.to ? notification.to.toString() : '';
    notification.cc = notification.cc ? notification.cc.toString() : '';
    notification.bcc = notification.bcc ? notification.bcc.toString() : '';

    const filesReads = [];

    const templateDir = path.resolve(notification.templateDir, notification.template);
    const htmlTemplate = path.resolve(templateDir, 'html.html');
    const txtTemplate = path.resolve(templateDir, 'text.txt');

    filesReads.push(this.readFilePromise('html', htmlTemplate));
    filesReads.push(this.readFilePromise('text', txtTemplate));

    try {
      Promise.all(filesReads).then(res => {
        let html_data;
        let text_data;

        if (res[0].hasOwnProperty('html')) {
          [html_data, text_data] = [res[0].html.toString(), res[1].text.toString()];
        } else {
          [html_data, text_data] = [res[1].html.toString(), res[0].text.toString()];
        }

        const textData = [];
        textData.push(interpreter(html_data, notification));
        textData.push(interpreter(text_data, notification));

        Promise.all(textData).then(res => {
          const [html, text] = res;
          const mailOptions = {
            from: notification.from,
            to: notification.to,
            cc: notification.cc,
            bcc: notification.bcc,
            subject: notification.subject,
            text: text,
            html: html,
            attachments: notification.attachments
          };

          if (notification.disable) {
            this.logger('warn', 'Mail sender is disable.');
            endOptions.messageLog = 'Mail sender is disable.';
            this.end(endOptions);
          } else {
            // SES Transport
            if (notification.transport?.service === 'SES') {
              if (!notification.transport.region) throw new Error('Must indicate the region to use SES transport');

              const ses = new aws.SES({
                apiVersion: '2010-12-01',
                region: notification.transport.region,
                defaultProvider
              });

              const transport = nodemailer.createTransport({ SES: { ses, aws } });

              if (notification.transport.ses) {
                mailOptions.ses = Object.assign(notification.transport.ses, notification.ses);
              }
              transport.sendMail(mailOptions, err => {
                if (err) {
                  endOptions.messageLog = 'Mail sender:' + JSON.stringify(err);
                  this.end(endOptions);
                } else {
                  this.end();
                }
              });
            } else {
              // SMTP Transport
              const transport = nodemailer.createTransport(notification.transport);
              transport.sendMail(mailOptions, err => {
                if (err) {
                  endOptions.messageLog = 'Mail sender:' + JSON.stringify(err);
                  this.end(endOptions);
                } else {
                  this.end();
                }
              });
            }
          }
        });
      });
    } catch (err) {
      endOptions.end = 'error';
      endOptions.messageLog = 'Mail sender:' + JSON.stringify(err);
      this.end(endOptions);
    }
  }

  readFilePromise(type, file) {
    return new Promise((resolve, reject) => {
      fs.readFile(file, (err, data) => {
        const res = {};
        if (err) {
          res[type] = err;
          reject(res);
        } else {
          res[type] = data;
          resolve(res);
        }
      });
    });
  }
}

module.exports = mailNotifier;