Fix wrong download state (#418)

* fix: wrong download state * fix: incorect state of download all switch
openedx · May 3, 2024 · 90bde33 · 90bde33
1 parent de1f373
commit 90bde33
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/...se/Presentation/Subviews/CourseVideoDownloadBarView/CourseVideoDownloadBarViewModel.swift b/...se/Presentation/Subviews/CourseVideoDownloadBarView/CourseVideoDownloadBarViewModel.swift
@@ -194,17 +194,17 @@ final class CourseVideoDownloadBarViewModel: ObservableObject {
 
     // MARK: - Private intents
 
-    private func toggleStateIsOn() {
-        let totalCount = courseViewModel.downloadableVerticals.count
-        let availableCount = courseViewModel.downloadableVerticals.filter { $0.state == .available }.count
-        let finishedCount = courseViewModel.downloadableVerticals.filter { $0.state == .finished }.count
-        let downloadingCount = courseViewModel.downloadableVerticals.filter { $0.state == .downloading }.count
+    private func toggleStateIsOn(downloadableVerticals: Set<VerticalsDownloadState>) {
+        let totalCount = downloadableVerticals.count
+        let availableCount = downloadableVerticals.filter { $0.state == .available }.count
+        let finishedCount = downloadableVerticals.filter { $0.state == .finished }.count
+        let downloadingCount = downloadableVerticals.filter { $0.state == .downloading }.count
 
-        if downloadingCount == totalCount {
+        if downloadingCount == totalCount, totalCount > 0 {
             self.isOn = true
             return
         }
-        if totalCount == finishedCount {
+        if totalCount == finishedCount, totalCount > 0 {
             self.isOn = true
             return
         }
@@ -217,18 +217,18 @@ final class CourseVideoDownloadBarViewModel: ObservableObject {
             return
         }
 
-        let isOn = totalCount - finishedCount == downloadingCount
+        let isOn = totalCount - finishedCount == downloadingCount && totalCount > 0
         self.isOn = isOn
     }
 
     private func observers() {
         currentDownloadTask = courseViewModel.manager.currentDownloadTask
-        toggleStateIsOn()
+        toggleStateIsOn(downloadableVerticals: courseViewModel.downloadableVerticals)
         courseViewModel.$downloadableVerticals
-            .sink { [weak self] _ in
+            .sink { [weak self] value in
                 guard let self else { return }
                 self.currentDownloadTask = self.courseViewModel.manager.currentDownloadTask
-                self.toggleStateIsOn()
+                self.toggleStateIsOn(downloadableVerticals: value)
         }
         .store(in: &cancellables)
         courseViewModel.manager.eventPublisher()
@@ -237,7 +237,7 @@ final class CourseVideoDownloadBarViewModel: ObservableObject {
                 if case .progress = state {
                     self.currentDownloadTask = self.courseViewModel.manager.currentDownloadTask
                 }
-                self.toggleStateIsOn()
+                self.toggleStateIsOn(downloadableVerticals: self.courseViewModel.downloadableVerticals)
             }
             .store(in: &cancellables)
     }