Implement multi account commands for dbus client

[signal-cli] / src / main / java / org / asamk / signal / dbus / DbusMultiAccountManagerImpl.java

diff --git a/src/main/java/org/asamk/signal/dbus/DbusMultiAccountManagerImpl.java b/src/main/java/org/asamk/signal/dbus/DbusMultiAccountManagerImpl.java
new file mode 100644 (file)
index 0000000..9659fea
--- /dev/null
+++ b/src/main/java/org/asamk/signal/dbus/DbusMultiAccountManagerImpl.java
@@ -0,0 +1,98 @@
+package org.asamk.signal.dbus;
+
+import org.asamk.Signal;
+import org.asamk.SignalControl;
+import org.asamk.signal.DbusConfig;
+import org.asamk.signal.manager.Manager;
+import org.asamk.signal.manager.MultiAccountManager;
+import org.asamk.signal.manager.ProvisioningManager;
+import org.asamk.signal.manager.RegistrationManager;
+import org.freedesktop.dbus.DBusPath;
+import org.freedesktop.dbus.connections.impl.DBusConnection;
+import org.freedesktop.dbus.exceptions.DBusException;
+import org.freedesktop.dbus.interfaces.DBusInterface;
+
+import java.io.IOException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.concurrent.TimeoutException;
+import java.util.function.Consumer;
+
+/**
+ * This class implements the MultiAccountManager interface using the DBus Signal interface, where possible.
+ * It's used for the signal-cli dbus client mode (--dbus, --dbus-system)
+ */
+public class DbusMultiAccountManagerImpl implements MultiAccountManager {
+
+    private final SignalControl signalControl;
+    private final DBusConnection connection;
+    // TODO add listeners for added/removed accounts
+    private final Set<Consumer<Manager>> onManagerAddedHandlers = new HashSet<>();
+    private final Set<Consumer<Manager>> onManagerRemovedHandlers = new HashSet<>();
+
+    public DbusMultiAccountManagerImpl(final SignalControl signalControl, DBusConnection connection) {
+        this.signalControl = signalControl;
+        this.connection = connection;
+    }
+
+    @Override
+    public List<String> getAccountNumbers() {
+        return signalControl.listAccounts()
+                .stream()
+                .map(a -> getRemoteObject(a, Signal.class).getSelfNumber())
+                .toList();
+    }
+
+    @Override
+    public void addOnManagerAddedHandler(final Consumer<Manager> handler) {
+        synchronized (onManagerAddedHandlers) {
+            onManagerAddedHandlers.add(handler);
+        }
+    }
+
+    @Override
+    public void addOnManagerRemovedHandler(final Consumer<Manager> handler) {
+        synchronized (onManagerRemovedHandlers) {
+            onManagerRemovedHandlers.add(handler);
+        }
+    }
+
+    @Override
+    public Manager getManager(final String phoneNumber) {
+        return new DbusManagerImpl(getRemoteObject(signalControl.getAccount(phoneNumber), Signal.class), connection);
+    }
+
+    @Override
+    public URI getNewProvisioningDeviceLinkUri() throws TimeoutException, IOException {
+        try {
+            return new URI(signalControl.startLink());
+        } catch (URISyntaxException e) {
+            throw new IOException(e);
+        }
+    }
+
+    @Override
+    public ProvisioningManager getProvisioningManagerFor(final URI deviceLinkUri) {
+        return new DbusProvisioningManagerImpl(signalControl, connection, deviceLinkUri);
+    }
+
+    @Override
+    public RegistrationManager getNewRegistrationManager(final String account) throws IOException {
+        return new DbusRegistrationManagerImpl(account, signalControl, connection);
+    }
+
+    @Override
+    public void close() {
+    }
+
+    private <T extends DBusInterface> T getRemoteObject(final DBusPath path, final Class<T> type) {
+        try {
+            return connection.getRemoteObject(DbusConfig.getBusname(), path.getPath(), type);
+        } catch (DBusException e) {
+            throw new AssertionError(e);
+        }
+    }
+}