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

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.inf.Namespace;
   4 import net.sourceforge.argparse4j.inf.Subparser;
   5
   6 import org.asamk.Signal;
   7 import org.asamk.signal.PlainTextWriterImpl;
   8 import org.asamk.signal.manager.groups.GroupIdFormatException;
   9 import org.asamk.signal.util.Util;
  10 import org.freedesktop.dbus.exceptions.DBusExecutionException;
  11 import org.slf4j.Logger;
  12 import org.slf4j.LoggerFactory;
  13
  14 import java.io.IOException;
  15 import java.util.ArrayList;
  16 import java.util.Base64;
  17 import java.util.List;
  18
  19 import static org.asamk.signal.util.ErrorUtils.handleAssertionError;
  20 import static org.asamk.signal.util.ErrorUtils.handleGroupIdFormatException;
  21
  22 public class UpdateGroupCommand implements DbusCommand {
  23
  24     private final static Logger logger = LoggerFactory.getLogger(UpdateGroupCommand.class);
  25
  26     @Override
  27     public void attachToSubparser(final Subparser subparser) {
  28         subparser.addArgument("-g", "--group").help("Specify the recipient group ID.");
  29         subparser.addArgument("-n", "--name").help("Specify the new group name.");
  30         subparser.addArgument("-a", "--avatar").help("Specify a new group avatar image file");
  31         subparser.addArgument("-m", "--member").nargs("*").help("Specify one or more members to add to the group");
  32     }
  33
  34     @Override
  35     public int handleCommand(final Namespace ns, final Signal signal) {
  36         final var writer = new PlainTextWriterImpl(System.out);
  37         byte[] groupId = null;
  38         if (ns.getString("group") != null) {
  39             try {
  40                 groupId = Util.decodeGroupId(ns.getString("group")).serialize();
  41             } catch (GroupIdFormatException e) {
  42                 handleGroupIdFormatException(e);
  43                 return 1;
  44             }
  45         }
  46         if (groupId == null) {
  47             groupId = new byte[0];
  48         }
  49
  50         var groupName = ns.getString("name");
  51         if (groupName == null) {
  52             groupName = "";
  53         }
  54
  55         List<String> groupMembers = ns.getList("member");
  56         if (groupMembers == null) {
  57             groupMembers = new ArrayList<>();
  58         }
  59
  60         var groupAvatar = ns.getString("avatar");
  61         if (groupAvatar == null) {
  62             groupAvatar = "";
  63         }
  64
  65         try {
  66             var newGroupId = signal.updateGroup(groupId, groupName, groupMembers, groupAvatar);
  67             if (groupId.length != newGroupId.length) {
  68                 try {
  69                     writer.println("Creating new group \"{}\" …", Base64.getEncoder().encodeToString(newGroupId));
  70                 } catch (IOException e) {
  71                     e.printStackTrace();
  72                     return 3;
  73                 }
  74             }
  75             return 0;
  76         } catch (AssertionError e) {
  77             handleAssertionError(e);
  78             return 1;
  79         } catch (Signal.Error.AttachmentInvalid e) {
  80             System.err.println("Failed to add avatar attachment for group\": " + e.getMessage());
  81             return 1;
  82         } catch (DBusExecutionException e) {
  83             System.err.println("Failed to send message: " + e.getMessage());
  84             return 2;
  85         }
  86     }
  87 }