From aa0712b967291d423454b8a7b2a274fe3f895a2a Mon Sep 17 00:00:00 2001
From: kingbri <bdashore3@proton.me>
Date: Thu, 13 Jun 2024 09:44:13 -0400
Subject: [PATCH] Debrid: Fix cleanup of unrestrict and task

The task can be set to nil after completion, and the unrestrict
flag should also be set to nil when the batch sheet is dismissed.

Signed-off-by: kingbri <bdashore3@proton.me>
---
 Ferrite/ViewModels/DebridManager.swift         | 1 +
 Ferrite/Views/SheetViews/BatchChoiceView.swift | 1 +
 2 files changed, 2 insertions(+)

diff --git a/Ferrite/ViewModels/DebridManager.swift b/Ferrite/ViewModels/DebridManager.swift
index 01cd4b1..823b228 100644
--- a/Ferrite/ViewModels/DebridManager.swift
+++ b/Ferrite/ViewModels/DebridManager.swift
@@ -375,6 +375,7 @@ class DebridManager: ObservableObject {
         defer {
             logManager?.hideIndeterminateToast()
             requiresUnrestrict = false
+            currentDebridTask = nil
         }
 
         logManager?.updateIndeterminateToast("Loading content", cancelAction: {
diff --git a/Ferrite/Views/SheetViews/BatchChoiceView.swift b/Ferrite/Views/SheetViews/BatchChoiceView.swift
index 81d7dca..a18d355 100644
--- a/Ferrite/Views/SheetViews/BatchChoiceView.swift
+++ b/Ferrite/Views/SheetViews/BatchChoiceView.swift
@@ -50,6 +50,7 @@ struct BatchChoiceView: View {
                             try? await Task.sleep(seconds: 1)
 
                             debridManager.clearSelectedDebridItems()
+                            debridManager.requiresUnrestrict = false
                         }
                     }
                 }