From 0509686fd90e938b97d15b9d1a16e81f12f6441c Mon Sep 17 00:00:00 2001
From: Prathamesh More <prathameshmm02@gmail.com>
Date: Sun, 20 Mar 2022 12:46:21 +0530
Subject: [PATCH] Code cleanup

---
 .../name/monkey/retromusic/dialogs/AddToPlaylistDialog.kt  | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/code/name/monkey/retromusic/dialogs/AddToPlaylistDialog.kt b/app/src/main/java/code/name/monkey/retromusic/dialogs/AddToPlaylistDialog.kt
index 1557e0212..1440270c0 100644
--- a/app/src/main/java/code/name/monkey/retromusic/dialogs/AddToPlaylistDialog.kt
+++ b/app/src/main/java/code/name/monkey/retromusic/dialogs/AddToPlaylistDialog.kt
@@ -23,8 +23,6 @@ import code.name.monkey.retromusic.EXTRA_PLAYLISTS
 import code.name.monkey.retromusic.EXTRA_SONG
 import code.name.monkey.retromusic.R
 import code.name.monkey.retromusic.db.PlaylistEntity
-import code.name.monkey.retromusic.db.toSongs
-import code.name.monkey.retromusic.db.toSongsEntity
 import code.name.monkey.retromusic.extensions.colorButtons
 import code.name.monkey.retromusic.extensions.extraNotNull
 import code.name.monkey.retromusic.extensions.materialDialog
@@ -73,12 +71,11 @@ class AddToPlaylistDialog : DialogFragment() {
                 if (which == 0) {
                     showCreateDialog(songs)
                 } else {
-                    val playlistName = playlistEntities[which - 1].playlistName
-                    val songEntities = songs.toSongsEntity(playlistEntities[which - 1])
-                    libraryViewModel.addToPlaylist(playlistName, songEntities.toSongs())
+                    libraryViewModel.addToPlaylist(playlistNames[which], songs)
                 }
                 dialog.dismiss()
             }
+            .setNegativeButton(R.string.action_cancel, null)
             .create().colorButtons()
     }