nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/SendReactionCommand.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 SendReactionCommand implements DbusCommand, JsonRpcLocalCommand {
  27
  28     @Override
  29     public String getName() {
  30         return "sendReaction";
  31     }
  32
  33     @Override
  34     public void attachToSubparser(final Subparser subparser) {
  35         subparser.help("Send reaction to a previously received or sent message.");
  36         subparser.addArgument("-g", "--group-id", "--group").help("Specify the recipient group ID.").nargs("*");
  37         subparser.addArgument("recipient").help("Specify the recipients' phone number.").nargs("*");
  38         subparser.addArgument("--note-to-self")
  39                 .help("Send the reaction to self without notification.")
  40                 .action(Arguments.storeTrue());
  41         subparser.addArgument("-e", "--emoji")
  42                 .required(true)
  43                 .help("Specify the emoji, should be a single unicode grapheme cluster.");
  44         subparser.addArgument("-a", "--target-author")
  45                 .required(true)
  46                 .help("Specify the number of the author of the message to which to react.");
  47         subparser.addArgument("-t", "--target-timestamp")
  48                 .required(true)
  49                 .type(long.class)
  50                 .help("Specify the timestamp of the message to which to react.");
  51         subparser.addArgument("-r", "--remove").help("Remove a reaction.").action(Arguments.storeTrue());
  52     }
  53
  54     @Override
  55     public void handleCommand(
  56             final Namespace ns, final Manager m, final OutputWriter outputWriter
  57     ) throws CommandException {
  58         final var isNoteToSelf = ns.getBoolean("note-to-self");
  59         final var recipientStrings = ns.<String>getList("recipient");
  60         final var groupIdStrings = ns.<String>getList("group-id");
  61
  62         final var recipientIdentifiers = CommandUtil.getRecipientIdentifiers(m,
  63                 isNoteToSelf,
  64                 recipientStrings,
  65                 groupIdStrings);
  66
  67         final var emoji = ns.getString("emoji");
  68         final var isRemove = ns.getBoolean("remove");
  69         final var targetAuthor = ns.getString("target-author");
  70         final var targetTimestamp = ns.getLong("target-timestamp");
  71
  72         try {
  73             final var results = m.sendMessageReaction(emoji,
  74                     isRemove,
  75                     CommandUtil.getSingleRecipientIdentifier(targetAuthor, m.getSelfNumber()),
  76                     targetTimestamp,
  77                     recipientIdentifiers);
  78             outputResult(outputWriter, results.getTimestamp());
  79             ErrorUtils.handleSendMessageResults(results.getResults());
  80         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
  81             throw new UserErrorException(e.getMessage());
  82         } catch (IOException e) {
  83             throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
  84                     .getSimpleName() + ")", e);
  85         }
  86     }
  87
  88     @Override
  89     public void handleCommand(
  90             final Namespace ns, final Signal signal, final OutputWriter outputWriter
  91     ) throws CommandException {
  92         final var recipients = ns.<String>getList("recipient");
  93         final var groupIdStrings = ns.<String>getList("group-id");
  94
  95         final var noRecipients = recipients == null || recipients.isEmpty();
  96         final var noGroups = groupIdStrings == null || groupIdStrings.isEmpty();
  97         if (noRecipients && noGroups) {
  98             throw new UserErrorException("No recipients given");
  99         }
 100         if (!noRecipients && !noGroups) {
 101             throw new UserErrorException("You cannot specify recipients by phone number and groups at the same time");
 102         }
 103
 104         final var emoji = ns.getString("emoji");
 105         final var isRemove = ns.getBoolean("remove");
 106         final var targetAuthor = ns.getString("target-author");
 107         final var targetTimestamp = ns.getLong("target-timestamp");
 108
 109         try {
 110             long timestamp = 0;
 111             if (!noGroups) {
 112                 final var groupIds = CommandUtil.getGroupIds(groupIdStrings);
 113                 for (final var groupId : groupIds) {
 114                     timestamp = signal.sendGroupMessageReaction(emoji,
 115                             isRemove,
 116                             targetAuthor,
 117                             targetTimestamp,
 118                             groupId.serialize());
 119                 }
 120             } else {
 121                 timestamp = signal.sendMessageReaction(emoji, isRemove, targetAuthor, targetTimestamp, recipients);
 122             }
 123             outputResult(outputWriter, timestamp);
 124         } catch (UnknownObject e) {
 125             throw new UserErrorException("Failed to find dbus object, maybe missing the -u flag: " + e.getMessage());
 126         } catch (Signal.Error.InvalidNumber e) {
 127             throw new UserErrorException("Invalid number: " + e.getMessage());
 128         } catch (Signal.Error.GroupNotFound e) {
 129             throw new UserErrorException("Failed to send to group: " + e.getMessage());
 130         } catch (DBusExecutionException e) {
 131             throw new UnexpectedErrorException("Failed to send message: " + e.getMessage() + " (" + e.getClass()
 132                     .getSimpleName() + ")", e);
 133         }
 134     }
 135
 136     private void outputResult(final OutputWriter outputWriter, final long timestamp) {
 137         if (outputWriter instanceof PlainTextWriter) {
 138             final var writer = (PlainTextWriter) outputWriter;
 139             writer.println("{}", timestamp);
 140         } else {
 141             final var writer = (JsonWriter) outputWriter;
 142             writer.write(Map.of("timestamp", timestamp));
 143         }
 144     }
 145 }