node-red-contrib-chatbot/nodes/mc-graphql.js

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

const _ = require('lodash');
const gql = require('graphql-tag');

const Client = require('../database/client');
const lcd = require('../lib/lcd/index');
const MessageTemplate = require('../lib/message-template-async');
const {
  isValidMessage,
  extractValue
} = require('../lib/helpers/utils');

const isEmptyResponse = response => {
  const keys = Object.keys(response.data);
  return keys.length === 1 && _.isEmpty(response.data[keys[0]]);
}

const isMutation = query => query.includes('mutation(');


module.exports = function(RED) {
  const client = Client(RED);

  function MissionControlGraphQL(config) {
    RED.nodes.createNode(this, config);
    const node = this;
    this.query = config.query;
    this.debug = config.debug;
    this.name = config.name;

    this.on('input', async function(msg, send, done) {
      // send/done compatibility for node-red < 1.0
      send = send || function() { node.send.apply(node, arguments) };
      done = done || function(error) { node.error.call(node, error, msg) };

      const debug = extractValue('boolean', 'debug', node, msg, false);
      const name = extractValue('string', 'name', node, msg, false);
      // get params
      if (_.isEmpty(node.query)) {
        done('GraphQL query is empty');
        return;
      }
      let variables;
      if (_.isObject(msg.variables)) {
        variables = msg.variables;
      } else if (msg.payload != null && _.isObject(msg.payload.variables)) {
        variables = msg.payload.variables;
      }
      // translate query
      let translatedQuery = node.query;
      if (isValidMessage(msg, null, { silent: true })) {
        const template = MessageTemplate(msg, node);
        translatedQuery = await template(node.query);
      }
      if (debug) {
        // eslint-disable-next-line no-console
        console.log(
          lcd.green('GraphQL')
          + ' (id: ' + lcd.grey(this.id)
          + (!_.isEmpty(name) ? `, name: ${lcd.grey(name)}` : '')
          + ')');
        // eslint-disable-next-line no-console
        console.log(lcd.prettify(translatedQuery, { indent: 2 }))
      }

      const mutate = isMutation(translatedQuery);

      try {
        const query = gql`${translatedQuery}`;

        const response = mutate ?
          await client.mutate({ mutation: query, variables }) :
          await client.query({ query, variables, fetchPolicy: 'network-only' });
        if (!isEmptyResponse(response)) {
          send([{
            ...msg,
            payload: response.data,
            previous: msg.payload
          }, null, null]);
          done();
        } else {
          send([null, msg, null]);
          done();
        }

      } catch(error) {
        lcd.graphQLError(error, node);
        done(lcd.textGraphQLError(error));
        send([null, null, { ...msg, payload: lcd.textGraphQLError(error) }]);
      }
    });
  }

  RED.nodes.registerType('mc-graphql', MissionControlGraphQL);
};