tempus/app/src/main/java/com/cappielloantonio/play/viewmodel/HomeViewModel.java

package com.cappielloantonio.play.viewmodel;

import android.app.Application;

import androidx.annotation.NonNull;
import androidx.lifecycle.AndroidViewModel;
import androidx.lifecycle.LiveData;
import androidx.lifecycle.ViewModel;

import com.cappielloantonio.play.model.Song;
import com.cappielloantonio.play.repository.SongRepository;

import java.util.ArrayList;
import java.util.List;

public class HomeViewModel extends AndroidViewModel {
    private SongRepository songRepository;

    private LiveData<List<Song>> dicoverSongSample;
    private LiveData<List<Song>> recentlyPlayedSongSample;
    private LiveData<List<Song>> recentlyAddedSongSample;
    private LiveData<List<Song>> mostPlayedSongSample;

    public HomeViewModel(@NonNull Application application) {
        super(application);

        songRepository = new SongRepository(application);

        dicoverSongSample = songRepository.getListLiveDiscoverSampleSong();
        recentlyPlayedSongSample = songRepository.getListLiveRecentlyPlayedSampleSong();
        recentlyAddedSongSample = songRepository.getListLiveRecentlyAddedSampleSong();
        mostPlayedSongSample = songRepository.getListLiveMostPlayedSampleSong();
    }


    public LiveData<List<Song>> getDiscoverSongList() {
        return dicoverSongSample;
    }

    public LiveData<List<Song>> getRecentlyAddedSongList() {
        return recentlyAddedSongSample;
    }

    public LiveData<List<Song>> getRecentlyPlayedSongList() {
        return recentlyPlayedSongSample;
    }

    public LiveData<List<Song>> getMostPlayedSongList() {
        return mostPlayedSongSample;
    }
}