253 lines
8.0 KiB
TypeScript
253 lines
8.0 KiB
TypeScript
import * as db from "./database";
|
|
import {config} from "./config";
|
|
import {io} from "./http";
|
|
import * as irc from "irc";
|
|
import * as discord from "discord.js";
|
|
import * as twitch from "dank-twitch-irc";
|
|
|
|
var ircClient;
|
|
var xmppClient;
|
|
var twitchClient;
|
|
var twitchArr: Array<string> = [];
|
|
var discordClient;
|
|
var liveUsers: Array<any>;
|
|
var chatIntegration: Array<any>;
|
|
|
|
async function init() {
|
|
setInterval(updateUsers, 20000);
|
|
setInterval(updateInteg, 60000);
|
|
if(config['chat']['discord']['enabled']){
|
|
discordClient = new discord.Client();
|
|
discordClient.once('ready', ()=>{ console.log('Discord bot ready')});
|
|
discordClient.on('message', async (msg) => {
|
|
if(msg['author']['bot']) return;
|
|
var lu = getUsr(msg['channel']['name'], 'discord')
|
|
for(var i=0;i<lu.length;i++){
|
|
sendAll(lu[i], [msg['author']['username'], await normalizeDiscordMsg(msg)], "discord");
|
|
}
|
|
});
|
|
discordClient.login(config['chat']['discord']['token']);
|
|
}
|
|
if(config['chat']['irc']['enabled']){
|
|
ircClient = new irc.Client(config['chat']['irc']['server'], config['chat']['irc']['nickname'], {
|
|
userName: config['chat']['irc']['username'],
|
|
realName: config['chat']['irc']['realname'],
|
|
port: config['chat']['irc']['port'],
|
|
secure: config['chat']['irc']['tls'],
|
|
sasl: config['chat']['irc']['sasl'],
|
|
password: config['chat']['irc']['password'],
|
|
});
|
|
ircClient.addListener('error', (message) => {
|
|
console.log('IRC Client Error: ', message);
|
|
});
|
|
ircClient.once('registered', () => {
|
|
console.log("IRC Client Ready");
|
|
});
|
|
ircClient.on('message', (from, to, msg) => {
|
|
var lu = getUsr(to, 'irc');
|
|
for(var i=0;i<lu.length;i++){
|
|
sendAll(lu[i], [from, msg], "irc")
|
|
}
|
|
});
|
|
}
|
|
if(config['chat']['xmpp']['enabled']){
|
|
|
|
}
|
|
if(config['chat']['twitch']['enabled']){
|
|
twitchClient = new twitch.ChatClient({
|
|
username: config['chat']['twitch']['username'],
|
|
password: config['chat']['twitch']['password'],
|
|
});
|
|
twitchClient.on('ready', () => {
|
|
console.log("Twitch Client Ready");
|
|
});
|
|
twitchClient.on("error", (error) => {
|
|
if (error != null) {
|
|
console.error("Twitch Client Error: ", error);
|
|
}
|
|
});
|
|
twitchClient.on("PRIVMSG", (msg) => {
|
|
if(msg['senderUserID'] === twitchClient['userStateTracker']['globalState']['userID']) return;
|
|
var lu = getUsr(msg['channelName'], 'twitch');
|
|
for(var i=0;i<lu.length;i++){
|
|
sendAll(lu[i], [msg['displayName'], msg['messageText']], "twitch");
|
|
}
|
|
});
|
|
//this library doesn't internally track which channels are currently joined, so we have to do it ourself
|
|
twitchClient.on('JOIN', (m) => {
|
|
if(twitchArr.indexOf(m['channelName']) === -1)
|
|
twitchArr.push(m['channelName']);
|
|
});
|
|
twitchClient.on('PART', (m) => {
|
|
if(twitchArr.indexOf(m['channelName']) !== -1)
|
|
twitchArr.splice(twitchArr.indexOf(m['channelName']), 1);
|
|
});
|
|
twitchClient.connect();
|
|
}
|
|
}
|
|
|
|
async function updateUsers() {
|
|
liveUsers = await db.query('SELECT username FROM user_meta WHERE live=true');
|
|
return;
|
|
}
|
|
|
|
async function updateInteg() {
|
|
if(liveUsers.length === 0) {
|
|
chatIntegration = [];
|
|
return;
|
|
}
|
|
if(liveUsers.length === 1) {
|
|
chatIntegration = await db.query('SELECT * FROM chat_integration WHERE username='+db.raw.escape(liveUsers[0]['username']));
|
|
updateIRCChan();
|
|
updateTwitchChan();
|
|
return;
|
|
}
|
|
var qs: string;
|
|
for(var u in liveUsers) {
|
|
qs += db.raw.escape(u['username']) + " OR username=";
|
|
}
|
|
qs = qs.substring(0, qs.length - 13);
|
|
chatIntegration = await db.query('SELECT * FROM chat_integration WHERE username='+qs);
|
|
updateIRCChan();
|
|
updateTwitchChan();
|
|
}
|
|
|
|
async function sendAll(user: string, msg: Array<string>, src: string) {
|
|
//msg should be an array containing first the username of the user who sent the message
|
|
//followed by the message text
|
|
//[sender, message]
|
|
|
|
//user string is the user whose chat is being mirrored
|
|
|
|
if(user === null) return;
|
|
|
|
if(src !== "irc") sendIRC(getCh(user, "irc"), '['+src.toUpperCase()+']'+msg[0]+': '+msg[1]);
|
|
if(src !== "twitch") sendTwitch(getCh(user, "twitch"), '['+src.toUpperCase()+']'+msg[0]+': '+msg[1]);
|
|
if(src !== "discord") sendDiscord(getCh(user, "discord"), '['+src.toUpperCase()+']'+msg[0]+': '+msg[1]);
|
|
//if(src !== "xmpp") sendXMPP();
|
|
if(src !== "web") sendWeb(user, ['['+src.toUpperCase()+']'+msg[0], msg[1]]);
|
|
}
|
|
|
|
async function sendIRC(channel: string, msg: string) {
|
|
if(!config['chat']['irc']['enabled']) return;
|
|
if(channel === null) return;
|
|
ircClient.say(channel, msg);
|
|
}
|
|
|
|
async function sendDiscord(channel: string, msg: string) {
|
|
if(!config['chat']['discord']['enabled']) return;
|
|
if(channel === null) return;
|
|
var ch = discordClient.channels.find('name', channel);
|
|
ch.send(msg);
|
|
}
|
|
|
|
async function sendXMPP(channel: string, msg: string) {
|
|
if(!config['chat']['xmpp']['enabled']) return;
|
|
if(channel === null) return;
|
|
}
|
|
|
|
async function sendTwitch(channel: string, msg: string) {
|
|
if(!config['chat']['twitch']['enabled']) return;
|
|
if(channel === null) return;
|
|
twitchClient.say(channel, msg);
|
|
}
|
|
|
|
async function sendWeb(channel: string, msg: Array<string>) {
|
|
if(channel === null) return;
|
|
io.to(channel).emit('MSG', {nick: msg[0], msg: msg[1], room: channel});
|
|
}
|
|
|
|
function getCh(usr: string, itype: string): string{
|
|
for(var i=0;i<chatIntegration.length;i++){
|
|
if(chatIntegration[i]['username'] === usr){
|
|
if(chatIntegration[i][itype].trim() !== "" && chatIntegration[i][itype] !== null) return chatIntegration[i][itype];
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function getUsr(channel: string, ctype: string): Array<string>{
|
|
var list: Array<string> = [];
|
|
for(var i=0;i<chatIntegration.length;i++){
|
|
if(chatIntegration[i][ctype] === channel) list.push(chatIntegration[i]['username']);
|
|
}
|
|
return list;
|
|
}
|
|
|
|
async function updateIRCChan() {
|
|
var ilist: Array<string> = [];
|
|
for(var i=0;i<chatIntegration.length;i++){
|
|
if(chatIntegration[i]['irc'].trim() !== "" && chatIntegration[i]['irc'] !== null) {
|
|
ilist.push(chatIntegration[i]['irc']);
|
|
}
|
|
}
|
|
//do this to avoid duplicate channel joins
|
|
//and leave IRC channels when done
|
|
var tmp: Array<string> = ilist.filter(n => !Object.keys(ircClient.chans).includes(n)); //channels to join
|
|
for(var i=0;i<tmp.length;i++){
|
|
ircClient.join(tmp[i]);
|
|
}
|
|
tmp = Object.keys(ircClient.chans).filter(n => !ilist.includes(n)); //channels to part
|
|
for(var i=0;i<tmp.length;i++){
|
|
ircClient.part(tmp[i]);
|
|
}
|
|
}
|
|
|
|
async function updateTwitchChan() {
|
|
var ilist: Array<string> = [];
|
|
for(var i=0;i<chatIntegration.length;i++){
|
|
if(chatIntegration[i]['twitch'].trim() !== "" && chatIntegration[i]['twitch'] !== null) ilist.push(chatIntegration[i]['twitch']);
|
|
}
|
|
//do this to avoid duplicate channel joins
|
|
//and leave twitch channels when done
|
|
var tmp: Array<string> = ilist.filter(n => !twitchArr.includes(n)); //channels to join
|
|
for(var i=0;i<tmp.length;i++){
|
|
twitchClient.join(tmp[i]);
|
|
}
|
|
tmp = twitchArr.filter(n => !ilist.includes(n)); //channels to part
|
|
for(var i=0;i<tmp.length;i++){
|
|
twitchClient.part(tmp[i]);
|
|
}
|
|
}
|
|
|
|
async function normalizeDiscordMsg(msg): Promise<string>{
|
|
var nmsg: string=msg['content'];
|
|
|
|
//normalize user mentions
|
|
var uarray = await msg['mentions']['users'].array();
|
|
var karray = await msg['mentions']['users'].keyArray();
|
|
for(var i=0;i<karray.length;i++){
|
|
var usr = uarray[i];
|
|
nmsg = nmsg.replace(new RegExp('<@!'+karray[i]+'>', 'g'), '@'+usr['username']);
|
|
}
|
|
|
|
//normalize emoji
|
|
var e = nmsg.match(new RegExp('<:\\w+:[0-9]+>', 'g'));
|
|
//<:.+:.+>
|
|
if(e !== null)
|
|
for (var i=0;i<e.length;i++){
|
|
nmsg = nmsg.replace(e[i], e[i].match(new RegExp(':\\w+:', 'g'))[0]);
|
|
}
|
|
//in 10 minutes, I will have forgot what all of this regex does.
|
|
|
|
//normalize role mentions
|
|
var uarray = await msg['mentions']['roles'].array();
|
|
var karray = await msg['mentions']['roles'].keyArray();
|
|
for(var i=0;i<karray.length;i++){
|
|
var role = uarray[i];
|
|
nmsg = nmsg.replace(new RegExp('<@&'+karray[i]+'>', 'g'), '@'+role['name']);
|
|
}
|
|
|
|
//normalize channel mentions
|
|
var c = nmsg.match(new RegExp('<#[0-9]+>', 'g'));
|
|
if(c !== null)
|
|
for(var i=0;i<c.length;i++){
|
|
nmsg = nmsg.replace(c[i], '#'+discordClient.channels.get(c[i].match(new RegExp('[0-9]+', 'g'))[0])['name']);
|
|
}
|
|
|
|
//fuck me this better work
|
|
|
|
return nmsg;
|
|
}
|
|
|
|
export { init, sendAll }; |