Mercury-IM/app/src/main/java/org/mercury_im/messenger/ui/roster/RosterViewModel.java

package org.mercury_im.messenger.ui.roster;

import android.app.Application;
import android.arch.lifecycle.AndroidViewModel;
import android.arch.lifecycle.LiveData;
import android.support.annotation.NonNull;

import org.mercury_im.messenger.MercuryImApplication;
import org.mercury_im.messenger.persistence.database.model.RosterEntryModel;
import org.mercury_im.messenger.persistence.repository.roster.RosterEntryRepository;

import java.util.List;

import javax.inject.Inject;

public class RosterViewModel extends AndroidViewModel {

    @Inject
    RosterEntryRepository rosterEntryRepository;

    private final LiveData<List<RosterEntryModel>> rosterEntryList;

    @Inject
    public RosterViewModel(@NonNull Application application) {
        super(application);
        MercuryImApplication.getApplication().getAppComponent().inject(this);
        this.rosterEntryList = rosterEntryRepository.getAllRosterEntries();
    }

    public LiveData<List<RosterEntryModel>> getRosterEntryList() {
        return rosterEntryList;
    }
}