initial WTVIRC
This commit is contained in:
365
zefie_wtvp_minisrv/includes/classes/WTVIRC.js
Normal file
365
zefie_wtvp_minisrv/includes/classes/WTVIRC.js
Normal file
@@ -0,0 +1,365 @@
|
||||
const net = require('net');
|
||||
|
||||
class WTVIRC {
|
||||
constructor(minisrv_config, host = 'localhost', port = 6667, debug = false) {
|
||||
this.minisrv_config = minisrv_config;
|
||||
this.version =
|
||||
this.host = host;
|
||||
this.port = port;
|
||||
this.debug = debug;
|
||||
this.server = null;
|
||||
this.clients = [];
|
||||
this.channels = new Map();
|
||||
this.channeltopics = new Map(); // channel -> topic
|
||||
this.nicknames = new Map(); // socket -> nickname
|
||||
this.awaymsgs = new Map(); // nickname -> away message
|
||||
this.nicklen = 30;
|
||||
this.servername = 'irc.local';
|
||||
this.caps = `AWAYLEN=200 CHANTYPES=# PREFIX=(ov)@+ CHANMODES=beI,k,l,imnp SAFELIST MAXLIST=b:100,e:100,i:100,k:100,l:50 CHANLIMIT=#:50 NICKLEN=${this.nicklen} TOPICLEN=390 KICKLEN=390`;
|
||||
}
|
||||
|
||||
start() {
|
||||
this.server = net.createServer(socket => {
|
||||
socket.setEncoding('utf8');
|
||||
this.clients.push(socket);
|
||||
|
||||
let registered = false;
|
||||
let nickname = '';
|
||||
let username = '';
|
||||
let channel = '';
|
||||
let host = socket.remoteAddress || 'minisrv.local';
|
||||
|
||||
if (this.debug) {
|
||||
const originalWrite = socket.write;
|
||||
socket.write = function (...args) {
|
||||
console.log(`[socket.write]`, ...args);
|
||||
return originalWrite.apply(socket, args);
|
||||
};
|
||||
}
|
||||
|
||||
socket.write(':irc.local NOTICE AUTH :Welcome to minisrv IRC Server\r\n');
|
||||
|
||||
socket.on('data', data => {
|
||||
const lines = data.split(/\r\n|\n/).filter(Boolean);
|
||||
for (let line of lines) {
|
||||
if (this.debug) {
|
||||
console.log(`Received data from client: ${line}`);
|
||||
}
|
||||
const [command, ...params] = line.trim().split(' ');
|
||||
switch (command.toUpperCase()) {
|
||||
case 'TOPIC':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (params.length < 1) {
|
||||
socket.write(`:${this.servername} 461 ${nickname} TOPIC :Not enough parameters\r\n`);
|
||||
break;
|
||||
}
|
||||
channel = params[0];
|
||||
if (params.length > 1) {
|
||||
var topic = params.slice(1).join(' ');
|
||||
if (topic.startsWith(':')) {
|
||||
topic = topic.slice(1);
|
||||
}
|
||||
this.channeltopics.set(channel, topic);
|
||||
socket.write(`:${this.servername} 332 ${nickname} ${channel} :${topic}\r\n`);
|
||||
this.broadcast(`:${nickname}!${username}@${host} TOPIC ${channel} :${topic}\r\n`, socket);
|
||||
} else {
|
||||
const topic = this.channeltopics.get(channel) || 'No topic set';
|
||||
socket.write(`:${this.servername} 331 ${nickname} ${channel} :${topic}\r\n`);
|
||||
}
|
||||
break;
|
||||
case "AWAY":
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (params.length > 0) {
|
||||
socket.write(`:${this.servername} 301 ${nickname} :You are now marked as away\r\n`);
|
||||
this.awaymsgs.set(nickname, params.join(' '));
|
||||
} else {
|
||||
socket.write(`:${this.servername} 305 ${nickname} :You are no longer marked as away\r\n`);
|
||||
this.awaymsgs.delete(nickname);
|
||||
}
|
||||
break;
|
||||
case 'CAP':
|
||||
// Minimal CAP support: just acknowledge LS
|
||||
if (params[0] && params[0].toUpperCase() === 'LS') {
|
||||
socket.write('CAP * LS :\r\n');
|
||||
}
|
||||
break;
|
||||
case 'NICK':
|
||||
var new_nickname = params[0];
|
||||
if (!new_nickname || new_nickname.length < 1) {
|
||||
socket.write(`:${this.servername} 431 * :No nickname\r\n`);
|
||||
break;
|
||||
}
|
||||
if (new_nickname.length > 30) {
|
||||
socket.write(`:${this.servername} 432 * ${new_nickname} :Erroneus nickname\r\n`);
|
||||
break;
|
||||
}
|
||||
var result = Array.from(this.nicknames.values()).find(nick => nick === new_nickname);
|
||||
if (result) {
|
||||
socket.write(`:${this.servername} 433 * ${new_nickname} :Nickname is already in use\r\n`);
|
||||
break;
|
||||
}
|
||||
if (!nickname) {
|
||||
nickname = new_nickname;
|
||||
}
|
||||
if (nickname && nickname !== new_nickname) {
|
||||
socket.write(`:${nickname}!${username}@${host} NICK :${new_nickname}\r\n`);
|
||||
this.broadcast(`:${nickname}!${username}@${host} NICK :${new_nickname}\r\n`, socket);
|
||||
// Update nickname in all channels
|
||||
for (const [ch, users] of this.channels.entries()) {
|
||||
if (users.has(nickname)) {
|
||||
users.delete(nickname);
|
||||
users.add(new_nickname);
|
||||
}
|
||||
}
|
||||
// Update away message mapping if present
|
||||
if (this.awaymsgs.has(nickname)) {
|
||||
const msg = this.awaymsgs.get(nickname);
|
||||
this.awaymsgs.delete(nickname);
|
||||
this.awaymsgs.set(new_nickname, msg);
|
||||
}
|
||||
}
|
||||
nickname = new_nickname;
|
||||
this.nicknames.set(socket, nickname);
|
||||
if (!registered && nickname && username) {
|
||||
registered = true;
|
||||
socket.write(`:${this.servername} 001 ${nickname} :Welcome to the IRC server, ${nickname}\r\n`);
|
||||
socket.write(`:${this.servername} 002 ${nickname} :Your host is ${this.servername}, running version minisrv ${this.minisrv_config.version}\r\n`);
|
||||
socket.write(`:${this.servername} 003 ${nickname} :This server is ready to accept commands\r\n`);
|
||||
socket.write(`:${this.servername} 004 ${nickname} ${this.servername} minisrv ${this.minisrv_config.version} :End of /MOTD command\r\n`);
|
||||
socket.write(`:${this.servername} 005 ${this.caps}\r\n`);
|
||||
}
|
||||
break;
|
||||
case 'USER':
|
||||
username = params[0];
|
||||
if (!registered && nickname && username) {
|
||||
registered = true;
|
||||
socket.write(`:${this.servername} 001 ${nickname} :Welcome to the IRC server, ${nickname}\r\n`);
|
||||
socket.write(`:${this.servername} 002 ${nickname} :Your host is ${this.servername}, running version minisrv ${this.minisrv_config.version}\r\n`);
|
||||
socket.write(`:${this.servername} 003 ${nickname} :This server is ready to accept commands\r\n`);
|
||||
socket.write(`:${this.servername} 004 ${nickname} ${this.servername} minisrv ${this.minisrv_config.version} :End of /MOTD command\r\n`);
|
||||
socket.write(`:${this.servername} 005 ${this.caps}\r\n`);
|
||||
}
|
||||
break;
|
||||
case 'JOIN':
|
||||
if (!registered) {
|
||||
socket.write(`:irc.local 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
channel = params[0];
|
||||
if (channel.includes(',')) {
|
||||
var channels = channel.split(',');
|
||||
} else {
|
||||
var channels = [channel];
|
||||
}
|
||||
for (const ch of channels) {
|
||||
// Recursively process each channel join
|
||||
const joinLine = `JOIN ${ch}`;
|
||||
// Simulate a JOIN command for each channel
|
||||
const [command, ...params] = joinLine.trim().split(' ');
|
||||
// Reuse the JOIN logic for each channel
|
||||
// Only run the code after $PLACEHOLDER$ for each channel
|
||||
// (excluding the code before $PLACEHOLDER$ to avoid duplicate checks)
|
||||
// You can refactor this logic into a helper if needed
|
||||
socket.write(`:${nickname}!${username}@${host} JOIN ${ch}\r\n`);
|
||||
this.broadcast(`:${nickname}!${username}@${host} JOIN ${ch}\r\n`, socket);
|
||||
if (this.channeltopics.has(ch)) {
|
||||
const topic = this.channeltopics.get(ch);
|
||||
socket.write(`:${this.servername} 332 ${nickname} ${ch} :${topic}\r\n`);
|
||||
} else {
|
||||
socket.write(`:${this.servername} 331 ${nickname} ${ch} :No topic is set\r\n`);
|
||||
}
|
||||
if (!this.channels.has(ch)) {
|
||||
this.channels.set(ch, new Set());
|
||||
}
|
||||
this.channels.get(ch).add(nickname);
|
||||
var users = this.getUsersInChannel(ch);
|
||||
if (users.length > 0) {
|
||||
socket.write(`:${this.servername} 353 ${nickname} = ${ch} :${users.join(' ')}\r\n`);
|
||||
}
|
||||
socket.write(`:${this.servername} 366 ${nickname} ${ch} :End of /NAMES list\r\n`);
|
||||
}
|
||||
break;
|
||||
case 'NAMES':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (params.length < 1) {
|
||||
socket.write(`:${this.servername} 461 ${nickname} NAMES :Not enough parameters\r\n`);
|
||||
break;
|
||||
}
|
||||
channel = params[0];
|
||||
if (!this.channels.has(channel)) {
|
||||
socket.write(`:${this.servername} 403 ${nickname} ${channel} :No such channel\r\n`);
|
||||
break;
|
||||
}
|
||||
var users = this.getUsersInChannel(channel);
|
||||
if (users.length > 0) {
|
||||
socket.write(`:${this.servername} 353 ${nickname} = ${channel} :${users.join(' ')}\r\n`);
|
||||
}
|
||||
socket.write(`:${this.servername} 366 ${nickname} ${channel} :End of /NAMES list\r\n`);
|
||||
break;
|
||||
case 'PART':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
channel = params[0];
|
||||
if (!this.channels.has(channel) || !this.channels.get(channel).has(nickname)) {
|
||||
socket.write(`:${this.servername} 442 ${nickname} ${channel} :You're not on that channel\r\n`);
|
||||
break;
|
||||
}
|
||||
socket.write(`:${nickname}!${username}@${host} PART ${channel}\r\n`);
|
||||
this.broadcast(`:${nickname}!${username}@${host} PART ${channel}\r\n`, socket);
|
||||
if (this.channels.has(channel)) {
|
||||
this.channels.get(channel).delete(nickname);
|
||||
if (this.channels.get(channel).size === 0) {
|
||||
this.channels.delete(channel);
|
||||
}
|
||||
}
|
||||
break;
|
||||
case 'LIST':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
let channelsToList;
|
||||
if (params.length > 0 && params[0]) {
|
||||
channelsToList = params[0].split(',').filter(ch => ch.length > 0);
|
||||
} else {
|
||||
channelsToList = Array.from(this.channels.keys());
|
||||
}
|
||||
socket.write(`:${this.servername} 321 ${nickname} Channel :Users\r\n`);
|
||||
for (const channel of channelsToList) {
|
||||
const users = this.getUsersInChannel(channel);
|
||||
const topic = this.channeltopics.get(channel) || 'No topic is set';
|
||||
socket.write(`:${this.servername} 322 ${nickname} ${channel} ${users.length} :${topic}\r\n`);
|
||||
}
|
||||
socket.write(`:${this.servername} 323 ${nickname} :End of /LIST\r\n`);
|
||||
break;
|
||||
case 'WHO':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (!params[0]) {
|
||||
socket.write(`:${this.servername} 461 ${nickname} WHO :Not enough parameters\r\n`);
|
||||
break;
|
||||
}
|
||||
const target = params[0];
|
||||
if (target.startsWith('#')) {
|
||||
// WHO for channel
|
||||
if (this.channels.has(target)) {
|
||||
const users = this.getUsersInChannel(target);
|
||||
for (const user of users) {
|
||||
const sock = Array.from(this.nicknames.keys()).find(s => this.nicknames.get(s) === user);
|
||||
if (sock) {
|
||||
socket.write(`:${this.servername} 352 ${nickname} * ${user} ${sock.remoteAddress} ${this.servername} ${user} H :0 ${user}\r\n`);
|
||||
}
|
||||
}
|
||||
}
|
||||
socket.write(`:${this.servername} 315 ${nickname} ${target} :End of /WHO list\r\n`);
|
||||
} else {
|
||||
// WHO for nickname
|
||||
let found = false;
|
||||
for (const [sock, nick] of this.nicknames.entries()) {
|
||||
if (nick === target) {
|
||||
found = true;
|
||||
socket.write(`:${this.servername} 352 ${nickname} * ${nick} ${sock.remoteAddress} ${this.servername} ${nick} H :0 ${nick}\r\n`);
|
||||
break;
|
||||
}
|
||||
}
|
||||
socket.write(`:${this.servername} 315 ${nickname} ${target} :End of /WHO list\r\n`);
|
||||
if (!found) {
|
||||
// Optionally, could send no such nick/channel
|
||||
}
|
||||
}
|
||||
break;
|
||||
case 'PRIVMSG':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (params[0]) {
|
||||
const msg = line.slice(line.indexOf(':', 1) + 1);
|
||||
this.broadcast(`:${nickname}!${username}@${host} PRIVMSG ${params[0]} :${msg}\r\n`, socket);
|
||||
}
|
||||
break;
|
||||
case 'NOTICE':
|
||||
if (!registered) {
|
||||
socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
|
||||
break;
|
||||
}
|
||||
if (params[0]) {
|
||||
const msg = line.slice(line.indexOf(':', 1) + 1);
|
||||
this.broadcast(`:${nickname}!${username}@${host} NOTICE ${params[0]} :${msg}\r\n`, socket);
|
||||
}
|
||||
break;
|
||||
case 'PING':
|
||||
socket.write(`PONG ${params.join(' ')}\r\n`);
|
||||
break;
|
||||
case 'QUIT':
|
||||
socket.end();
|
||||
break;
|
||||
default:
|
||||
// Ignore unknown commands
|
||||
break;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('end', () => {
|
||||
this.clients = this.clients.filter(c => c !== socket);
|
||||
this.nicknames.delete(socket);
|
||||
});
|
||||
|
||||
socket.on('error', () => {
|
||||
this.clients = this.clients.filter(c => c !== socket);
|
||||
this.nicknames.delete(socket);
|
||||
});
|
||||
});
|
||||
|
||||
this.server.listen(this.port, this.host, () => {
|
||||
if (this.debug) {
|
||||
console.log(`IRC server started on port ${this.host}:${this.port}`);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
sendWebTVNotice(message) {
|
||||
this.broadcast(`:${this.servername} NOTICE * :${message}\r\n`);
|
||||
}
|
||||
|
||||
sendToChannelAs(username, channel, message) {
|
||||
const users = this.getUsersInChannel(channel);
|
||||
for (const user of users) {
|
||||
const socket = Array.from(this.nicknames.keys()).find(s => this.nicknames.get(s) === user);
|
||||
if (socket) {
|
||||
socket.write(`:${username}!${username}@${this.servername} PRIVMSG ${channel} :${message}\r\n`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
getUsersInChannel(channel) {
|
||||
if (this.channels.has(channel)) {
|
||||
return Array.from(this.channels.get(channel));
|
||||
}
|
||||
return [];
|
||||
}
|
||||
|
||||
broadcast(message, exceptSocket = null) {
|
||||
for (const client of this.clients) {
|
||||
if (client !== exceptSocket) {
|
||||
client.write(message);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = WTVIRC;
|
||||
Reference in New Issue
Block a user