const dbPool = require('./database') /** * @param {WebSocket} ws * @param {string} msg */ const msgSplit = async function (ws, msg) { if (!msg || typeof msg !== 'string') return null let txtarr = msg.split(' ') if (txtarr.length > 2) { if (/^ping$/i.test(txtarr[0])) { ws.send(`PONG ${txtarr[1]}`) return } // chat message if (/^privmsg$/i.test(txtarr[1])) { let user = txtarr[0].split('!')[0].substr(1) if (user === 'mtfosbot') return let channel = txtarr[2] txtarr = txtarr.slice(3, txtarr.length) let m = txtarr.join(' ').substr(1) let result = await parseCMD(user, channel, m) if (result && typeof result === 'string') { ws.send(`PRIVMSG ${channel} :${result}`) } } } } /** * * @param {string} user * @param {string} channel * @param {string} msg */ const parseCMD = async function (user, channel, msg) { let db = await dbPool.connect() let m = null try { let txt = msg.trim().split(' ') let ch = channel[0] === '#' ? channel.substr(1) : channel if (txt.length < 1) { await db.end() return null } // add cmd if (/^!\+/.test(txt[0])) { let cmd = txt[0].substr(2) let tmpmsg = txt.slice(1, txt.length).join(' ') let query = `insert into "public"."commands" ("command", "channel", "message") values ($1, $2, $3) on conflict ("command", "channel") do update set "message" = $3, "active" = true` let param = [cmd, ch, tmpmsg] await db.query(query, param) } else if (/^!-/.test(txt[0])) { let cmd = txt[0].substr(2) let query = `update "public"."commands" set "active" = $1 where "command" = $2 and "channel" = $3` let param = [false, cmd, ch] await db.query(query, param) m = `command "${cmd}" has removed` } else if (/^!/.test(txt[0])) { let cmd = txt[0].substr(1) let query = `select "message", "command" from "public"."commands" where "command" = $1 and "channel" = $2 and "active" = $3 limit 1` let param = [cmd, ch, true] let result = await db.query(query, param) if (result === null || result.rows.length === 0) return null m = result.rows[0].message } } catch (err) { console.error(err) } finally { await db.end() } return m } module.exports = { msgSplit }