✨ 终于支持导出pdf分页了

Author: tw93

Mac/View/MPreviewView.swift

@@ -1,4 +1,5 @@
 import Carbon.HIToolbox
+import PDFKit
 import WebKit
 
 #if os(iOS)
@@ -83,30 +84,90 @@ class MPreviewView: WKWebView, WKUIDelegate, WKNavigationDelegate {
 
     public func exportPdf() {
         guard let vc = ViewController.shared() else { return }
+        // 获取 WKWebView 的内容大小
+        var a4Size = CGSize(width: 800, height: 1120)
+        if UserDefaultsManagement.isOnExportPPT {
+            a4Size = CGSize(width: 1536, height: 957)
+        }
 
-        if #available(macOS 11.0.0, *) {
-            DispatchQueue.main.asyncAfter(deadline: .now() + 0.1) {
-                let config = WKPDFConfiguration()
-                // Render the PDF
-                super.createPDF(configuration: config) { result in
-                    switch result {
-                    case .success(let data):
-                        if let path = NSSearchPathForDirectoriesInDomains(.downloadsDirectory, .userDomainMask, true).first {
-                            let currentName = self.note?.getTitle()
-                            let filePath: String = path + "/" + (currentName ?? "MiaoYan") + ".pdf"
-                            try! data.write(to: URL(fileURLWithPath: filePath))
-                            vc.toastExport(status: true)
-                        }
+        let pageWidth = a4Size.width
+        let pageHeight = a4Size.height
+        let maxHeight: CGFloat = 14355.0
+
+        super.frame.size.width = pageWidth
+
+        super.evaluateJavaScript("document.body.scrollHeight", completionHandler: { height, error in
+            guard let height = height as? CGFloat, error == nil else {
+                vc.toastExport(status: false)
+                return
+            }
+            if #available(macOS 11.0, *) {
+                let numberOfParts = Int(ceil(height / maxHeight))
+                let newDocument = PDFDocument()
+
+                // 设置 PDF 分页
+                func processPart(_ partIndex: Int) {
+                    let partY = CGFloat(partIndex) * maxHeight
+                    let partHeight = min(maxHeight, height - partY)
+
+                    let pdfConfiguration = WKPDFConfiguration()
+                    pdfConfiguration.rect = CGRect(x: 0, y: partY, width: super.bounds.width, height: partHeight)
+
+                    super.createPDF(configuration: pdfConfiguration) { result in
+                        switch result {
+                        case .success(let pdfData):
+                            // 使用 PDFKit 进行分页
+
+                            let pdfDocument = PDFDocument(data: pdfData)
+
+                            for i in 0 ..< (pdfDocument?.pageCount ?? 0) {
+                                guard let page = pdfDocument?.page(at: i) else {
+                                    return
+                                }
+
+                                let pageBounds = page.bounds(for: .cropBox)
+                                let subPageCount = Int(ceil(pageBounds.height / a4Size.height))
+
+                                for j in 0 ..< subPageCount {
+                                    let subPageRect = CGRect(x: 0, y: CGFloat(subPageCount - j - 1) * a4Size.height, width: pageWidth, height: a4Size.height)
 
-                    case .failure(let error):
-                        print(error)
-                        vc.toastExport(status: false)
+                                    guard let subPage = page.copy() as? PDFPage else {
+                                        return
+                                    }
+
+                                    subPage.setBounds(subPageRect, for: .cropBox)
+                                    newDocument.insert(subPage, at: newDocument.pageCount)
+                                }
+                            }
+
+                            if partIndex < numberOfParts - 1 {
+                                processPart(partIndex + 1)
+                            } else {
+                                if let newPDFData = newDocument.dataRepresentation() {
+                                    if let path = NSSearchPathForDirectoriesInDomains(.downloadsDirectory, .userDomainMask, true).first {
+                                        vc.toastExport(status: true)
+                                        let currentName = self.note?.getTitle()
+                                        let filePath: String = path + "/" + (currentName ?? "MiaoYan") + ".pdf"
+                                        try! newPDFData.write(to: URL(fileURLWithPath: filePath))
+                                    }
+
+                                } else {}
+                            }
+
+                        case .failure:
+                            vc.toastExport(status: false)
+                            return
+                        }
                     }
                 }
+
+                processPart(0)
+
+            } else {
+                // Fallback on earlier versions
+                vc.toastExport(status: false)
             }
-        } else {
-            vc.toastExport(status: false)
-        }
+        })
     }
 
     public func slideTo(index: Int) {

Mac/ViewController.swift

@@ -2552,7 +2552,7 @@ class ViewController:
         UserDefaultsManagement.isOnExportPPT = true
         toast(message: NSLocalizedString("🙊 Starting export~", comment: ""))
         enableMiaoYanPPT()
-        DispatchQueue.main.asyncAfter(deadline: .now() + 2.5) {
+        DispatchQueue.main.asyncAfter(deadline: .now() + 4) {
             self.editArea.markdownView?.exportPdf()
             UserDefaultsManagement.isOnExport = false
             UserDefaultsManagement.isOnExportPPT = false

Resources/DownView.bundle/ppt/dist/reveal.css

@@ -2024,32 +2024,15 @@ html.print-pdf .reveal .slides .pdf-page {
     overflow: hidden;
     z-index: 1;
     page-break-after: always;
-    border: 8px solid #d1d6ee;
     border-top: none;
     box-sizing: border-box;
     background-color: #fff !important;
 }
 
-html.print-pdf .reveal .slides .pdf-page:first-child {
-    border-top: 8px solid #d1d6ee;
-}
-
-html.print-pdf .reveal-viewport {
-    background-color: #d1d6ee;
-}
 
 @media (prefers-color-scheme: dark) {
     html.print-pdf .reveal .slides .pdf-page {
         background-color: #21262b !important;
-        border: 8px solid #fff;
-    }
-
-    html.print-pdf .reveal .slides .pdf-page:first-child {
-        border-top: 8px solid #fff;
-    }
-
-    html.print-pdf .reveal-viewport {
-        background-color: #fff;
     }
 }