safesupport-bot/src/index.js

154 lines
4.6 KiB
JavaScript
Raw Normal View History

2020-02-23 00:30:27 +00:00
const path = require("path")
const fs = require("fs")
const os = require("os")
const util = require("util")
const config = require('config');
const winston = require('winston');
const uuid = require('uuidv4').uuid;
const LocalStorage = require('node-localstorage').LocalStorage;
global.Olm = require('olm');
const matrix = require('matrix-js-sdk');
const ENCRYPTION_CONFIG = { "algorithm": "m.megolm.v1.aes-sha2" };
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()
}));
}
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"
let awaitingAgreement = {}
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"))
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir);
}
const localStoragePath = path.resolve(path.join(dir, storageLoc))
localStorage = new LocalStorage(localStoragePath);
}
let deviceId = localStorage.getItem('deviceId')
const sendMessage = (client, roomId, msgText) => {
client.sendTextMessage(roomId, msgText)
.then((res) => {
logger.log('info', "Message sent")
logger.log('info', res)
})
.catch((err) => {
logger.log('error', "Error sending message");
logger.log('error', err);
switch (err["name"]) {
case "UnknownDeviceError":
Object.keys(err.devices).forEach((userId) => {
Object.keys(err.devices[userId]).map((deviceId) => {
client.setDeviceVerified(userId, deviceId, true);
});
});
sendMessage(client, roomId, msgText)
break;
default:
// logger.log('error', err);
break;
}
})
}
const notifyFacilitators = (client, roomId) => {
}
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)
})
.catch(err => {
logger.log('error', `Login error: ${err}`)
})
.then(() => client.initCrypto())
.then(() => {
// Automatically join all room invitations
client.on("RoomMember.membership", (event, member) => {
if (member.membership === "invite" && member.userId === userId) {
client.joinRoom(member.roomId)
.then(() => client.setRoomEncryption(member.roomId, ENCRYPTION_CONFIG))
.then(() => {
sendMessage(client, member.roomId, 'Do you want to continue?')
awaitingAgreement[member.roomId] = true
})
}
});
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
if (sender !== userId && awaitingAgreement[roomId]) {
if (body.toLowerCase().startsWith('yes')) {
sendMessage(client, roomId, "A facilitator will be with you soon.")
notifyFacilitators(client, roomId)
awaitingAgreement[roomId] = false
} else {
sendMessage(client, roomId, "Ok, bye")
awaitingAgreement[roomId] = false
}
}
}
});
})
.finally(() => client.startClient())