Files
minisrv/zefie_wtvp_minisrv/includes/classes/WTVIRC.js
2025-06-12 23:18:22 -04:00

431 lines
24 KiB
JavaScript

const net = require('net');
class WTVIRC {
/*
* WTVIRC - A simple IRC server implementation for WebTV
* Tested with WebTV and KvIRC
* This is a basic implementation and does not cover all IRC features.
* It supports basic commands like NICK, USER, JOIN, PART, PRIVMSG, NOTICE, TOPIC, AWAY, and PING.
* TODO: KICK, BAN, MODE.
*/
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.usernames = new Map(); // nickname -> username
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(`:${this.servername} 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) {
// If no nickname is set, set it now
nickname = new_nickname;
}
this.nicknames.set(socket, nickname);
if (nickname && this.usernames.has(nickname)) {
this.usernames.delete(nickname);
}
this.usernames.set(nickname, username);
if (nickname && nickname !== new_nickname) {
socket.write(`:${nickname}!${username}@${host} NICK :${new_nickname}\r\n`);
this.broadcastUser(nickname, `:${nickname}!${username}@${host} NICK :${new_nickname}\r\n`, socket);
nickname = new_nickname;
this.nicknames.set(socket, nickname);
// 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);
}
}
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;
this.usernames.set(nickname, username);
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`);
if (!this.channels.has(ch)) {
this.channels.set(ch, new Set());
}
this.channels.get(ch).add(nickname);
this.broadcastUser(nickname, `:${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`);
}
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.broadcastUser(nickname, `:${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 'WHOIS':
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} WHOIS :Not enough parameters\r\n`);
break;
}
const whoisNick = params[0];
const whoisSocket = Array.from(this.nicknames.keys()).find(s => this.nicknames.get(s) === whoisNick);
if (whoisSocket) {
const whois_username = this.usernames.get(whoisNick);
socket.write(`:${this.servername} 311 ${nickname} ${whoisNick} ${whois_username} ${whoisSocket.remoteAddress} * ${whoisNick}\r\n`);
if (this.awaymsgs.has(whoisNick)) {
socket.write(`:${this.servername} 301 ${nickname} ${whoisNick} :${this.awaymsgs.get(whoisNick)}\r\n`);
}
const userChannels = [];
for (const [ch, users] of this.channels.entries()) {
if (users.has(whoisNick)) {
userChannels.push(ch);
}
}
if (userChannels.length > 0) {
socket.write(`:${this.servername} 319 ${nickname} ${whoisNick} :${userChannels.join(' ')}\r\n`);
}
socket.write(`:${this.servername} 318 ${nickname} ${whoisNick} :End of /WHOIS list\r\n`);
} else {
socket.write(`:${this.servername} 401 ${nickname} ${whoisNick} :No such nick/channel\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`);
}
sendWebTVNoticeTo(username, message) {
const socket = Array.from(this.nicknames.keys()).find(s => this.nicknames.get(s) === username);
if (socket) {
socket.write(`:${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 [];
}
broadcastUser(username, message, exceptSocket = null) {
for (const [channel, users] of this.channels.entries()) {
if (users.has(username)) {
for (const user of users) {
const sock = Array.from(this.nicknames.keys()).find(s => this.nicknames.get(s) === user);
if (sock && sock !== exceptSocket) {
sock.write(message);
}
}
}
}
}
broadcast(message, exceptSocket = null) {
for (const client of this.clients) {
if (client !== exceptSocket) {
client.write(message);
}
}
}
}
module.exports = WTVIRC;