nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/internal/MultiAccountManagerImpl.java

   1 package org.asamk.signal.manager.internal;
   2
   3 import org.asamk.signal.manager.Manager;
   4 import org.asamk.signal.manager.MultiAccountManager;
   5 import org.asamk.signal.manager.ProvisioningManager;
   6 import org.asamk.signal.manager.RegistrationManager;
   7 import org.asamk.signal.manager.SignalAccountFiles;
   8 import org.slf4j.Logger;
   9 import org.slf4j.LoggerFactory;
  10
  11 import java.io.IOException;
  12 import java.net.URI;
  13 import java.util.ArrayList;
  14 import java.util.Collection;
  15 import java.util.HashMap;
  16 import java.util.HashSet;
  17 import java.util.List;
  18 import java.util.Map;
  19 import java.util.Set;
  20 import java.util.concurrent.TimeoutException;
  21 import java.util.function.Consumer;
  22
  23 public class MultiAccountManagerImpl implements MultiAccountManager {
  24
  25     private final static Logger logger = LoggerFactory.getLogger(MultiAccountManagerImpl.class);
  26
  27     private final Set<Consumer<Manager>> onManagerAddedHandlers = new HashSet<>();
  28     private final Set<Consumer<Manager>> onManagerRemovedHandlers = new HashSet<>();
  29     private final Set<Manager> managers = new HashSet<>();
  30     private final Map<URI, ProvisioningManager> provisioningManagers = new HashMap<>();
  31     private final SignalAccountFiles signalAccountFiles;
  32
  33     public MultiAccountManagerImpl(final Collection<Manager> managers, final SignalAccountFiles signalAccountFiles) {
  34         this.signalAccountFiles = signalAccountFiles;
  35         this.managers.addAll(managers);
  36         managers.forEach(m -> m.addClosedListener(() -> this.removeManager(m)));
  37     }
  38
  39     @Override
  40     public List<String> getAccountNumbers() {
  41         synchronized (managers) {
  42             return managers.stream().map(Manager::getSelfNumber).toList();
  43         }
  44     }
  45
  46     @Override
  47     public List<Manager> getManagers() {
  48         synchronized (managers) {
  49             return new ArrayList<>(managers);
  50         }
  51     }
  52
  53     void addManager(final Manager m) {
  54         synchronized (managers) {
  55             if (managers.contains(m)) {
  56                 return;
  57             }
  58             managers.add(m);
  59             m.addClosedListener(() -> this.removeManager(m));
  60         }
  61         synchronized (onManagerAddedHandlers) {
  62             for (final var handler : onManagerAddedHandlers) {
  63                 handler.accept(m);
  64             }
  65         }
  66     }
  67
  68     @Override
  69     public void addOnManagerAddedHandler(final Consumer<Manager> handler) {
  70         synchronized (onManagerAddedHandlers) {
  71             onManagerAddedHandlers.add(handler);
  72         }
  73     }
  74
  75     void removeManager(final Manager m) {
  76         synchronized (managers) {
  77             if (!managers.remove(m)) {
  78                 return;
  79             }
  80         }
  81         synchronized (onManagerRemovedHandlers) {
  82             for (final var handler : onManagerRemovedHandlers) {
  83                 handler.accept(m);
  84             }
  85         }
  86     }
  87
  88     @Override
  89     public void addOnManagerRemovedHandler(final Consumer<Manager> handler) {
  90         synchronized (onManagerRemovedHandlers) {
  91             onManagerRemovedHandlers.add(handler);
  92         }
  93     }
  94
  95     @Override
  96     public Manager getManager(final String number) {
  97         synchronized (managers) {
  98             return managers.stream().filter(m -> m.getSelfNumber().equals(number)).findFirst().orElse(null);
  99         }
 100     }
 101
 102     @Override
 103     public URI getNewProvisioningDeviceLinkUri() throws TimeoutException, IOException {
 104         final var provisioningManager = getNewProvisioningManager();
 105         final var deviceLinkUri = provisioningManager.getDeviceLinkUri();
 106         provisioningManagers.put(deviceLinkUri, provisioningManager);
 107         return deviceLinkUri;
 108     }
 109
 110     @Override
 111     public ProvisioningManager getProvisioningManagerFor(final URI deviceLinkUri) {
 112         return provisioningManagers.remove(deviceLinkUri);
 113     }
 114
 115     private ProvisioningManager getNewProvisioningManager() {
 116         return signalAccountFiles.initProvisioningManager(this::addManager);
 117     }
 118
 119     @Override
 120     public RegistrationManager getNewRegistrationManager(String number) throws IOException {
 121         return signalAccountFiles.initRegistrationManager(number, this::addManager);
 122     }
 123
 124     @Override
 125     public void close() {
 126         synchronized (managers) {
 127             for (var m : new ArrayList<>(managers)) {
 128                 try {
 129                     m.close();
 130                 } catch (IOException e) {
 131                     logger.warn("Cleanup failed", e);
 132                 }
 133             }
 134             managers.clear();
 135         }
 136     }
 137 }