node-red-contrib-chatbot/nodes/chatbot-video.js

REDBot a Chat bot for a full featured chat bot for Telegram, Facebook Messenger and Slack. Almost no coding skills required

var _ = require('underscore');
var fs = require('fs');
var Path = require('path');
var utils = require('../lib/helpers/utils');
var sanitize = require('sanitize-filename');
var mime = require('mime');

var ValidExtensions = {
  'facebook': ['.mp4'],
  'telegram': ['.mp4'],
  'slack': ['.mp4']
};

module.exports = function(RED) {

  function ChatBotVideo(config) {
    RED.nodes.createNode(this, config);
    var node = this;
    this.filename = config.filename;
    this.name = config.name;
    this.caption = config.caption;
    this.transports = ['facebook', 'telegram', 'slack'];

    this.on('input', function(msg) {

      var name = node.name;
      var chatId = utils.getChatId(msg);
      var messageId = utils.getMessageId(msg);
      var transport = utils.getTransport(msg);
      var validExtensions = ValidExtensions[transport];
      var file = null;

      // check if valid message
      if (!utils.isValidMessage(msg, node)) {
        return;
      }
      // check transport compatibility
      if (!utils.matchTransport(node, msg)) {
        return;
      }

      var defaultFilename = null;
      if (!_.isEmpty(msg.filename)) {
        defaultFilename = msg.filename;
      } else if (!_.isEmpty(name)) {
        defaultFilename = sanitize(name);
      } else if (msg.payload != null && !_.isEmpty(msg.payload.filename)) {
        defaultFilename = msg.payload.filename;
      } else if (msg.payload != null && !_.isEmpty(msg.filename)) {
        defaultFilename = msg.filename;
      }

      var path = null;
      if (!_.isEmpty(node.filename)) {
        path = node.filename;
      } else if (!_.isEmpty(msg.filename)) {
        path = msg.filename;
      }

      if (!_.isEmpty(path)) {
        if (!fs.existsSync(path)) {
          node.error('File doesn\'t exist: ' + path);
          return;
        }
        file = {
          filename: Path.basename(path),
          extension: Path.extname(path),
          mimeType: mime.lookup(path),
          buffer: fs.readFileSync(path)
        };
      } else if (msg.payload instanceof Buffer) {
        // handle a file buffer passed through payload
        if (_.isEmpty(defaultFilename) || _.isEmpty(Path.extname(defaultFilename))) {
          node.error('Unknown file type, use the "name" parameter to specify the file name and extension as default');
          return;
        }
        file = {
          filename: Path.basename(defaultFilename),
          extension: Path.extname(defaultFilename),
          mimeType: mime.lookup(defaultFilename),
          buffer: msg.payload
        };
      } else if (_.isObject(msg.payload) && msg.payload.file instanceof Buffer) {
        // handle a buffer passed by another video node
        if (_.isEmpty(defaultFilename) || _.isEmpty(Path.extname(defaultFilename))) {
          node.error('Unknown file type, use the "name" parameter to specify the file name and extension as default');
          return;
        }
        file = {
          filename: Path.basename(defaultFilename),
          extension: Path.extname(defaultFilename),
          mimeType: mime.lookup(defaultFilename),
          buffer: msg.payload.file
        };
      } else {
        node.error('Unable to find a video in the input message.');
        return;
      }

      // get caption
      var caption = null;
      if (!_.isEmpty(node.caption)) {
        caption = node.caption;
      } else if (_.isObject(msg.payload) && _.isString(msg.payload.caption) && !_.isEmpty(msg.payload.caption)) {
        caption = msg.payload.caption;
      }

      // if the file has a not a valid extension, stop it
      if (!_(validExtensions).contains(file.extension)) {
        node.error('Unsupported file format for video node, allowed formats: ' + validExtensions.join(', '));
        return;
      }

      // send out the message
      msg.payload = {
        type: 'video',
        content: file.buffer,
        filename: file.filename,
        caption: caption,
        chatId: chatId,
        messageId: messageId,
        inbound: false,
        mimeType: file.mimeType
      };
      // send out reply
      node.send(msg);

    });
  }

  RED.nodes.registerType('chatbot-video', ChatBotVideo);
};