From 7804da72b3c5c9a77d01b4c8095b82adacd9679a Mon Sep 17 00:00:00 2001
From: Luna712 <142361265+Luna712@users.noreply.github.com>
Date: Sat, 6 Jul 2024 19:01:00 -0600
Subject: [PATCH] Fix addSelected and cancel bugs

---
 .../lagradost/cloudstream3/ui/download/DownloadFragment.kt | 4 ++++
 .../cloudstream3/ui/download/DownloadViewModel.kt          | 7 +++----
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadFragment.kt b/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadFragment.kt
index e96d86f3..858af9fe 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadFragment.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadFragment.kt
@@ -200,6 +200,10 @@ class DownloadFragment : Fragment() {
                 context?.let { ctx -> downloadsViewModel.handleMultiDelete(ctx) }
             }
             binding?.downloadDeleteToolbar?.btnCancel?.setOnClickListener {
+                adapter?.setDeleteCheckboxVisibility(false)
+                adapter?.clearSelectedIds()
+                adapter?.notifyItemRangeChanged(0, adapter.itemCount)
+
                 downloadsViewModel.clearSelectedIds()
             }
 
diff --git a/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadViewModel.kt b/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadViewModel.kt
index d656ef04..65370bbd 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadViewModel.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/ui/download/DownloadViewModel.kt
@@ -45,10 +45,9 @@ class DownloadViewModel : ViewModel() {
     private var previousVisual: List<VisualDownloadHeaderCached>? = null
 
     fun addSelected(id: Int, name: String) {
-        selectedIds.value?.let { selectedIds ->
-            selectedIds[id] = name
-            _selectedIds.postValue(selectedIds)
-        }
+        val currentSelected = selectedIds.value ?: HashMap()
+        currentSelected[id] = name
+        _selectedIds.postValue(currentSelected)
     }
 
     fun setSelected(selected: HashMap<Int, String>) {