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.manager.Manager;
   7 import org.asamk.signal.manager.MultiAccountManager;
   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 MultiAccountManager c;
  25
  26     private final String objectPath;
  27
  28     public DbusSignalControlImpl(final MultiAccountManager 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             registrationManager.verifyAccount(verificationCode, pin);
  79         } catch (IOException | PinLockedException | IncorrectPinException e) {
  80             throw new SignalControl.Error.Failure(e.getClass().getSimpleName() + " " + e.getMessage());
  81         }
  82     }
  83
  84     @Override
  85     public String link(final String newDeviceName) throws Error.Failure {
  86         try {
  87             final ProvisioningManager provisioningManager = c.getNewProvisioningManager();
  88             final URI deviceLinkUri = provisioningManager.getDeviceLinkUri();
  89             new Thread(() -> {
  90                 try {
  91                     provisioningManager.finishDeviceLink(newDeviceName);
  92                 } catch (IOException | TimeoutException | UserAlreadyExists e) {
  93                     e.printStackTrace();
  94                 }
  95             }).start();
  96             return deviceLinkUri.toString();
  97         } catch (TimeoutException | IOException e) {
  98             throw new SignalControl.Error.Failure(e.getClass().getSimpleName() + " " + e.getMessage());
  99         }
 100     }
 101
 102     @Override
 103     public String version() {
 104         return BaseConfig.PROJECT_VERSION;
 105     }
 106
 107     @Override
 108     public List<DBusPath> listAccounts() {
 109         return c.getAccountNumbers()
 110                 .stream()
 111                 .map(u -> new DBusPath(DbusConfig.getObjectPath(u)))
 112                 .collect(Collectors.toList());
 113     }
 114 }