nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/ListGroupsCommand.java

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.impl.Arguments;
   4 import net.sourceforge.argparse4j.inf.Namespace;
   5 import net.sourceforge.argparse4j.inf.Subparser;
   6
   7 import org.asamk.signal.JsonWriter;
   8 import org.asamk.signal.OutputWriter;
   9 import org.asamk.signal.PlainTextWriter;
  10 import org.asamk.signal.commands.exceptions.CommandException;
  11 import org.asamk.signal.manager.Manager;
  12 import org.asamk.signal.manager.api.Group;
  13 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
  14 import org.slf4j.Logger;
  15 import org.slf4j.LoggerFactory;
  16
  17 import java.util.Set;
  18 import java.util.UUID;
  19 import java.util.stream.Collectors;
  20
  21 public class ListGroupsCommand implements JsonRpcLocalCommand {
  22
  23     private final static Logger logger = LoggerFactory.getLogger(ListGroupsCommand.class);
  24
  25     @Override
  26     public String getName() {
  27         return "listGroups";
  28     }
  29
  30     @Override
  31     public void attachToSubparser(final Subparser subparser) {
  32         subparser.help("List group information including names, ids, active status, blocked status and members");
  33         subparser.addArgument("-d", "--detailed")
  34                 .action(Arguments.storeTrue())
  35                 .help("List the members and group invite links of each group. If output=json, then this is always set");
  36     }
  37
  38     private static Set<String> resolveMembers(Set<RecipientAddress> addresses) {
  39         return addresses.stream().map(RecipientAddress::getLegacyIdentifier).collect(Collectors.toSet());
  40     }
  41
  42     private static Set<JsonGroupMember> resolveJsonMembers(Set<RecipientAddress> addresses) {
  43         return addresses.stream()
  44                 .map(address -> new JsonGroupMember(address.getNumber().orElse(null),
  45                         address.getUuid().map(UUID::toString).orElse(null)))
  46                 .collect(Collectors.toSet());
  47     }
  48
  49     private static void printGroupPlainText(
  50             PlainTextWriter writer, Group group, boolean detailed
  51     ) {
  52         if (detailed) {
  53             final var groupInviteLink = group.getGroupInviteLinkUrl();
  54
  55             writer.println(
  56                     "Id: {} Name: {} Description: {} Active: {} Blocked: {} Members: {} Pending members: {} Requesting members: {} Admins: {} Message expiration: {} Link: {}",
  57                     group.getGroupId().toBase64(),
  58                     group.getTitle(),
  59                     group.getDescription(),
  60                     group.isMember(),
  61                     group.isBlocked(),
  62                     resolveMembers(group.getMembers()),
  63                     resolveMembers(group.getPendingMembers()),
  64                     resolveMembers(group.getRequestingMembers()),
  65                     resolveMembers(group.getAdminMembers()),
  66                     group.getMessageExpirationTime() == 0 ? "disabled" : group.getMessageExpirationTime() + "s",
  67                     groupInviteLink == null ? '-' : groupInviteLink.getUrl());
  68         } else {
  69             writer.println("Id: {} Name: {}  Active: {} Blocked: {}",
  70                     group.getGroupId().toBase64(),
  71                     group.getTitle(),
  72                     group.isMember(),
  73                     group.isBlocked());
  74         }
  75     }
  76
  77     @Override
  78     public void handleCommand(
  79             final Namespace ns, final Manager m, final OutputWriter outputWriter
  80     ) throws CommandException {
  81         final var groups = m.getGroups();
  82
  83         if (outputWriter instanceof JsonWriter) {
  84             final var jsonWriter = (JsonWriter) outputWriter;
  85
  86             var jsonGroups = groups.stream().map(group -> {
  87                 final var groupInviteLink = group.getGroupInviteLinkUrl();
  88
  89                 return new JsonGroup(group.getGroupId().toBase64(),
  90                         group.getTitle(),
  91                         group.getDescription(),
  92                         group.isMember(),
  93                         group.isBlocked(),
  94                         group.getMessageExpirationTime(),
  95                         resolveJsonMembers(group.getMembers()),
  96                         resolveJsonMembers(group.getPendingMembers()),
  97                         resolveJsonMembers(group.getRequestingMembers()),
  98                         resolveJsonMembers(group.getAdminMembers()),
  99                         groupInviteLink == null ? null : groupInviteLink.getUrl());
 100             }).collect(Collectors.toList());
 101
 102             jsonWriter.write(jsonGroups);
 103         } else {
 104             final var writer = (PlainTextWriter) outputWriter;
 105             boolean detailed = ns.getBoolean("detailed");
 106             for (var group : groups) {
 107                 printGroupPlainText(writer, group, detailed);
 108             }
 109         }
 110     }
 111
 112     private static final class JsonGroup {
 113
 114         public final String id;
 115         public final String name;
 116         public final String description;
 117         public final boolean isMember;
 118         public final boolean isBlocked;
 119         public final int messageExpirationTime;
 120
 121         public final Set<JsonGroupMember> members;
 122         public final Set<JsonGroupMember> pendingMembers;
 123         public final Set<JsonGroupMember> requestingMembers;
 124         public final Set<JsonGroupMember> admins;
 125         public final String groupInviteLink;
 126
 127         public JsonGroup(
 128                 String id,
 129                 String name,
 130                 String description,
 131                 boolean isMember,
 132                 boolean isBlocked,
 133                 final int messageExpirationTime,
 134                 Set<JsonGroupMember> members,
 135                 Set<JsonGroupMember> pendingMembers,
 136                 Set<JsonGroupMember> requestingMembers,
 137                 Set<JsonGroupMember> admins,
 138                 String groupInviteLink
 139         ) {
 140             this.id = id;
 141             this.name = name;
 142             this.description = description;
 143             this.isMember = isMember;
 144             this.isBlocked = isBlocked;
 145             this.messageExpirationTime = messageExpirationTime;
 146
 147             this.members = members;
 148             this.pendingMembers = pendingMembers;
 149             this.requestingMembers = requestingMembers;
 150             this.admins = admins;
 151             this.groupInviteLink = groupInviteLink;
 152         }
 153     }
 154
 155     private static final class JsonGroupMember {
 156
 157         public final String number;
 158         public final String uuid;
 159
 160         private JsonGroupMember(final String number, final String uuid) {
 161             this.number = number;
 162             this.uuid = uuid;
 163         }
 164     }
 165 }