File: coma.py 1 #!/usr/bin/python3 2 3 # The MIT License (MIT) 4 # 5 # Copyright © 2024 pacman64 6 # 7 # Permission is hereby granted, free of charge, to any person obtaining a copy 8 # of this software and associated documentation files (the “Software”), to deal 9 # in the Software without restriction, including without limitation the rights 10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 11 # copies of the Software, and to permit persons to whom the Software is 12 # furnished to do so, subject to the following conditions: 13 # 14 # The above copyright notice and this permission notice shall be included in 15 # all copies or substantial portions of the Software. 16 # 17 # THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 23 # SOFTWARE. 24 25 26 from re import compile, Match, Pattern 27 from sys import argv, exit, stderr, stdin, stdout 28 from typing import List, Tuple 29 30 31 info = ''' 32 coma [regex/style pairs...] 33 34 COlor MAtches colors/styles regex matches everywhere they're found, 35 using the named-style associated to the regex in its argument-pair. 36 37 Lines not matching any regex stay verbatim. 38 39 The colors/styles available are: 40 blue 41 bold 42 gray 43 green 44 inverse 45 magenta 46 orange 47 purple 48 red 49 underline 50 51 Some style aliases are: 52 b blue 53 g green 54 m magenta 55 o orange 56 p purple 57 r red 58 u underline 59 hi inverse (highlight) 60 ''' 61 62 # ansi_re matches ANSI-style sequences, so they're only matched `around` 63 ansi_re = compile('\x1b\\[[0-9;]*[A-Za-z]') 64 65 66 def fail(msg, code: int = 1) -> None: 67 'Show the error message given, and quit the app right away.' 68 print(f'\x1b[31m{msg}\x1b[0m', file=stderr) 69 exit(code) 70 71 72 def match(src: str, start: int, stop: int) -> Tuple[Match, str]: 73 first = None 74 style = '' 75 for expr, st in pairs: 76 m = expr.search(src, start, stop) 77 if (not first) or (m and m.start() < first.start()): 78 first = m 79 style = st 80 return first, style 81 82 83 def style_line(w, s: str) -> None: 84 # start is used outside the regex-match loop to handle trailing parts 85 # in lines 86 start = 0 87 88 # replace all regex-matches on the line by surrounding each matched 89 # substring with ANSI styles/resets 90 while True: 91 m = ansi_re.search(s, start) 92 if not m: 93 start = style_chunk(w, s, start, len(s)) 94 break 95 96 stop = m.start() 97 start = style_chunk(w, s, start, stop) 98 # don't forget the last part of the line, or the whole line 99 stop = m.end() 100 w.write(s[start:stop]) 101 start = stop 102 103 # don't forget the last part of the line, or the whole line 104 w.write(s[start:]) 105 w.write('\n') 106 107 108 def style_chunk(w, s: str, start: int, stop: int) -> int: 109 while True: 110 m, style = match(s, start, stop) 111 if not m: 112 return start 113 114 i = m.start() 115 j = m.end() 116 117 # part before match 118 w.write(s[start:i]) 119 120 # current match 121 w.write(style) 122 w.write(s[i:j]) 123 w.write('\x1b[0m') 124 125 # the end of the match is the start of the `rest` of the string 126 start = j 127 128 129 # names_aliases normalizes lookup keys for table names2styles 130 names_aliases = { 131 'b': 'blue', 132 'g': 'green', 133 'm': 'magenta', 134 'o': 'orange', 135 'p': 'purple', 136 'r': 'red', 137 'u': 'underline', 138 139 'bb': 'blueback', 140 'bg': 'greenback', 141 'bm': 'magentaback', 142 'bo': 'orangeback', 143 'bp': 'purpleback', 144 'br': 'redback', 145 146 'gb': 'greenback', 147 'mb': 'magentaback', 148 'ob': 'orangeback', 149 'pb': 'purpleback', 150 'rb': 'redback', 151 152 'hi': 'inverse', 153 'inv': 'inverse', 154 'mag': 'magenta', 155 156 'du': 'doubleunderline', 157 158 'flip': 'inverse', 159 'swap': 'inverse', 160 161 'reset': 'plain', 162 'highlight': 'inverse', 163 'hilite': 'inverse', 164 'invert': 'inverse', 165 'inverted': 'inverse', 166 'swapped': 'inverse', 167 168 'dunderline': 'doubleunderline', 169 'dunderlined': 'doubleunderline', 170 171 'strikethrough': 'strike', 172 'strikethru': 'strike', 173 'struck': 'strike', 174 175 'underlined': 'underline', 176 177 'bblue': 'blueback', 178 'bgray': 'grayback', 179 'bgreen': 'greenback', 180 'bmagenta': 'magback', 181 'bmagenta': 'magentaback', 182 'borange': 'orangeback', 183 'bpurple': 'purpleback', 184 'bred': 'redback', 185 186 'bgblue': 'blueback', 187 'bggray': 'grayback', 188 'bggreen': 'greenback', 189 'bgmag': 'magentaback', 190 'bgmagenta': 'magentaback', 191 'bgorange': 'orangeback', 192 'bgpurple': 'purpleback', 193 'bgred': 'redback', 194 195 'bluebg': 'blueback', 196 'graybg': 'grayback', 197 'greenbg': 'greenback', 198 'magbg': 'magentaback', 199 'magentabg': 'magentaback', 200 'orangebg': 'orangeback', 201 'purplebg': 'purpleback', 202 'redbg': 'redback', 203 204 'backblue': 'blueback', 205 'backgray': 'grayback', 206 'backgreen': 'greenback', 207 'backmag': 'magentaback', 208 'backmagenta': 'magentaback', 209 'backorange': 'orangeback', 210 'backpurple': 'purpleback', 211 'backred': 'redback', 212 } 213 214 # names2styles matches color/style names to their ANSI-style strings 215 names2styles = { 216 'blue': '\x1b[38;2;0;95;215m', 217 'bold': '\x1b[1m', 218 'doubleunderline': '\x1b[21m', 219 'gray': '\x1b[38;2;168;168;168m', 220 'green': '\x1b[38;2;0;135;95m', 221 'inverse': '\x1b[7m', 222 'magenta': '\x1b[38;2;215;0;255m', 223 'orange': '\x1b[38;2;215;95;0m', 224 'plain': '\x1b[0m', 225 'purple': '\x1b[38;2;135;95;255m', 226 'red': '\x1b[38;2;204;0;0m', 227 'strike': '\x1b[9m', 228 'underline': '\x1b[4m', 229 230 'blueback': '\x1b[48;2;0;95;215m\x1b[38;2;238;238;238m', 231 'grayback': '\x1b[48;2;168;168;168m\x1b[38;2;238;238;238m', 232 'greenback': '\x1b[48;2;0;135;95m\x1b[38;2;238;238;238m', 233 'magentaback': '\x1b[48;2;215;0;255m\x1b[38;2;238;238;238m', 234 'orangeback': '\x1b[48;2;215;95;0m\x1b[38;2;238;238;238m', 235 'purpleback': '\x1b[48;2;135;95;255m\x1b[38;2;238;238;238m', 236 'redback': '\x1b[48;2;204;0;0m\x1b[38;2;238;238;238m', 237 } 238 239 240 # ensure an even number of args, after any leading options 241 if len(argv) % 2 != 1: 242 pre = 'expected an even number of args as regex/style pairs' 243 fail(f'{pre}, but got {len(argv) - 1} instead') 244 245 # make regex/ANSI-style pairs which are directly usable 246 pairs: List[Tuple[Pattern, str]] = [] 247 start_args = 1 248 while start_args + 1 < len(argv): 249 # compile regex 250 try: 251 expr = compile(argv[start_args + 0]) 252 except Exception as e: 253 fail(e) 254 255 # lookup style name 256 name = argv[start_args + 1] 257 if name in names_aliases: 258 name = names_aliases[name] 259 if not name in names2styles: 260 fail(f'style named {name} not supported') 261 262 # remember both for later 263 pairs.append((expr, names2styles[name])) 264 # previous check ensures args has an even number of items 265 start_args += 2 266 267 try: 268 for line in stdin: 269 # ignore trailing carriage-returns and/or line-feeds in input lines 270 line = line.rstrip('\r\n').rstrip('\n') 271 style_line(stdout, line) 272 except BrokenPipeError: 273 # quit quietly, instead of showing a confusing error message 274 pass 275 except KeyboardInterrupt: 276 exit(2) 277 except Exception as e: 278 fail(e, 1)