From 8f7fe94d21e5ca109228457a888e17c7de37d5b1 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
                         }
                     }
                 }