From 0706d45aba3f2892fd389daac5c520ded625f36d Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Fri, 7 Feb 2020 11:01:53 +0100
Subject: [PATCH] Handle channel mode changes

---
 upstream.go | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/upstream.go b/upstream.go
index e063340..7e1862b 100644
--- a/upstream.go
+++ b/upstream.go
@@ -18,6 +18,7 @@ type upstreamChannel struct {
 	TopicWho  string
 	TopicTime time.Time
 	Status    channelStatus
+	modes     modeSet
 	Members   map[string]membership
 	complete  bool
 }
@@ -105,10 +106,29 @@ func (c *upstreamConn) handleMessage(msg *irc.Message) error {
 		if len(msg.Params) < 2 {
 			return newNeedMoreParamsError(msg.Command)
 		}
-		if nick := msg.Params[0]; nick != c.upstream.Nick {
-			return fmt.Errorf("received MODE message for unknow nick %q", nick)
+		name := msg.Params[0]
+		modeStr := msg.Params[1]
+
+		if name == msg.Prefix.Name { // user mode change
+			if name != c.upstream.Nick {
+				return fmt.Errorf("received MODE message for unknow nick %q", name)
+			}
+			return c.modes.Apply(modeStr)
+		} else { // channel mode change
+			ch, err := c.getChannel(name)
+			if err != nil {
+				return err
+			}
+			if err := ch.modes.Apply(modeStr); err != nil {
+				return err
+			}
+
+			c.srv.lock.Lock()
+			for _, dc := range c.srv.downstreamConns {
+				dc.messages <- msg
+			}
+			c.srv.lock.Unlock()
 		}
-		return c.modes.Apply(msg.Params[1])
 	case "NOTICE":
 		c.logger.Print(msg)
 	case irc.RPL_WELCOME: