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

   1 package org.asamk.signal.commands;
   2
   3 import com.fasterxml.jackson.core.type.TypeReference;
   4
   5 import net.sourceforge.argparse4j.impl.Arguments;
   6 import net.sourceforge.argparse4j.inf.Namespace;
   7 import net.sourceforge.argparse4j.inf.Subparser;
   8
   9 import org.asamk.signal.OutputType;
  10 import org.asamk.signal.ReceiveMessageHandler;
  11 import org.asamk.signal.Shutdown;
  12 import org.asamk.signal.commands.exceptions.CommandException;
  13 import org.asamk.signal.commands.exceptions.IOErrorException;
  14 import org.asamk.signal.commands.exceptions.UserErrorException;
  15 import org.asamk.signal.json.JsonReceiveMessageHandler;
  16 import org.asamk.signal.manager.Manager;
  17 import org.asamk.signal.manager.api.AlreadyReceivingException;
  18 import org.asamk.signal.manager.api.ReceiveConfig;
  19 import org.asamk.signal.output.JsonWriter;
  20 import org.asamk.signal.output.OutputWriter;
  21 import org.asamk.signal.output.PlainTextWriter;
  22 import org.slf4j.Logger;
  23 import org.slf4j.LoggerFactory;
  24
  25 import java.io.IOException;
  26 import java.time.Duration;
  27 import java.util.ArrayList;
  28 import java.util.List;
  29 import java.util.Optional;
  30
  31 public class ReceiveCommand implements LocalCommand, JsonRpcSingleCommand<ReceiveCommand.ReceiveParams> {
  32
  33     private static final Logger logger = LoggerFactory.getLogger(ReceiveCommand.class);
  34
  35     @Override
  36     public String getName() {
  37         return "receive";
  38     }
  39
  40     @Override
  41     public void attachToSubparser(final Subparser subparser) {
  42         subparser.help("Query the server for new messages.");
  43         subparser.addArgument("-t", "--timeout")
  44                 .type(double.class)
  45                 .setDefault(3.0)
  46                 .help("Number of seconds to wait for new messages (negative values disable timeout)");
  47         subparser.addArgument("--max-messages")
  48                 .type(int.class)
  49                 .setDefault(-1)
  50                 .help("Maximum number of messages to receive, before returning.");
  51         subparser.addArgument("--ignore-attachments")
  52                 .help("Don’t download attachments of received messages.")
  53                 .action(Arguments.storeTrue());
  54         subparser.addArgument("--ignore-stories")
  55                 .help("Don’t receive story messages from the server.")
  56                 .action(Arguments.storeTrue());
  57         subparser.addArgument("--send-read-receipts")
  58                 .help("Send read receipts for all incoming data messages (in addition to the default delivery receipts)")
  59                 .action(Arguments.storeTrue());
  60     }
  61
  62     @Override
  63     public List<OutputType> getSupportedOutputTypes() {
  64         return List.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  65     }
  66
  67     @Override
  68     public void handleCommand(
  69             final Namespace ns,
  70             final Manager m,
  71             final OutputWriter outputWriter
  72     ) throws CommandException {
  73         Shutdown.installHandler();
  74         final var timeout = ns.getDouble("timeout");
  75         final var maxMessagesRaw = ns.getInt("max-messages");
  76         final var ignoreAttachments = Boolean.TRUE.equals(ns.getBoolean("ignore-attachments"));
  77         final var ignoreStories = Boolean.TRUE.equals(ns.getBoolean("ignore-stories"));
  78         final var sendReadReceipts = Boolean.TRUE.equals(ns.getBoolean("send-read-receipts"));
  79         m.setReceiveConfig(new ReceiveConfig(ignoreAttachments, ignoreStories, sendReadReceipts));
  80         try {
  81             final var handler = switch (outputWriter) {
  82                 case JsonWriter writer -> new JsonReceiveMessageHandler(m, writer);
  83                 case PlainTextWriter writer -> new ReceiveMessageHandler(m, writer);
  84             };
  85             final var duration = timeout < 0 ? null : Duration.ofMillis((long) (timeout * 1000));
  86             final var maxMessages = maxMessagesRaw < 0 ? null : maxMessagesRaw;
  87             Shutdown.registerShutdownListener(m::stopReceiveMessages);
  88             m.receiveMessages(Optional.ofNullable(duration), Optional.ofNullable(maxMessages), handler);
  89         } catch (IOException e) {
  90             throw new IOErrorException("Error while receiving messages: " + e.getMessage(), e);
  91         } catch (AlreadyReceivingException e) {
  92             throw new UserErrorException("Receive command cannot be used if messages are already being received.", e);
  93         }
  94     }
  95
  96     @Override
  97     public TypeReference<ReceiveParams> getRequestType() {
  98         return new TypeReference<>() {};
  99     }
 100
 101     @Override
 102     public void handleCommand(
 103             final ReceiveParams request,
 104             final Manager m,
 105             final JsonWriter jsonWriter
 106     ) throws CommandException {
 107         final var timeout = request.timeout() == null ? 3.0 : request.timeout();
 108         final var maxMessagesRaw = request.maxMessages() == null ? -1 : request.maxMessages();
 109
 110         try {
 111             final var messages = new ArrayList<>();
 112             final var handler = new JsonReceiveMessageHandler(m, messages::add);
 113             final var duration = timeout < 0 ? null : Duration.ofMillis((long) (timeout * 1000));
 114             final var maxMessages = maxMessagesRaw < 0 ? null : maxMessagesRaw;
 115             m.receiveMessages(Optional.ofNullable(duration), Optional.ofNullable(maxMessages), handler);
 116             jsonWriter.write(messages);
 117         } catch (IOException e) {
 118             throw new IOErrorException("Error while receiving messages: " + e.getMessage(), e);
 119         } catch (AlreadyReceivingException e) {
 120             throw new UserErrorException("Receive command cannot be used if messages are already being received.", e);
 121         }
 122     }
 123
 124     public record ReceiveParams(Double timeout, Integer maxMessages) {}
 125 }