Github/safesupport-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactor into a bot class

Browse Source
This commit is contained in:
Sharon Kennedy 2020-02-23 13:58:09 -05:00
parent 5570f79e69
commit febd3feb61
5 changed files with 701 additions and 9866 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.babelrc Normal file
View File

@ -0,0 +1,8 @@
{
"presets": [
"@babel/preset-env"
],
"plugins": [
"dynamic-import-node"
]
}

70
package.json
View File

@ -4,11 +4,7 @@
"description": "Chatbot to manage interactions on OCRCC client chatbots", "description": "Chatbot to manage interactions on OCRCC client chatbots",
"main": "dist/ocrcc-chatbot.js", "main": "dist/ocrcc-chatbot.js",
"scripts": { "scripts": {
"build": "NODE_ENV=production webpack-cli --mode production", "start": "nodemon --exec babel-node src/index.js"
"start": "webpack-dev-server",
"develop": "webpack --mode development --watch",
"test": "jest",
"lint": "./node_modules/.bin/eslint ."
}, },
"author": "", "author": "",
"license": "ISC", "license": "ISC",
@ -22,64 +18,10 @@
"winston": "^3.2.1" "winston": "^3.2.1"
}, },
"devDependencies": { "devDependencies": {
"@babel/core": "7.7.7", "@babel/core": "^7.8.4",
"@babel/plugin-proposal-class-properties": "7.7.4", "@babel/node": "^7.8.4",
"@babel/plugin-proposal-decorators": "7.7.4", "@babel/preset-env": "^7.8.4",
"@babel/plugin-proposal-export-namespace-from": "7.7.4", "babel-plugin-dynamic-import-node": "^2.3.0",
"@babel/plugin-proposal-function-sent": "7.7.4", "nodemon": "^2.0.2"
"@babel/plugin-proposal-json-strings": "7.7.4",
"@babel/plugin-proposal-numeric-separator": "7.7.4",
"@babel/plugin-proposal-throw-expressions": "7.7.4",
"@babel/plugin-syntax-dynamic-import": "7.7.4",
"@babel/plugin-syntax-import-meta": "7.7.4",
"@storybook/addon-actions": "5.2.8",
"@storybook/addon-links": "5.2.8",
"@storybook/addons": "5.2.8",
"@storybook/react": "5.2.8",
"babel-core": "7.0.0-bridge.0",
"babel-eslint": "10.0.3",
"babel-jest": "24.9.0",
"babel-loader": "8.0.6",
"babel-preset-airbnb": "4.4.0",
"clean-webpack-plugin": "3.0.0",
"copy-webpack-plugin": "5.1.1",
"enzyme": "3.11.0",
"enzyme-adapter-react-16": "1.15.2",
"enzyme-to-json": "3.4.3",
"eslint": "6.8.0",
"eslint-config-airbnb": "18.0.1",
"eslint-import-resolver-webpack": "0.12.0",
"eslint-loader": "3.0.3",
"eslint-plugin-import": "2.19.1",
"eslint-plugin-jsx-a11y": "6.2.3",
"eslint-plugin-react": "7.17.0",
"jest": "24.9.0",
"jest-cli": "24.9.0",
"webpack": "4.41.5",
"webpack-cli": "3.3.10",
"webpack-dev-server": "3.10.1",
"webpack-obfuscator": "0.22.0",
"webpack-serve": "3.2.0"
},
"babel": {
"presets": [
"airbnb"
],
"plugins": [
"@babel/plugin-syntax-dynamic-import",
"@babel/plugin-syntax-import-meta",
"@babel/plugin-proposal-class-properties",
"@babel/plugin-proposal-json-strings",
[
"@babel/plugin-proposal-decorators",
{
"legacy": true
}
],
"@babel/plugin-proposal-function-sent",
"@babel/plugin-proposal-export-namespace-from",
"@babel/plugin-proposal-numeric-separator",
"@babel/plugin-proposal-throw-expressions"
]
} }
} }

363
src/index.js
View File

@ -1,206 +1,205 @@
const path = require("path") import * as fs from 'fs'
const fs = require("fs") import * as os from 'os'
const os = require("os") import * as path from 'path'
const util = require("util") import * as util from 'util'
import { LocalStorage } from "node-localstorage";
import { uuid } from "uuidv4"
import config from 'config';
const config = require('config');
const winston = require('winston');
const uuid = require('uuidv4').uuid;
const LocalStorage = require('node-localstorage').LocalStorage;
global.Olm = require('olm'); global.Olm = require('olm');
const matrix = require('matrix-js-sdk');
import * as matrix from "matrix-js-sdk";
import logger from './logger'
const ENCRYPTION_CONFIG = { "algorithm": "m.megolm.v1.aes-sha2" }; const ENCRYPTION_CONFIG = { "algorithm": "m.megolm.v1.aes-sha2" };
const logger = winston.createLogger({ class OcrccBot {
level: 'info', constructor() {
format: winston.format.json(), this.awaitingAgreement = {}
defaultMeta: { service: 'user-service' }, this.awaitingFacilitator = {}
transports: [ this.client = matrix.createClient(config.get('homeserverUrl'))
// }
// - Write all logs with level `error` and below to `error.log`
// - Write all logs with level `info` and below to `combined.log`
//
new winston.transports.File({ filename: 'error.log', level: 'error' }),
new winston.transports.File({ filename: 'combined.log' })
]
});
// createLocalStorage() {
// If we're not in production then log to the `console` with the format: const storageLoc = `matrix-chatbot-${config.get('username')}`
// `${info.level}: ${info.message} JSON.stringify({ ...rest }) `
//
if (process.env.NODE_ENV !== 'production') {
logger.add(new winston.transports.Console({
format: winston.format.simple()
}));
}
const homeserverUrl = 'https://matrix.rhok.space'
const accessToken = 'MDAxOGxvY2F0aW9uIHJob2suc3BhY2UKMDAxM2lkZW50aWZpZXIga2V5CjAwMTBjaWQgZ2VuID0gMQowMDI3Y2lkIHVzZXJfaWQgPSBAaGVscC1ib3Q6cmhvay5zcGFjZQowMDE2Y2lkIHR5cGUgPSBhY2Nlc3MKMDAyMWNpZCBub25jZSA9IGZBOCsjWWQ4MF9LeTssaF8KMDAyZnNpZ25hdHVyZSA370YUvuoVD3r08AwdgGV9sE0aNWBRTrKvB1me8Bm0tQo'
const botName = 'Help Bot'
const username = 'help-bot'
const password = 'ocrccdemo'
const userId = "@help-bot:rhok.space"
const waitingRoomId = '!pYVVPyFKacZeKZbWyz:rhok.space'
const introMessage = 'This chat application does not collect any of your personal data or any data from your use of this service.'
const termsUrl = 'https://tosdr.org/'
const agreementMessage = 'Do you want to continue?'
const confirmationMessage = 'A faciltator will be with you soon.'
const exitMessage = 'That chat was not started. You can close this chatbox.'
let awaitingAgreement = {}
let awaitingFacilitator = {}
let client = matrix.createClient(homeserverUrl)
let localStorage = global.localStorage;
if (typeof localStorage === "undefined" || localStorage === null) {
const storageLoc = `matrix-chatbot-${username}`
const dir = path.resolve(path.join(os.homedir(), ".local-storage")) const dir = path.resolve(path.join(os.homedir(), ".local-storage"))
if (!fs.existsSync(dir)) { if (!fs.existsSync(dir)) {
fs.mkdirSync(dir); fs.mkdirSync(dir);
} }
const localStoragePath = path.resolve(path.join(dir, storageLoc)) const localStoragePath = path.resolve(path.join(dir, storageLoc))
localStorage = new LocalStorage(localStoragePath); return new LocalStorage(localStoragePath);
} }
let deviceId = localStorage.getItem('deviceId') sendMessage(roomId, msgText) {
return this.client.sendTextMessage(roomId, msgText)
const sendMessage = (roomId, msgText) => { .then((res) => {
return client.sendTextMessage(roomId, msgText) logger.log('info', "Message sent")
.then((res) => { logger.log('info', res)
logger.log('info', "Message sent") })
logger.log('info', res) .catch((err) => {
}) switch (err["name"]) {
.catch((err) => { case "UnknownDeviceError":
switch (err["name"]) { Object.keys(err.devices).forEach((userId) => {
case "UnknownDeviceError": Object.keys(err.devices[userId]).map((deviceId) => {
Object.keys(err.devices).forEach((userId) => { this.client.setDeviceVerified(userId, deviceId, true);
Object.keys(err.devices[userId]).map((deviceId) => { });
client.setDeviceVerified(userId, deviceId, true); });
}); return this.sendMessage(roomId, msgText)
}); break;
return sendMessage(roomId, msgText) default:
break; logger.log('error', "Error sending message");
default: logger.log('error', err);
logger.log('error', "Error sending message"); break;
logger.log('error', err);
break;
}
})
}
const inviteFacilitators = (roomId) => {
awaitingFacilitator[roomId] = true
client.getJoinedRoomMembers(waitingRoomId)
.then((members) => {
logger.log("info", "MEMBERS")
logger.log("info", members)
Object.keys(members["joined"]).forEach((member) => {
if (member !== userId)
client.invite(roomId, member)
})
})
// const notif = `There is a support seeker waiting. Go to https://riot.im/app/#/room/${roomId} to respond.`
// sendMessage(waitingRoomId, notif)
}
const kickFacilitators = (roomId) => {
awaitingFacilitator[roomId] = false
client.getJoinedRoomMembers(waitingRoomId)
.then((allFacilitators) => {
client.getJoinedRoomMembers(roomId)
.then((roomMembers) => {
const membersIds = Object.keys(roomMembers["joined"])
const facilitatorsIds = Object.keys(allFacilitators["joined"])
facilitatorsIds.forEach((f) => {
if (!membersIds.includes(f)) {
logger.log("info", "kicking out " + f + " from " + roomId)
client.kick(roomId, f, "A facilitator has already joined this chat.")
.then(() => {
logger.log("info", "Kick success")
})
.catch((err) => {
logger.log("error", err)
})
} }
}) })
})
})
}
client.login('m.login.password', {
user: username,
password: password,
initial_device_display_name: botName,
deviceId: deviceId,
})
.then((data) => {
const accessToken = data.access_token
const deviceId = data.device_id
localStorage.setItem('deviceId', data.device_id)
// create new client with full options
let opts = {
baseUrl: homeserverUrl,
accessToken: accessToken,
userId: userId,
deviceId: deviceId,
sessionStore: new matrix.WebStorageSessionStore(localStorage),
} }
client = matrix.createClient(opts) sendHtmlMessage(roomId, msgText, msgHtml) {
}) return this.client.sendHtmlMessage(roomId, msgText, msgHtml)
.catch(err => { .then((res) => {
logger.log('error', `Login error: ${err}`) logger.log('info', "Message sent")
}) logger.log('info', res)
.then(() => client.initCrypto()) })
.then(() => { .catch((err) => {
switch (err["name"]) {
// Automatically join all room invitations case "UnknownDeviceError":
client.on("RoomMember.membership", (event, member) => { Object.keys(err.devices).forEach((userId) => {
if (member.membership === "invite" && member.userId === userId) { Object.keys(err.devices[userId]).map((deviceId) => {
logger.log("info", "Auto-joining room " + member.roomId) this.client.setDeviceVerified(userId, deviceId, true);
client.joinRoom(member.roomId) });
.then(() => client.setRoomEncryption(member.roomId, ENCRYPTION_CONFIG)) });
.then(() => { return this.sendHtmlMessage(roomId, msgText, msgHtml)
if (member.roomId !== waitingRoomId) { break;
sendMessage(member.roomId, introMessage) default:
.then(() => sendMessage(member.roomId, `Please read the terms and conditions at ${termsUrl}`)) logger.log('error', "Error sending message");
.then(() => sendMessage(member.roomId, agreementMessage)) logger.log('error', err);
.then(() => awaitingAgreement[member.roomId] = true) break;
} }
}) })
} }
logger.log("info", "Membership event: " + JSON.stringify(member)) inviteFacilitators(roomId) {
logger.log("info", "Awaiting facilitator: " + awaitingFacilitator[member.roomId]) this.awaitingFacilitator[roomId] = true
this.client.getJoinedRoomMembers(config.get('waitingRoomId'))
.then((members) => {
Object.keys(members["joined"]).forEach((member) => {
if (member !== config.get('userId'))
this.client.invite(roomId, member)
})
})
// const notif = `There is a support seeker waiting. Go to https://riot.im/app/#/room/${roomId} to respond.`
// sendMessage(waitingRoomId, notif)
}
if (member.membership === 'join' && awaitingFacilitator[member.roomId]) { uninviteFacilitators(roomId) {
kickFacilitators(member.roomId) this.awaitingFacilitator[roomId] = false
} this.client.getJoinedRoomMembers(config.get('waitingRoomId'))
}); .then((allFacilitators) => {
this.client.getJoinedRoomMembers(roomId)
.then((roomMembers) => {
const membersIds = Object.keys(roomMembers["joined"])
const facilitatorsIds = Object.keys(allFacilitators["joined"])
facilitatorsIds.forEach((f) => {
if (!membersIds.includes(f)) {
logger.log("info", "kicking out " + f + " from " + roomId)
this.client.kick(roomId, f, "A facilitator has already joined this chat.")
.then(() => {
logger.log("info", "Kick success")
})
.catch((err) => {
logger.log("error", err)
})
}
})
})
})
}
client.on('Event.decrypted', (event) => { start() {
if (event.getType() === 'm.room.message') { const localStorage = this.createLocalStorage()
const roomId = event.getRoomId() let deviceId = localStorage.getItem('deviceId')
const sender = event.getSender()
const content = event.getContent()
const body = content.body
if (sender !== userId && awaitingAgreement[roomId]) { this.client.login('m.login.password', {
if (body.toLowerCase().startsWith('yes')) { user: config.get('username'),
sendMessage(roomId, confirmationMessage) password: config.get('password'),
inviteFacilitators(roomId) initial_device_display_name: config.get('botName'),
awaitingAgreement[roomId] = false deviceId: deviceId,
} else { })
sendMessage(roomId, exitMessage) .then((data) => {
awaitingAgreement[roomId] = false const accessToken = data.access_token
} const deviceId = data.device_id
localStorage.setItem('deviceId', data.device_id)
// create new client with full options
let opts = {
baseUrl: config.get('homeserverUrl'),
accessToken: accessToken,
userId: config.get('userId'),
deviceId: deviceId,
sessionStore: new matrix.WebStorageSessionStore(localStorage),
} }
}
});
})
.finally(() => client.startClient())
this.client = matrix.createClient(opts)
})
.catch(err => {
logger.log('error', `Login error: ${err}`)
})
.then(() => this.client.initCrypto())
.then(() => {
// Automatically accept all room invitations
// On joining a room, send the intro messages and wait for agreement to continue
this.client.on("RoomMember.membership", (event, member) => {
if (member.membership === "invite" && member.userId === config.get('userId')) {
logger.log("info", "Auto-joining room " + member.roomId)
this.client.joinRoom(member.roomId)
.then(() => this.client.setRoomEncryption(member.roomId, ENCRYPTION_CONFIG))
.then(() => {
if (member.roomId !== config.get('waitingRoomId')) {
this.sendMessage(member.roomId, config.get('introMessage'))
.then(() => this.sendHtmlMessage(member.roomId, `Please read the terms and conditions at ${config.get('termsUrl')}`, `Please read the full <a href="${config.get('termsUrl')}">terms and conditions</a>.`))
.then(() => this.sendMessage(member.roomId, config.get('agreementMessage')))
.then(() => this.awaitingAgreement[member.roomId] = true)
}
})
}
// When the first facilitator joins a support session, uninvite the other facilitators
if (member.membership === 'join' && this.awaitingFacilitator[member.roomId]) {
this.uninviteFacilitators(member.roomId)
}
});
// Listen for incoming messages
this.client.on('Event.decrypted', (event) => {
if (event.getType() === 'm.room.message') {
const roomId = event.getRoomId()
const sender = event.getSender()
const content = event.getContent()
const body = content.body
// Listen for the user to agree to continue, then invite facilitators to join
if (sender !== config.get('userId') && this.awaitingAgreement[roomId]) {
if (body.toLowerCase().startsWith('yes')) {
this.sendMessage(roomId, config.get('confirmationMessage'))
this.inviteFacilitators(roomId)
this.awaitingAgreement[roomId] = false
} else {
this.sendMessage(roomId, config.get('exitMessage'))
this.awaitingAgreement[roomId] = false
}
}
}
});
})
.finally(() => this.client.startClient())
}
}
const bot = new OcrccBot();
bot.start()

27
src/logger.js Normal file
View File

@ -0,0 +1,27 @@
import winston from 'winston'
const logger = winston.createLogger({
level: 'info',
format: winston.format.json(),
defaultMeta: { service: 'user-service' },
transports: [
//
// - Write all logs with level `error` and below to `error.log`
// - Write all logs with level `info` and below to `combined.log`
//
new winston.transports.File({ filename: 'error.log', level: 'error' }),
new winston.transports.File({ filename: 'combined.log' })
]
});
//
// If we're not in production then log to the `console` with the format:
// `${info.level}: ${info.message} JSON.stringify({ ...rest }) `
//
if (process.env.NODE_ENV !== 'production') {
logger.add(new winston.transports.Console({
format: winston.format.simple()
}));
}
export default logger;

10099
yarn.lock
View File

File diff suppressed because it is too large Load Diff