Mercury-IM/app/src/main/java/org/mercury_im/messenger/android/ui/account/AndroidAccountDetailsViewModel.java

package org.mercury_im.messenger.android.ui.account;

import android.app.Application;

import androidx.annotation.NonNull;
import androidx.lifecycle.AndroidViewModel;
import androidx.lifecycle.LiveData;
import androidx.lifecycle.MutableLiveData;
import androidx.lifecycle.ViewModel;
import androidx.lifecycle.ViewModelProvider;

import org.jxmpp.jid.EntityBareJid;
import org.mercury_im.messenger.android.MercuryImApplication;
import org.mercury_im.messenger.android.ui.MercuryAndroidViewModel;
import org.mercury_im.messenger.core.SchedulersFacade;
import org.mercury_im.messenger.core.util.Optional;
import org.mercury_im.messenger.core.viewmodel.accounts.AccountDetailsViewModel;
import org.mercury_im.messenger.core.viewmodel.accounts.AccountFingerprintViewItem;
import org.pgpainless.key.OpenPgpV4Fingerprint;

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

import javax.inject.Inject;

public class AndroidAccountDetailsViewModel extends AndroidViewModel implements MercuryAndroidViewModel<AccountDetailsViewModel> {

    @Inject
    SchedulersFacade schedulers;

    @Inject
    AccountDetailsViewModel commonViewModel;

    private final UUID accountId;
    private MutableLiveData<OpenPgpV4Fingerprint> localFingerprint = new MutableLiveData<>();
    private MutableLiveData<List<AccountFingerprintViewItem>> remoteFingerprints = new MutableLiveData<>(new ArrayList<>());
    private MutableLiveData<EntityBareJid> jid;

    public AndroidAccountDetailsViewModel(@NonNull Application application, UUID accountId) {
        super(application);
        this.accountId = accountId;

        ((MercuryImApplication) application).getAppComponent().inject(this);

        addDisposable(getCommonViewModel().observeLocalFingerprint(accountId)
                .compose(schedulers.executeUiSafeObservable())
                .filter(Optional::isPresent)
                .map(Optional::getItem)
                .subscribe(localFingerprint::setValue));

        addDisposable(getCommonViewModel().observeRemoteFingerprints(accountId)
                .compose(schedulers.executeUiSafeObservable())
                .subscribe(list -> remoteFingerprints.setValue(list)));

        //addDisposable(getCommonViewModel().getJid(accountId).subscribe(jid::setValue));
    }

    @Override
    public AccountDetailsViewModel getCommonViewModel() {
        return commonViewModel;
    }

    public LiveData<EntityBareJid> getJid() {
        return jid;
    }

    public void markFingerprintTrusted(OpenPgpV4Fingerprint fingerprint, boolean trusted) {
        getCommonViewModel().markFingerprintTrusted(accountId, fingerprint, trusted);
    }

    public LiveData<OpenPgpV4Fingerprint> getLocalFingerprint() {
        return localFingerprint;
    }

    public LiveData<List<AccountFingerprintViewItem>> getRemoteFingerprints() {
        return remoteFingerprints;
    }

    public static class AndroidAccountDetailsViewModelFactory implements ViewModelProvider.Factory {

        private final Application application;
        private final UUID accountId;

        public AndroidAccountDetailsViewModelFactory(Application application, UUID accountId) {
            this.application = application;
            this.accountId = accountId;
        }

        @NonNull
        @Override
        public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
            return (T) new AndroidAccountDetailsViewModel(application, accountId);
        }
    }
}