nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/VerifyCommand.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 import net.sourceforge.argparse4j.inf.Subparser;
   7
   8 import org.asamk.signal.OutputType;
   9 import org.asamk.signal.commands.exceptions.CommandException;
  10 import org.asamk.signal.commands.exceptions.IOErrorException;
  11 import org.asamk.signal.commands.exceptions.UserErrorException;
  12 import org.asamk.signal.manager.RegistrationManager;
  13 import org.asamk.signal.manager.api.IncorrectPinException;
  14 import org.asamk.signal.manager.api.PinLockMissingException;
  15 import org.asamk.signal.manager.api.PinLockedException;
  16 import org.asamk.signal.output.JsonWriter;
  17 import org.slf4j.Logger;
  18 import org.slf4j.LoggerFactory;
  19
  20 import java.io.IOException;
  21 import java.util.List;
  22
  23 public class VerifyCommand implements RegistrationCommand, JsonRpcRegistrationCommand<VerifyCommand.VerifyParams> {
  24
  25     private static final Logger logger = LoggerFactory.getLogger(VerifyCommand.class);
  26
  27     @Override
  28     public String getName() {
  29         return "verify";
  30     }
  31
  32     @Override
  33     public void attachToSubparser(final Subparser subparser) {
  34         subparser.help("Verify the number using the code received via SMS or voice.");
  35         subparser.addArgument("verification-code").help("The verification code you received via sms or voice call.");
  36         subparser.addArgument("-p", "--pin").help("The registration lock PIN, that was set by the user (Optional)");
  37     }
  38
  39     @Override
  40     public void handleCommand(final Namespace ns, final RegistrationManager m) throws CommandException {
  41         var verificationCode = ns.getString("verification-code");
  42         var pin = ns.getString("pin");
  43
  44         verify(m, verificationCode, pin);
  45     }
  46
  47     @Override
  48     public TypeReference<VerifyParams> getRequestType() {
  49         return new TypeReference<>() {};
  50     }
  51
  52     @Override
  53     public List<OutputType> getSupportedOutputTypes() {
  54         return List.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  55     }
  56
  57     @Override
  58     public void handleCommand(
  59             final VerifyParams request,
  60             final RegistrationManager m,
  61             final JsonWriter jsonWriter
  62     ) throws CommandException {
  63         verify(m, request.verificationCode(), request.pin());
  64     }
  65
  66     private void verify(
  67             final RegistrationManager m,
  68             final String verificationCode,
  69             final String pin
  70     ) throws UserErrorException, IOErrorException {
  71         try {
  72             m.verifyAccount(verificationCode, pin);
  73         } catch (PinLockedException e) {
  74             throw new UserErrorException(
  75                     "Verification failed! This number is locked with a pin. Hours remaining until reset: "
  76                             + (e.getTimeRemaining() / 1000 / 60 / 60)
  77                             + "\nUse '--pin PIN_CODE' to specify the registration lock PIN");
  78         } catch (IncorrectPinException e) {
  79             throw new UserErrorException("Verification failed! Invalid pin, tries remaining: " + e.getTriesRemaining());
  80         } catch (PinLockMissingException e) {
  81             throw new UserErrorException("Account is pin locked, but pin data has been deleted on the server.");
  82         } catch (IOException e) {
  83             throw new IOErrorException("Verify error: " + e.getMessage(), e);
  84         }
  85     }
  86
  87     public record VerifyParams(String verificationCode, String pin) {}
  88 }