Export to PDF now opens ActivityViewController

neoneye · neoneye · commit 84f031bf0ed7 · 2023-01-12T16:47:10.000+01:00
diff --git a/Source/TriangleDrawMain/Canvas/Menu/HCMenuView.swift b/Source/TriangleDrawMain/Canvas/Menu/HCMenuView.swift
@@ -5,7 +5,8 @@ import TTProgressHUD
 
 enum ShareData {
     case none
-    case exportPNG(image: UIImage, filename: String?)
+    case exportPNG(image: UIImage, filename: String)
+    case exportPDF(pdfData: Data, filename: String)
 }
 
 struct HCMenuView: View {
@@ -43,6 +44,25 @@ struct HCMenuView: View {
         }
     }
 
+    var exportPDFButton: some View {
+        Button("Vector PDF") {
+            hudVisible = true
+            model.exportToPDF() { status in
+                switch status {
+                case let HCMenuViewModel.ExportPDFStatus.ok(pdfData, filename):
+                    hudVisible = false
+                    self.shareData = ShareData.exportPDF(pdfData: pdfData, filename: filename)
+                    self.isSharePresented = true
+                case HCMenuViewModel.ExportPDFStatus.progress(let progress):
+                    log.debug("progress: \(progress)")
+                case HCMenuViewModel.ExportPDFStatus.error(let message):
+                    log.error("Unable to export to PDF. \(message)")
+                    hudVisible = false
+                }
+            }
+        }
+    }
+
     var emailDeveloperButton: some View {
         #if os(iOS)
         let view = MailButtonView(
@@ -81,7 +101,7 @@ struct HCMenuView: View {
                 }
                 Section(header: Text("Export")) {
                     exportPNGButton
-                    Button("Vector PDF") {}
+                    exportPDFButton
                     Button("Vector SVG") {}
                 }
                 Section(header: Text("Feedback")) {
@@ -126,6 +146,9 @@ struct ActivityViewController: UIViewControllerRepresentable {
         case let .exportPNG(image, filename):
             log.debug("share: png")
             return HCMenuViewController.createSharePNGActivityViewController(image: image, filename: filename, triangleCount: 0)
+        case let .exportPDF(pdfData, filename):
+            log.debug("share: pdf")
+            return HCMenuViewController.createSharePDFActivityViewController(pdfData: pdfData, filename: filename, triangleCount: 0)
         }
     }
 
diff --git a/Source/TriangleDrawMain/Canvas/Menu/HCMenuViewModel.swift b/Source/TriangleDrawMain/Canvas/Menu/HCMenuViewModel.swift
@@ -30,7 +30,7 @@ public class HCMenuViewModel: ObservableObject {
 
     public enum ExportPNGStatus {
         case progress(progress: Float)
-        case ok(image: UIImage, filename: String?)
+        case ok(image: UIImage, filename: String)
         case error(message: String)
     }
 
@@ -56,4 +56,33 @@ public class HCMenuViewModel: ObservableObject {
             callback(ExportPNGStatus.ok(image: image, filename: filename))
         }
     }
+
+    public enum ExportPDFStatus {
+        case progress(progress: Float)
+        case ok(pdfData: Data, filename: String)
+        case error(message: String)
+    }
+
+    func exportToPDF(callback: @escaping (ExportPDFStatus) -> Void) {
+        guard let canvas: E2Canvas = self.initialCanvas else {
+            callback(ExportPDFStatus.error(message: "exportToPDF - Expected document.canvas to be non-nil, but got nil"))
+            return
+        }
+        log.debug("exportToPDF initiate")
+        let t0 = CFAbsoluteTimeGetCurrent()
+        let filename = document_displayName ?? ""
+
+        PDFExporter.createPDF(from: canvas, progress: { progress in
+            callback(ExportPDFStatus.progress(progress: progress))
+        }) { pdfData in // swiftlint:disable:this multiple_closures_with_trailing_closure
+            guard pdfData.count > 0 else {
+                callback(ExportPDFStatus.error(message: "exportToPDF - Expected size of pdf to be greater than 0 bytes"))
+                return
+            }
+            let t1 = CFAbsoluteTimeGetCurrent()
+            let elapsed: Double = t1 - t0
+            log.debug("exportToPDF - ready for sharing. elapsed: \(elapsed.string2)")
+            callback(ExportPDFStatus.ok(pdfData: pdfData, filename: filename))
+        }
+    }
 }
