mirror of
https://github.com/ItsDrike/nixdots
synced 2024-12-30 23:24:35 +00:00
116 lines
4.4 KiB
Python
Executable file
116 lines
4.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
"""This is a utility script for regex remaps on window names.
|
|
|
|
This is done in python, because of the complex regex logic, which would be pretty hard to
|
|
recreate in bash. This python script is expected to be called from the bash script controlling
|
|
the window names. Window name and class are obtained from piped stdin, to prevent having to
|
|
needlessly keep restarting this program, which takes a while due to the interpreter starting
|
|
overhead.
|
|
"""
|
|
|
|
import json
|
|
import re
|
|
import sys
|
|
from typing import Iterator, Optional
|
|
|
|
|
|
class RemapRule:
|
|
__slots__ = ("name_pattern", "output_pattern", "class_pattern")
|
|
|
|
def __init__(
|
|
self,
|
|
name_pattern: str,
|
|
output_pattern: str,
|
|
class_pattern: Optional[str] = None,
|
|
):
|
|
self.name_pattern = re.compile(name_pattern)
|
|
self.output_pattern = output_pattern
|
|
self.class_pattern = re.compile(class_pattern) if class_pattern else None
|
|
|
|
def apply(self, window_name: str, window_class: str) -> str:
|
|
"""Returns new name after this remap rule was applied."""
|
|
if self.class_pattern is not None:
|
|
if not self.class_pattern.fullmatch(window_class):
|
|
# Rule doesn't apply, class mismatch, return original name
|
|
return window_name
|
|
|
|
res = self.name_pattern.fullmatch(window_name)
|
|
if not res:
|
|
# Rule doesn't apply, name mismatch, return original name
|
|
return window_name
|
|
|
|
# NOTE: This is potentially unsafe, since output_pattern might be say {0.__class__}, etc.
|
|
# meaning this allows arbitrary attribute access, however it doesn't allow running functions
|
|
# here. That said, code could still end up being run if there's some descriptor defined,
|
|
# and generally I wouldn't trust this in production. However, this code is for my personal
|
|
# use here, and all of the output patterns are hard-coded in this file, so in this case, it's
|
|
# fine. But if you see this code and you'd like to use it in your production code, maybe don't
|
|
return self.output_pattern.format(*res.groups())
|
|
|
|
|
|
# Rules will be applied in specified order
|
|
REMAP_RULES: list[RemapRule] = [
|
|
RemapRule(r"", "", ""),
|
|
RemapRule(r"(.*) — Mozilla Firefox", " {}", "firefox"),
|
|
RemapRule(r"Mozilla Firefox", " Mozilla Firefox", "firefox"),
|
|
RemapRule(r"Alacritty", " Alacritty", "Alacritty"),
|
|
RemapRule(
|
|
r"zsh;#toggleterm#1 - \(term:\/\/(.+)\/\/(\d+):(.+)\) - N?VIM",
|
|
" Terminal: {0}",
|
|
),
|
|
RemapRule(r"(.+) \+ \((.+)\) - N?VIM", " {0} ({1}) [MODIFIED]"),
|
|
RemapRule(r"(.+) \((.+)\) - N?VIM", " {0} ({1})"),
|
|
RemapRule(r"(?:\[\d+\] )?\*?WebCord - (.+)", " {}", "WebCord"),
|
|
RemapRule(r"(.+) - Discord", " {}", "discord"),
|
|
RemapRule(r"(?:\(\d+\) )?Discord \| (.+)", " {}", "vesktop"),
|
|
RemapRule(r"(.+) - mpv", " {}", "mpv"),
|
|
RemapRule(r"Stremio - (.+)", " Stremio - {}", r"(Stremio)|(com.stremio.stremio)"),
|
|
RemapRule(r"Spotify((?: Premium)?)", " Spotify{}", "[Ss]potify"),
|
|
RemapRule(r"pulsemixer", " Pulsemixer"),
|
|
RemapRule(r"(.*)", " {}", "Pcmanfm"),
|
|
RemapRule(r"(.*)", " {}", "pcmanfm-qt"),
|
|
# Needs to be last
|
|
RemapRule(r"(.*)", " {}", "kitty"),
|
|
]
|
|
|
|
MAX_LENGTH = 65
|
|
|
|
|
|
def iter_window() -> Iterator[tuple[str, str]]:
|
|
"""Listen for the window parameters from stdin/pipe, yields (window_name, window_class)."""
|
|
for line in sys.stdin:
|
|
line = line.removesuffix("\n")
|
|
els = line.split(",", maxsplit=1)
|
|
if len(els) != 2:
|
|
raise ValueError(
|
|
f"Expected 2 arguments from stdin line (name, class), but got {len(els)}"
|
|
)
|
|
yield els[1], els[0]
|
|
|
|
|
|
def main() -> None:
|
|
for window_name, window_class in iter_window():
|
|
formatted_name = window_name
|
|
for remap_rule in REMAP_RULES:
|
|
new_name = remap_rule.apply(formatted_name, window_class)
|
|
if new_name != formatted_name:
|
|
formatted_name = new_name
|
|
break
|
|
|
|
if len(formatted_name) > MAX_LENGTH:
|
|
formatted_name = formatted_name[: MAX_LENGTH - 3] + "..."
|
|
|
|
ret = json.dumps(
|
|
{
|
|
"name": window_name,
|
|
"class": window_class,
|
|
"formatted_name": formatted_name,
|
|
}
|
|
)
|
|
print(ret)
|
|
sys.stdout.flush()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|