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

src/main/java/im/rosetta/service/services/BufferService.java

@@ -0,0 +1,99 @@
+package im.rosetta.service.services;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+
+import im.rosetta.client.tags.ECIAuthentificate;
+import im.rosetta.database.QuerySession;
+import im.rosetta.database.entity.Buffer;
+import im.rosetta.database.repository.BufferRepository;
+import im.rosetta.exception.UnauthorizedExeception;
+import im.rosetta.service.Service;
+
+import io.orprotocol.ProtocolException;
+import io.orprotocol.client.Client;
+import io.orprotocol.packet.Packet;
+import io.orprotocol.packet.PacketManager;
+
+public class BufferService extends Service<BufferRepository> {
+
+    private PacketManager packetManager;
+
+    public BufferService(BufferRepository repository, PacketManager packetManager) {
+        super(repository);
+        this.packetManager = packetManager;
+    }
+
+    /**
+     * Получить пакеты из буфера для клиента, которые были добавлены в буфер после fromTimestampMs. Если клиент не авторизован, возвращает пустой список.
+     * @param client
+     * @param fromTimestampMs
+     * @return
+     * @throws ProtocolException
+     */
+    public List<Packet> getPacketsFromTime(Client client, long fromTimestampMs) throws ProtocolException, UnauthorizedExeception {
+        ECIAuthentificate eciAuthentificate = client.getTag(ECIAuthentificate.class);
+        if(eciAuthentificate == null || !eciAuthentificate.hasAuthorized()){
+            /**
+             * Если клиент не авторизован, то он не может получать пакеты из буфера, возвращаем пустой список
+             */
+            throw new UnauthorizedExeception("Unauthorized client cannot get packets from buffer");
+        }
+        String toPublicKey = eciAuthentificate.getPublicKey();
+        String hql = "FROM Buffer WHERE to = :to AND timestamp > :timestamp ORDER BY timestamp ASC";
+        HashMap<String, Object> parameters = new HashMap<>();
+        parameters.put("to", toPublicKey);
+        parameters.put("timestamp", fromTimestampMs);
+        List<Packet> packets = new ArrayList<>();
+        try(QuerySession<Buffer> querySession = this.getRepository().buildQuery(hql, parameters)){
+            List<Buffer> buffers = querySession.getQuery().list();
+            for(Buffer buffer : buffers) {
+                byte[] packetBytes = buffer.getPacket();
+                Packet packet = this.packetManager.createPacket(packetBytes);
+                packets.add(packet);
+            }
+        }
+        return packets;
+    }
+
+    /**
+     * Добавить пакет в буфер для клиента с публичным ключом to. Если клиент не авторизован, выбрасывает UnauthorizedExeception
+     * @param from публичный ключ отправителя пакета
+     * @param to публичный ключ получателя пакета
+     * @param packet пакет для добавления в буфер
+     */
+    public void pushPacketToBuffer(String from, String to, Packet packet) {    
+        int packetId = this.packetManager.getPacketIdByClass(packet.getClass());
+        packet.packetId = packetId;
+        byte[] packetBytes = packet.write().getBuffer();
+        Buffer buffer = new Buffer();
+        buffer.setFrom(from);
+        buffer.setTo(to);
+        buffer.setTimestamp(System.currentTimeMillis());
+        buffer.setPacketId(packetId);
+        buffer.setPacket(packetBytes);
+        this.getRepository().save(buffer);
+    }
+
+    /**
+     * Удаляет из буфера все пакеты для определенного клиента с публичным ключом to, которые были добавлены
+     * в буфер после fromTimestampMs и имееют такой же тип пакета как и переданный packet
+     * @param to публичный ключ получателя пакета
+     * @param packet пакет, по типу которого будет происходить удаление из буфера
+     * @param fromTimestampMs метка времени в миллисекундах, после которой были добавлены пакеты, которые нужно удалить
+     */
+    public void deletePacketsFromBuffer(String to, Packet packet, long fromTimestampMs) {
+        int packetId = this.packetManager.getPacketIdByClass(packet.getClass());
+        String hql = "DELETE FROM Buffer WHERE to = :to AND packetId = :packetId AND timestamp > :timestamp";
+        HashMap<String, Object> parameters = new HashMap<>();
+        parameters.put("to", to);
+        parameters.put("packetId", packetId);
+        parameters.put("timestamp", fromTimestampMs);
+        try(QuerySession<?> querySession = this.getRepository().buildQuery(hql, parameters, true)){
+            querySession.getQuery().executeUpdate();
+            querySession.commit();
+        }
+    }
+
+}

src/main/java/im/rosetta/service/services/DeviceService.java

@@ -0,0 +1,48 @@
+package im.rosetta.service.services;
+
+import java.util.List;
+
+import im.rosetta.database.entity.Device;
+import im.rosetta.database.entity.User;
+import im.rosetta.database.repository.DeviceRepository;
+import im.rosetta.service.Service;
+
+public class DeviceService extends Service<DeviceRepository> {
+
+    public DeviceService(DeviceRepository repository) {
+        super(repository);
+    }
+
+    /**
+     * Проверяет, верифицировано ли устройство с deviceId для пользователя user
+     * @param deviceId ID устройства
+     * @param user пользователь
+     * @return true если устройство верифицировано, иначе false
+     */
+    public boolean isDeviceVerifiedByUser(String deviceId, User user) {
+        List<Device> devices = this.getRepository().findAll(user);
+        if(devices.size() == 0) {
+            /**
+             * Если у пользователя нет устройств, значит текущее устройство верифицировано
+             * такого быть не может, это избыточная проверка
+             */
+            return true;
+        }
+        for(Device device : devices) {
+            if(device.getDeviceId().equals(deviceId)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    /**
+     * Получить список устройств для публичного ключа
+     * @param publicKey публичный ключ пользователя, для которого нужно получить список устройств
+     * @return список устройств для публичного ключа
+     */
+    public List<Device> getDevicesByPK(String publicKey) {
+        return this.getRepository().findAllByField("publicKey", publicKey);
+    }
+
+}

src/main/java/im/rosetta/service/services/UserService.java

@@ -0,0 +1,93 @@
+package im.rosetta.service.services;
+
+import java.util.HashMap;
+import java.util.List;
+
+import im.rosetta.client.tags.ECIAuthentificate;
+import im.rosetta.database.QuerySession;
+import im.rosetta.database.entity.User;
+import im.rosetta.database.repository.UserRepository;
+import im.rosetta.service.Service;
+
+import io.orprotocol.client.Client;
+
+public class UserService extends Service<UserRepository> {
+
+    public UserService(UserRepository repository) {
+        super(repository);
+    }
+
+    /**
+     * Поиск пользователей по части имени пользователя и публичному ключу.
+     * @param query часть имени пользователя
+     * @param take сколько пользователей отдать
+     * @return список пользователей, соответствующих запросу
+     */
+    public List<User> searchUsers(String query, int take) {
+        String hql = "FROM User WHERE username LIKE :query OR publicKey = :queryExact ORDER BY verified ASC";
+        HashMap<String, Object> parameters = new HashMap<>();
+        parameters.put("query", "%" + query + "%");
+        parameters.put("queryExact", query);
+        try(QuerySession<User> querySession = this.getRepository().buildQuery(hql, parameters)){
+            return querySession.getQuery().setMaxResults(take).list();
+        }
+    }
+
+    /**
+     * Получает User из клиента, так же на всякий случай проверяется авторизован ли пользователь,
+     * если нет то User не будет найден
+     * @param client сетевой клиент
+     * @return пользователь
+     */
+    public User fromClient(Client client) {
+        ECIAuthentificate eciAuthentificate = client.getTag(ECIAuthentificate.class);
+        if(eciAuthentificate == null){
+            return null;
+        }
+        if(!eciAuthentificate.hasAuthorized()){
+            return null;
+        }
+        return this.getRepository().findByField("publicKey", eciAuthentificate.getPublicKey());
+    }
+
+    /**
+     * Проверяет занятость имени пользователя
+     * @param username имя пользователя
+     * @return true если имя занято, иначе false
+     */
+    public boolean isUsernameTaken(String username) {
+        User user = this.getRepository().findByField("username", username);
+        return user != null;
+    }
+
+    /**
+     * Подписывает пользователя на пуш уведомления, добавляя токен в его список токенов. Если токен уже был добавлен, то ничего не произойдет.
+     * @param user пользователь, которого нужно подписать на пуш уведомления
+     * @param notificationToken токен пуш уведомлений, который нужно добавить пользователю. Если токен уже был добавлен, то ничего не произойдет
+     */
+    public void subscribeToPushNotifications(User user, String notificationToken) {
+        List<String> tokens = user.getNotificationsTokens();
+        if(tokens.contains(notificationToken)){
+            return;
+        }
+        tokens.add(notificationToken);
+        user.setNotificationsTokens(tokens);
+        this.getRepository().update(user);
+    }
+
+    /**
+     * Отписывает пользователя от пуш уведомлений, удаляя токен из его списка токенов. Если токена не было, то ничего не произойдет.
+     * @param user пользователь, которого нужно отписать от пуш уведомлений
+     * @param notificationToken токен пуш уведомлений, который нужно удалить у пользователя. Если токена не было, то ничего не произойдет
+     */
+    public void unsubscribeFromPushNotifications(User user, String notificationToken) {
+        List<String> tokens = user.getNotificationsTokens();
+        if(!tokens.contains(notificationToken)){
+            return;
+        }
+        tokens.remove(notificationToken);
+        user.setNotificationsTokens(tokens);
+        this.getRepository().update(user);
+    }
+
+}