nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/JsonRpcDispatcherCommand.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.OutputType;
   8 import org.asamk.signal.commands.exceptions.CommandException;
   9 import org.asamk.signal.jsonrpc.SignalJsonRpcDispatcherHandler;
  10 import org.asamk.signal.manager.Manager;
  11 import org.asamk.signal.manager.MultiAccountManager;
  12 import org.asamk.signal.output.JsonWriter;
  13 import org.asamk.signal.output.OutputWriter;
  14 import org.asamk.signal.util.IOUtils;
  15 import org.slf4j.Logger;
  16 import org.slf4j.LoggerFactory;
  17
  18 import java.io.InputStreamReader;
  19 import java.util.List;
  20 import java.util.function.Supplier;
  21
  22 import static org.asamk.signal.util.CommandUtil.getReceiveConfig;
  23
  24 public class JsonRpcDispatcherCommand implements LocalCommand, MultiLocalCommand {
  25
  26     private final static Logger logger = LoggerFactory.getLogger(JsonRpcDispatcherCommand.class);
  27
  28     @Override
  29     public String getName() {
  30         return "jsonRpc";
  31     }
  32
  33     @Override
  34     public void attachToSubparser(final Subparser subparser) {
  35         subparser.help("Take commands from standard input as line-delimited JSON RPC while receiving messages.");
  36         subparser.addArgument("--ignore-attachments")
  37                 .help("Don’t download attachments of received messages.")
  38                 .action(Arguments.storeTrue());
  39         subparser.addArgument("--ignore-stories")
  40                 .help("Don’t receive story messages from the server.")
  41                 .action(Arguments.storeTrue());
  42         subparser.addArgument("--send-read-receipts")
  43                 .help("Send read receipts for all incoming data messages (in addition to the default delivery receipts)")
  44                 .action(Arguments.storeTrue());
  45         subparser.addArgument("--receive-mode")
  46                 .help("Specify when to start receiving messages.")
  47                 .type(Arguments.enumStringType(ReceiveMode.class))
  48                 .setDefault(ReceiveMode.ON_START);
  49     }
  50
  51     @Override
  52     public List<OutputType> getSupportedOutputTypes() {
  53         return List.of(OutputType.JSON);
  54     }
  55
  56     @Override
  57     public void handleCommand(
  58             final Namespace ns, final Manager m, final OutputWriter outputWriter
  59     ) throws CommandException {
  60         final var receiveMode = ns.<ReceiveMode>get("receive-mode");
  61         final var receiveConfig = getReceiveConfig(ns);
  62         m.setReceiveConfig(receiveConfig);
  63
  64         final var jsonOutputWriter = (JsonWriter) outputWriter;
  65         final var lineSupplier = getLineSupplier();
  66
  67         final var handler = new SignalJsonRpcDispatcherHandler(jsonOutputWriter,
  68                 lineSupplier,
  69                 receiveMode == ReceiveMode.MANUAL);
  70         handler.handleConnection(m);
  71     }
  72
  73     @Override
  74     public void handleCommand(
  75             final Namespace ns, final MultiAccountManager c, final OutputWriter outputWriter
  76     ) throws CommandException {
  77         final var receiveMode = ns.<ReceiveMode>get("receive-mode");
  78         final var receiveConfig = getReceiveConfig(ns);
  79         c.getManagers().forEach(m -> m.setReceiveConfig(receiveConfig));
  80         c.addOnManagerAddedHandler(m -> m.setReceiveConfig(receiveConfig));
  81
  82         final var jsonOutputWriter = (JsonWriter) outputWriter;
  83         final var lineSupplier = getLineSupplier();
  84
  85         final var handler = new SignalJsonRpcDispatcherHandler(jsonOutputWriter,
  86                 lineSupplier,
  87                 receiveMode == ReceiveMode.MANUAL);
  88         handler.handleConnection(c);
  89     }
  90
  91     private static Supplier<String> getLineSupplier() {
  92         return IOUtils.getLineSupplier(new InputStreamReader(System.in, IOUtils.getConsoleCharset()));
  93     }
  94 }