aboutsummaryrefslogtreecommitdiffstats
path: root/filters/colorize
blob: 9d2a736353bf6db8e3bf5c3f0188cde094451c72 (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
176
177
#!/usr/bin/awk -f
# Copyright (c) 2022 Robin Jarry
#
# A filter for the aerc mail program to colorize messages / attachments.
# Basic colour themes are supported. To use a theme set the theme variable
# in your aerc.conf accordingly, for example:
#
# text/plain=colorize -v theme=solarized

BEGIN {
	if (theme == "solarized") {
		# R;G;B colors
		url = "\033[38;2;181;137;0m" # yellow
		header = "\033[38;2;211;54;130m" # magenta
		signature = "\033[38;2;211;54;130m" # magenta
		diff_meta = "\033[1;38;2;131;148;150m" # bold brblue
		diff_chunk = "\033[38;2;42;161;152m" # cyan
		diff_add = "\033[38;2;133;153;0m" # green
		diff_del = "\033[38;2;220;50;47m" # red
		quote_1 = "\033[38;2;38;139;210m"  # blue
		quote_2 = "\033[38;2;203;75;22m" # brred
		quote_3 = "\033[38;2;211;54;130m" # magenta
		quote_4 = "\033[38;2;108;113;196m" # brmagenta
		quote_x = "\033[38;2;147;161;161m" # brcyan
		bold = "\033[1m"
		reset = "\033[0m"
	} else if (theme == "" || theme == "default") {
		# R;G;B colors
		url = "\033[38;2;255;255;175m" # yellow
		header = "\033[38;2;175;135;255m" # purple
		signature = "\033[38;2;175;135;255m" # purple
		diff_meta = "\033[1;38;2;255;255;255m" # bold white
		diff_chunk = "\033[38;2;0;205;205m" # cyan
		diff_add = "\033[38;2;0;205;0m" # green
		diff_del = "\033[38;2;205;0;0m" # red
		quote_1 = "\033[38;2;95;175;255m"  # blue
		quote_2 = "\033[38;2;255;135;0m" # orange
		quote_3 = "\033[38;2;175;135;255m" # purple
		quote_4 = "\033[38;2;255;95;215m" # pink
		quote_x = "\033[38;2;128;128;128m" # gray
		bold = "\033[1m"
		reset = "\033[0m"
	} else if (theme == "terminal") {
		# terminal respects the users configured terminal color theme
		url = "\033[4;34m" # underline blue
		header = "\033[35m" # magenta
		signature = "\033[35m" # magenta
		diff_meta = "\033[2m" # faint
		diff_chunk = "\033[36m" # cyan
		diff_add = "\033[32m" # green
		diff_del = "\033[31m" # red
		quote_1 = "\033[37m"  # grey
		quote_2 = "\033[34m" # blue
		quote_3 = "\033[2;37m" # faint grey
		quote_4 = "\033[2;34m" # faint blue
		quote_x = "\033[2;37m" # faint grey
		bold = "\033[1m"
		reset = "\033[0m"
	} else {
		print "error: unknown theme " theme > "/dev/stderr"
		exit 1
	}
	# state
	in_diff = 0
	in_signature = 0
	in_headers = 0
	in_body = 0
	# patterns
	header_pattern = "^[A-Z][[:alnum:]-]+:"
	url_pattern = "[[:lower:]]+://[[:graph:]]+|(mailto:)?[[:alnum:]_\\+\\.~/-]*[[:alnum:]_]@[[:lower:]][[:alnum:]\\.-]*[[:lower:]]"
	meta_pattern = "^(diff --git|(new|deleted) file|similarity index|(rename|copy) (to|from)|index|---|\\+\\+\\+) "
}
function color_quote(line) {
	level = 0
	quotes = ""
	while (line ~ /^>/) {
		level += 1
		quotes = quotes ">"
		line = substr(line, 2)
		while (line ~ /^ /) {
			quotes = quotes " "
			line = substr(line, 2)
		}
	}
	if (level == 1) {
		color = quote_1
	} else if (level == 2) {
		color = quote_2
	} else if (level == 3) {
		color = quote_3
	} else if (level == 4) {
		color = quote_4
	} else {
		color = quote_x
	}
	if (match(line, meta_pattern)) {
		return color quotes bold line reset
	} else if (line ~ /^\+/) {
		return color quotes diff_add line reset
	} else if (line ~ /^-/) {
		return color quotes diff_del line reset
	}
	gsub(url_pattern, url "&" color, line)
	return color quotes line reset
}
{
	# Strip carriage returns from line
	sub(/\r$/, "")

	if (in_diff) {
		if ($0 ~ /^-- ?$/) {
			in_diff = 0
			in_signature = 1
			$0 = signature $0 reset
		} else if ($0 ~ /^@@ /) {
			gsub(/^@@[^@]+@@/, diff_chunk "&" reset)
		} else if (match($0, meta_pattern)) {
			$0 = diff_meta $0 reset
		} else if ($0 ~ /^\+/) {
			$0 = diff_add $0 reset
		} else if ($0 ~ /^-/) {
			$0 = diff_del $0 reset
		} else if ($0 !~ /^ / && $0 !~ /^$/) {
			in_diff = 0
			in_body = 1
			if ($0 ~ /^>/) {
				$0 = color_quote($0)
			} else {
				gsub(url_pattern, url "&" reset)
			}
		}
	} else if (in_signature) {
		gsub(url_pattern, url "&" signature)
		$0 = signature $0 reset
	} else if (in_headers) {
		if ($0 ~ /^$/) {
			in_headers = 0
			in_body = 1
		} else {
			sub(header_pattern, header "&" reset)
			gsub(url_pattern, url "&" reset)
		}
	} else if (in_body) {
		if ($0 ~ /^>/) {
			$0 = color_quote($0)
		} else if ($0 ~ /^diff --git /) {
			in_body = 0
			in_diff = 1
			$0 = diff_meta $0 reset
		} else if ($0 ~ /^-- ?$/) {
			in_body = 0
			in_signature = 1
			$0 = signature $0 reset
		} else {
			gsub(url_pattern, url "&" reset)
		}
	} else if ($0 ~ /^diff --git /) {
		in_diff = 1
		$0 = diff_meta $0 reset
	} else if ($0 ~ /^-- ?$/) {
		in_signature = 1
		$0 = signature $0 reset
	} else if (match($0, header_pattern)) {
		in_headers = 1
		sub(header_pattern, header "&" reset)
		gsub(url_pattern, url "&" reset)
	} else {
		in_body = 1
		if ($0 ~ /^>/) {
			$0 = color_quote($0)
		} else {
			gsub(url_pattern, url "&" reset)
		}
	}

	print
}