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

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 void setSong(Song song) {
        this.song = song;
    }

    public Song getSong() {
        return song;
    }

    public void setFavorite() {
        if(song.isFavorite())
            song.setFavorite(false);
        else
            song.setFavorite(true);

        songRepository.setFavoriteStatus(song);
    }

    public Album getAlbum() {
        return albumRepository.getAlbumByID(song.getAlbumId());
    }

    public Artist getArtist() {
        return artistRepository.getArtistByID(song.getArtistId());
    }
}