Changeset 540 in code for trunk/downstream.go

Timestamp:

May 25, 2021, 6:24:45 PM (4 years ago)

Author:

contact

Message:

Add support for IRCv3 setname

References: ​https://todo.sr.ht/~emersion/soju/41

File:

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

trunk/downstream.go

@@ -123 +123 @@
         "sasl":          "PLAIN",
         "server-time":   "",
+        "setname":       "",
 
         "soju.im/bouncer-networks":        "",
@@ -155 +156 @@
         "MAXTARGETS": true,
         "MODES":      true,
+        "NAMELEN":    true,
         "NETWORK":    true,
         "NICKLEN":    true,
@@ -380 +382 @@
         if msg.Command == "JOIN" && !dc.caps["extended-join"] {
                 msg.Params = msg.Params[:1]
+        }
+        if msg.Command == "SETNAME" && !dc.caps["setname"] {
+                return
         }
 
@@ -460 +465 @@
         case "TOPIC":
                 msg.Params[0] = dc.marshalEntity(net, msg.Params[0])
-        case "QUIT":
+        case "QUIT", "SETNAME":
                 // This space is intentionally left blank
         default:
@@ -867 +872 @@
                 dc.nick = uc.nick
                 dc.nickCM = casemapASCII(dc.nick)
+        }
+}
+
+func (dc *downstreamConn) updateRealname() {
+        if uc := dc.upstream(); uc != nil && uc.realname != dc.realname && dc.caps["setname"] {
+                dc.SendMessage(&irc.Message{
+                        Prefix:  dc.prefix(),
+                        Command: "SETNAME",
+                        Params:  []string{uc.realname},
+                })
+                dc.realname = uc.realname
         }
 }
@@ -1063 +1079 @@
 
         dc.updateNick()
+        dc.updateRealname()
         dc.updateSupportedCaps()
 
@@ -1343 +1360 @@
                         dc.nick = nick
                         dc.nickCM = casemapASCII(dc.nick)
+                }
+        case "SETNAME":
+                var realname string
+                if err := parseMessageParams(msg, &realname); err != nil {
+                        return err
+                }
+
+                var storeErr error
+                var needUpdate []Network
+                dc.forEachNetwork(func(n *network) {
+                        // We only need to call updateNetwork for upstreams that don't
+                        // support setname
+                        if uc := n.conn; uc != nil && uc.caps["setname"] {
+                                uc.SendMessageLabeled(dc.id, &irc.Message{
+                                        Command: "SETNAME",
+                                        Params:  []string{realname},
+                                })
+
+                                n.Realname = realname
+                                if err := dc.srv.db.StoreNetwork(dc.user.ID, &n.Network); err != nil {
+                                        dc.logger.Printf("failed to store network realname: %v", err)
+                                        storeErr = err
+                                }
+                                return
+                        }
+
+                        record := n.Network // copy network record because we'll mutate it
+                        record.Realname = realname
+                        needUpdate = append(needUpdate, record)
+                })
+
+                // Walk the network list as a second step, because updateNetwork
+                // mutates the original list
+                for _, record := range needUpdate {
+                        if _, err := dc.user.updateNetwork(&record); err != nil {
+                                dc.logger.Printf("failed to update network realname: %v", err)
+                                storeErr = err
+                        }
+                }
+                if storeErr != nil {
+                        return ircError{&irc.Message{
+                                Command: "FAIL",
+                                Params:  []string{"SETNAME", "CANNOT_CHANGE_REALNAME", "Failed to update realname"},
+                        }}
+                }
+
+                if dc.upstream() == nil && dc.caps["setname"] {
+                        dc.SendMessage(&irc.Message{
+                                Prefix:  dc.prefix(),
+                                Command: "SETNAME",
+                                Params:  []string{realname},
+                        })
                 }
         case "JOIN":