@maildots/sdk/lib/SendingClient.ts

import * as Bluebird from 'bluebird';
import * as dotenv from'dotenv';

import LinkboxService from './core/LinkboxService';
import AttachmentService from './core/AttachmentService'
import LogService from './core/LogService'
import AWSService from './core/AWSService'
import Message from '../model/Message';
import Attachment from '../model/Attachment'

dotenv.config();

class SendingClient {
  private LOGS = process.env.LOGS;

  private BOT_ADDRESS = process.env.BOT_ADDRESS;
  private BOT_NAME = process.env.BOT_NAME;
  private ACCESS_TOKEN = process.env.ACCOUNT_TOKEN;

  private linkboxService: LinkboxService;
  private awsService: AWSService;
  private logService: LogService;

  constructor () {
    this.linkboxService = new LinkboxService();
    this.awsService = new AWSService();
    if (this.LOGS && this.LOGS === 'true') {
      this.logService = LogService.getInstance();
    }
  }

  async sendMessage (message: Message) {
    message
    .buildCore(this.ACCESS_TOKEN, this.BOT_ADDRESS, this.BOT_NAME)
    .setType('message');
    
    let msgData = await this.send(message);
    return msgData;
  }

  async sendInlineMessage (refMessage: Message, message: Message) {
    message
    .buildCore(this.ACCESS_TOKEN, this.BOT_ADDRESS, this.BOT_NAME)
    .setType('inline')
    .setInline(true)
    .setSubject(this.setSubject(refMessage.Subject, message.Subject))
    .setReferences([refMessage.MessageId].concat(refMessage.References))
    .setReceivers([refMessage.From])
    .setSentDate(refMessage.SentDate + 1)
    .setMessageId(message.getMessageId(refMessage.SentDate + 1, this.BOT_ADDRESS));

    let msgData = await this.send(message);
    return msgData;
  }

  async sendReply (refMessage: Message, message: Message) {
    message
    .buildCore(this.ACCESS_TOKEN, this.BOT_ADDRESS, this.BOT_NAME)
    .setType('message')
    .setSubject(this.setSubject(refMessage.Subject, message.Subject))
    .setReferences([refMessage.MessageId].concat(refMessage.References))
    .setReceivers([refMessage.From])
    .setSentDate(refMessage.SentDate + 1)
    .setMessageId(message.getMessageId(refMessage.SentDate + 1, this.BOT_ADDRESS))

    let msgData = await this.send(message);
    return msgData;
  }

  private setSubject (refMessageSubject: string, messageSubject: string): string {
    let subject = (messageSubject !== '' && messageSubject) ? messageSubject : refMessageSubject;
    if (!subject.toUpperCase().startsWith("RE:")) {
      subject = "Re: " + subject;
    }
    return subject;
  }

  private async send (message: Message) {
    let _ = await this.uploadAttachments(message.getAttachments());
    let jsonMessage = message.toJSON();
    let response = await this.linkboxService.postMessage(jsonMessage);
    if (this.LOGS && this.LOGS === 'true') {
      let bd = await this.logService.saveMessage(jsonMessage, this.BOT_ADDRESS);
    }
    return jsonMessage;
  }

  private async uploadAttachments (attachments: Attachment[]) {
    let keys = await Bluebird.map(attachments, async (attachment: Attachment) => {
      let attachmentService = new AttachmentService(attachment.getSrc());
      let attachmentData = await attachmentService.download();
      let uploadData = await this.awsService.upload(attachmentData.buffer, attachment.getKey());
      attachment.setSize(attachmentData.size);
      let response = await this.linkboxService.postAttachment({
        access_token: this.ACCESS_TOKEN,
        attachment_key: attachment.getKey(),
        attachment_name: attachment.getName(),
        size: attachment.getSize()
      });
      return response;
    })
    return keys;
  }
}

export default SendingClient;