Changeset 660 in code for trunk/downstream.go

Timestamp:

Nov 2, 2021, 5:25:43 PM (4 years ago)

Author:

contact

Message:

Add support for WHOX

This adds support for WHOX, without bothering about flags and mask2
because Solanum and Ergo [1] don't support it either.

The motivation is to allow clients to reliably query account names.

It's not possible to use WHOX tokens to route replies to the right
client, because RPL_ENDOFWHO doesn't contain it.

[1]: ​https://github.com/ergochat/ergo/pull/1184

Closes: ​https://todo.sr.ht/~emersion/soju/135

File:

• trunk/downstream.go (modified) (5 diffs)

trunk/downstream.go

@@ -237 +237 @@
         "USERLEN":       true,
         "UTF8ONLY":      true,
+        "WHOX":          true,
 }
 
@@ -1156 +1157 @@
         if dc.network != nil {
                 isupport = append(isupport, fmt.Sprintf("BOUNCER_NETID=%v", dc.network.ID))
+        }
+
+        if dc.network == nil && dc.caps["soju.im/bouncer-networks"] {
+                isupport = append(isupport, "WHOX")
         }
 
@@ -1883 +1888 @@
                         }
                 }
+        // For WHOX docs, see:
+        // - http://faerion.sourceforge.net/doc/irc/whox.var
+        // - https://github.com/quakenet/snircd/blob/master/doc/readme.who
+        // Note, many features aren't widely implemented, such as flags and mask2
         case "WHO":
                 if len(msg.Params) == 0 {
@@ -1894 +1903 @@
                 }
 
-                // TODO: support WHO masks
-                entity := msg.Params[0]
-                entityCM := casemapASCII(entity)
-
-                if dc.network == nil && entityCM == dc.nickCM {
+                // Clients will use the first mask to match RPL_ENDOFWHO
+                endOfWhoToken := msg.Params[0]
+
+                // TODO: add support for WHOX mask2
+                mask := msg.Params[0]
+                var options string
+                if len(msg.Params) > 1 {
+                        options = msg.Params[1]
+                }
+
+                optionsParts := strings.SplitN(options, "%", 2)
+                // TODO: add support for WHOX flags in optionsParts[0]
+                var fields, whoxToken string
+                if len(optionsParts) == 2 {
+                        optionsParts := strings.SplitN(optionsParts[1], ",", 2)
+                        fields = strings.ToLower(optionsParts[0])
+                        if len(optionsParts) == 2 && strings.Contains(fields, "t") {
+                                whoxToken = optionsParts[1]
+                        }
+                }
+
+                // TODO: support mixed bouncer/upstream WHO queries
+                maskCM := casemapASCII(mask)
+                if dc.network == nil && maskCM == dc.nickCM {
                         // TODO: support AWAY (H/G) in self WHO reply
                         flags := "H"
@@ -1904 +1932 @@
                                 flags += "*"
                         }
-                        dc.SendMessage(&irc.Message{
-                                Prefix:  dc.srv.prefix(),
-                                Command: irc.RPL_WHOREPLY,
-                                Params:  []string{dc.nick, "*", dc.user.Username, dc.hostname, dc.srv.Hostname, dc.nick, flags, "0 " + dc.realname},
-                        })
+                        info := whoxInfo{
+                                Token:    whoxToken,
+                                Username: dc.user.Username,
+                                Hostname: dc.hostname,
+                                Server:   dc.srv.Hostname,
+                                Nickname: dc.nick,
+                                Flags:    flags,
+                                Realname: dc.realname,
+                        }
+                        dc.SendMessage(generateWHOXReply(dc.srv.prefix(), dc.nick, fields, &info))
                         dc.SendMessage(&irc.Message{
                                 Prefix:  dc.srv.prefix(),
                                 Command: irc.RPL_ENDOFWHO,
-                                Params:  []string{dc.nick, dc.nick, "End of /WHO list"},
+                                Params:  []string{dc.nick, endOfWhoToken, "End of /WHO list"},
                         })
                         return nil
                 }
-                if entityCM == serviceNickCM {
-                        dc.SendMessage(&irc.Message{
-                                Prefix:  dc.srv.prefix(),
-                                Command: irc.RPL_WHOREPLY,
-                                Params:  []string{serviceNick, "*", servicePrefix.User, servicePrefix.Host, dc.srv.Hostname, serviceNick, "H*", "0 " + serviceRealname},
-                        })
+                if maskCM == serviceNickCM {
+                        info := whoxInfo{
+                                Token:    whoxToken,
+                                Username: servicePrefix.User,
+                                Hostname: servicePrefix.Host,
+                                Server:   dc.srv.Hostname,
+                                Nickname: serviceNick,
+                                Flags:    "H*",
+                                Realname: serviceRealname,
+                        }
+                        dc.SendMessage(generateWHOXReply(dc.srv.prefix(), dc.nick, fields, &info))
                         dc.SendMessage(&irc.Message{
                                 Prefix:  dc.srv.prefix(),
                                 Command: irc.RPL_ENDOFWHO,
-                                Params:  []string{dc.nick, serviceNick, "End of /WHO list"},
+                                Params:  []string{dc.nick, endOfWhoToken, "End of /WHO list"},
                         })
                         return nil
                 }
 
-                uc, upstreamName, err := dc.unmarshalEntity(entity)
+                // TODO: properly support WHO masks
+                uc, upstreamMask, err := dc.unmarshalEntity(mask)
                 if err != nil {
                         return err
                 }
 
-                var params []string
-                if len(msg.Params) == 2 {
-                        params = []string{upstreamName, msg.Params[1]}
-                } else {
-                        params = []string{upstreamName}
+                params := []string{upstreamMask}
+                if options != "" {
+                        params = append(params, options)
                 }