diff --git a/README.md b/README.md index 0e8d8e1..640eab4 100644 --- a/README.md +++ b/README.md @@ -9,7 +9,7 @@

  - + --- @@ -45,38 +45,125 @@ Dev | 6612/tcp fervent_carson +``` -3. **[Download](https://github.com/zekroTJA/yuri/archive/master.zip)** this repository or clone it simply with
-`$ git clone https://github.com/zekroTJA/yuri.git` +To stop the container, just use the stop command with the container ID of this output: +``` +$ sudo docker stop 1bdf62abc781 +``` +### 2 - Manual configuration of your node.js setup -4. Install all **npm dependencies** with the command
-`$ npm i` +> Why you need to do all this stuff: Simply, the sound library of yuri has some problems with later versions of node.js then 9.4.0. So, you need to install all dependencies on this version and run the bot with it. +1. Clone the repository +``` +$ git clone https://github.com/zekroTJA/yuri.git && cd yuri +``` -5. Open the **config.json** file and enter there your discord **bot app token**, your discord **accound id**, the **prefix** you want to control the bot with and the **sound files location** *(absolute or relative)*, where the sound files are the bot should have access to. +2. You need to install **ffmpeg** on your system. To do this, just follow the steps described in [**this blog post**](https://superuser.com/questions/286675/how-to-install-ffmpeg-on-debian#865744). -6. Now, you can start the bot with
-`$ npm start`
-or if you want that the bot should run 24/7, then use [**screen**](https://linux.die.net/man/1/screen)
-`$ screen -L -S soundboard npm start` +3. Now, you need to install npm package `n`, which will be used to manage multiple node.js versions +``` +$ sudo npm i -g n +``` +4. Install node 9.4.0 and check, that the version has changed +``` +$ sudo n 9.4.0 +$ node -v +v9.4.0 +``` -7. Then, take a look in the console and copy the **invite link** and open it in the browser to invite the soundboard bot to your server.
-**This is only possible if you have admin permission on the guild!** +5. Now, install all node modules +``` +$ npm i +``` +6. Then, open the `package.json` file and change the `"start"` command to `"n use 9.4.0 src/main.js"` +```js +"scripts": { + "start": "n use 9.4.0 src/main.js" +}, +``` + +7. After that, start the bot once to create the `config.json` located in the `expose` folder. +``` +$ npm start +``` + +8. Open the config file, enter your Bot application token got from [**here**](http://discordapp.com/developers/applications/me), your prefered prefix the bot will listen to, your ID to identify you as the owner of the bot and the websocket password/token. Also, set the sound files directory where you want to store your sound files. Attention: This directory will **not** be automatically created from the bot, so you will need to create the directory manually! + +9. Finally, you need to start the bot with the `runner.sh` script to ensure that the bot restarts after using the restart function. In best case, you should run all of this in a screen session +``` +$ screen -dmLS yuri bash runner.sh +``` +*If you do not want to log the output of the screen, just do not use the `L` argument.* + +10. At last, you should change back to your latest version of node +``` +$ sudo n latest +``` +*...or enter the version you prefer to use normally.* -8. Place some sound files in the defined location and enter `.help` in discord to see all commands of the soundboard bot! ;) --- ## Web Interface @@ -99,6 +186,12 @@ After login, you can play sounds by clicking the tiles, search for sounds or man --- +## To Do + +[**Here**](https://github.com/zekroTJA/yuri/projects/1) you can find current tasks to do and ideas to implement. + +--- + ## Used 3rd party dependencies - [body-parser](https://www.npmjs.com/package/body-parser) @@ -108,3 +201,8 @@ After login, you can play sounds by clicking the tiles, search for sounds or man - [express-handlebars](https://www.npmjs.com/package/express-handlebars) - [node-opus](https://www.npmjs.com/package/node-opus) - [socket.io](https://www.npmjs.com/package/socket.io) + +--- + +© 2018 zekro Development (Ringo Hoffmann) +[zekro.de](https://zekro.de) | contact[at]zekro.de diff --git a/src/core/websocket.js b/src/core/websocket.js index 3e95525..cfe0c8b 100644 --- a/src/core/websocket.js +++ b/src/core/websocket.js @@ -1,650 +1,650 @@ -const Main = require('../main') -const Logger = require('../util/logger') -const { players, guildLog, Player } = require('../core/player') -const { info, error } = require('../util/msgs') -const path = require('path') -const EventEmitter = require('events'); -const DicordOAuth = require('../util/discordOAuth') - -const express = require('express') -const hbs = require('express-handlebars') -const socketio = require('socket.io') -const http = require('http') - - -const WEBINTERFACE_VERSION = '1.10.0' -const SESSION_TIMEOUT = 1800 * 1000 // 30 Minutes -const LOGIN_TIMEOUT = 30 * 1000 // 30 Seconds -const EXPOSE_PORT = process.argv.includes("--1337") ? 1337 : 6612 - -const STATUS = { - ERROR: 'ERROR', - OK: 'OK' -} - -const ERRCODE = { - OK: 0, - INVALID_TOKEN: 1, - INVALID_GUILD: 2, - INVALID_FILE: 3, - PLAYER_ERROR: 4, - INVALID_LOGIN: 5, - SESSION_NOT_LOGGED_IN: 6, - NO_VC: 7, - LOGIN_TIMED_OUT: 8 -} - -class Session { - constructor(userid, token) { - return new Promise((res, reject) => { - this.userid = userid - this.token = token - Main.client.fetchUser(userid, true) - .then(user => { - this.user = user - Main.client.guilds.forEach(g => { - let memb = g.members.find(m => m.id == this.user.id && m.voiceChannel) - if (memb) { - this.guild = g - this.player = players[g.id] - this.member = memb - } - }) - if (this.member) { - res(this) - this.timer = new SessionTimer(SESSION_TIMEOUT) - } - else - reject({message: 'User not found in any voice channel!'}) - }) - .catch(reject) - }) - } - - get vc() { - return this.member.voiceChannel - } - -} - - -class SessionTimer extends EventEmitter { - constructor(timeout) { - super() - this.timeout = timeout - this.create() - } - - create() { - this.timer = setTimeout(() => { - this.emit('elapsed') - }, this.timeout) - } - - refresh() { - clearTimeout(this.timer) - this.create() - } -} - - -function removeFromArrayIfExists(array, element) { - let i = array.indexOf(element) - if (i > -1) - array.splice(i, 1) -} - -class Websocket { - - constructor() { - this.sessions = {} - this.ipregister = {} - this.authorizedids = [] - this.oauth = new DicordOAuth({ - clientid: Main.config.client.id, - clientsecret: Main.config.client.secret, - serveraddr: Main.config.serveraddr - }) - this.app = express() - this.app.engine('hbs', hbs({ - extname: 'hbs', - defaultLayout: 'layout', - layoutsDir: __dirname + '/webinterface/layouts' - })) - this.app.set('views', path.join(__dirname, 'webinterface')) - this.app.set('view engine', 'hbs') - this.app.use(express.static(path.join(__dirname, 'webinterface'))) - this.server = new http.Server(this.app) - this.io = socketio(this.server) - this.token = Main.config.wstoken - - if (!this.token || this.token == "") { - Logger.error('Can not set up Websocket API! Missing token in config!') - return - } - - /////////////////////// - //// WEB INTERFACE //// - /////////////////////// - - // WEBINTERFACE - this.app.get('/', (req, res) => { - var user = req.query.user ? req.query.user : this.ipregister[req.connection.remoteAddress] - if (!user) { - this.oauth.redirectToAuth('authorize', res) - return - } - var session = this.sessions[user] - var sortbydate = (req.query.sortbydate == 1) - if (session && session.vc) { - var fileList = Player.getFilelist(sortbydate) - .map(f => f.split('.')[0]) - - res.render('index', { - fileList, - user, - usertag: session.user.tag, - voicechannel: { - id: session.vc.id, - name: session.vc.name - }, - guild: { - id: session.guild.id, - name: session.guild.name - }, - sortbydate, - token: session.token, - inchannel: (session.guild.me.voiceChannel != null), - WEBINTERFACE_VERSION - }) - } else { - this.oauth.redirectToAuth('authorize', res) - } - }) - - // WEBINTERFACE LOGIN - this.app.get('/wilogin', (req, res) => { - var token = req.query.token - var user = req.query.user - var passedByToken = req.query.passedByToken - - if (!user || !token || user == "" || token == "") { - res.render('error', { - code: ERRCODE.INVALID_LOGIN, - reason: 'Invalid login credentials.' - }) - return - } - - if (!this.authorizedids.includes(user)) { - res.render('error', { - code: ERRCODE.LOGIN_TIMED_OUT, - reason: 'Login timed out.' - }) - return - } - - if (!this._checkToken(token)) { - if (passedByToken == '1') { - res.render('resetTokenCookie', { - code: ERRCODE.INVALID_TOKEN, - reason: 'Invalid token.' - }) - return - } - res.render('error', { - code: ERRCODE.INVALID_TOKEN, - reason: 'Invalid token.' - }) - return - } - - new Session(user, token) - .then(session => { - session.timer.on('elapsed', () => { - Logger.info(`[WS Session Expired] CID: ${user} | TAG: ${session.user.tag}`) - let socket = session.socket - if (socket) - socket.emit('logout') - this.sessions[user] = null - }) - this.sessions[user] = session - this.ipregister[req.connection.remoteAddress] = user - Logger.info(`[WS Login (WEB)] CID: ${user} | TAG: ${session.user.tag}`) - res.redirect('/?user=' + user) - }) - .catch(e => { - res.render('error', { - code: ERRCODE.INVALID_LOGIN, - reason: e.message - }) - }) - }) - - // WEBINTERFACE LOGOUT - this.app.get('/wilogout', (req, res) => { - var user = req.query.user - - if (!this._checkToken(req.query.token)) { - res.send("") - return - } - - Logger.info(`[WS Logout (WEB)] CID: ${user} | TAG: ${this.sessions[user].user.tag}`) - this.sessions[user] = null - this.ipregister[req.connection.remoteAddress] = null - res.redirect('/') - }) - - // WEBINTERFACE RESTART BOT - this.app.get('/wirestart', (req, res) => { - var token = req.query.token - var user = req.query.user - if (!this._checkToken(token)) { - res.send("") - return - } - - var session = this.sessions[user] - - new Promise((res, rej) => { - if (session && session.vc) - session.vc.leave() - .then(res()).catch(rej()) - else - res() - }).then(() => { - process.exit(); - }).catch(() => { - process.exit(); - }) - - - }) - - // WEBINTERFACE SHOW LOG - this.app.get('/wishowlog', (req, res) => { - var guild = req.query.guild - var user = req.query.user - - if (!this._checkToken(req.query.token)) { - res.send("") - return - } - - var log = guildLog[guild] - if (log == null) { - res.render('error', { - code: 7, - reason: 'Guild not found in players log list.' - }) - return - } - - var session = this.sessions[user] - if (!session) { - res.render('error', { - code: ERRCODE.INVALID_LOGIN, - reason: 'Invalid session.' - }) - return - } - - session.timer.refresh() - - log = log - .map(s => s.replace(/[`\*]/gm, '')) - - res.render('log', { - guild: { - id: session.guild.id, - name: session.guild.name - }, - log - }) - }) - - // LOGG IN AND CREATE SESSION FOR USER ID - this.app.get('/login', (req, res) => { - res.set('Content-Type', 'application/json') - - var token = req.query.token - var userID = req.query.user - - if (!this._checkToken(token)) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - return - } - - new Session(userID, token) - .then(session => { - this._sendStatus(res, STATUS.OK, ERRCODE.OK) - session.timer.on('elapsed', () => this.sessions[user] = null) - this.sessions[userID] = session - this.ipregister[req.connection.remoteAddress] = userID - Logger.info(`[WS Login] CID: ${userID} | TAG: ${session.user.tag}`) - }) - .catch(e => { - console.log(e) - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_LOGIN, e.message) - }) - }) - - // LOGGOUT - this.app.get('/logout', (req, res) => { - res.set('Content-Type', 'application/json') - - var token = req.query.token - var userID = req.query.user - - if (!this._checkToken(token)) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - return - } - - var session = this.sessions[userID] - if (session) { - this.sessions[userID] = null - this.ipregister[req.connection.remoteAddress] = null - this._sendStatus(res, STATUS.OK, ERRCODE.OK) - Logger.info(`[WS Logout] CID: ${userID} | TAG: ${session.user.tag}`) - } - else - this._sendStatus(res, STATUS.ERROR, ERRCODE.SESSION_NOT_LOGGED_IN) - }) - - // PLAY SOUND METHOD - this.app.get('/play', (req, res) => { - res.set('Content-Type', 'application/json') - - var token = req.query.token - var userID = req.query.user - var soundFile = req.query.file - - if (!this._checkToken(token)) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - return - } - - var session = this.sessions[userID] - - if (!session) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.SESSION_NOT_LOGGED_IN) - return - } - - if (!soundFile || soundFile == "") { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_GUILD) - return - } - - if (!session.vc) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.NO_VC) - return - } - - session.timer.refresh() - - new Promise(res => { - // var player = players[session.guild.id] - var player = session.player - if (player) - res(player) - else - new Player(session.vc) - .then(p => res(p)) - }).then(player => { - session.player = player - player.play(soundFile, session.member).then(() => { - this._sendStatus(res, STATUS.OK, 0) - }).catch(e => { - this._sendStatus(res, STATUS.ERROR, ERRCODE.PLAYER_ERROR, e.message) - }) - }).catch(e => { - this._sendStatus(res, STATUS.ERROR, ERRCODE.PLAYER_ERROR, e.message) - }) - }) - - // GET SOUND FILES - this.app.get('/sounds', (req, res) => { - res.set('Content-Type', 'application/json') - var token = req.query.token - - if (!this._checkToken(token)) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - return - } - - var fileList = Player.getFilelist() - .map(f => f.split('.')[0]) - - res.send(JSON.stringify({ - status: STATUS.OK, - code: 0, - desc: { - n: fileList.length, - sounds: fileList - } - }, 0, 2)) - }) - - // SET GUILDS AND IDS - this.app.get('/guilds', (req, res) => { - res.set('Content-Type', 'application/json') - var token = req.query.token - - if (!this._checkToken(token)) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - return - } - - var servers = Main.client.guilds - .map(g => [g.name, g.id]) - - res.send(JSON.stringify({ - status: STATUS.OK, - code: 0, - desc: { - n: servers.length, - servers: servers - } - }, 0, 2)) - }) - - // CHECK TOKEN - this.app.get('/token', (req, res) => { - res.set('Content-Type', 'application/json') - var token = req.query.token - - if (!this._checkToken(token)) - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) - else - this._sendStatus(res, STATUS.OK, 0) - - }) - - this.app.get('/authorize', (req, res) => { - let code = req.query.code; - let resetToken = req.query.resetToken - - if (resetToken) { - let user = req.query.user - res.render('login', { user, resetToken }) - } - - this.oauth.getId(req, (err, user) => { - if (err) { - this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_LOGIN) - return - } - var sortbydate = (req.query.sortbydate == 1) - var session = this.sessions[user] - this.authorizedids.push(user) - setTimeout(() => { - removeFromArrayIfExists(this.authorizedids, user) - }, LOGIN_TIMEOUT) - if (!session) { - res.render('login', { user }) - return - } - if (!session.vc) { - this.sessions[user] = null - res.render('login', { user }) - return - } - res.redirect('/?user=' + user) - }) - }) - - this.server.listen(EXPOSE_PORT, () => { - Logger.info('WS API set up at port ' + this.server.address().port) - }) - - - this.io.on('connection', (socket) => { - var session - Logger.info('WS onnection etablished: ' + socket.id) - - socket.on('thatsMe', (data) => { - session = this.sessions[data.user] - if (!session) { - socket.emit('logout') - socket.disconnect() - return; - } - let user = session.member.user - socket.join(session.guild.id) - session.socket = socket - let guildsessions = Object.keys(this.sessions) - .filter(k => this.sessions[k] && this.sessions[k].guild && this.sessions[k].guild.id == session.guild.id && this.sessions[k].socket) - .map(k => this.sessions[k].member.user) - let alreadyConnected = [] - guildsessions.forEach(u => alreadyConnected.push({ id: u.id, tag: u.tag, avatarURL: u.avatarURL })) - this.io.to(session.guild.id).emit('userConnected', alreadyConnected) - }) - - socket.on('disconnecting', (socket) => { - Logger.info("WS onnection closed") - if (session) { - let user = session.member.user - this.io.to(session.guild.id).emit('userDisconnected', { id: user.id, tag: user.tag }) - session.socket = null - } - }) - - socket.emit('whoAreYou') - - socket.on('changeChannelStatus', () => { - if (session && session.vc) { - session.timer.refresh() - var currchan = session.guild.me.voiceChannel - if (currchan) { - if (session.player) - session.player.destroy() - else - currchan.leave() - } - else { - session.vc.join().catch() - } - } - }) - - socket.on('stopSound', () => { - if (session && session.player) { - session.player.stop() - session.timer.refresh() - } - }) - - // PLAY EVENT - socket.on('playSound', (data) => { - var user = data.user - var soundFile = data.sound - - if (!user || !session) - return - - session.timer.refresh() - - new Promise((res, rej) => { - var player = players[session.guild.id] - if (player) - res(player) - else - new Player(session.vc) - .then(p => res(p)) - }).then(player => { - session.player = player - let sockets = this.io.to(session.guild.id) //.of(session.guild.id) - player.play(soundFile, session.member).then(() => { - sockets.emit('soundPlaying', { user, sound: soundFile }) - player.dispatcher.on('end', () => { - sockets.emit('soundStopped', { user, sound: soundFile }) - }) - }).catch(e => { - sockets.emit('playError', { user, sound: soundFile, error: e }) - console.log(e) - }) - }).catch(e => { - sockets.emit('playError', { user, sound: soundFile, error: e }) - console.log(e) - }) - }) - - Main.client.on('voiceStateUpdate', (oldMemb, newMemb) => { - let session = this.sessions[oldMemb.id] - if (session && session.socket && oldMemb.voiceChannel && !newMemb.voiceChannel) { - session.socket.emit('logout') - } - if (oldMemb.id != Main.client.user.id) - return - if (!oldMemb.voiceChannel && newMemb.voiceChannel) { - let vc = newMemb.voiceChannel - this.io.emit('channelJoined', vc.id) - } - else if (oldMemb.voiceChannel && !newMemb.voiceChannel) { - let vc = oldMemb.voiceChannel - this.io.emit('channelLeft', vc.id) - } - }) - - Main.eventEmiter.on('closing', () => { - this.io.emit('logout') - }) - }) - } - - _sendStatus(res, status, code, msg) { - let desc = (() => { - switch (code) { - case ERRCODE.INVALID_TOKEN: - return 'Invalid token.' - case ERRCODE.INVALID_GUILD: - return 'Invalid guild ID or no player active on this guild currently.' - case ERRCODE.INVALID_FILE: - return 'File name not set.' - case ERRCODE.PLAYER_ERROR: - return msg ? msg : 'Unknown player error.' - case ERRCODE.INVALID_LOGIN: - return msg ? msg : 'Invalid login.' - case ERRCODE.SESSION_NOT_LOGGED_IN: - return 'Session not logged in.' - case ERRCODE.NO_VC: - return 'USer not in voice channel.' - default: - return msg ? msg : "OK" - } - })() - res.send(JSON.stringify({ status, code, desc }, 0, 2)) - } - - _checkToken(token) { - return token == this.token - } - -} - - - -module.exports = Websocket +const Main = require('../main') +const Logger = require('../util/logger') +const { players, guildLog, Player } = require('../core/player') +const { info, error } = require('../util/msgs') +const path = require('path') +const EventEmitter = require('events'); +const DicordOAuth = require('../util/discordOAuth') + +const express = require('express') +const hbs = require('express-handlebars') +const socketio = require('socket.io') +const http = require('http') + + +const WEBINTERFACE_VERSION = '1.10.0' +const SESSION_TIMEOUT = 1800 * 1000 // 30 Minutes +const LOGIN_TIMEOUT = 30 * 1000 // 30 Seconds +const EXPOSE_PORT = process.argv.includes("--1337") ? 1337 : 6612 + +const STATUS = { + ERROR: 'ERROR', + OK: 'OK' +} + +const ERRCODE = { + OK: 0, + INVALID_TOKEN: 1, + INVALID_GUILD: 2, + INVALID_FILE: 3, + PLAYER_ERROR: 4, + INVALID_LOGIN: 5, + SESSION_NOT_LOGGED_IN: 6, + NO_VC: 7, + LOGIN_TIMED_OUT: 8 +} + +class Session { + constructor(userid, token) { + return new Promise((res, reject) => { + this.userid = userid + this.token = token + Main.client.fetchUser(userid, true) + .then(user => { + this.user = user + Main.client.guilds.forEach(g => { + let memb = g.members.find(m => m.id == this.user.id && m.voiceChannel) + if (memb) { + this.guild = g + this.player = players[g.id] + this.member = memb + } + }) + if (this.member) { + res(this) + this.timer = new SessionTimer(SESSION_TIMEOUT) + } + else + reject({message: 'User not found in any voice channel!'}) + }) + .catch(reject) + }) + } + + get vc() { + return this.member.voiceChannel + } + +} + + +class SessionTimer extends EventEmitter { + constructor(timeout) { + super() + this.timeout = timeout + this.create() + } + + create() { + this.timer = setTimeout(() => { + this.emit('elapsed') + }, this.timeout) + } + + refresh() { + clearTimeout(this.timer) + this.create() + } +} + + +function removeFromArrayIfExists(array, element) { + let i = array.indexOf(element) + if (i > -1) + array.splice(i, 1) +} + +class Websocket { + + constructor() { + this.sessions = {} + this.ipregister = {} + this.authorizedids = [] + this.oauth = new DicordOAuth({ + clientid: Main.config.client.id, + clientsecret: Main.config.client.secret, + serveraddr: Main.config.serveraddr + }) + this.app = express() + this.app.engine('hbs', hbs({ + extname: 'hbs', + defaultLayout: 'layout', + layoutsDir: __dirname + '/webinterface/layouts' + })) + this.app.set('views', path.join(__dirname, 'webinterface')) + this.app.set('view engine', 'hbs') + this.app.use(express.static(path.join(__dirname, 'webinterface'))) + this.server = new http.Server(this.app) + this.io = socketio(this.server) + this.token = Main.config.wstoken + + if (!this.token || this.token == "") { + Logger.error('Can not set up Websocket API! Missing token in config!') + return + } + + /////////////////////// + //// WEB INTERFACE //// + /////////////////////// + + // WEBINTERFACE + this.app.get('/', (req, res) => { + var user = req.query.user ? req.query.user : this.ipregister[req.connection.remoteAddress] + if (!user) { + this.oauth.redirectToAuth('authorize', res) + return + } + var session = this.sessions[user] + var sortbydate = (req.query.sortbydate == 1) + if (session && session.vc) { + var fileList = Player.getFilelist(sortbydate) + .map(f => f.split('.')[0]) + + res.render('index', { + fileList, + user, + usertag: session.user.tag, + voicechannel: { + id: session.vc.id, + name: session.vc.name + }, + guild: { + id: session.guild.id, + name: session.guild.name + }, + sortbydate, + token: session.token, + inchannel: (session.guild.me.voiceChannel != null), + WEBINTERFACE_VERSION + }) + } else { + this.oauth.redirectToAuth('authorize', res) + } + }) + + // WEBINTERFACE LOGIN + this.app.get('/wilogin', (req, res) => { + var token = req.query.token + var user = req.query.user + var passedByToken = req.query.passedByToken + + if (!user || !token || user == "" || token == "") { + res.render('error', { + code: ERRCODE.INVALID_LOGIN, + reason: 'Invalid login credentials.' + }) + return + } + + if (!this.authorizedids.includes(user)) { + res.render('error', { + code: ERRCODE.LOGIN_TIMED_OUT, + reason: 'Login timed out.' + }) + return + } + + if (!this._checkToken(token)) { + if (passedByToken == '1') { + res.render('resetTokenCookie', { + code: ERRCODE.INVALID_TOKEN, + reason: 'Invalid token.' + }) + return + } + res.render('error', { + code: ERRCODE.INVALID_TOKEN, + reason: 'Invalid token.' + }) + return + } + + new Session(user, token) + .then(session => { + session.timer.on('elapsed', () => { + Logger.info(`[WS Session Expired] CID: ${user} | TAG: ${session.user.tag}`) + let socket = session.socket + if (socket) + socket.emit('logout') + this.sessions[user] = null + }) + this.sessions[user] = session + this.ipregister[req.connection.remoteAddress] = user + Logger.info(`[WS Login (WEB)] CID: ${user} | TAG: ${session.user.tag}`) + res.redirect('/?user=' + user) + }) + .catch(e => { + res.render('error', { + code: ERRCODE.INVALID_LOGIN, + reason: e.message + }) + }) + }) + + // WEBINTERFACE LOGOUT + this.app.get('/wilogout', (req, res) => { + var user = req.query.user + + if (!this._checkToken(req.query.token)) { + res.send("") + return + } + + Logger.info(`[WS Logout (WEB)] CID: ${user} | TAG: ${this.sessions[user].user.tag}`) + this.sessions[user] = null + this.ipregister[req.connection.remoteAddress] = null + res.redirect('/') + }) + + // WEBINTERFACE RESTART BOT + this.app.get('/wirestart', (req, res) => { + var token = req.query.token + var user = req.query.user + if (!this._checkToken(token)) { + res.send("") + return + } + + var session = this.sessions[user] + + new Promise((res, rej) => { + if (session && session.vc) + session.vc.leave() + .then(res()).catch(rej()) + else + res() + }).then(() => { + process.exit(); + }).catch(() => { + process.exit(); + }) + + + }) + + // WEBINTERFACE SHOW LOG + this.app.get('/wishowlog', (req, res) => { + var guild = req.query.guild + var user = req.query.user + + if (!this._checkToken(req.query.token)) { + res.send("") + return + } + + var log = guildLog[guild] + if (log == null) { + res.render('error', { + code: 7, + reason: 'Guild not found in players log list.' + }) + return + } + + var session = this.sessions[user] + if (!session) { + res.render('error', { + code: ERRCODE.INVALID_LOGIN, + reason: 'Invalid session.' + }) + return + } + + session.timer.refresh() + + log = log + .map(s => s.replace(/[`\*]/gm, '')) + + res.render('log', { + guild: { + id: session.guild.id, + name: session.guild.name + }, + log + }) + }) + + // LOGG IN AND CREATE SESSION FOR USER ID + this.app.get('/login', (req, res) => { + res.set('Content-Type', 'application/json') + + var token = req.query.token + var userID = req.query.user + + if (!this._checkToken(token)) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + return + } + + new Session(userID, token) + .then(session => { + this._sendStatus(res, STATUS.OK, ERRCODE.OK) + session.timer.on('elapsed', () => this.sessions[user] = null) + this.sessions[userID] = session + this.ipregister[req.connection.remoteAddress] = userID + Logger.info(`[WS Login] CID: ${userID} | TAG: ${session.user.tag}`) + }) + .catch(e => { + console.log(e) + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_LOGIN, e.message) + }) + }) + + // LOGGOUT + this.app.get('/logout', (req, res) => { + res.set('Content-Type', 'application/json') + + var token = req.query.token + var userID = req.query.user + + if (!this._checkToken(token)) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + return + } + + var session = this.sessions[userID] + if (session) { + this.sessions[userID] = null + this.ipregister[req.connection.remoteAddress] = null + this._sendStatus(res, STATUS.OK, ERRCODE.OK) + Logger.info(`[WS Logout] CID: ${userID} | TAG: ${session.user.tag}`) + } + else + this._sendStatus(res, STATUS.ERROR, ERRCODE.SESSION_NOT_LOGGED_IN) + }) + + // PLAY SOUND METHOD + this.app.get('/play', (req, res) => { + res.set('Content-Type', 'application/json') + + var token = req.query.token + var userID = req.query.user + var soundFile = req.query.file + + if (!this._checkToken(token)) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + return + } + + var session = this.sessions[userID] + + if (!session) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.SESSION_NOT_LOGGED_IN) + return + } + + if (!soundFile || soundFile == "") { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_GUILD) + return + } + + if (!session.vc) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.NO_VC) + return + } + + session.timer.refresh() + + new Promise(res => { + // var player = players[session.guild.id] + var player = session.player + if (player) + res(player) + else + new Player(session.vc) + .then(p => res(p)) + }).then(player => { + session.player = player + player.play(soundFile, session.member).then(() => { + this._sendStatus(res, STATUS.OK, 0) + }).catch(e => { + this._sendStatus(res, STATUS.ERROR, ERRCODE.PLAYER_ERROR, e.message) + }) + }).catch(e => { + this._sendStatus(res, STATUS.ERROR, ERRCODE.PLAYER_ERROR, e.message) + }) + }) + + // GET SOUND FILES + this.app.get('/sounds', (req, res) => { + res.set('Content-Type', 'application/json') + var token = req.query.token + + if (!this._checkToken(token)) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + return + } + + var fileList = Player.getFilelist() + .map(f => f.split('.')[0]) + + res.send(JSON.stringify({ + status: STATUS.OK, + code: 0, + desc: { + n: fileList.length, + sounds: fileList + } + }, 0, 2)) + }) + + // SET GUILDS AND IDS + this.app.get('/guilds', (req, res) => { + res.set('Content-Type', 'application/json') + var token = req.query.token + + if (!this._checkToken(token)) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + return + } + + var servers = Main.client.guilds + .map(g => [g.name, g.id]) + + res.send(JSON.stringify({ + status: STATUS.OK, + code: 0, + desc: { + n: servers.length, + servers: servers + } + }, 0, 2)) + }) + + // CHECK TOKEN + this.app.get('/token', (req, res) => { + res.set('Content-Type', 'application/json') + var token = req.query.token + + if (!this._checkToken(token)) + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_TOKEN) + else + this._sendStatus(res, STATUS.OK, 0) + + }) + + this.app.get('/authorize', (req, res) => { + let code = req.query.code; + let resetToken = req.query.resetToken + + if (resetToken) { + let user = req.query.user + res.render('login', { user, resetToken }) + } + + this.oauth.getId(req, (err, user) => { + if (err) { + this._sendStatus(res, STATUS.ERROR, ERRCODE.INVALID_LOGIN) + return + } + var sortbydate = (req.query.sortbydate == 1) + var session = this.sessions[user] + this.authorizedids.push(user) + setTimeout(() => { + removeFromArrayIfExists(this.authorizedids, user) + }, LOGIN_TIMEOUT) + if (!session) { + res.render('login', { user }) + return + } + if (!session.vc) { + this.sessions[user] = null + res.render('login', { user }) + return + } + res.redirect('/?user=' + user) + }) + }) + + this.server.listen(EXPOSE_PORT, () => { + Logger.info('WS API set up at port ' + this.server.address().port) + }) + + + this.io.on('connection', (socket) => { + var session + Logger.info('WS onnection etablished: ' + socket.id) + + socket.on('thatsMe', (data) => { + session = this.sessions[data.user] + if (!session) { + socket.emit('logout') + socket.disconnect() + return; + } + let user = session.member.user + socket.join(session.guild.id) + session.socket = socket + let guildsessions = Object.keys(this.sessions) + .filter(k => this.sessions[k] && this.sessions[k].guild && this.sessions[k].guild.id == session.guild.id && this.sessions[k].socket) + .map(k => this.sessions[k].member.user) + let alreadyConnected = [] + guildsessions.forEach(u => alreadyConnected.push({ id: u.id, tag: u.tag, avatarURL: u.avatarURL })) + this.io.to(session.guild.id).emit('userConnected', alreadyConnected) + }) + + socket.on('disconnecting', (socket) => { + Logger.info("WS onnection closed") + if (session) { + let user = session.member.user + this.io.to(session.guild.id).emit('userDisconnected', { id: user.id, tag: user.tag }) + session.socket = null + } + }) + + socket.emit('whoAreYou') + + socket.on('changeChannelStatus', () => { + if (session && session.vc) { + session.timer.refresh() + var currchan = session.guild.me.voiceChannel + if (currchan) { + if (session.player) + session.player.destroy() + else + currchan.leave() + } + else { + session.vc.join().catch() + } + } + }) + + socket.on('stopSound', () => { + if (session && session.player) { + session.player.stop() + session.timer.refresh() + } + }) + + // PLAY EVENT + socket.on('playSound', (data) => { + var user = data.user + var soundFile = data.sound + + if (!user || !session) + return + + session.timer.refresh() + + new Promise((res, rej) => { + var player = players[session.guild.id] + if (player) + res(player) + else + new Player(session.vc) + .then(p => res(p)) + }).then(player => { + session.player = player + let sockets = this.io.to(session.guild.id) //.of(session.guild.id) + player.play(soundFile, session.member).then(() => { + sockets.emit('soundPlaying', { user, sound: soundFile }) + player.dispatcher.on('end', () => { + sockets.emit('soundStopped', { user, sound: soundFile }) + }) + }).catch(e => { + sockets.emit('playError', { user, sound: soundFile, error: e }) + console.log(e) + }) + }).catch(e => { + sockets.emit('playError', { user, sound: soundFile, error: e }) + console.log(e) + }) + }) + + Main.client.on('voiceStateUpdate', (oldMemb, newMemb) => { + let session = this.sessions[oldMemb.id] + if (session && session.socket && oldMemb.voiceChannel && !newMemb.voiceChannel) { + session.socket.emit('logout') + } + if (oldMemb.id != Main.client.user.id) + return + if (!oldMemb.voiceChannel && newMemb.voiceChannel) { + let vc = newMemb.voiceChannel + this.io.emit('channelJoined', vc.id) + } + else if (oldMemb.voiceChannel && !newMemb.voiceChannel) { + let vc = oldMemb.voiceChannel + this.io.emit('channelLeft', vc.id) + } + }) + + Main.eventEmiter.on('closing', () => { + this.io.emit('logout') + }) + }) + } + + _sendStatus(res, status, code, msg) { + let desc = (() => { + switch (code) { + case ERRCODE.INVALID_TOKEN: + return 'Invalid token.' + case ERRCODE.INVALID_GUILD: + return 'Invalid guild ID or no player active on this guild currently.' + case ERRCODE.INVALID_FILE: + return 'File name not set.' + case ERRCODE.PLAYER_ERROR: + return msg ? msg : 'Unknown player error.' + case ERRCODE.INVALID_LOGIN: + return msg ? msg : 'Invalid login.' + case ERRCODE.SESSION_NOT_LOGGED_IN: + return 'Session not logged in.' + case ERRCODE.NO_VC: + return 'USer not in voice channel.' + default: + return msg ? msg : "OK" + } + })() + res.send(JSON.stringify({ status, code, desc }, 0, 2)) + } + + _checkToken(token) { + return token == this.token + } + +} + + + +module.exports = Websocket