Add logic for download actions

Automattic · Oct 24, 2024 · 0961656 · 0961656
1 parent cf658d5
commit 0961656
Show file tree

Hide file tree

Showing 7 changed files with 71 additions and 7 deletions.
diff --git a/Modules/DataModel/Sources/PocketCastsDataModel/Public/Model/BaseEpisode.swift b/Modules/DataModel/Sources/PocketCastsDataModel/Public/Model/BaseEpisode.swift
@@ -67,3 +67,9 @@ import Foundation
     /// Whether this is a regular episode, or an user episode (File)
     var isUserEpisode: Bool { get }
 }
+
+extension BaseEpisode {
+    public var isInDownloadProcess: Bool {
+        return downloading() || queued() || waitingForWifi()
+    }
+}
diff --git a/podcasts/Enumerations.swift b/podcasts/Enumerations.swift
@@ -350,7 +350,10 @@ extension PlayerAction: AnalyticsDescribable {
         case .transcript:
             return L10n.transcript
         case .download:
-            return L10n.download
+            guard let episode else {
+                return L10n.download
+            }
+            return episode.downloaded(pathFinder: DownloadManager.shared) ? L10n.removeDownload : (episode.isInDownloadProcess ? L10n.statusDownloading : L10n.download)
         }
     }
 
@@ -390,7 +393,10 @@ extension PlayerAction: AnalyticsDescribable {
         case .transcript:
             return "transcript"
         case .download:
-            return "download"
+            guard let episode else {
+                return "episode-download"
+            }
+            return episode.downloaded(pathFinder: DownloadManager.shared) ? "episode-downloaded" : "episode-download"
         }
     }
 
@@ -419,7 +425,10 @@ extension PlayerAction: AnalyticsDescribable {
         case .transcript:
             return "transcript"
         case .download:
-            return "download"
+            guard let episode else {
+                return "episode-download"
+            }
+            return episode.downloaded(pathFinder: DownloadManager.shared) ? "episode-downloaded" : "episode-download"
         }
     }
 

diff --git a/podcasts/NowPlayingPlayerItemViewController+Shelf.swift b/podcasts/NowPlayingPlayerItemViewController+Shelf.swift
@@ -28,7 +28,7 @@ extension NowPlayingPlayerItemViewController: NowPlayingActionsDelegate {
         let actions = Settings.playerActions()
 
         // don't reload the actions unless we need to
-        if !lastShelfLoadState.updateRequired(shelfActions: actions, episodeUuid: playingEpisode.uuid, effectsOn: PlaybackManager.shared.effects().effectsEnabled(), sleepTimerOn: PlaybackManager.shared.sleepTimerActive(), episodeStarred: playingEpisode.keepEpisode) { return }
+        if !lastShelfLoadState.updateRequired(shelfActions: actions, episodeUuid: playingEpisode.uuid, effectsOn: PlaybackManager.shared.effects().effectsEnabled(), sleepTimerOn: PlaybackManager.shared.sleepTimerActive(), episodeStarred: playingEpisode.keepEpisode, episodeStatus: playingEpisode.episodeStatus) { return }
 
         // load the first 4 actions into the player, followed by an overflow icon
         playerControlsStackView.removeAllSubviews()
@@ -246,7 +246,38 @@ extension NowPlayingPlayerItemViewController: NowPlayingActionsDelegate {
     }
 
     func downloadTapped() {
-        return
+        guard let episode = PlaybackManager.shared.currentEpisode() as? Episode else { return }
+
+        AnalyticsEpisodeHelper.shared.currentSource = analyticsSource
+
+        if episode.downloaded(pathFinder: DownloadManager.shared) {
+            let confirmation = OptionsPicker(title: L10n.podcastDetailsRemoveDownload)
+            let yesAction = OptionAction(label: L10n.remove, icon: nil) {
+                self.deleteDownloadedFile()
+                Toast.show(L10n.playerEpisodeWasRemoved)
+            }
+            yesAction.destructive = true
+            confirmation.addAction(action: yesAction)
+
+            confirmation.show(statusBarStyle: preferredStatusBarStyle)
+        } else if episode.isInDownloadProcess {
+            PlaybackActionHelper.stopDownload(episodeUuid: episode.uuid)
+            Toast.show(L10n.playerEpisodeDownloadCancelled)
+        } else {
+            PlaybackActionHelper.download(episodeUuid: episode.uuid)
+            Toast.show(L10n.playerEpisodeQueuedForDownload)
+        }
+    }
+
+    private func deleteDownloadedFile() {
+        guard let episode = PlaybackManager.shared.currentEpisode() as? Episode else { return }
+
+        EpisodeManager.analyticsHelper.currentSource = analyticsSource
+
+        PlaybackManager.shared.removeIfPlayingOrQueued(episode: episode, fireNotification: true, userInitiated: false)
+        EpisodeManager.deleteDownloadedFiles(episode: episode, userInitated: true)
+
+        NotificationCenter.postOnMainThread(notification: Constants.Notifications.episodeDownloadStatusChanged, object: episode.uuid)
     }
 
     // MARK: - Player Actions

diff --git a/podcasts/NowPlayingPlayerItemViewController+Update.swift b/podcasts/NowPlayingPlayerItemViewController+Update.swift
@@ -20,6 +20,7 @@ extension NowPlayingPlayerItemViewController {
         addCustomObserver(Constants.Notifications.playerActionsUpdated, selector: #selector(reloadShelfActions))
         addCustomObserver(UIApplication.willEnterForegroundNotification, selector: #selector(update))
         addCustomObserver(Constants.Notifications.episodeStarredChanged, selector: #selector(reloadShelfActions))
+        addCustomObserver(Constants.Notifications.episodeDownloadStatusChanged, selector: #selector(reloadShelfActions))
     }
 
     @objc private func playbackTrackChanged() {

diff --git a/podcasts/ShelfLoadState.swift b/podcasts/ShelfLoadState.swift
@@ -7,9 +7,10 @@ struct ShelfLoadState {
     private var effectsAreOn = false
     private var sleepTimerIsOn = false
     private var episodeIsStarred = false
+    private var episodeStatus: Int32 = 0
 
-    mutating func updateRequired(shelfActions: [PlayerAction], episodeUuid: String, effectsOn: Bool, sleepTimerOn: Bool, episodeStarred: Bool) -> Bool {
-        if lastShelfActionsLoaded == shelfActions, lastShelfEpisodeUuid == episodeUuid, effectsAreOn == effectsOn, sleepTimerIsOn == sleepTimerOn, episodeIsStarred == episodeStarred {
+    mutating func updateRequired(shelfActions: [PlayerAction], episodeUuid: String, effectsOn: Bool, sleepTimerOn: Bool, episodeStarred: Bool, episodeStatus: Int32) -> Bool {
+        if lastShelfActionsLoaded == shelfActions, lastShelfEpisodeUuid == episodeUuid, effectsAreOn == effectsOn, sleepTimerIsOn == sleepTimerOn, episodeIsStarred == episodeStarred, episodeStatus == self.episodeStatus {
             return false
         }
 
@@ -18,6 +19,7 @@ struct ShelfLoadState {
         effectsAreOn = effectsOn
         sleepTimerIsOn = sleepTimerOn
         episodeIsStarred = episodeStarred
+        self.episodeStatus = episodeStatus
 
         return true
     }

diff --git a/podcasts/Strings+Generated.swift b/podcasts/Strings+Generated.swift
diff --git a/podcasts/en.lproj/Localizable.strings b/podcasts/en.lproj/Localizable.strings
@@ -4508,4 +4508,13 @@
 /* Auto Downloads Setting - Limits downloads to a number of show episodes. `%1$@' is a placeholder for the number of episodes*/
 "auto_download_limit_number_of_episodes_show" = "%1$@ Latest Episodes per Show";
 
+/* Toast message when episode download is removed*/
+"player_episode_was_removed" = "Episode was removed";
+
+/* Toast message when episode is queued for download*/
+"player_episode_queued_for_download" = "Episode queued for download";
+
+/* Toast message when episode download is cancelled*/
+"player_episode_download_cancelled" = "Episode download cancelled";
+