Изменение домена с rosetta-im.com на rosetta.im

src/main/java/im/rosetta/client/ClientManager.java

@@ -0,0 +1,112 @@
+package im.rosetta.client;
+
+import java.util.HashSet;
+import java.util.List;
+
+import im.rosetta.client.tags.ECIAuthentificate;
+
+import io.orprotocol.ProtocolException;
+import io.orprotocol.Server;
+import io.orprotocol.client.Client;
+import io.orprotocol.index.ClientIndexer;
+import io.orprotocol.packet.Packet;
+
+/**
+ * Менеджер клиентов
+ */
+public class ClientManager {
+
+    private Server server;
+    private ClientIndexer clientIndexer;
+
+    public ClientManager(Server server) {
+        this.server = server;
+        this.clientIndexer = server.getClientIndexer();
+    }
+    
+    public Server getServer() {
+        return this.server;
+    }
+
+    public boolean isClientConnected(String publicKey) {
+        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        if(clients == null){
+            /**
+             * Нет клиентов с таким публичным ключом
+             */
+            return false;
+        }
+        if(clients.size() <= 0){
+            /**
+             * Нет клиентов с таким публичным ключом
+             */
+            return false;
+        }
+        /**
+         * Есть клиенты с таким публичным ключом
+         */
+        return true;
+    }
+
+    /**
+     * Отправить пакет всем АВТОРИЗОВАННЫМ клиентам с публичным ключом publicKey
+     * @param publicKey публичный ключ получателя
+     * @param packet пакет для отправки
+     * @throws ProtocolException если произошла ошибка при отправке пакета клиенту
+     */
+    public void sendPacketToAuthorizedPK(String publicKey, Packet packet) throws ProtocolException {
+        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        if(clients == null){
+            /**
+             * Нет клиентов с таким публичным ключом, значит отправлять некому
+             */
+            return;
+        }
+        for(Client client : clients){
+            ECIAuthentificate eciAuthentificate = client.getTag(ECIAuthentificate.class);
+            if(eciAuthentificate == null || !eciAuthentificate.hasAuthorized()){
+                /**
+                 * Если клиент не авторизован, пропускаем его, он не должен получать пакеты,
+                 * если нужно отправить пакет неавторизованному клиенту, нужно отправить его напрямую посредством client.send(packet), 
+                 * а не через этот метод
+                 */
+                continue;
+            }
+            /**
+             * Отправляем пакет каждому клиенту с таким публичным ключом (то есть всем его авторизованным сессиям/устройствам)
+             */
+            client.send(packet);
+        }
+    }
+
+    /**
+     * Отправить пакет всем клиентам с публичными ключами из списка publicKeys
+     * @param publicKeys список публичных ключей получателей
+     * @param packet пакет для отправки
+     * @throws ProtocolException если произошла ошибка при отправке пакета клиенту
+     */
+    public void sendPacketToAuthorizedPK(List<String> publicKeys, Packet packet) throws ProtocolException {
+        for(String publicKey : publicKeys){
+            this.sendPacketToAuthorizedPK(publicKey, packet);
+        }
+    }
+
+    /**
+     * Получить список клиентов по публичному ключу (get PublicKey clients), могут быть неавторизованные клиенты
+     * @param publicKey публичный ключ клиента
+     * @return список клиентов с таким публичным ключом, может быть пустым, если клиентов с таким публичным ключом нет
+     */
+    public List<Client> getPKClients(String publicKey) {
+        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        if(clients == null){
+            /**
+             * Нет клиентов с таким публичным ключом
+             */
+            return List.of();
+        }
+        return List.copyOf(clients);
+    }
+    
+
+
+}