#!/usr/bin/env python3
import colorsys
import sys

def hex_to_rgb(hex_color):
    hex_color = hex_color.lstrip('#')
    return tuple(int(hex_color[i:i+2], 16)/255.0 for i in (0, 2, 4))

def rgb_to_hex(r, g, b):
    r = max(0, min(1, r))
    g = max(0, min(1, g)) 
    b = max(0, min(1, b))
    return f"#{int(r*255):02x}{int(g*255):02x}{int(b*255):02x}"

def generate_palette(base_color, is_light=False, honor_primary=None):
    r, g, b = hex_to_rgb(base_color)
    h, s, v = colorsys.rgb_to_hsv(r, g, b)
    
    palette = []
    
    if is_light:
        palette.append("#f8f8f8")
    else:
        palette.append("#1a1a1a")
    
    red_h = 0.0  # Force true red hue (0 degrees)
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(red_h, 0.75, 0.85)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(red_h, 0.65, 1.0)))
    
    green_h = 0.33  # Force true green hue (120 degrees / 360 = 0.33)
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(green_h, max(s * 0.8, 0.65), v * 0.9)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(green_h, max(s * 0.7, 0.6), v * 1.1)))
    
    yellow_h = 0.16  # Force true yellow hue (60 degrees / 360 = 0.16)
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(yellow_h, max(s * 0.7, 0.55), v * 1.2)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(yellow_h, max(s * 0.5, 0.45), v * 1.4)))
    
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(h, max(s * 0.9, 0.7), v * 1.1)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(h, max(s * 0.7, 0.6), min(v * 1.3, 0.9))))
    
    mag_h = h + 0.15 if h + 0.15 <= 1.0 else h + 0.15 - 1.0  # Derive from primary hue for harmony
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(mag_h, max(s * 0.8, 0.65), v * 0.95)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(mag_h, max(s * 0.65, 0.55), min(v * 1.15, 0.8))))
    
    cyan_h = h + 0.08
    if honor_primary and not is_light:
        palette.append(honor_primary)
    elif is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(cyan_h, max(s * 0.8, 0.65), v * 1.05)))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(cyan_h, max(s * 0.6, 0.5), min(v * 1.25, 0.85))))
    
    if is_light:
        palette.append("#2e2e2e")
        palette.append("#4a4a4a")
    else:
        palette.append("#abb2bf")
        palette.append("#5c6370")
    
    if is_light:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(red_h, 0.6, 0.9)))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(green_h, max(s * 0.7, 0.6), v * 1.25)))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(yellow_h, max(s * 0.6, 0.5), v * 1.35)))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(h, max(s * 0.8, 0.7), min(v * 1.3, 1.0))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(mag_h, max(s * 0.9, 0.75), min(v * 1.25, 1.0))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(cyan_h, max(s * 0.75, 0.65), min(v * 1.25, 1.0))))
    else:
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(red_h, 0.45, min(1.0, 0.9))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(green_h, max(s * 0.5, 0.4), min(v * 1.5, 0.9))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(yellow_h, max(s * 0.4, 0.35), min(v * 1.6, 0.95))))
        if honor_primary:
            hr, hg, hb = hex_to_rgb(honor_primary)
            hh, hs, hv = colorsys.rgb_to_hsv(hr, hg, hb)
            palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(hh, min(hs * 1.2, 1.0), min(hv * 1.1, 1.0))))
        else:
            palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(h, max(s * 0.6, 0.5), min(v * 1.5, 0.9))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(mag_h, max(s * 0.65, 0.55), min(v * 1.4, 0.85))))
        palette.append(rgb_to_hex(*colorsys.hsv_to_rgb(cyan_h, max(s * 0.55, 0.45), min(v * 1.4, 0.85))))
    
    if is_light:
        palette.append("#1a1a1a")
    else:
        palette.append("#ffffff")
    
    return palette

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Usage: b16.py <hex_color> [--light] [--kitty] [--honor-primary HEX]", file=sys.stderr)
        sys.exit(1)
    
    base = sys.argv[1]
    if not base.startswith('#'):
        base = '#' + base
    
    is_light = "--light" in sys.argv
    is_kitty = "--kitty" in sys.argv
    
    honor_primary = None
    if "--honor-primary" in sys.argv:
        try:
            honor_idx = sys.argv.index("--honor-primary")
            if honor_idx + 1 < len(sys.argv):
                honor_primary = sys.argv[honor_idx + 1]
                if not honor_primary.startswith('#'):
                    honor_primary = '#' + honor_primary
        except (ValueError, IndexError):
            print("Error: --honor-primary requires a hex color", file=sys.stderr)
            sys.exit(1)
    
    colors = generate_palette(base, is_light, honor_primary)
    
    if is_kitty:
        # Kitty color format mapping
        kitty_colors = [
            ("color0", colors[0]),
            ("color1", colors[1]),
            ("color2", colors[2]),
            ("color3", colors[3]),
            ("color4", colors[4]),
            ("color5", colors[5]),
            ("color6", colors[6]),
            ("color7", colors[7]),
            ("color8", colors[8]),
            ("color9", colors[9]),
            ("color10", colors[10]),
            ("color11", colors[11]),
            ("color12", colors[12]),
            ("color13", colors[13]),
            ("color14", colors[14]),
            ("color15", colors[15])
        ]
        
        for name, color in kitty_colors:
            print(f"{name}   {color}")
    else:
        for i, color in enumerate(colors):
            print(f"palette = {i}={color}")