package com.cappielloantonio.play.viewmodel; import android.app.Application; import androidx.annotation.NonNull; import androidx.lifecycle.AndroidViewModel; import androidx.lifecycle.LiveData; import com.cappielloantonio.play.model.Song; import com.cappielloantonio.play.repository.SongRepository; import java.util.List; public class HomeViewModel extends AndroidViewModel { private static final String TAG = "HomeViewModel"; private SongRepository songRepository; private List dicoverSongSample; private LiveData> recentlyPlayedSongSample; private LiveData> recentlyAddedSongSample; private LiveData> mostPlayedSongSample; public HomeViewModel(@NonNull Application application) { super(application); songRepository = new SongRepository(application); dicoverSongSample = songRepository.getRandomSample(5); recentlyPlayedSongSample = songRepository.getListLiveRecentlyPlayedSampleSong(20); recentlyAddedSongSample = songRepository.getListLiveRecentlyAddedSampleSong(20); mostPlayedSongSample = songRepository.getListLiveMostPlayedSampleSong(20); } public List getDiscoverSongList() { if(dicoverSongSample.isEmpty()) { dicoverSongSample = songRepository.getRandomSample(5); } return dicoverSongSample; } public LiveData> getRecentlyAddedSongList() { return recentlyAddedSongSample; } public LiveData> getRecentlyPlayedSongList() { return recentlyPlayedSongSample; } public LiveData> getMostPlayedSongList() { return mostPlayedSongSample; } }