111 lines
3.1 KiB
Python
Executable file
111 lines
3.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
"""
|
|
List distributions supported by neofetch
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import string
|
|
import textwrap
|
|
from dataclasses import dataclass
|
|
|
|
from pathlib import Path
|
|
|
|
import regex
|
|
|
|
RE_SPLIT = regex.compile('EOF[ \n]*?;;')
|
|
RE_COLORS = regex.compile("""(?<=set_colors )[a-z\\d ]+(?=\n)""")
|
|
|
|
|
|
@dataclass
|
|
class AsciiArt:
|
|
match: str
|
|
color: str
|
|
ascii: str
|
|
|
|
def get_friendly_name(self) -> str:
|
|
return self.match.split("|")[0].strip(string.punctuation + '* ')\
|
|
.replace('"', '').replace('*', '')
|
|
|
|
|
|
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)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print(generate_help(100, ' ' * 32))
|
|
print(generate_help(100, '# '))
|