Merge pull request #143 from tanhakabir/issue-138

Experimental set download location
tanhakabir · Aug 17, 2021 · 43e887b · 43e887b
2 parents abb0a29 + 006b94e
commit 43e887b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/Source/Model/AudioDataManager.swift b/Source/Model/AudioDataManager.swift
@@ -30,10 +30,12 @@ protocol AudioDataManagable {
     var numberOfActive: Int { get }
 
     var allowCellular: Bool { get set }
+    var downloadDirectory: FileManager.SearchPathDirectory { get }
 
     func setHTTPHeaderFields(_ fields: [String: String]?)
     func setBackgroundCompletionHandler(_ completionHandler: @escaping () -> ())
     func setAllowCellularDownloadPreference(_ preference: Bool)
+    func setDownloadDirectory(_ dir: FileManager.SearchPathDirectory)
 
     func clear()
 
@@ -54,6 +56,7 @@ protocol AudioDataManagable {
 
 class AudioDataManager: AudioDataManagable {
     var allowCellular: Bool = true
+    var downloadDirectory: FileManager.SearchPathDirectory = .documentDirectory
 
     static let shared: AudioDataManagable = AudioDataManager()
 
@@ -110,6 +113,10 @@ class AudioDataManager: AudioDataManagable {
         allowCellular = preference
     }
 
+    func setDownloadDirectory(_ dir: FileManager.SearchPathDirectory) {
+        downloadDirectory = dir
+    }
+
     func attach(callback: @escaping (_ id: ID, _ progress: Double)->()) {
         globalDownloadProgressCallback = callback
     }

diff --git a/Source/Model/Downloading/FileStorage.swift b/Source/Model/Downloading/FileStorage.swift
@@ -64,7 +64,7 @@ struct FileStorage {
 // MARK:- Audio
 extension FileStorage {
     struct Audio {
-        private static let directory: FileManager.SearchPathDirectory = .documentDirectory
+        private static let directory: FileManager.SearchPathDirectory = AudioDataManager.shared.downloadDirectory
         private init() {}
 
         static func isStored(_ id: ID) -> Bool {
@@ -103,7 +103,7 @@ extension FileStorage {
         }
 
         static func locate(_ id: ID) -> URL? {
-            let folderUrls = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)
+            let folderUrls = FileManager.default.urls(for: directory, in: .userDomainMask)
             guard folderUrls.count != 0 else { return nil }
 
             if let urls = try? FileManager.default.contentsOfDirectory(at: folderUrls[0], includingPropertiesForKeys: nil) {

diff --git a/Source/SAPlayerDownloader.swift b/Source/SAPlayerDownloader.swift
@@ -109,5 +109,14 @@ extension SAPlayer {
                 AudioDataManager.shared.setAllowCellularDownloadPreference(allowUsingCellularData)
             }
         }
+
+        /**
+         EXPERIMENTAL!
+         */
+        public static var downloadDirectory: FileManager.SearchPathDirectory = .documentDirectory {
+            didSet {
+                AudioDataManager.shared.setDownloadDirectory(downloadDirectory)
+            }
+        }
     }
 }