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

   1 package org.asamk.signal.commands;
   2
   3 import com.fasterxml.jackson.annotation.JsonAutoDetect;
   4 import com.fasterxml.jackson.annotation.PropertyAccessor;
   5 import com.fasterxml.jackson.core.JsonGenerator;
   6 import com.fasterxml.jackson.databind.DeserializationFeature;
   7 import com.fasterxml.jackson.databind.ObjectMapper;
   8
   9 import net.sourceforge.argparse4j.impl.Arguments;
  10 import net.sourceforge.argparse4j.inf.Namespace;
  11 import net.sourceforge.argparse4j.inf.Subparser;
  12
  13 import org.asamk.signal.manager.Manager;
  14
  15 import java.io.IOException;
  16 import java.util.ArrayList;
  17 import java.util.HashSet;
  18 import java.util.Set;
  19 import java.util.Map;
  20 import java.util.List;
  21
  22 public class GetUserStatusCommand implements LocalCommand {
  23
  24     @Override
  25     public void attachToSubparser(final Subparser subparser) {
  26         subparser.addArgument("number").help("Phone number").nargs("+");
  27         subparser.help("Check if the specified phone number/s have been registered");
  28         subparser.addArgument("--json")
  29                 .help("Output received messages in json format, one json object per line.")
  30                 .action(Arguments.storeTrue());
  31     }
  32
  33     @Override
  34     public int handleCommand(final Namespace ns, final Manager m) {
  35         if (!m.isRegistered()) {
  36             System.err.println("User is not registered.");
  37             return 1;
  38         }
  39
  40         // Setup the json object mapper
  41         ObjectMapper jsonProcessor = new ObjectMapper();
  42         jsonProcessor.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY); // disable autodetect
  43         jsonProcessor.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
  44         jsonProcessor.disable(JsonGenerator.Feature.AUTO_CLOSE_TARGET);
  45
  46         // Get a map of registration statuses
  47         Map<String, Boolean> registered;
  48         try {
  49             registered = m.areUsersRegistered(new HashSet<>(ns.<String>getList("number")));
  50         } catch (IOException e) {
  51             System.err.println("Unable to check if users are registered");
  52             return 1;
  53         }
  54
  55         // Output
  56         if (ns.getBoolean("json")) {
  57             List<JsonIsRegistered> objects = new ArrayList<>();
  58             for (Map.Entry<String, Boolean> entry : registered.entrySet()) {
  59                 objects.add(new JsonIsRegistered(entry.getKey(), entry.getValue()));
  60             }
  61
  62             try {
  63                 System.out.println(jsonProcessor.writeValueAsString(objects));
  64             } catch (IOException e) {
  65                 System.err.println(e.getMessage());
  66             }
  67
  68         } else {
  69             for (Map.Entry<String, Boolean> entry : registered.entrySet()) {
  70                 System.out.println(entry.getKey() + ": " + entry.getValue());
  71             }
  72         }
  73
  74         return 0;
  75     }
  76
  77     private class JsonIsRegistered {
  78         String name;
  79         boolean isRegistered;
  80
  81         public JsonIsRegistered(String name, boolean isRegistered) {
  82             this.name = name;
  83             this.isRegistered = isRegistered;
  84         }
  85     }
  86
  87 }