]> nmode's Git Repositories - signal-cli/blobdiff - src/main/java/org/asamk/signal/storage/groups/GroupInfo.java
Move group classes to separate package
[signal-cli] / src / main / java / org / asamk / signal / storage / groups / GroupInfo.java
index d24a2694251ef85215559c24d544f85ab7d4ff13..fe72514172c8b3a945e6c4ed8aad2e690b19b285 100644 (file)
@@ -1,25 +1,26 @@
 package org.asamk.signal.storage.groups;
 
 import com.fasterxml.jackson.annotation.JsonIgnore;
-import com.fasterxml.jackson.annotation.JsonProperty;
 
+import org.asamk.signal.manager.groups.GroupId;
+import org.asamk.signal.manager.groups.GroupInviteLinkUrl;
 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
 
-import java.util.HashSet;
 import java.util.Set;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 public abstract class GroupInfo {
 
-    @JsonProperty
-    public final byte[] groupId;
-
-    public GroupInfo(byte[] groupId) {
-        this.groupId = groupId;
-    }
+    @JsonIgnore
+    public abstract GroupId getGroupId();
 
     @JsonIgnore
     public abstract String getTitle();
 
+    @JsonIgnore
+    public abstract GroupInviteLinkUrl getGroupInviteLink();
+
     @JsonIgnore
     public abstract Set<SignalServiceAddress> getMembers();
 
@@ -44,18 +45,29 @@ public abstract class GroupInfo {
 
     @JsonIgnore
     public Set<SignalServiceAddress> getMembersWithout(SignalServiceAddress address) {
-        Set<SignalServiceAddress> members = new HashSet<>();
+        return getMembers().stream().filter(member -> !member.matches(address)).collect(Collectors.toSet());
+    }
+
+    @JsonIgnore
+    public Set<SignalServiceAddress> getMembersIncludingPendingWithout(SignalServiceAddress address) {
+        return Stream.concat(getMembers().stream(), getPendingMembers().stream())
+                .filter(member -> !member.matches(address))
+                .collect(Collectors.toSet());
+    }
+
+    @JsonIgnore
+    public boolean isMember(SignalServiceAddress address) {
         for (SignalServiceAddress member : getMembers()) {
-            if (!member.matches(address)) {
-                members.add(member);
+            if (member.matches(address)) {
+                return true;
             }
         }
-        return members;
+        return false;
     }
 
     @JsonIgnore
-    public boolean isMember(SignalServiceAddress address) {
-        for (SignalServiceAddress member : getMembers()) {
+    public boolean isPendingMember(SignalServiceAddress address) {
+        for (SignalServiceAddress member : getPendingMembers()) {
             if (member.matches(address)) {
                 return true;
             }