[_]: Add v1 and legacy download methods

internxt · Jun 21, 2024 · dde342b · dde342b
1 parent b4aede9
commit dde342b
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 3 deletions.
diff --git a/Sources/InternxtSwiftCore/Extensions/Sequence.swift b/Sources/InternxtSwiftCore/Extensions/Sequence.swift
@@ -0,0 +1,17 @@
+//
+//  Sequence.swift
+//  
+//
+//  Created by Robert Garcia on 21/6/24.
+//
+
+import Foundation
+extension Sequence {
+    func asyncForEach(
+        _ operation: (Element) async throws -> Void
+    ) async rethrows {
+        for element in self {
+            try await operation(element)
+        }
+    }
+}
diff --git a/Sources/InternxtSwiftCore/Services/Network/Download.swift b/Sources/InternxtSwiftCore/Services/Network/Download.swift
@@ -16,6 +16,7 @@ enum DownloadError: Error {
     case InvalidBucketId
     case MissingShards
     case V1DownloadDetected
+    case NoMirrorsFound
 }
 
 public struct DownloadResult {
@@ -62,7 +63,25 @@ public class Download: NSObject {
         let info = try await networkAPI.getFileInfo(bucketId: bucketId, fileId: fileId, debug: debug)
 
         if info.version == 1 {
-            _ = try await networkAPI.getFileMirrors(bucketId: bucketId, fileId: fileId, debug: debug)
+            let mirrors = try await networkAPI.getFileMirrors(bucketId: bucketId, fileId: fileId, debug: debug)
+            guard let mirror = mirrors.first else {
+                throw DownloadError.NoMirrorsFound
+            }
+
+            if mirrors.count > 1 {
+                // Legacy download here
+
+                try await mirrors.asyncForEach{mirror in
+                    _ = try await downloadEncryptedFile(downloadUrl: mirror.url, destinationURL: destination)
+                }
+
+                return DownloadResult(url: destination, expectedContentHash: "NO_CONTENT_HASH", index: info.index)
+            }
+
+
+            let url = try await downloadEncryptedFile(downloadUrl: mirror.url, destinationURL: destination, progressHandler: progressHandler)
+
+            return DownloadResult(url: url, expectedContentHash: mirror.hash, index: info.index)
         }
 
         guard let shards = info.shards else {

diff --git a/Sources/InternxtSwiftCore/Services/Network/NetworkFacade.swift b/Sources/InternxtSwiftCore/Services/Network/NetworkFacade.swift
@@ -215,7 +215,7 @@ public struct NetworkFacade {
         return decryptedFileURL
     }
 
-    public func decryptFile(bucketId: String, destinationURL: URL, progressHandler: ProgressHandler, encryptedFileDownloadResult: DownloadResult) async throws -> URL {
+    public func decryptFile(bucketId: String, destinationURL: URL, progressHandler: ProgressHandler, encryptedFileDownloadResult: DownloadResult, ignoreHashMissmatchCheck: Bool = false) async throws -> URL {
 
         if encryptedFileDownloadResult.url.fileSize == 0 {
             throw NetworkFacadeError.FileIsEmpty
@@ -234,7 +234,7 @@ public struct NetworkFacade {
 
 
         let hashMatch = encryptedContentHash.toHexString() == encryptedFileDownloadResult.expectedContentHash
-        if hashMatch == false {
+        if hashMatch == false && ignoreHashMissmatchCheck != false {
             throw NetworkFacadeError.HashMissmatch
         }