Fix saving images

metabolist · May 10, 2021 · 99678e4 · 99678e4
1 parent f7ff1b8
commit 99678e4
Showing 1 changed file with 35 additions and 22 deletions.
diff --git a/View Controllers/ImageViewController.swift b/View Controllers/ImageViewController.swift
@@ -193,17 +193,26 @@ extension ImageViewController {
     }
 
     func presentActivityViewController() {
-        if let image = imageView.image {
-            let activityViewController = UIActivityViewController(
-                activityItems: [NSItemProvider(object: image)],
-                applicationActivities: [])
-
-            if UIDevice.current.userInterfaceIdiom == .pad {
-                activityViewController.popoverPresentationController?
-                    .barButtonItem = parent?.navigationItem.rightBarButtonItem
-            }
+        if let imageData = imageView.image?.sd_imageData(), let url = imageURL ?? viewModel?.attachment.url.url {
+            let tempURL = URL(fileURLWithPath: NSTemporaryDirectory(), isDirectory: true)
+                .appendingPathComponent(url.lastPathComponent)
+
+            do {
+                try imageData.write(to: tempURL)
 
-            present(activityViewController, animated: true)
+                let activityViewController = UIActivityViewController(
+                    activityItems: [tempURL],
+                    applicationActivities: [])
+
+                if UIDevice.current.userInterfaceIdiom == .pad {
+                    activityViewController.popoverPresentationController?
+                        .barButtonItem = parent?.navigationItem.rightBarButtonItem
+                }
+
+                present(activityViewController, animated: true)
+            } catch {
+                alertUnableToExportMedia()
+            }
         } else if let asset = playerView.player?.currentItem?.asset as? AVURLAsset {
             asset.exportWithoutAudioTrack { result in
                 DispatchQueue.main.async {
@@ -224,18 +233,7 @@ extension ImageViewController {
 
                         self.present(activityViewController, animated: true)
                     case .failure:
-                        let alertController = UIAlertController(
-                            title: nil,
-                            message: NSLocalizedString("attachment.unable-to-export-media", comment: ""),
-                            preferredStyle: .alert)
-
-                        let okAction = UIAlertAction(
-                            title: NSLocalizedString("ok", comment: ""),
-                            style: .default) { _ in }
-
-                        alertController.addAction(okAction)
-
-                        self.present(alertController, animated: true)
+                        self.alertUnableToExportMedia()
                     }
                 }
             }
@@ -291,4 +289,19 @@ private extension ImageViewController {
                 scrollView.setZoomScale(scrollView.minimumZoomScale, animated: true)
             }
     }
+
+    func alertUnableToExportMedia() {
+        let alertController = UIAlertController(
+            title: nil,
+            message: NSLocalizedString("attachment.unable-to-export-media", comment: ""),
+            preferredStyle: .alert)
+
+        let okAction = UIAlertAction(
+            title: NSLocalizedString("ok", comment: ""),
+            style: .default) { _ in }
+
+        alertController.addAction(okAction)
+
+        self.present(alertController, animated: true)
+    }
 }