Changeset 729 in code for trunk/upstream.go

Timestamp:

Nov 30, 2021, 11:02:54 AM (4 years ago)

Author:

contact

Message:

Add support for draft/account-registration proxying

This adds support for the draft/account-registration extension [1].
This allows downstreams to register on upstream networks.

[1]: ​https://ircv3.net/specs/extensions/account-registration

File:

• trunk/upstream.go (modified) (6 diffs)

trunk/upstream.go

@@ -36 +36 @@
         "setname":          true,
 
-        "draft/extended-monitor": true,
+        "draft/account-registration": true,
+        "draft/extended-monitor":     true,
 }
 
@@ -301 +302 @@
                                         Params:  []string{dc.nick, "SASL authentication aborted"},
                                 })
+                        case "REGISTER", "VERIFY":
+                                dc.SendMessage(&irc.Message{
+                                        Prefix:  dc.srv.prefix(),
+                                        Command: "FAIL",
+                                        Params:  []string{pendingCmd.msg.Command, "TEMPORARILY_UNAVAILABLE", pendingCmd.msg.Params[0], "Command aborted"},
+                                })
                         default:
                                 panic(fmt.Errorf("Unsupported pending command %q", pendingCmd.msg.Command))
@@ -319 +326 @@
 func (uc *upstreamConn) enqueueCommand(dc *downstreamConn, msg *irc.Message) {
         switch msg.Command {
-        case "LIST", "WHO", "AUTHENTICATE":
+        case "LIST", "WHO", "AUTHENTICATE", "REGISTER", "VERIFY":
                 // Supported
         default:
@@ -633 +640 @@
                                 Params:  []string{"END"},
                         })
+                }
+        case "REGISTER", "VERIFY":
+                if dc, cmd := uc.dequeueCommand(msg.Command); dc != nil {
+                        if msg.Command == "REGISTER" {
+                                var account, password string
+                                if err := parseMessageParams(msg, nil, &account); err != nil {
+                                        return err
+                                }
+                                if err := parseMessageParams(cmd, nil, nil, &password); err != nil {
+                                        return err
+                                }
+                                uc.network.autoSaveSASLPlain(context.TODO(), account, password)
+                        }
+
+                        dc.SendMessage(msg)
                 }
         case irc.RPL_WELCOME:
@@ -1570 +1592 @@
                 }
 
-                if command == "LIST" || command == "WHO" {
-                        dc, _ := uc.dequeueCommand(command)
-                        if dc != nil && downstreamID == 0 {
-                                downstreamID = dc.id
-                        }
+                if dc, _ := uc.dequeueCommand(command); dc != nil && downstreamID == 0 {
+                        downstreamID = dc.id
                 }
 
@@ -1583 +1602 @@
                                 Params:  []string{dc.nick, command, reason},
                         })
+                })
+        case "FAIL":
+                var command string
+                if err := parseMessageParams(msg, &command); err != nil {
+                        return err
+                }
+
+                if dc, _ := uc.dequeueCommand(command); dc != nil && downstreamID == 0 {
+                        downstreamID = dc.id
+                }
+
+                uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
+                        dc.SendMessage(msg)
                 })
         case "ACK":