bot.js

import { RtmClient, WebClient, CLIENT_EVENTS, RTM_EVENTS } from '@slack/client';
import axios from 'axios';
import express from 'express';
import { messageConfirmation, getQueryParams } from './constants';
import { User } from './models';

const router = express.Router();
const botToken = process.env.SLACK_BOT_TOKEN || '';
const rtm = new RtmClient(botToken);
const web = new WebClient(botToken);

let channel = 'T6AVBE3GX';

// The client will emit an RTM.AUTHENTICATED event on successful connection, with the `rtm.start` payload
rtm.on(CLIENT_EVENTS.RTM.AUTHENTICATED, (rtmStartData) => {
  for (const c of rtmStartData.channels) {
    if (c.is_member && c.name === 'general') { channel = c.id; }
  }
  console.log(`Logged in as ${rtmStartData.self.name} of team ${rtmStartData.team.name}, but not yet connected to a channel`);
});

// you need to wait for the client to fully connect before you can send messages
rtm.on(CLIENT_EVENTS.RTM.RTM_CONNECTION_OPENED, () => {
  // things to do when the bot connects to slack
});

const mapping = {};

rtm.on(RTM_EVENTS.MESSAGE, (msg) => {
  const dm = rtm.dataStore.getDMByUserId(msg.user);
  if (!dm || dm.id !== msg.channel || msg.type !== 'message') {
    return;
  }
  const bool = msg.text.includes('<@');
  if(bool) {
    const arr = [];
    let temp = [];
    for(let a = 0; a < msg.text.length; a++) {
      if(msg.text[a] === '@') {
        temp.push(a);
      }
      if(msg.text[a] === '>') {
        temp.push(a);
        arr.push(temp);
        temp = [];
      }
    }

    console.log(msg.text);
    for(let b = 0; b < arr.length; b++) {
      const temp = arr[b];
      console.log(temp);
    }

    for(let b = 0; b < arr.length; b++) {
      const i = msg.text.indexOf('@');
      const j = msg.text.indexOf('>');
      console.log(i, j, msg.text);
      const id = msg.text.slice(i + 1, j);
      console.log("id", id);
      const username = rtm.dataStore.getUserById(id).profile.first_name;
      const reg = /(\<.*?\>)/i;
      mapping[username] = id;
      const newMessage = msg.text.replace(reg, username);
      msg.text = newMessage;
      console.log(username, newMessage);
    }

    // const i = msg.text.indexOf('@');
    // const j = msg.text.indexOf('>');
    // const id = msg.text.slice(i + 1, j);
    // const username = rtm.dataStore.getUserById(id).profile.first_name;
    // const reg = /(\<.*?\>)/gi;
    // mapping[username] = id;
    // const newMessage = msg.text.replace(reg, username);
    // console.log(username, newMessage);
    // msg.text = newMessage;

  }
  User.findOne({slackId: msg.user})
    .then((user) => {
      if (!user) {
        return new User(
          {
            slackId: msg.user,
            slackDmId: msg.channel,
            google: {}
          }).save();
      }
      return user;
    })
    .then((user) => {
      console.log('USER is', rtm.dataStore.getUserById(user.slackId));
      if (!user.google) {
        rtm.sendMessage(`Hello this is scheduler bot. I need to schedule reminders. Please visit http://glacial-shelf-50059.herokuapp.com/connect?user=${user._id} to setup Google Calendar`, msg.channel);
      } else {
        getQuery(msg.text, msg.user)
          .then(({ data }) => {
            // if (JSON.parse(user).pending && (JSON.parse(user.pending).type === 'meeting' || JSON.parse(user.pending).type === 'reminder') ) {
            //   rtm.sendMessage("Please select a choice before moving on", msg.channel);
            //   return;
            // }
            switch (data.result.action) {
              case 'meeting.add':
                if (data.result.actionIncomplete) {
                  rtm.sendMessage(data.result.fulfillment.speech, msg.channel);
                } else {
                  console.log('Finish', data);
                  const ids = [];
                  for (const key in mapping) {
                    if (mapping.hasOwnProperty(key)) {
                      ids.push(mapping[key]);
                    }
                  }
                  // let text = data.result.fulfillment.speech;
                  // const startIndex = text.indexOf('with');
                  // const endIndex = text.indexOf('at');
                  // text = text.slice(startIndex + 5, endIndex - 1).trim();
                  // console.log(mapping);
                  web.chat.postMessage(msg.channel, data.result.fulfillment.speech, messageConfirmation(data.result.fulfillment.speech, "remember to add code to actaully cancel the meeting/not schedule one"));
                  user.pending = JSON.stringify(Object.assign({}, data.result.parameters, { type: 'meeting', ids: ids }));
                  user.save();
                }
                break;
              case 'reminder.add':
                if (data.result.actionIncomplete) {
                  rtm.sendMessage(data.result.fulfillment.speech, msg.channel);
                } else {
                  console.log('Finish', data);
                  // global_state = data.result.parameters;
                  web.chat.postMessage(msg.channel, data.result.fulfillment.speech, messageConfirmation(data.result.fulfillment.speech, "remember to add code to actaully cancel the meeting/not schedule one"));
                  // David removed the next 3 lines. I don't think that's right, but if the program is broken try removing them
                  console.log('data.result.parameters is ', data.result.parameters);
                  user.pending = JSON.stringify(Object.assign({}, data.result.parameters, { type: 'reminder' }));
                  user.save();
                  // the above 3 lines
                }
                break;
              default:
                console.log('default statement');
                console.log(data.result.action);
                if (data.result.action === 'bestbot.reply' || data.result.action.startsWith('smalltalk.')) {
                  rtm.sendMessage(data.result.fulfillment.speech, msg.channel);
                }
                return;
            }
            return;
          })
          .catch((err) => {
            console.log('error is ', err);
          });
      }
    });
});

function getQuery(msg, sessionId) {
  return axios.get('https://api.api.ai/api/query', {
    params: getQueryParams(msg, sessionId),
    headers: {
      Authorization: `Bearer ${process.env.API_AI_TOKEN}`
    }
  });
}

export { web, rtm };