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
|
// SPDX-License-Identifier: GPL-3.0-only
// (c) Hare authors <https://harelang.org>
use fmt;
use hare::unparse;
use os;
use regex;
// Colors/Renditions with defaults; SGR parameters for ANSI escape sequences.
const default_colors = [
"_", // ident
"1", // comment
"_", // constant
"_", // function
"_", // global
"_", // typedef
"_", // import_alias
"_", // secondary
"94", // keyword
"96", // type
"33", // attribute
"1", // operator
"_", // punctuation
"91", // rune_string
"95", // number
"_", // label
];
let colors: [len(default_colors)]str = [""...];
let normal_color = "";
let primary_color = "";
fn init_colors() (void | error) = {
const env_colors = os::tryenv("HAREDOC_COLORS", "");
const expr = regex::compile(`([a-z_]+)=(_|[0-9;]*)`)!;
defer regex::finish(&expr);
const matches = regex::findall(&expr, env_colors)!;
defer regex::result_freeall(matches);
for (let m .. matches) {
let (k, v) = (m[1].content, m[2].content);
let idx = 0z;
let out: *str = switch (k) {
case "ident" =>
yield &colors[unparse::synkind::IDENT];
case "comment" =>
yield &colors[unparse::synkind::COMMENT];
case "constant" =>
yield &colors[unparse::synkind::CONSTANT];
case "function" =>
yield &colors[unparse::synkind::FUNCTION];
case "global" =>
yield &colors[unparse::synkind::GLOBAL];
case "typedef" =>
yield &colors[unparse::synkind::TYPEDEF];
case "import_alias" =>
yield &colors[unparse::synkind::IMPORT_ALIAS];
case "secondary" =>
yield &colors[unparse::synkind::SECONDARY];
case "keyword" =>
yield &colors[unparse::synkind::KEYWORD];
case "type" =>
yield &colors[unparse::synkind::TYPE];
case "attribute" =>
yield &colors[unparse::synkind::ATTRIBUTE];
case "operator" =>
yield &colors[unparse::synkind::OPERATOR];
case "punctuation" =>
yield &colors[unparse::synkind::PUNCTUATION];
case "rune_string" =>
yield &colors[unparse::synkind::RUNE_STRING];
case "number" =>
yield &colors[unparse::synkind::NUMBER];
case "label" =>
yield &colors[unparse::synkind::LABEL];
case "normal" =>
yield &normal_color;
case "primary" =>
yield &primary_color;
case =>
static let err: [64]u8 = [0...];
if (len(k) > len(err)) {
return "": haredoc_colors_error;
};
return fmt::bsprint(err, k): haredoc_colors_error;
};
*out = if (v == "_" && k == "normal") "0" else v;
};
};
fn color(kind: unparse::synkind) str = {
const color = if (colors[kind] != "") colors[kind]
else default_colors[kind];
if (color != "_") {
return color;
};
if (primary_color != "" && primary_color != "_") {
switch (kind) {
case unparse::synkind::CONSTANT,
unparse::synkind::FUNCTION,
unparse::synkind::GLOBAL,
unparse::synkind::TYPEDEF =>
return primary_color;
case => void;
};
};
return if (normal_color == "") "0" else normal_color;
};
|
