File: ANSIAnnotation.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (60 lines) | stat: -rw-r--r-- 1,610 bytes parent folder | download
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
/*
 This source file is part of the Swift.org open source project

 Copyright (c) 2023 Apple Inc. and the Swift project authors
 Licensed under Apache License v2.0 with Runtime Library Exception

 See https://swift.org/LICENSE.txt for license information
 See https://swift.org/CONTRIBUTORS.txt for Swift project authors
*/

import Foundation

struct ANSIAnnotation {
    enum Color: UInt8 {
        case normal = 0
        case red = 31
        case green = 32
        case yellow = 33
        case `default` = 39
    }
    
    enum Trait: UInt8 {
        case normal = 0
        case bold = 1
        case italic = 3
    }
    
    private var color: Color
    private var trait: Trait
    
    /// The textual representation of the annotation.
    private var code: String {
        "\u{001B}[\(trait.rawValue);\(color.rawValue)m"
    }
    
    init(color: Color, trait: Trait = .normal) {
        self.color = color
        self.trait = trait
    }
    
    func applied(to message: some StringProtocol) -> String {
        guard !message.isEmpty else {
            return ""
        }
        return "\(code)\(message)\(ANSIAnnotation.normal.code)"
    }
    
    static var normal: ANSIAnnotation {
        self.init(color: .normal, trait: .normal)
    }
    
    /// Annotation used for highlighting source text.
    static var sourceHighlight: ANSIAnnotation {
        ANSIAnnotation(color: .green, trait: .bold)
    }
    /// Annotation used for highlighting source suggestion.
    static var sourceSuggestionHighlight: ANSIAnnotation {
        ANSIAnnotation(color: .default, trait: .bold)
    }
}