Mercury-IM/app/src/main/java/org/mercury_im/messenger/ui/login/AccountsViewModel.java

package org.mercury_im.messenger.ui.login;

import android.app.Application;
import android.widget.Toast;

import androidx.lifecycle.AndroidViewModel;
import androidx.lifecycle.LiveData;
import androidx.lifecycle.MutableLiveData;
import androidx.lifecycle.ViewModel;

import io.reactivex.Scheduler;
import io.reactivex.android.schedulers.AndroidSchedulers;
import io.reactivex.disposables.CompositeDisposable;
import io.reactivex.functions.Consumer;
import io.reactivex.schedulers.Schedulers;

import org.jivesoftware.smack.XMPPConnection;
import org.mercury_im.messenger.MercuryImApplication;
import org.mercury_im.messenger.core.centers.ConnectionCenter;
import org.mercury_im.messenger.core.connection.MercuryConnection;
import org.mercury_im.messenger.persistence.model.AccountModel;
import org.mercury_im.messenger.persistence.repository.AccountRepository;

import java.util.List;

import javax.inject.Inject;

public class AccountsViewModel extends AndroidViewModel {

    @Inject
    AccountRepository repository;

    @Inject
    ConnectionCenter connectionCenter;

    private final MutableLiveData<List<AccountModel>> accounts = new MutableLiveData<>();
    private final CompositeDisposable compositeDisposable = new CompositeDisposable();

    @Inject
    public AccountsViewModel(Application application) {
        super(application);
        MercuryImApplication.getApplication().getAppComponent().inject(this);
        compositeDisposable.add(repository.getAllAccounts()
                .subscribeOn(Schedulers.io())
                .observeOn(AndroidSchedulers.mainThread())
                .subscribe((Consumer<List<AccountModel>>) accounts::setValue));
    }

    @Override
    protected void onCleared() {
        super.onCleared();
        compositeDisposable.clear();
    }

    public LiveData<List<AccountModel>> getAccounts() {
        return accounts;
    }

    public void toggleAccountEnabled(AccountModel accountModel) {
        MercuryConnection connection = connectionCenter.getConnection(accountModel);
        if (connection == null) {
            Toast.makeText(this.getApplication(), "MercuryConnection is null!", Toast.LENGTH_LONG).show();
            return;
        }

        accountModel.setEnabled(!accountModel.getEnabled());
        repository.updateAccount(accountModel)
                .subscribeOn(Schedulers.io())
                .subscribe();
    }
}