Changeset 292 in code for trunk/irc.go

Timestamp:

May 11, 2020, 10:25:49 AM (5 years ago)

Author:

delthas

Message:

Add support for multiple user channel memberships

User channel memberships are actually a set of memberships, not a single
value. This introduces memberships, a type representing a set of
memberships, stored as an array of memberships ordered by descending
rank.

This also adds multi-prefix to the permanent downstream and upstream
capabilities, so that we try to get all possible channel memberships.

File:

• trunk/irc.go (modified) (1 diff)

trunk/irc.go

@@ -177 +177 @@
 }
 
-func (m *membership) String() string {
-        if m == nil {
-                return ""
-        }
-        return string(m.Prefix)
+// memberships always sorted by descending membership rank
+type memberships []membership
+
+func (m *memberships) Add(availableMemberships []membership, newMembership membership) {
+        l := *m
+        i := 0
+        for _, availableMembership := range availableMemberships {
+                if i >= len(l) {
+                        break
+                }
+                if l[i] == availableMembership {
+                        if availableMembership == newMembership {
+                                // we already have this membership
+                                return
+                        }
+                        i++
+                        continue
+                }
+                if availableMembership == newMembership {
+                        break
+                }
+        }
+        // insert newMembership at i
+        l = append(l, membership{})
+        copy(l[i+1:], l[i:])
+        l[i] = newMembership
+        *m = l
+}
+
+func (m *memberships) Remove(oldMembership membership) {
+        l := *m
+        for i, currentMembership := range l {
+                if currentMembership == oldMembership {
+                        *m = append(l[:i], l[i+1:]...)
+                        return
+                }
+        }
+}
+
+func (m memberships) Format(dc *downstreamConn) string {
+        if !dc.caps["multi-prefix"] {
+                if len(m) == 0 {
+                        return ""
+                }
+                return string(m[0].Prefix)
+        }
+        prefixes := make([]byte, len(m))
+        for i, membership := range m {
+                prefixes[i] = membership.Prefix
+        }
+        return string(prefixes)
 }