From 9e839feef30cb1ee28698de1a73e848507b20568 Mon Sep 17 00:00:00 2001
From: vfsfitvnm <vfsfitvnm@protonmail.com>
Date: Wed, 3 Aug 2022 21:44:10 +0200
Subject: [PATCH] Implement #127

---
 .../ui/components/themed/MediaItemMenu.kt        | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/components/themed/MediaItemMenu.kt b/app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/components/themed/MediaItemMenu.kt
index 0f13a21..a528d12 100644
--- a/app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/components/themed/MediaItemMenu.kt
+++ b/app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/components/themed/MediaItemMenu.kt
@@ -70,7 +70,7 @@ fun InFavoritesMediaItemMenu(
         onDismiss = onDismiss,
         onRemoveFromFavorites = {
             query {
-                Database.like(song.id, if (song.likedAt == null) System.currentTimeMillis() else null)
+                Database.like(song.id, null)
             }
         },
         modifier = modifier
@@ -341,6 +341,20 @@ fun MediaItemMenu(
                     }
 
                     onAddToPlaylist?.let { onAddToPlaylist ->
+                        if (onRemoveFromFavorites == null) {
+                            MenuEntry(
+                                icon = R.drawable.heart,
+                                text = "Favorites",
+                                onClick = {
+                                    onDismiss()
+                                    query {
+                                        Database.insert(mediaItem)
+                                        Database.like(mediaItem.mediaId, System.currentTimeMillis())
+                                    }
+                                }
+                            )
+                        }
+
                         playlistPreviews.forEach { playlistPreview ->
                             MenuEntry(
                                 icon = R.drawable.playlist,