Add bottomsheet everywhere

app/src/main/java/com/cappielloantonio/play/viewmodel/AlbumBottomSheetViewModel.java

@@ -5,35 +5,30 @@ import android.app.Application;
 import androidx.annotation.NonNull;
 import androidx.lifecycle.AndroidViewModel;
 
-import com.cappielloantonio.play.model.Song;
-import com.cappielloantonio.play.repository.SongRepository;
+import com.cappielloantonio.play.model.Album;
+import com.cappielloantonio.play.model.Artist;
+import com.cappielloantonio.play.repository.ArtistRepository;
 
 public class AlbumBottomSheetViewModel extends AndroidViewModel {
-    private SongRepository songRepository;
-    private Song song;
+    private ArtistRepository artistRepository;
+
+    private Album album;
 
     public AlbumBottomSheetViewModel(@NonNull Application application) {
         super(application);
 
-        songRepository = new SongRepository(application);
+        artistRepository = new ArtistRepository(application);
     }
 
-    public void setSong(Song song) {
-        this.song = song;
+    public Album getAlbum() {
+        return album;
     }
 
-    public Song getSong() {
-        return song;
+    public void setAlbum(Album album) {
+        this.album = album;
     }
 
-    public void setFavorite() {
-        if(song.isFavorite()) {
-            song.setFavorite(false);
-        }
-        else {
-            song.setFavorite(true);
-        }
-
-        songRepository.setFavoriteStatus(song);
+    public Artist getArtist() {
+        return artistRepository.getArtistByID(album.getArtistId());
     }
 }

app/src/main/java/com/cappielloantonio/play/viewmodel/ArtistBottomSheetViewModel.java

@@ -0,0 +1,24 @@
+package com.cappielloantonio.play.viewmodel;
+
+import android.app.Application;
+
+import androidx.annotation.NonNull;
+import androidx.lifecycle.AndroidViewModel;
+
+import com.cappielloantonio.play.model.Artist;
+
+public class ArtistBottomSheetViewModel extends AndroidViewModel {
+    private Artist artist;
+
+    public ArtistBottomSheetViewModel(@NonNull Application application) {
+        super(application);
+    }
+
+    public void setArtist(Artist artist) {
+        this.artist = artist;
+    }
+
+    public Artist getArtist() {
+        return artist;
+    }
+}

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

@@ -0,0 +1,54 @@
+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());
+    }
+}