nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/RegisterCommand.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.commands.exceptions.CommandException;
  11 import org.asamk.signal.commands.exceptions.IOErrorException;
  12 import org.asamk.signal.commands.exceptions.RateLimitErrorException;
  13 import org.asamk.signal.commands.exceptions.UserErrorException;
  14 import org.asamk.signal.manager.RegistrationManager;
  15 import org.asamk.signal.manager.api.CaptchaRequiredException;
  16 import org.asamk.signal.manager.api.NonNormalizedPhoneNumberException;
  17 import org.asamk.signal.manager.api.RateLimitException;
  18 import org.asamk.signal.manager.api.VerificationMethodNotAvailableException;
  19 import org.asamk.signal.output.JsonWriter;
  20 import org.asamk.signal.util.CommandUtil;
  21
  22 import java.io.IOException;
  23 import java.util.List;
  24
  25 public class RegisterCommand implements RegistrationCommand, JsonRpcRegistrationCommand<RegisterCommand.RegistrationParams> {
  26
  27     @Override
  28     public String getName() {
  29         return "register";
  30     }
  31
  32     @Override
  33     public void attachToSubparser(final Subparser subparser) {
  34         subparser.help("Register a phone number with SMS or voice verification.");
  35         subparser.addArgument("-v", "--voice")
  36                 .help("The verification should be done over voice, not SMS.")
  37                 .action(Arguments.storeTrue());
  38         subparser.addArgument("--captcha")
  39                 .help("The captcha token, required if registration failed with a captcha required error.");
  40         subparser.addArgument("--reregister")
  41                 .action(Arguments.storeTrue())
  42                 .help("Register even if account is already registered");
  43     }
  44
  45     @Override
  46     public void handleCommand(final Namespace ns, final RegistrationManager m) throws CommandException {
  47         final boolean voiceVerification = Boolean.TRUE.equals(ns.getBoolean("voice"));
  48         final var captcha = ns.getString("captcha");
  49         final var reregister = Boolean.TRUE.equals(ns.getBoolean("reregister"));
  50
  51         register(m, voiceVerification, captcha, reregister);
  52     }
  53
  54     @Override
  55     public TypeReference<RegistrationParams> getRequestType() {
  56         return new TypeReference<>() {};
  57     }
  58
  59     @Override
  60     public List<OutputType> getSupportedOutputTypes() {
  61         return List.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  62     }
  63
  64     @Override
  65     public void handleCommand(
  66             final RegistrationParams request,
  67             final RegistrationManager m,
  68             final JsonWriter jsonWriter
  69     ) throws CommandException {
  70         register(m, Boolean.TRUE.equals(request.voice()), request.captcha(), Boolean.TRUE.equals(request.reregister()));
  71     }
  72
  73     private void register(
  74             final RegistrationManager m,
  75             final boolean voiceVerification,
  76             final String captcha,
  77             final boolean reregister
  78     ) throws CommandException {
  79         try {
  80             m.register(voiceVerification, captcha, reregister);
  81         } catch (RateLimitException e) {
  82             final var message = CommandUtil.getRateLimitMessage(e);
  83             throw new RateLimitErrorException(message, e);
  84         } catch (CaptchaRequiredException e) {
  85             final var message = CommandUtil.getCaptchaRequiredMessage(e, captcha != null);
  86             throw new UserErrorException(message);
  87         } catch (NonNormalizedPhoneNumberException e) {
  88             throw new UserErrorException("Failed to register: " + e.getMessage(), e);
  89         } catch (IOException e) {
  90             throw new IOErrorException("Failed to register: %s (%s)".formatted(e.getMessage(),
  91                     e.getClass().getSimpleName()), e);
  92         } catch (VerificationMethodNotAvailableException e) {
  93             throw new UserErrorException("Failed to register: " + e.getMessage() + (
  94                     voiceVerification
  95                             ? ": Before requesting voice verification you need to request SMS verification and wait a minute."
  96                             : ""
  97             ), e);
  98         }
  99     }
 100
 101     public record RegistrationParams(Boolean voice, String captcha, Boolean reregister) {}
 102 }