From 33f4f9b24bfd6d4cca2cd0a726ea6c9cd8b9610b Mon Sep 17 00:00:00 2001
From: zefie <zefie@zefie.net>
Date: Fri, 13 Jun 2025 12:33:59 -0400
Subject: [PATCH] allow part/quit messages

---
 zefie_wtvp_minisrv/includes/classes/WTVIRC.js | 30 +++++++++++++++++--
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/zefie_wtvp_minisrv/includes/classes/WTVIRC.js b/zefie_wtvp_minisrv/includes/classes/WTVIRC.js
index 393119fd..c33a9a51 100644
--- a/zefie_wtvp_minisrv/includes/classes/WTVIRC.js
+++ b/zefie_wtvp_minisrv/includes/classes/WTVIRC.js
@@ -6,7 +6,7 @@ class WTVIRC {
         * 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, MODE, KICK, and PING.
-        * TODO: Enforce Bans, channel mode support, enforce invite only channel mode.
+        * TODO: Proper channel mode support, implement invite, enforce invite only channel mode.
     */ 
     constructor(minisrv_config, host = 'localhost', port = 6667, debug = false) {
         this.minisrv_config = minisrv_config;
@@ -433,8 +433,17 @@ class WTVIRC {
                                 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 (params.length == 2) {
+                                let reason = params.join(' ');
+                                if (reason.startsWith(':')) {
+                                    reason = reason.slice(1);
+                                }
+                                socket.write(`:${nickname}!${username}@${host} PART ${channel} :${reason}\r\n`);
+                                this.broadcastUser(nickname, `:${nickname}!${username}@${host} PART ${channel} :${reason}\r\n`, socket);
+                            } else {
+                                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) {
@@ -565,6 +574,21 @@ class WTVIRC {
                             }
                             break;
                         case 'QUIT':
+                            if (!registered) {
+                                socket.write(`:${this.servername} 451 ${nickname} :You have not registered\r\n`);
+                            } else {
+                                if (params.length > 0) {
+                                    let reason = params.join(' ');
+                                    if (reason.startsWith(':')) {
+                                        reason = reason.slice(1);
+                                    }
+                                    socket.write(`:${nickname}!${username}@${host} QUIT :${reason}\r\n`);
+                                    this.broadcastUser(nickname, `:${nickname}!${username}@${host} QUIT :${reason}\r\n`, socket);
+                                } else {
+                                    socket.write(`:${nickname}!${username}@${host} QUIT\r\n`);
+                                    this.broadcastUser(nickname, `:${nickname}!${username}@${host} QUIT\r\n`, socket);
+                                }
+                            }
                             socket.end();
                             break;
                         default: