Menno van Leeuwen
62954eb986
Some checks failed
Nix Format Check / check-format (push) Failing after 37s
98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import math
|
|
import random
|
|
try:
|
|
import pyfiglet
|
|
except ImportError:
|
|
pyfiglet = None
|
|
|
|
# Color codes for terminal output
|
|
COLORS = {
|
|
"black": "\033[0;30m",
|
|
"red": "\033[0;31m",
|
|
"green": "\033[0;32m",
|
|
"yellow": "\033[0;33m",
|
|
"blue": "\033[0;34m",
|
|
"purple": "\033[0;35m",
|
|
"cyan": "\033[0;36m",
|
|
"white": "\033[0;37m",
|
|
"reset": "\033[0m"
|
|
}
|
|
|
|
def printfe(color, message):
|
|
"""Print a formatted message with the specified color"""
|
|
color_code = COLORS.get(color.lower(), COLORS["reset"])
|
|
print(f"{color_code}{message}{COLORS['reset']}")
|
|
|
|
def println(message, color=None):
|
|
"""Print a line with optional color"""
|
|
if color:
|
|
printfe(color, message)
|
|
else:
|
|
print(message)
|
|
|
|
def _rainbow_color(text, freq=0.1, offset=0):
|
|
"""Apply rainbow colors to text similar to lolcat"""
|
|
colored_text = ""
|
|
for i, char in enumerate(text):
|
|
if char.strip(): # Only color non-whitespace characters
|
|
# Calculate RGB values using sine waves with phase shifts
|
|
r = int(127 * math.sin(freq * i + offset + 0) + 128)
|
|
g = int(127 * math.sin(freq * i + offset + 2 * math.pi / 3) + 128)
|
|
b = int(127 * math.sin(freq * i + offset + 4 * math.pi / 3) + 128)
|
|
|
|
# Apply the RGB color to the character
|
|
colored_text += f"\033[38;2;{r};{g};{b}m{char}\033[0m"
|
|
else:
|
|
colored_text += char
|
|
|
|
return colored_text
|
|
|
|
def logo(continue_after=False):
|
|
"""Display the dotfiles logo"""
|
|
try:
|
|
# Try to read logo file first for backward compatibility
|
|
logo_path = f"{os.environ.get('DOTFILES_PATH', os.path.expanduser('~/.dotfiles'))}/bin/resources/logo.txt"
|
|
if os.path.exists(logo_path):
|
|
with open(logo_path, "r") as f:
|
|
logo_text = f.read()
|
|
print(logo_text)
|
|
elif pyfiglet:
|
|
# Generate ASCII art with pyfiglet and rainbow colors
|
|
ascii_art = pyfiglet.figlet_format("Menno's Dotfiles", font='slant')
|
|
print("\n") # Add some space before the logo
|
|
|
|
# Use a random offset to vary the rainbow start position
|
|
random_offset = random.random() * 2 * math.pi
|
|
line_offset = 0
|
|
|
|
for line in ascii_art.splitlines():
|
|
# Add a little variation to each line
|
|
print(_rainbow_color(line, offset=random_offset + line_offset))
|
|
line_offset += 0.1
|
|
|
|
print("\n") # Add some space after the logo
|
|
else:
|
|
# Fallback if pyfiglet is not available
|
|
printfe("yellow", "\n\n *** Menno's Dotfiles ***\n\n")
|
|
printfe("cyan", " Note: Install pyfiglet for better logo display")
|
|
printfe("cyan", " (pip install pyfiglet)\n")
|
|
|
|
if not continue_after:
|
|
sys.exit(0)
|
|
except Exception as e:
|
|
printfe("red", f"Error displaying logo: {e}")
|
|
|
|
def run_command(command, shell=False):
|
|
"""Run a shell command and return the result"""
|
|
try:
|
|
result = subprocess.run(command, shell=shell, check=True, text=True,
|
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
return True, result.stdout.strip()
|
|
except subprocess.CalledProcessError as e:
|
|
return False, e.stderr.strip()
|