Mercury-IM/data/src/main/java/org/mercury_im/messenger/data/repository/RxGroupChatRepository.java

package org.mercury_im.messenger.data.repository;

import org.mercury_im.messenger.core.data.repository.GroupChatRepository;
import org.mercury_im.messenger.core.util.Optional;
import org.mercury_im.messenger.data.mapping.GroupChatMapping;
import org.mercury_im.messenger.data.model.GroupChatModel;
import org.mercury_im.messenger.data.repository.dao.GroupChatDao;
import org.mercury_im.messenger.entity.Account;
import org.mercury_im.messenger.entity.chat.GroupChat;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import javax.inject.Inject;

import io.reactivex.Completable;
import io.reactivex.Maybe;
import io.reactivex.Observable;
import io.reactivex.Single;
import io.requery.Persistable;
import io.requery.query.ResultDelegate;
import io.requery.reactivex.ReactiveEntityStore;

public class RxGroupChatRepository
        extends RequeryRepository
        implements GroupChatRepository {

    private final GroupChatMapping groupChatMapping;

    private final GroupChatDao dao;

    @Inject
    public RxGroupChatRepository(
            ReactiveEntityStore<Persistable> data,
            GroupChatMapping groupChatMapping) {
        super(data);
        this.groupChatMapping = groupChatMapping;
        this.dao = new GroupChatDao(data);
    }

    @Override
    public Single<GroupChat> insertGroupChat(GroupChat chat) {
        return Single.just(chat)
                .map(groupChatMapping::toModel)
                .flatMap(dao::insert)
                .map(model -> groupChatMapping.toEntity(model, chat));
    }

    @Override
    public Observable<Optional<GroupChat>> observeGroupChat(UUID chatId) {
        return dao.get(chatId).observableResult()
                .map(result -> new Optional<>(result.firstOrNull()))
                .map(groupChatMapping::toEntity);
    }

    @Override
    public Maybe<GroupChat> getGroupChat(UUID chatId) {
        return dao.get(chatId).maybe()
                .map(groupChatMapping::toEntity);
    }

    @Override
    public Single<GroupChat> getOrCreateGroupChat(Account account, String roomAddress) {
        return getGroupChatByRoomAddress(account, roomAddress)
                .switchIfEmpty(Single.just(new GroupChat())
                        .map(chat -> {
                            chat.setAccount(account);
                            chat.setRoomAddress(roomAddress);
                            return chat;
                        })
                        .flatMap(this::insertGroupChat));
    }

    @Override
    public Observable<Optional<GroupChat>> observeGroupChatByRoomAddress(UUID accountId, String roomAddress) {
        return dao.get(accountId, roomAddress).observableResult()
                .map(result -> new Optional<>(result.firstOrNull()))
                .map(groupChatMapping::toEntity);
    }

    @Override
    public Maybe<GroupChat> getGroupChatByRoomAddress(UUID accountId, String roomAddress) {
        return dao.get(accountId, roomAddress).maybe()
                .map(groupChatMapping::toEntity);
    }

    @Override
    public Observable<List<GroupChat>> observeAllGroupChats() {
        return dao.getAll().observableResult()
                .map(ResultDelegate::toList)
                .map(this::modelsToEntities);
    }

    @Override
    public Observable<List<GroupChat>> observeAllGroupChatsOfAccount(UUID accountId) {
        return data().select(GroupChatModel.class)
                .where(GroupChatModel.ACCOUNT_ID.eq(accountId))
                .get().observableResult()
                .map(ResultDelegate::toList)
                .map(this::modelsToEntities);
    }

    private List<GroupChat> modelsToEntities(List<GroupChatModel> models) {
        List<GroupChat> entities = new ArrayList<>(models.size());
        for (GroupChatModel model : models) {
            entities.add(groupChatMapping.toEntity(model));
        }
        return entities;
    }

    @Override
    public Single<GroupChat> updateGroupChat(GroupChat chat) {
        return dao.get(chat.getId()).maybe().toSingle()
                .map(model -> groupChatMapping.toModel(chat, model))
                .flatMap(data()::update)
                .map(model -> groupChatMapping.toEntity(model, chat));
    }

    @Override
    public Single<GroupChat> upsertGroupChat(GroupChat chat) {
        return dao.get(chat.getId()).maybe()
                .switchIfEmpty(Single.just(chat).map(groupChatMapping::toModel)
                        .flatMap(dao::insert))
                .map(model -> groupChatMapping.toModel(chat, model))
                .flatMap(data()::update)
                .map(model -> groupChatMapping.toEntity(model, chat));
    }

    @Override
    public Completable deleteGroupChat(UUID chatId) {
        return dao.delete(chatId).ignoreElement();
    }
}