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.OutputType;
   9 import org.asamk.signal.PlainTextWriter;
  10 import org.asamk.signal.PlainTextWriterImpl;
  11 import org.asamk.signal.commands.exceptions.CommandException;
  12 import org.asamk.signal.manager.Manager;
  13 import org.asamk.signal.manager.storage.groups.GroupInfo;
  14 import org.asamk.signal.manager.storage.recipients.RecipientId;
  15 import org.slf4j.Logger;
  16 import org.slf4j.LoggerFactory;
  17 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  18
  19 import java.util.ArrayList;
  20 import java.util.Set;
  21 import java.util.stream.Collectors;
  22
  23 public class ListGroupsCommand implements LocalCommand {
  24
  25     private final static Logger logger = LoggerFactory.getLogger(ListGroupsCommand.class);
  26
  27     private static Set<String> resolveMembers(Manager m, Set<RecipientId> addresses) {
  28         return addresses.stream()
  29                 .map(m::resolveSignalServiceAddress)
  30                 .map(SignalServiceAddress::getLegacyIdentifier)
  31                 .collect(Collectors.toSet());
  32     }
  33
  34     private static void printGroupPlainText(
  35             PlainTextWriter writer, Manager m, GroupInfo group, boolean detailed
  36     ) {
  37         if (detailed) {
  38             final var groupInviteLink = group.getGroupInviteLink();
  39
  40             writer.println(
  41                     "Id: {} Name: {} Description: {} Active: {} Blocked: {} Members: {} Pending members: {} Requesting members: {} Link: {}",
  42                     group.getGroupId().toBase64(),
  43                     group.getTitle(),
  44                     group.getDescription(),
  45                     group.isMember(m.getSelfRecipientId()),
  46                     group.isBlocked(),
  47                     resolveMembers(m, group.getMembers()),
  48                     resolveMembers(m, group.getPendingMembers()),
  49                     resolveMembers(m, group.getRequestingMembers()),
  50                     groupInviteLink == null ? '-' : groupInviteLink.getUrl());
  51         } else {
  52             writer.println("Id: {} Name: {}  Active: {} Blocked: {}",
  53                     group.getGroupId().toBase64(),
  54                     group.getTitle(),
  55                     group.isMember(m.getSelfRecipientId()),
  56                     group.isBlocked());
  57         }
  58     }
  59
  60     @Override
  61     public void attachToSubparser(final Subparser subparser) {
  62         subparser.addArgument("-d", "--detailed")
  63                 .action(Arguments.storeTrue())
  64                 .help("List the members and group invite links of each group. If output=json, then this is always set");
  65
  66         subparser.help("List group information including names, ids, active status, blocked status and members");
  67     }
  68
  69     @Override
  70     public Set<OutputType> getSupportedOutputTypes() {
  71         return Set.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  72     }
  73
  74     @Override
  75     public void handleCommand(final Namespace ns, final Manager m) throws CommandException {
  76         if (ns.get("output") == OutputType.JSON) {
  77             final var jsonWriter = new JsonWriter(System.out);
  78
  79             var jsonGroups = new ArrayList<JsonGroup>();
  80             for (var group : m.getGroups()) {
  81                 final var groupInviteLink = group.getGroupInviteLink();
  82
  83                 jsonGroups.add(new JsonGroup(group.getGroupId().toBase64(),
  84                         group.getTitle(),
  85                         group.getDescription(),
  86                         group.isMember(m.getSelfRecipientId()),
  87                         group.isBlocked(),
  88                         resolveMembers(m, group.getMembers()),
  89                         resolveMembers(m, group.getPendingMembers()),
  90                         resolveMembers(m, group.getRequestingMembers()),
  91                         groupInviteLink == null ? null : groupInviteLink.getUrl()));
  92             }
  93
  94             jsonWriter.write(jsonGroups);
  95         } else {
  96             final var writer = new PlainTextWriterImpl(System.out);
  97             boolean detailed = ns.getBoolean("detailed");
  98             for (var group : m.getGroups()) {
  99                 printGroupPlainText(writer, m, group, detailed);
 100             }
 101         }
 102     }
 103
 104     private static final class JsonGroup {
 105
 106         public String id;
 107         public String name;
 108         public String description;
 109         public boolean isMember;
 110         public boolean isBlocked;
 111
 112         public Set<String> members;
 113         public Set<String> pendingMembers;
 114         public Set<String> requestingMembers;
 115         public String groupInviteLink;
 116
 117         public JsonGroup(
 118                 String id,
 119                 String name,
 120                 String description,
 121                 boolean isMember,
 122                 boolean isBlocked,
 123                 Set<String> members,
 124                 Set<String> pendingMembers,
 125                 Set<String> requestingMembers,
 126                 String groupInviteLink
 127         ) {
 128             this.id = id;
 129             this.name = name;
 130             this.description = description;
 131             this.isMember = isMember;
 132             this.isBlocked = isBlocked;
 133
 134             this.members = members;
 135             this.pendingMembers = pendingMembers;
 136             this.requestingMembers = requestingMembers;
 137             this.groupInviteLink = groupInviteLink;
 138         }
 139     }
 140 }