[+] Run neofetch with a preset

hyfetch/neofetch_util.py

@@ -1,8 +1,13 @@
 import os
+import re
+from pathlib import Path
 from subprocess import check_output
+from tempfile import TemporaryDirectory
 
 import pkg_resources
 
+from .presets import ColorProfile
+
 
 def get_command_path() -> str:
     """
@@ -13,11 +18,34 @@ def get_command_path() -> str:
     return pkg_resources.resource_filename(__name__, 'scripts/neofetch_mod.sh')
 
 
-def get_distro_ascii_lines() -> int:
+def get_distro_ascii() -> str:
     """
-    Get how many lines are in the distro ascii
+    Get the distro ascii
 
-    :return: Number of lines
+    :return: Distro ascii
     """
-    out = check_output([get_command_path(), "print_ascii"])
-    return len(out.decode().strip().split('\n'))
+    return check_output([get_command_path(), "print_ascii"]).decode().strip()
+
+
+def run(preset: ColorProfile):
+    # Get existing ascii
+    asc = get_distro_ascii()
+
+    # Remove existing colors
+    asc = re.sub('\\${.*?}', '', asc)
+
+    # Add new colors
+    lines = asc.split('\n')
+    colors = preset.with_length(len(lines))
+    asc = '\n'.join([colors[i].to_ansi_rgb() + l for i, l in enumerate(lines)])
+
+    # Write temp file
+    with TemporaryDirectory() as tmp_dir:
+        tmp_dir = Path(tmp_dir)
+        path = tmp_dir / 'ascii.txt'
+        path.write_text(asc)
+
+        # Call neofetch with the temp file
+        os.environ['ascii_len'] = str(max(len(l) for l in lines))
+        os.environ['ascii_lines'] = str(len(lines))
+        os.system(get_command_path() + f' --ascii --source {path.absolute()} --ascii-colors')