From f12f10a7bb2164d3f1509a857fc841ddc95eb11b Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Wed, 5 Apr 2023 10:54:43 +0200
Subject: [PATCH] downstream: return channel in cached WHO reply

References: https://github.com/ircdocs/modern-irc/pull/211
---
 downstream.go |  3 +++
 xirc/whox.go  | 16 ++++++++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/downstream.go b/downstream.go
index bdcc8a2..b091e76 100644
--- a/downstream.go
+++ b/downstream.go
@@ -2255,6 +2255,9 @@ func (dc *downstreamConn) handleMessageRegistered(ctx context.Context, msg *irc.
 					Account:  uu.Account,
 					Realname: uu.Realname,
 				}
+				if uc.isChannel(mask) {
+					info.Channel = mask
+				}
 				dc.SendMessage(xirc.GenerateWHOXReply(dc.srv.prefix(), dc.nick, fields, &info))
 			}
 			dc.SendMessage(&irc.Message{
diff --git a/xirc/whox.go b/xirc/whox.go
index 0f7b537..a9eb1e8 100644
--- a/xirc/whox.go
+++ b/xirc/whox.go
@@ -13,6 +13,7 @@ var whoxFields = []byte("tcuihsnfdlaor")
 
 type WHOXInfo struct {
 	Token    string
+	Channel  string
 	Username string
 	Hostname string
 	Server   string
@@ -27,7 +28,11 @@ func (info *WHOXInfo) get(k byte) string {
 	case 't':
 		return info.Token
 	case 'c':
-		return "*"
+		channel := info.Channel
+		if channel == "" {
+			channel = "*"
+		}
+		return channel
 	case 'u':
 		return info.Username
 	case 'i':
@@ -69,6 +74,8 @@ func (info *WHOXInfo) set(k byte, v string) {
 	switch k {
 	case 't':
 		info.Token = v
+	case 'c':
+		info.Channel = v
 	case 'u':
 		info.Username = v
 	case 'h':
@@ -96,10 +103,15 @@ func GenerateWHOXReply(prefix *irc.Prefix, nick, fields string, info *WHOXInfo)
 			hostname = "0" + hostname
 		}
 
+		channel := info.Channel
+		if channel == "" {
+			channel = "*"
+		}
+
 		return &irc.Message{
 			Prefix:  prefix,
 			Command: irc.RPL_WHOREPLY,
-			Params:  []string{nick, "*", info.Username, hostname, info.Server, info.Nickname, info.Flags, "0 " + info.Realname},
+			Params:  []string{nick, channel, info.Username, hostname, info.Server, info.Nickname, info.Flags, "0 " + info.Realname},
 		}
 	}