Custom Page Template in PDF Editor using Swift for iOS

1
//
2
//  Copyright © 2018-2025 PSPDFKit GmbH. All rights reserved.
3
//
4
//  The Nutrient sample applications are licensed with a modified BSD license.
5
//  Please see License for details. This notice may not be removed from this file.
6
//
7

8
import PSPDFKit
9
import PSPDFKitUI
10

11
class DocumentEditorCustomTemplatesExample: Example {
12
    override init() {
13
        super.init()
14
        title = "Add New Page from Custom Template"
15
        contentDescription = "Use custom templates to add new pages to a document."
16
        category = .documentEditing
17
        priority = 3
18
    }
19

20
    override func invoke(with delegate: ExampleRunnerDelegate) -> UIViewController {
21
        // Let's create a new writable document every time we invoke the example.
22
        let document = AssetLoader.writableDocument(for: .welcome, overrideIfExists: true)
23
        let pdfController = EditingPDFController(document: document)
24

25
        // Add a single button that triggers preset document editing actions.
26
        let editButtonItem = UIBarButtonItem(image: SDK.imageNamed("document_editor"), style: .plain, target: pdfController, action: #selector(EditingPDFController.edit))
27
        pdfController.navigationItem.rightBarButtonItems = [editButtonItem]
28

29
        return pdfController
30
    }
31

32
    // MARK: Controller
33

34
    class EditingPDFController: PDFViewController, PDFNewPageViewControllerDelegate {
35
        // MARK: Actions
36

37
        @objc func edit(_ sender: AnyObject) {
38
            let url = AssetLoader.document(for: .about).fileURL!
39
            let document = Document(url: url)
40
            let customTemplate = PageTemplate(document: document, sourcePageIndex: 0)
41
            let newPageViewController = PDFNewPageViewController(documentEditorConfiguration: PDFDocumentEditor.Configuration {
42
                $0.pageTemplates.append(contentsOf: [customTemplate])
43
            })
44
            newPageViewController.delegate = self
45
            newPageViewController.modalPresentationStyle = .popover
46

47
            let options = [.inNavigationController: true, .closeButton: true] as [PresentationOption: Any]
48
            present(newPageViewController, options: options, animated: true, sender: sender)
49
        }
50

51
        func newPageController(_ controller: PDFNewPageViewController, didFinishSelecting configuration: PDFNewPageConfiguration?, pageCount: PageCount) {
52
            dismiss(animated: true, completion: nil)
53

54
            guard let document, let configuration, let editor = PDFDocumentEditor(document: document) else { return }
55

56
            editor.addPages(in: NSRange(location: 0, length: Int(pageCount)), with: configuration)
57

58
            // Save and overwrite the document.
59
            editor.save { _, error in
60
                if let error {
61
                    print("Document editing failed: \(error)")
62
                    return
63
                }
64

65
                // Access the UI on the main thread.
66
                DispatchQueue.main.async {
67
                    self.pdfController.reloadData()
68
                }
69
            }
70
        }
71
    }
72
}