Update PHPicker.swift

tsuzukihashi · May 18, 2024 · 9895157 · 9895157
1 parent 743dbaf
commit 9895157
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/Sources/ImagePickerSwiftUI/PHPicker.swift b/Sources/ImagePickerSwiftUI/PHPicker.swift
@@ -51,6 +51,11 @@ public struct PHPicker: UIViewControllerRepresentable {
 
     public func picker(_ picker: PHPickerViewController, didFinishPicking results: [PHPickerResult]) {
       Task {
+        if results.isEmpty {
+          parent.isPresented = false
+          return
+        }
+
         for result in results {
           do {
             if let image = try await loadImage(result: result) {
@@ -59,12 +64,14 @@ public struct PHPicker: UIViewControllerRepresentable {
             if let videoURL = try await loadVideo(result: result) {
               parent.videoURLs.append(videoURL)
             }
+
+            if results.last == result {
+              parent.isPresented = false
+            }
           } catch {
             print(error.localizedDescription)
           }
         }
-
-        parent.isPresented = false
       }
     }