104 lines
4.3 KiB
Java
104 lines
4.3 KiB
Java
package org.mercury_im.messenger.core.viewmodel.chat;
|
|
|
|
import org.jxmpp.jid.EntityBareJid;
|
|
import org.mercury_im.messenger.core.Messenger;
|
|
import org.mercury_im.messenger.core.SchedulersFacade;
|
|
import org.mercury_im.messenger.core.data.repository.DirectChatRepository;
|
|
import org.mercury_im.messenger.core.data.repository.MessageRepository;
|
|
import org.mercury_im.messenger.core.data.repository.PeerRepository;
|
|
import org.mercury_im.messenger.core.util.Optional;
|
|
import org.mercury_im.messenger.core.viewmodel.MercuryViewModel;
|
|
import org.mercury_im.messenger.entity.chat.DirectChat;
|
|
import org.mercury_im.messenger.entity.contact.Peer;
|
|
import org.mercury_im.messenger.entity.message.Message;
|
|
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
|
|
import io.reactivex.Observable;
|
|
import io.reactivex.Single;
|
|
import io.reactivex.subjects.BehaviorSubject;
|
|
import lombok.Getter;
|
|
|
|
public class ChatViewModel implements MercuryViewModel {
|
|
|
|
private static final Logger LOGGER = Logger.getLogger(ChatViewModel.class.getName());
|
|
|
|
private final Messenger messenger;
|
|
private final PeerRepository contactRepository;
|
|
private final DirectChatRepository directChatRepository;
|
|
private final MessageRepository messageRepository;
|
|
private final SchedulersFacade schedulers;
|
|
|
|
@Getter
|
|
private BehaviorSubject<Optional<Peer>> peer = BehaviorSubject.create();
|
|
|
|
private Single<DirectChat> directChat;
|
|
|
|
@Getter
|
|
private BehaviorSubject<DirectChat> chat = BehaviorSubject.create();
|
|
|
|
@Getter
|
|
private Observable<List<Message>> messages;
|
|
|
|
@Getter
|
|
private Observable<String> contactDisplayName;
|
|
|
|
private final BehaviorSubject<Observable<List<Message>>> messageQueryObservable = BehaviorSubject.create();
|
|
|
|
|
|
public ChatViewModel(Messenger messenger,
|
|
PeerRepository peerRepository,
|
|
DirectChatRepository directChatRepository,
|
|
MessageRepository messageRepository,
|
|
SchedulersFacade schedulers) {
|
|
this.messenger = messenger;
|
|
this.contactRepository = peerRepository;
|
|
this.directChatRepository = directChatRepository;
|
|
this.messageRepository = messageRepository;
|
|
this.schedulers = schedulers;
|
|
}
|
|
|
|
public void init(UUID accountId, EntityBareJid contactJid) {
|
|
Single<Peer> peerSingle = contactRepository.getOrCreatePeer(accountId, contactJid);
|
|
peerSingle.flatMapObservable(contactRepository::observePeer).subscribe(peer);
|
|
directChat = peerSingle.flatMap(directChatRepository::getOrCreateChatWithPeer);
|
|
|
|
directChat.toObservable().compose(schedulers.executeUiSafeObservable()).subscribe(chat);
|
|
|
|
Observable<List<Message>> allMessagesObservable = directChat.flatMapObservable(messageRepository::observeMessages);
|
|
messageQueryObservable.onNext(allMessagesObservable);
|
|
messages = Observable.switchOnNext(messageQueryObservable);
|
|
contactDisplayName = peer.filter(Optional::isPresent).map(Optional::getItem)
|
|
.map(Peer::getDisplayName);
|
|
}
|
|
|
|
public void onQueryTextChanged(String query) {
|
|
if (query.trim().isEmpty()) {
|
|
messageQueryObservable.onNext(directChat.flatMapObservable(messageRepository::observeMessages));
|
|
} else {
|
|
messageQueryObservable.onNext(directChat.flatMapObservable(c -> messageRepository.findMessagesWithBody(c, query)));
|
|
}
|
|
}
|
|
|
|
public void deleteContact() {
|
|
addDisposable(peer.single(new Optional<>())
|
|
.filter(Optional::isPresent)
|
|
.map(Optional::getItem)
|
|
.flatMapCompletable(messenger::deleteContact)
|
|
.compose(schedulers.executeUiSafeCompletable())
|
|
.subscribe(
|
|
() -> LOGGER.log(Level.INFO, "Contact deleted."),
|
|
e -> LOGGER.log(Level.SEVERE, "Error deleting contact.", e)));
|
|
}
|
|
|
|
public void sendMessage(String body) {
|
|
addDisposable(messenger.sendEncryptedMessage(peer.getValue().getItem(), body)
|
|
.compose(schedulers.executeUiSafeCompletable())
|
|
.subscribe(() -> LOGGER.log(Level.INFO, "Successfully sent encrypted message."),
|
|
e -> LOGGER.log(Level.SEVERE, "Error sending encrypted message.", e)));
|
|
}
|
|
}
|