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

   1 package org.asamk.signal.commands;
   2
   3 import com.fasterxml.jackson.core.type.TypeReference;
   4
   5 import net.sourceforge.argparse4j.inf.Namespace;
   6
   7 import org.asamk.signal.OutputType;
   8 import org.asamk.signal.commands.exceptions.CommandException;
   9 import org.asamk.signal.manager.Manager;
  10 import org.asamk.signal.util.Util;
  11
  12 import java.util.List;
  13 import java.util.Map;
  14 import java.util.Set;
  15
  16 public interface JsonRpcLocalCommand extends JsonRpcCommand<Map<String, Object>> {
  17
  18     void handleCommand(Namespace ns, Manager m) throws CommandException;
  19
  20     default TypeReference<Map<String, Object>> getRequestType() {
  21         return new TypeReference<>() {
  22         };
  23     }
  24
  25     default void handleCommand(Map<String, Object> request, Manager m) throws CommandException {
  26         Namespace commandNamespace = new JsonRpcNamespace(request == null ? Map.of() : request);
  27         handleCommand(commandNamespace, m);
  28     }
  29
  30     default Set<OutputType> getSupportedOutputTypes() {
  31         return Set.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  32     }
  33
  34     /**
  35      * Namepace implementation, that defaults booleans to false and converts camel case keys to dashed strings
  36      */
  37     final class JsonRpcNamespace extends Namespace {
  38
  39         public JsonRpcNamespace(final Map<String, Object> attrs) {
  40             super(attrs);
  41         }
  42
  43         public <T> T get(String dest) {
  44             final T value = super.get(dest);
  45             if (value != null) {
  46                 return value;
  47             }
  48
  49             final var camelCaseString = Util.dashSeparatedToCamelCaseString(dest);
  50             return super.get(camelCaseString);
  51         }
  52
  53         @Override
  54         public <E> List<E> getList(final String dest) {
  55             final List<E> value = super.getList(dest);
  56             if (value != null) {
  57                 return value;
  58             }
  59
  60             return super.getList(dest + "s");
  61         }
  62
  63         @Override
  64         public Boolean getBoolean(String dest) {
  65             Boolean maybeGotten = this.get(dest);
  66             if (maybeGotten == null) {
  67                 maybeGotten = false;
  68             }
  69             return maybeGotten;
  70         }
  71     }
  72 }