#!/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()