From 0251dda8add24c5c43b100d9a9fcdd324c4a403c Mon Sep 17 00:00:00 2001
From: Sheogorath <sheogorath@shivering-isles.com>
Date: Wed, 17 Jul 2019 21:45:29 +0200
Subject: [PATCH 1/2] Add sendMsg function to channel class

This function implements sending messages to all users in a channel
while sending the channel name as target to the user's socket.
---
 src/channel.js  |  7 +++++++
 test/channel.js | 17 +++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/channel.js b/src/channel.js
index cf0eddf..b6cb5ee 100644
--- a/src/channel.js
+++ b/src/channel.js
@@ -20,6 +20,13 @@ function Channel(name) {
           this.userlist.splice(index, 1)
       }
   }
+
+  this.sendMsg = function(from, message) {
+      const channel = this
+      this.userlist.forEach(function(item) {
+          item.sendMsg(from, message, channel.name)
+      })
+  }
 }
 
 module.exports = Channel
diff --git a/test/channel.js b/test/channel.js
index 68accce..870a640 100644
--- a/test/channel.js
+++ b/test/channel.js
@@ -38,4 +38,21 @@ describe('Channel', function () {
             channel.part(mockedUser)
         })
     })
+
+    describe('#sendMsg(from, message)', function () {
+        it("should remove a user from a channel and send a PART command to the user", function (done) {
+            const channel = new Channel("#testchan")
+            let mockedUser = {nickname: "some_nick", sendRaw: function() {}}
+            mockedUser.sendMsg = function(from, message, to) {
+                assert.equal(from, mockedUser)
+                assert.equal(message, "test message")
+                assert.equal(to, "#testchan")
+                done()
+            }
+            // we can't part a channel without joining it first
+            channel.join(mockedUser)
+
+            channel.sendMsg(mockedUser, "test message")
+        })
+    })
 })

From b0d5e41bd9f3675ef8442c0d0e99f92639110d74 Mon Sep 17 00:00:00 2001
From: Sheogorath <sheogorath@shivering-isles.com>
Date: Wed, 17 Jul 2019 22:37:14 +0200
Subject: [PATCH 2/2] Add ability to send messages to channel

---
 src/server.js           | 17 ++++++++++++-----
 test/command_privmsg.js | 24 ++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 5 deletions(-)

diff --git a/src/server.js b/src/server.js
index 0d12a46..6ee62f9 100644
--- a/src/server.js
+++ b/src/server.js
@@ -74,15 +74,22 @@ server.create = function create(config = {}) {
                         socket.write(`:irc.example.com 001 ${user.nickname} :Welcome to the example IRC Project ${user.nickname}!~${user.username}@${address}\r\n`, "ascii")
                         break;
                     case "PRIVMSG":
-                        let target = userlist[tokenized[1]]
+                        let target
                         let message = tokenized[2]
-                        if (target.registered === false) {
-                            socket.write(`:irc.example.com ${ERR_NOSUCHNICK} ${target} no such nick/channel`)
-
+                        if (typeof tokenized[1] === 'string' && tokenized[1].charAt(0) === "#") {
+                            target = channellist[tokenized[1]]
                         } else {
+                            let targetUser = userlist[tokenized[1]]
+                            if (targetUser && targetUser.registered) {
+                                target = targetUser
+                            }
+                        }
+                        if (target) {
                             target.sendMsg(user, message)
-
+                        } else {
+                            user.sendRaw(`:irc.example.com ${ERR_NOSUCHNICK} ${target} no such nick/channel`)
                         }
+
                         break;
                     case "JOIN":
                         let channelname = tokenized[1]
diff --git a/test/command_privmsg.js b/test/command_privmsg.js
index 10df41e..d1d433f 100644
--- a/test/command_privmsg.js
+++ b/test/command_privmsg.js
@@ -101,4 +101,28 @@ describe("PRIVMSG OK", function () {
 
         mockedSock1.emit('data', Buffer.from("PRIVMSG other_nick :I'm a message\r\n", "ascii"))
     })
+
+    it("should allow to send message to channel -> PRIVMSG #testchan :I'm a message", function (done) {
+        const server = IRCServer.create()
+        let mockedSock = new EventEmitter()
+        mockedSock.address = function () {
+            return {port: 12346, family: 'IPv4', address: '127.0.0.1'}
+        }
+        mockedSock.write = function (data) {
+            let answer = data.toString("ascii")
+            if (answer.indexOf("PRIVMSG") >= 0) {
+                assert.equal(answer, ":some_nick PRIVMSG #testchan :I'm a message\r\n")
+                done()
+            }
+        }
+        mockedSock.destroy = function () {
+            done("Destroyed socket without answering")
+        }
+
+        server.emit("connection", mockedSock)
+        mockedSock.emit('data', Buffer.from("NICK some_nick\r\n", "ascii"))
+        mockedSock.emit('data', Buffer.from("USER guest tolmoon tolsun :Ronnie Reagan\r\n", "ascii"))
+        mockedSock.emit('data', Buffer.from("JOIN #testchan\r\n", "ascii"))
+        mockedSock.emit('data', Buffer.from("PRIVMSG #testchan :I'm a message\r\n", "ascii"))
+    })
 })