package com.cappielloantonio.play.viewmodel; import android.app.Application; import androidx.annotation.NonNull; import androidx.lifecycle.AndroidViewModel; import com.cappielloantonio.play.model.Album; import com.cappielloantonio.play.model.Artist; import com.cappielloantonio.play.model.Song; import com.cappielloantonio.play.repository.AlbumRepository; import com.cappielloantonio.play.repository.ArtistRepository; import com.cappielloantonio.play.repository.SongRepository; public class SongBottomSheetViewModel extends AndroidViewModel { private SongRepository songRepository; private AlbumRepository albumRepository; private ArtistRepository artistRepository; private Song song; public SongBottomSheetViewModel(@NonNull Application application) { super(application); songRepository = new SongRepository(application); albumRepository = new AlbumRepository(application); artistRepository = new ArtistRepository(application); } public Song getSong() { return song; } public void setSong(Song song) { this.song = song; } public void setFavorite() { song.setFavorite(!song.isFavorite()); // songRepository.setFavoriteStatus(song); } public Album getAlbum() { // return albumRepository.getAlbumByID(song.getAlbumId()); return null; } public Artist getArtist() { // return artistRepository.getArtistByID(song.getArtistId()); return null; } }