Changeset 127 in code for trunk/downstream.go

Timestamp:

Mar 19, 2020, 11:29:03 PM (5 years ago)

Author:

delthas

Message:

Add WHO support

File:

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

trunk/downstream.go

@@ -170 +170 @@
 }
 
-func (dc *downstreamConn) unmarshalChannel(name string) (*upstreamConn, string, error) {
+func (dc *downstreamConn) unmarshalEntity(name string) (*upstreamConn, string, error) {
         if uc := dc.upstream(); uc != nil {
                 return uc, name, nil
         }
 
-        network := ""
+        var conn *upstreamConn
         if i := strings.LastIndexByte(name, '/'); i >= 0 {
-                network = name[i+1:]
+                network := name[i+1:]
                 name = name[:i]
-        }
-
-        if network != "" {
-                var conn *upstreamConn
+
                 dc.forEachUpstream(func(uc *upstreamConn) {
                         if network != uc.network.GetName() {
@@ -189 +186 @@
                         conn = uc
                 })
-                return conn, name, nil
-        }
-
-        var channel *upstreamChannel
-        var err error
-        dc.forEachUpstream(func(uc *upstreamConn) {
-                if err != nil {
-                        return
-                }
-                if ch, ok := uc.channels[name]; ok {
-                        if channel != nil {
-                                err = fmt.Errorf("ambiguous channel name %q", name)
-                        } else {
-                                channel = ch
-                        }
-                }
-        })
-        if channel == nil {
+        }
+
+        if conn == nil {
                 return nil, "", ircError{&irc.Message{
                         Command: irc.ERR_NOSUCHCHANNEL,
@@ -212 +194 @@
                 }}
         }
-        return channel.conn, channel.Name, nil
+        return conn, name, nil
 }
 
@@ -844 +826 @@
                 }
 
-                uc, upstreamName, err := dc.unmarshalChannel(name)
+                uc, upstreamName, err := dc.unmarshalEntity(name)
                 if err != nil {
                         return ircError{&irc.Message{
@@ -886 +868 @@
 
                 if msg.Prefix.Name != name {
-                        uc, upstreamName, err := dc.unmarshalChannel(name)
+                        uc, upstreamName, err := dc.unmarshalEntity(name)
                         if err != nil {
                                 return err
@@ -934 +916 @@
                         }
                 }
+        case "WHO":
+                if len(msg.Params) == 0 {
+                        // TODO: support WHO without parameters
+                        dc.SendMessage(&irc.Message{
+                                Prefix:  dc.srv.prefix(),
+                                Command: irc.RPL_ENDOFWHO,
+                                Params:  []string{dc.nick, "*", "End of /WHO list."},
+                        })
+                        return nil
+                }
+
+                // TODO: support WHO masks
+                entity := msg.Params[0]
+
+                uc, upstreamName, err := dc.unmarshalEntity(entity)
+                if err != nil {
+                        return err
+                }
+
+                var params []string
+                if len(msg.Params) == 2 {
+                        params = []string{upstreamName, msg.Params[1]}
+                } else {
+                        params = []string{upstreamName}
+                }
+
+                uc.SendMessage(&irc.Message{
+                        Command: "WHO",
+                        Params:  params,
+                })
         case "PRIVMSG":
                 var targetsStr, text string
@@ -946 +958 @@
                         }
 
-                        uc, upstreamName, err := dc.unmarshalChannel(name)
+                        uc, upstreamName, err := dc.unmarshalEntity(name)
                         if err != nil {
                                 return err