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 androidx.annotation.NonNull;
import androidx.lifecycle.AndroidViewModel;
import androidx.lifecycle.LiveData;

import org.mercury_im.messenger.MercuryImApplication;
import org.mercury_im.messenger.persistence.repository.ContactRepository;
import org.mercury_im.messenger.persistence.room.model.RoomContactModel;

import java.util.List;

import javax.inject.Inject;

public class RosterViewModel extends AndroidViewModel {

    @Inject
    ContactRepository contactRepository;

    private final LiveData<List<RoomContactModel>> rosterEntryList;

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

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