hyfetch/tools/list_distros.py
2022-08-20 09:43:52 -04:00

128 lines
3.5 KiB
Python
Executable file

#!/usr/bin/env python3
"""
List distributions supported by neofetch
"""
from __future__ import annotations
import string
import sys
import textwrap
from pathlib import Path
import regex
sys.path.append(str(Path(__file__).parent.parent))
from hyfetch.distro import AsciiArt
RE_SPLIT = regex.compile('EOF[ \n]*?;;')
RE_COLORS = regex.compile("""(?<=set_colors )[a-z\\d ]+(?=\n)""")
def substr(s: str, start: str, end: str | None = None):
"""
Get substring between start and end
"""
start = s.index(start) + len(start)
if end is None:
return s[start:]
return s[start:s.index(end, start)]
def parse_ascii_distros() -> list[AsciiArt]:
"""
Parse ascii distros from neofetch script
"""
nf = (Path(__file__).parent.parent / 'neofetch').read_text()
# Get the content of "get_distro_ascii" function
nf = nf[nf.index('get_distro_ascii() {\n'):]
nf = nf[:nf.index('\n}\n')]
# Remove trailing spaces
while ' \n' in nf:
nf = nf.replace(' \n', '\n')
# Split by blocks
blocks = [sub.strip() for b in regex.split('case .*? in\n', nf) for sub in RE_SPLIT.split(b)]
# Parse blocks
def parse_block(block: str) -> AsciiArt:
try:
# Get ascii art
assert "'EOF'\n" in block
art = substr(block, "'EOF'\n")
# Join \
block = block.replace('\\\n', ' ')
# Get switch-case matching parameter
match = block.split('\n')[0].strip()
assert match.endswith(')')
match = match[:-1]
# Get colors
color = RE_COLORS.findall(block)[0]
if len(color) == 0:
raise Exception(block)
return AsciiArt(match, color, art)
except AssertionError:
pass
out = [parse_block(block) for block in blocks]
return [v for v in out if v]
def wrap(text: str, max_len: int, leading: str):
length = max_len - len(leading)
lines = [line for raw in text.split('\n') for line in textwrap.wrap(raw, length) or ['']]
return '\n'.join(leading + line if line else line for line in lines)
def generate_help(max_len: int, leading: str):
distros = sorted(list({a.get_friendly_name() for a in parse_ascii_distros()}), key=str.casefold)
smalls = [d.replace('_small', '') for d in distros if d.endswith('_small')]
olds = [d.replace('_old', '') for d in distros if d.endswith('_old')]
distros = [d for d in distros if not d.endswith('_small') and not d.endswith('_old')]
out = f"NOTE: {', '.join(distros)} have ascii logos.\n\n"\
f"NOTE: {', '.join(olds)} have 'old' logo variants, use {{distro}}_old to use them.\n\n" \
f"NOTE: {', '.join(smalls)} have 'small' logo variants, use {{distro}}_small to use them."
return wrap(out, max_len, leading)
def export_distro(d: AsciiArt):
"""
Export distro to a python script
"""
varname = d.name.lower()
for s in string.punctuation + ' ':
varname = varname.replace(s, '_')
ascii = d.ascii.replace('"""', '\\"""')
script = f"""
from hyfetch.distro import AsciiArt
{varname} = AsciiArt(match=r'''{d.match}''', color='{d.color}', ascii=r\"""
{ascii}
\""")
"""
Path('distros').mkdir(parents=True, exist_ok=True)
Path(f'distros/{varname}.py').write_text(script)
def export_distros():
distros = parse_ascii_distros()
[export_distro(d) for d in distros]
if __name__ == '__main__':
# print(generate_help(100, ' ' * 32))
# print(generate_help(100, '# '))
export_distros()