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;