mirror of
https://github.com/elicpeter/nyx.git
synced 2026-06-09 19:45:13 +02:00
37 lines
1 KiB
Python
37 lines
1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Merge consecutive ANSI SGR escape sequences into a single compound form.
|
||
|
|
|
||
|
|
freeze (charm.sh) renders only the most recently seen SGR escape, so
|
||
|
|
the `console` crate's habit of emitting separate `\x1b[34m\x1b[2m\x1b[4m`
|
||
|
|
sequences erases all but the last attribute. Pipe nyx output through
|
||
|
|
this filter to consolidate runs into `\x1b[34;2;4m` so freeze keeps
|
||
|
|
foreground, dim, and underline.
|
||
|
|
|
||
|
|
Usage: stream stdin -> stdout, e.g. `nyx scan | python3 sgr-merge.py`.
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
SGR_RUN = re.compile(r"(?:\x1b\[(\d+(?:;\d+)*)m){2,}")
|
||
|
|
|
||
|
|
|
||
|
|
def _merge(match: re.Match) -> str:
|
||
|
|
runs = re.findall(r"\x1b\[(\d+(?:;\d+)*)m", match.group(0))
|
||
|
|
return "\x1b[" + ";".join(runs) + "m"
|
||
|
|
|
||
|
|
|
||
|
|
def merge_sgr(s: str) -> str:
|
||
|
|
return SGR_RUN.sub(_merge, s)
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> int:
|
||
|
|
data = sys.stdin.buffer.read().decode("utf-8", errors="replace")
|
||
|
|
sys.stdout.buffer.write(merge_sgr(data).encode("utf-8"))
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|