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

   1 package org.asamk.signal.dbus;
   2
   3 import org.asamk.SignalControl;
   4 import org.asamk.signal.BaseConfig;
   5 import org.asamk.signal.DbusConfig;
   6 import org.asamk.signal.commands.SignalCreator;
   7 import org.asamk.signal.manager.Manager;
   8 import org.asamk.signal.manager.ProvisioningManager;
   9 import org.asamk.signal.manager.RegistrationManager;
  10 import org.asamk.signal.manager.UserAlreadyExists;
  11 import org.asamk.signal.manager.api.CaptchaRequiredException;
  12 import org.asamk.signal.manager.api.IncorrectPinException;
  13 import org.asamk.signal.manager.api.PinLockedException;
  14 import org.freedesktop.dbus.DBusPath;
  15
  16 import java.io.IOException;
  17 import java.net.URI;
  18 import java.util.List;
  19 import java.util.concurrent.TimeoutException;
  20 import java.util.stream.Collectors;
  21
  22 public class DbusSignalControlImpl implements org.asamk.SignalControl {
  23
  24     private final SignalCreator c;
  25
  26     private final String objectPath;
  27
  28     public DbusSignalControlImpl(final SignalCreator c, final String objectPath) {
  29         this.c = c;
  30         this.objectPath = objectPath;
  31     }
  32
  33     @Override
  34     public boolean isRemote() {
  35         return false;
  36     }
  37
  38     @Override
  39     public String getObjectPath() {
  40         return objectPath;
  41     }
  42
  43     @Override
  44     public void register(
  45             final String number, final boolean voiceVerification
  46     ) throws Error.Failure, Error.InvalidNumber {
  47         registerWithCaptcha(number, voiceVerification, null);
  48     }
  49
  50     @Override
  51     public void registerWithCaptcha(
  52             final String number, final boolean voiceVerification, final String captcha
  53     ) throws Error.Failure, Error.InvalidNumber {
  54         if (!Manager.isValidNumber(number, null)) {
  55             throw new SignalControl.Error.InvalidNumber(
  56                     "Invalid username (phone number), make sure you include the country code.");
  57         }
  58         try (final RegistrationManager registrationManager = c.getNewRegistrationManager(number)) {
  59             registrationManager.register(voiceVerification, captcha);
  60         } catch (CaptchaRequiredException e) {
  61             String message = captcha == null ? "Captcha required for verification." : "Invalid captcha given.";
  62             throw new SignalControl.Error.RequiresCaptcha(message);
  63         } catch (IOException e) {
  64             throw new SignalControl.Error.Failure(e.getClass().getSimpleName() + " " + e.getMessage());
  65         }
  66     }
  67
  68     @Override
  69     public void verify(final String number, final String verificationCode) throws Error.Failure, Error.InvalidNumber {
  70         verifyWithPin(number, verificationCode, null);
  71     }
  72
  73     @Override
  74     public void verifyWithPin(
  75             final String number, final String verificationCode, final String pin
  76     ) throws Error.Failure, Error.InvalidNumber {
  77         try (final RegistrationManager registrationManager = c.getNewRegistrationManager(number)) {
  78             final Manager manager = registrationManager.verifyAccount(verificationCode, pin);
  79             c.addManager(manager);
  80         } catch (IOException | PinLockedException | IncorrectPinException e) {
  81             throw new SignalControl.Error.Failure(e.getClass().getSimpleName() + " " + e.getMessage());
  82         }
  83     }
  84
  85     @Override
  86     public String link(final String newDeviceName) throws Error.Failure {
  87         try {
  88             final ProvisioningManager provisioningManager = c.getNewProvisioningManager();
  89             final URI deviceLinkUri = provisioningManager.getDeviceLinkUri();
  90             new Thread(() -> {
  91                 try {
  92                     final Manager manager = provisioningManager.finishDeviceLink(newDeviceName);
  93                     c.addManager(manager);
  94                 } catch (IOException | TimeoutException | UserAlreadyExists e) {
  95                     e.printStackTrace();
  96                 }
  97             }).start();
  98             return deviceLinkUri.toString();
  99         } catch (TimeoutException | IOException e) {
 100             throw new SignalControl.Error.Failure(e.getClass().getSimpleName() + " " + e.getMessage());
 101         }
 102     }
 103
 104     @Override
 105     public String version() {
 106         return BaseConfig.PROJECT_VERSION;
 107     }
 108
 109     @Override
 110     public List<DBusPath> listAccounts() {
 111         return c.getAccountNumbers()
 112                 .stream()
 113                 .map(u -> new DBusPath(DbusConfig.getObjectPath(u)))
 114                 .collect(Collectors.toList());
 115     }
 116 }