aboutsummaryrefslogtreecommitdiffstats
path: root/WireGuard/WireGuard/UI/macOS/View/ConfTextStorage.swift
blob: 7d32b7e15d67d0678e9d795c58aaa6fdc0854906 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
// SPDX-License-Identifier: MIT
// Copyright © 2018-2019 WireGuard LLC. All Rights Reserved.

import Cocoa

private let fontSize: CGFloat = 15

class ConfTextStorage: NSTextStorage {

    struct TextColorTheme {
        let plainText: NSColor
        let sections: NSColor
        let keyType: NSColor
        let key: NSColor
        let url: NSColor
        let urlAttribute: NSColor
        let comments: NSColor
        let number: NSColor
        let error: NSColor
    }

    let defaultFont = NSFontManager.shared.convertWeight(true, of: NSFont.systemFont(ofSize: fontSize))
    private let boldFont = NSFont.boldSystemFont(ofSize: fontSize)
    private lazy var italicFont = NSFontManager.shared.convert(defaultFont, toHaveTrait: .italicFontMask)

    private var defaultAttributes: [NSAttributedString.Key: Any]! //swiftlint:disable:this implicitly_unwrapped_optional
    private var highlightAttributes: [UInt32: [NSAttributedString.Key: Any]]! //swiftlint:disable:this implicitly_unwrapped_optional

    private let backingStore: NSMutableAttributedString
    private(set) var hasError = false
    private(set) var privateKeyString: String?

    override init() {
        backingStore = NSMutableAttributedString(string: "")
        super.init()
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    required init?(pasteboardPropertyList propertyList: Any, ofType type: NSPasteboard.PasteboardType) {
        fatalError("init(pasteboardPropertyList:ofType:) has not been implemented")
    }

    //swiftlint:disable:next function_body_length
    func updateAttributes(for theme: TextColorTheme) {
        self.defaultAttributes = [
            .foregroundColor: theme.plainText,
            .font: defaultFont
        ]

        self.highlightAttributes = [
            HighlightSection.rawValue: [
                .foregroundColor: theme.sections,
                .font: boldFont
            ],
            HighlightField.rawValue: [
                .foregroundColor: theme.keyType,
                .font: boldFont
            ],
            HighlightPublicKey.rawValue: [
                .foregroundColor: theme.key,
                .font: defaultFont
            ],
            HighlightPrivateKey.rawValue: [
                .foregroundColor: theme.key,
                .font: defaultFont
            ],
            HighlightPresharedKey.rawValue: [
                .foregroundColor: theme.key,
                .font: defaultFont
            ],
            HighlightIP.rawValue: [
                .foregroundColor: theme.url,
                .font: defaultFont
            ],
            HighlightCidr.rawValue: [
                .foregroundColor: theme.urlAttribute,
                .font: defaultFont
            ],
            HighlightHost.rawValue: [
                .foregroundColor: theme.url,
                .font: defaultFont
            ],
            HighlightPort.rawValue: [
                .foregroundColor: theme.urlAttribute,
                .font: defaultFont
            ],
            HighlightMTU.rawValue: [
                .foregroundColor: theme.number,
                .font: defaultFont
            ],
            HighlightKeepalive.rawValue: [
                .foregroundColor: theme.number,
                .font: defaultFont
            ],
            HighlightComment.rawValue: [
                .foregroundColor: theme.comments,
                .font: italicFont
            ],
            HighlightDelimiter.rawValue: [
                .foregroundColor: theme.plainText,
                .font: defaultFont
            ],
            HighlightError.rawValue: [
                .foregroundColor: theme.error,
                .font: defaultFont,
                .underlineStyle: 1
            ]
        ]

        highlightSyntax()
    }

    override var string: String {
        return backingStore.string
    }

    override func attributes(at location: Int, effectiveRange range: NSRangePointer?) -> [NSAttributedString.Key: Any] {
        return backingStore.attributes(at: location, effectiveRange: range)
    }

    override func replaceCharacters(in range: NSRange, with str: String) {
        beginEditing()
        backingStore.replaceCharacters(in: range, with: str)
        edited(.editedCharacters, range: range, changeInLength: str.count - range.length)
        endEditing()
    }

    override func replaceCharacters(in range: NSRange, with attrString: NSAttributedString) {
        beginEditing()
        backingStore.replaceCharacters(in: range, with: attrString)
        edited(.editedCharacters, range: range, changeInLength: attrString.length - range.length)
        endEditing()
    }

    override func setAttributes(_ attrs: [NSAttributedString.Key: Any]?, range: NSRange) {
        beginEditing()
        backingStore.setAttributes(attrs, range: range)
        edited(.editedAttributes, range: range, changeInLength: 0)
        endEditing()
    }

    func highlightSyntax() {
        hasError = false
        privateKeyString = nil

        backingStore.beginEditing()
        var spans = highlight_config(backingStore.string.cString(using: String.Encoding.utf8))!

        while spans.pointee.type != HighlightEnd {
            let span = spans.pointee

            let attributes = self.highlightAttributes[span.type.rawValue] ?? defaultAttributes
            backingStore.setAttributes(attributes, range: NSRange(location: span.start, length: span.len))

            if span.type == HighlightError {
                hasError = true
            }

            if span.type == HighlightPrivateKey {
                privateKeyString = backingStore.attributedSubstring(from: NSRange(location: span.start, length: span.len)).string
            }

            spans = spans.successor()
        }
        backingStore.endEditing()

        beginEditing()
        edited(.editedAttributes, range: NSRange(location: 0, length: (backingStore.string as NSString).length), changeInLength: 0)
        endEditing()
    }

}