nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/RemoteDeleteCommand.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;
   8 import org.asamk.signal.JsonWriter;
   9 import org.asamk.signal.OutputWriter;
  10 import org.asamk.signal.PlainTextWriter;
  11 import org.asamk.signal.commands.exceptions.CommandException;
  12 import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
  13 import org.asamk.signal.commands.exceptions.UserErrorException;
  14 import org.asamk.signal.manager.Manager;
  15 import org.asamk.signal.manager.groups.GroupNotFoundException;
  16 import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
  17 import org.asamk.signal.manager.groups.NotAGroupMemberException;
  18 import org.asamk.signal.util.CommandUtil;
  19 import org.asamk.signal.util.ErrorUtils;
  20 import org.freedesktop.dbus.errors.UnknownObject;
  21 import org.freedesktop.dbus.exceptions.DBusExecutionException;
  22
  23 import java.io.IOException;
  24 import java.util.Map;
  25
  26 public class RemoteDeleteCommand implements DbusCommand, JsonRpcLocalCommand {
  27
  28     @Override
  29     public String getName() {
  30         return "remoteDelete";
  31     }
  32
  33     @Override
  34     public void attachToSubparser(final Subparser subparser) {
  35         subparser.help("Remotely delete a previously sent message.");
  36         subparser.addArgument("-t", "--target-timestamp")
  37                 .required(true)
  38                 .type(long.class)
  39                 .help("Specify the timestamp of the message to delete.");
  40         subparser.addArgument("-g", "--group-id", "--group").help("Specify the recipient group ID.").nargs("*");
  41         subparser.addArgument("recipient").help("Specify the recipients' phone number.").nargs("*");
  42         subparser.addArgument("--note-to-self").action(Arguments.storeTrue());
  43     }
  44
  45     @Override
  46     public void handleCommand(
  47             final Namespace ns, final Manager m, final OutputWriter outputWriter
  48     ) throws CommandException {
  49         final var isNoteToSelf = ns.getBoolean("note-to-self");
  50         final var recipientStrings = ns.<String>getList("recipient");
  51         final var groupIdStrings = ns.<String>getList("group-id");
  52
  53         final var recipientIdentifiers = CommandUtil.getRecipientIdentifiers(m,
  54                 isNoteToSelf,
  55                 recipientStrings,
  56                 groupIdStrings);
  57
  58         final long targetTimestamp = ns.getLong("target-timestamp");
  59
  60         try {
  61             final var results = m.sendRemoteDeleteMessage(targetTimestamp, recipientIdentifiers);
  62             outputResult(outputWriter, results.getTimestamp());
  63             ErrorUtils.handleSendMessageResults(results.getResults());
  64         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
  65             throw new UserErrorException(e.getMessage());
  66         } catch (IOException e) {
  67             throw new UnexpectedErrorException("Failed to send message: " + e.getMessage());
  68         }
  69     }
  70
  71     @Override
  72     public void handleCommand(
  73             final Namespace ns, final Signal signal, final OutputWriter outputWriter
  74     ) throws CommandException {
  75         final var recipients = ns.<String>getList("recipient");
  76         final var groupIdStrings = ns.<String>getList("group-id");
  77
  78         final var noRecipients = recipients == null || recipients.isEmpty();
  79         final var noGroups = groupIdStrings == null || groupIdStrings.isEmpty();
  80         if (noRecipients && noGroups) {
  81             throw new UserErrorException("No recipients given");
  82         }
  83         if (!noRecipients && !noGroups) {
  84             throw new UserErrorException("You cannot specify recipients by phone number and groups at the same time");
  85         }
  86
  87         final long targetTimestamp = ns.getLong("target-timestamp");
  88
  89         try {
  90             long timestamp = 0;
  91             if (!noGroups) {
  92                 final var groupIds = CommandUtil.getGroupIds(groupIdStrings);
  93                 for (final var groupId : groupIds) {
  94                     timestamp = signal.sendGroupRemoteDeleteMessage(targetTimestamp, groupId.serialize());
  95                 }
  96             } else {
  97                 timestamp = signal.sendRemoteDeleteMessage(targetTimestamp, recipients);
  98             }
  99             outputResult(outputWriter, timestamp);
 100         } catch (UnknownObject e) {
 101             throw new UserErrorException("Failed to find dbus object, maybe missing the -u flag: " + e.getMessage());
 102         } catch (Signal.Error.InvalidNumber e) {
 103             throw new UserErrorException("Invalid number: " + e.getMessage());
 104         } catch (Signal.Error.GroupNotFound e) {
 105             throw new UserErrorException("Failed to send to group: " + e.getMessage());
 106         } catch (DBusExecutionException e) {
 107             throw new UnexpectedErrorException("Failed to send message: " + e.getMessage());
 108         }
 109     }
 110
 111     private void outputResult(final OutputWriter outputWriter, final long timestamp) {
 112         if (outputWriter instanceof PlainTextWriter) {
 113             final var writer = (PlainTextWriter) outputWriter;
 114             writer.println("{}", timestamp);
 115         } else {
 116             final var writer = (JsonWriter) outputWriter;
 117             writer.write(Map.of("timestamp", timestamp));
 118         }
 119     }
 120 }