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.AttachmentInvalidException;
   8 import org.asamk.signal.GroupIdFormatException;
   9 import org.asamk.signal.GroupNotFoundException;
  10 import org.asamk.signal.NotAGroupMemberException;
  11 import org.asamk.signal.util.Util;
  12 import org.whispersystems.signalservice.api.push.exceptions.EncapsulatedExceptions;
  13 import org.whispersystems.util.Base64;
  14
  15 import java.io.IOException;
  16 import java.util.ArrayList;
  17 import java.util.List;
  18
  19 import static org.asamk.signal.util.ErrorUtils.handleEncapsulatedExceptions;
  20 import static org.asamk.signal.util.ErrorUtils.handleGroupIdFormatException;
  21 import static org.asamk.signal.util.ErrorUtils.handleGroupNotFoundException;
  22 import static org.asamk.signal.util.ErrorUtils.handleIOException;
  23 import static org.asamk.signal.util.ErrorUtils.handleNotAGroupMemberException;
  24
  25 public class UpdateGroupCommand implements DbusCommand {
  26
  27     @Override
  28     public void attachToSubparser(final Subparser subparser) {
  29         subparser.addArgument("-g", "--group")
  30                 .help("Specify the recipient group ID.");
  31         subparser.addArgument("-n", "--name")
  32                 .help("Specify the new group name.");
  33         subparser.addArgument("-a", "--avatar")
  34                 .help("Specify a new group avatar image file");
  35         subparser.addArgument("-m", "--member")
  36                 .nargs("*")
  37                 .help("Specify one or more members to add to the group");
  38     }
  39
  40     @Override
  41     public int handleCommand(final Namespace ns, final Signal signal) {
  42         if (!signal.isRegistered()) {
  43             System.err.println("User is not registered.");
  44             return 1;
  45         }
  46
  47         try {
  48             byte[] groupId = null;
  49             if (ns.getString("group") != null) {
  50                 groupId = Util.decodeGroupId(ns.getString("group"));
  51             }
  52             if (groupId == null) {
  53                 groupId = new byte[0];
  54             }
  55             String groupName = ns.getString("name");
  56             if (groupName == null) {
  57                 groupName = "";
  58             }
  59             List<String> groupMembers = ns.getList("member");
  60             if (groupMembers == null) {
  61                 groupMembers = new ArrayList<>();
  62             }
  63             String groupAvatar = ns.getString("avatar");
  64             if (groupAvatar == null) {
  65                 groupAvatar = "";
  66             }
  67             byte[] newGroupId = signal.updateGroup(groupId, groupName, groupMembers, groupAvatar);
  68             if (groupId.length != newGroupId.length) {
  69                 System.out.println("Creating new group \"" + Base64.encodeBytes(newGroupId) + "\" …");
  70             }
  71             return 0;
  72         } catch (IOException e) {
  73             handleIOException(e);
  74             return 3;
  75         } catch (AttachmentInvalidException e) {
  76             System.err.println("Failed to add avatar attachment for group\": " + e.getMessage());
  77             System.err.println("Aborting sending.");
  78             return 1;
  79         } catch (GroupNotFoundException e) {
  80             handleGroupNotFoundException(e);
  81             return 1;
  82         } catch (NotAGroupMemberException e) {
  83             handleNotAGroupMemberException(e);
  84             return 1;
  85         } catch (EncapsulatedExceptions e) {
  86             handleEncapsulatedExceptions(e);
  87             return 3;
  88         } catch (GroupIdFormatException e) {
  89             handleGroupIdFormatException(e);
  90             return 1;
  91         }
  92     }
  93 }