Changeset 276 in code for trunk/upstream.go

Timestamp:

Apr 29, 2020, 5:34:44 PM (5 years ago)

Author:

contact

Message:

Add support for away-notify

This makes use of cap-notify to dynamically advertise support for
away-notify. away-notify is advertised to downstream connections if all
upstreams support it.

File:

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

trunk/upstream.go

@@ -47 +47 @@
         modes      userModes
         channels   map[string]*upstreamChannel
-        caps       map[string]string
+        caps       map[string]string // available capabilities
         batches    map[string]batch
         away       bool
 
-        tagsSupported   bool
-        labelsSupported bool
-        nextLabelID     uint64
+        tagsSupported       bool
+        awayNotifySupported bool
+        labelsSupported     bool
+        nextLabelID         uint64
 
         saslClient  sasl.Client
@@ -318 +319 @@
 
                         requestCaps := make([]string, 0, 16)
-                        for _, c := range []string{"message-tags", "batch", "labeled-response", "server-time"} {
+                        for _, c := range []string{"message-tags", "batch", "labeled-response", "server-time", "away-notify"} {
                                 if _, ok := uc.caps[c]; ok {
                                         requestCaps = append(requestCaps, c)
@@ -450 +451 @@
                 uc.registered = true
                 uc.logger.Printf("connection registered")
+
+                uc.forEachDownstream(func(dc *downstreamConn) {
+                        dc.updateSupportedCaps()
+                })
 
                 for _, ch := range uc.network.channels {
@@ -1149 +1154 @@
                         })
                 })
+        case "AWAY":
+                if msg.Prefix == nil {
+                        return fmt.Errorf("expected a prefix")
+                }
+
+                uc.forEachDownstream(func(dc *downstreamConn) {
+                        if !dc.caps["away-notify"] {
+                                return
+                        }
+                        dc.SendMessage(&irc.Message{
+                                Prefix:  dc.marshalUserPrefix(uc.network, msg.Prefix),
+                                Command: "AWAY",
+                                Params:  msg.Params,
+                        })
+                })
         case "TAGMSG":
                 // TODO: relay to downstream connections that accept message-tags
@@ -1263 +1283 @@
 
 func (uc *upstreamConn) handleCapAck(name string, ok bool) error {
-        auth := &uc.network.SASL
         switch name {
         case "sasl":
@@ -1271 +1290 @@
                 }
 
+                auth := &uc.network.SASL
                 switch auth.Mechanism {
                 case "PLAIN":
@@ -1287 +1307 @@
         case "labeled-response":
                 uc.labelsSupported = ok
+        case "away-notify":
+                uc.awayNotifySupported = ok
         case "batch", "server-time":
                 // Nothing to do