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

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

import java.util.List;

import javax.inject.Inject;

public class RosterViewModel extends AndroidViewModel {

    @Inject
    ContactAndEntityRepository contactAndEntityRepository;

    private final LiveData<List<RoomContactAndEntityModel>> rosterEntryList;

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

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