PDFViewController Controller State in Swift for iOS

1
//
2
//  Copyright © 2015-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
private enum ControllerState: String {
12
    case normal
13
    case empty
14
    case error
15
    case locked
16

17
    func nextState() -> ControllerState {
18
        switch self {
19
        case .normal: return .empty
20
        case .empty: return .error
21
        case .error: return .locked
22
        case .locked: return .normal
23
        }
24
    }
25
}
26

27
// MARK: CustomStringConvertible
28

29
extension ControllerState: CustomStringConvertible {
30

31
    var description: String {
32
        return "State: \(rawValue)"
33
    }
34
}
35

36
class ControllerStateExample: Example {
37

38
    // MARK: Properties
39

40
    private weak var pdfController: PDFViewController?
41
    private var toggleButton: UIBarButtonItem!
42

43
    private var displayState: ControllerState = .normal {
44
        didSet {
45
            guard let pdfController else { return }
46
            switch displayState {
47
            case .normal:
48
                pdfController.document = AssetLoader.document(for: .welcome)
49
            case .empty:
50
                pdfController.document = nil
51
            case .error:
52
                pdfController.document = Document(dataProviders: [DataContainerProvider(data: Data())])
53
            case .locked:
54
                pdfController.document = AssetLoader.document(for: "Password-Protected.pdf")
55
            }
56
            toggleButton.title = String(describing: displayState)
57
        }
58
    }
59

60
    override init() {
61
        super.init()
62

63
        title = "Controller States"
64
        contentDescription = "Shows default, empty, error, and locked states."
65
        category = .controllerCustomization
66
        toggleButton = UIBarButtonItem(title: "", style: .plain, target: self, action: #selector(ControllerStateExample.toggleButtonPressed(_:)))
67
    }
68

69
    @objc
70
    private func toggleButtonPressed(_ sender: UIBarButtonItem) {
71
        // If pressed, toggle to the next state.
72
        displayState = displayState.nextState()
73
    }
74

75
    override func invoke(with delegate: ExampleRunnerDelegate) -> UIViewController {
76
        let items = [toggleButton!]
77
        let document = AssetLoader.document(for: .welcome)
78

79
        pdfController = {
80
            let pdfController = PDFViewController(document: document)
81
            for viewMode: ViewMode in [.document, .documentEditor, .thumbnails] {
82
                pdfController.navigationItem.setRightBarButtonItems(items, for: viewMode, animated: false)
83
            }
84
            pdfController.barButtonItemsAlwaysEnabled = items
85
            return pdfController
86
        }()
87

88
        // Set displayState to trigger toggleButton title change
89
        displayState = .normal
90

91
        return pdfController!
92
    }
93
}