File: icoma.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, IGNORECASE, Match, Pattern 27 from sys import argv, exit, stderr, stdin, stdout 28 from typing import List, Tuple 29 30 31 info = ''' 32 icoma [regex/style pairs...] 33 34 Insensitive COlor MAtches colors/styles case-insensitive regex-matches 35 everywhere they're found, using the named-style associated to the regex 36 in its argument-pair. 37 38 Lines not matching any regex stay verbatim. 39 40 The colors/styles available are: 41 blue 42 bold 43 gray 44 green 45 inverse 46 magenta 47 orange 48 purple 49 red 50 underline 51 52 Some style aliases are: 53 b blue 54 g green 55 m magenta 56 o orange 57 p purple 58 r red 59 u underline 60 hi inverse (highlight) 61 ''' 62 63 # ansi_re matches ANSI-style sequences, so they're only matched `around` 64 ansi_re = compile('\x1b\\[[0-9;]*[A-Za-z]') 65 66 67 def fail(msg, code: int = 1) -> None: 68 'Show the error message given, and quit the app right away.' 69 print(f'\x1b[31m{msg}\x1b[0m', file=stderr) 70 exit(code) 71 72 73 def match(src: str, start: int, stop: int) -> Tuple[Match, str]: 74 first = None 75 style = '' 76 for expr, st in pairs: 77 m = expr.search(src, start, stop) 78 if (not first) or (m and m.start() < first.start()): 79 first = m 80 style = st 81 return first, style 82 83 84 def style_line(w, s: str) -> None: 85 # start is used outside the regex-match loop to handle trailing parts 86 # in lines 87 start = 0 88 89 # replace all regex-matches on the line by surrounding each matched 90 # substring with ANSI styles/resets 91 while True: 92 m = ansi_re.search(s, start) 93 if not m: 94 start = style_chunk(w, s, start, len(s)) 95 break 96 97 stop = m.start() 98 start = style_chunk(w, s, start, stop) 99 # don't forget the last part of the line, or the whole line 100 stop = m.end() 101 w.write(s[start:stop]) 102 start = stop 103 104 # don't forget the last part of the line, or the whole line 105 w.write(s[start:]) 106 w.write('\n') 107 108 109 def style_chunk(w, s: str, start: int, stop: int) -> int: 110 while True: 111 m, style = match(s, start, stop) 112 if not m: 113 return start 114 115 i = m.start() 116 j = m.end() 117 118 # part before match 119 w.write(s[start:i]) 120 121 # current match 122 w.write(style) 123 w.write(s[i:j]) 124 w.write('\x1b[0m') 125 126 # the end of the match is the start of the `rest` of the string 127 start = j 128 129 130 # names_aliases normalizes lookup keys for table names2styles 131 names_aliases = { 132 'b': 'blue', 133 'g': 'green', 134 'm': 'magenta', 135 'o': 'orange', 136 'p': 'purple', 137 'r': 'red', 138 'u': 'underline', 139 140 'bb': 'blueback', 141 'bg': 'greenback', 142 'bm': 'magentaback', 143 'bo': 'orangeback', 144 'bp': 'purpleback', 145 'br': 'redback', 146 147 'gb': 'greenback', 148 'mb': 'magentaback', 149 'ob': 'orangeback', 150 'pb': 'purpleback', 151 'rb': 'redback', 152 153 'hi': 'inverse', 154 'inv': 'inverse', 155 'mag': 'magenta', 156 157 'du': 'doubleunderline', 158 159 'flip': 'inverse', 160 'swap': 'inverse', 161 162 'reset': 'plain', 163 'highlight': 'inverse', 164 'hilite': 'inverse', 165 'invert': 'inverse', 166 'inverted': 'inverse', 167 'swapped': 'inverse', 168 169 'dunderline': 'doubleunderline', 170 'dunderlined': 'doubleunderline', 171 172 'strikethrough': 'strike', 173 'strikethru': 'strike', 174 'struck': 'strike', 175 176 'underlined': 'underline', 177 178 'bblue': 'blueback', 179 'bgray': 'grayback', 180 'bgreen': 'greenback', 181 'bmagenta': 'magback', 182 'bmagenta': 'magentaback', 183 'borange': 'orangeback', 184 'bpurple': 'purpleback', 185 'bred': 'redback', 186 187 'bgblue': 'blueback', 188 'bggray': 'grayback', 189 'bggreen': 'greenback', 190 'bgmag': 'magentaback', 191 'bgmagenta': 'magentaback', 192 'bgorange': 'orangeback', 193 'bgpurple': 'purpleback', 194 'bgred': 'redback', 195 196 'bluebg': 'blueback', 197 'graybg': 'grayback', 198 'greenbg': 'greenback', 199 'magbg': 'magentaback', 200 'magentabg': 'magentaback', 201 'orangebg': 'orangeback', 202 'purplebg': 'purpleback', 203 'redbg': 'redback', 204 205 'backblue': 'blueback', 206 'backgray': 'grayback', 207 'backgreen': 'greenback', 208 'backmag': 'magentaback', 209 'backmagenta': 'magentaback', 210 'backorange': 'orangeback', 211 'backpurple': 'purpleback', 212 'backred': 'redback', 213 } 214 215 # names2styles matches color/style names to their ANSI-style strings 216 names2styles = { 217 'blue': '\x1b[38;2;0;95;215m', 218 'bold': '\x1b[1m', 219 'doubleunderline': '\x1b[21m', 220 'gray': '\x1b[38;2;168;168;168m', 221 'green': '\x1b[38;2;0;135;95m', 222 'inverse': '\x1b[7m', 223 'magenta': '\x1b[38;2;215;0;255m', 224 'orange': '\x1b[38;2;215;95;0m', 225 'plain': '\x1b[0m', 226 'purple': '\x1b[38;2;135;95;255m', 227 'red': '\x1b[38;2;204;0;0m', 228 'strike': '\x1b[9m', 229 'underline': '\x1b[4m', 230 231 'blueback': '\x1b[48;2;0;95;215m\x1b[38;2;238;238;238m', 232 'grayback': '\x1b[48;2;168;168;168m\x1b[38;2;238;238;238m', 233 'greenback': '\x1b[48;2;0;135;95m\x1b[38;2;238;238;238m', 234 'magentaback': '\x1b[48;2;215;0;255m\x1b[38;2;238;238;238m', 235 'orangeback': '\x1b[48;2;215;95;0m\x1b[38;2;238;238;238m', 236 'purpleback': '\x1b[48;2;135;95;255m\x1b[38;2;238;238;238m', 237 'redback': '\x1b[48;2;204;0;0m\x1b[38;2;238;238;238m', 238 } 239 240 241 # ensure an even number of args, after any leading options 242 if len(argv) % 2 != 1: 243 pre = 'expected an even number of args as regex/style pairs' 244 fail(f'{pre}, but got {len(argv) - 1} instead') 245 246 # make regex/ANSI-style pairs which are directly usable 247 pairs: List[Tuple[Pattern, str]] = [] 248 start_args = 1 249 while start_args + 1 < len(argv): 250 # compile regex 251 try: 252 expr = compile(argv[start_args + 0], flags=IGNORECASE) 253 except Exception as e: 254 fail(e) 255 256 # lookup style name 257 name = argv[start_args + 1] 258 if name in names_aliases: 259 name = names_aliases[name] 260 if not name in names2styles: 261 fail(f'style named {name} not supported') 262 263 # remember both for later 264 pairs.append((expr, names2styles[name])) 265 # previous check ensures args has an even number of items 266 start_args += 2 267 268 try: 269 for line in stdin: 270 # ignore trailing carriage-returns and/or line-feeds in input lines 271 line = line.rstrip('\r\n').rstrip('\n') 272 style_line(stdout, line) 273 except BrokenPipeError: 274 # quit quietly, instead of showing a confusing error message 275 pass 276 except KeyboardInterrupt: 277 exit(2) 278 except Exception as e: 279 fail(e, 1)