import colorsys
from math import pow
from typing import Tuple
from datetime import datetime

# Function to calculate relative luminance of a color
def relative_luminance(rgb: Tuple[int, int, int]) -> float:
    def adjust(c: float) -> float:
        c = c / 255.0
        return c / 12.92 if c <= 0.03928 else pow((c + 0.055) / 1.055, 2.4)

    r, g, b = rgb
    return 0.2126 * adjust(r) + 0.7152 * adjust(g) + 0.0722 * adjust(b)

# Function to calculate contrast ratio
def contrast_ratio(foreground: Tuple[int, int, int], background: Tuple[int, int, int]) -> float:
    lum1 = relative_luminance(foreground)
    lum2 = relative_luminance(background)
    if lum1 > lum2:
        return (lum1 + 0.05) / (lum2 + 0.05)
    return (lum2 + 0.05) / (lum1 + 0.05)

# Generate HTML table
def generate_html(colors: list):
    html = '<style>* {font-family: sans-serif; text-shadow: none;} .textwhite {color: white;} .textblack {color: black;} .bgwhite.result:hover {text-shadow: 1px 1px 1px #000000;} .bgblack.result:hover {text-shadow: 1px 1px 1px #ffffff;} .label {padding: 0.5em; display: block;} .pass100 {color: #000000; background-color: #9cffac;} .pass80 {color: #000000; background-color: #f2ff9c;} .pass60 {color: #000000; background-color: #ffdb9c} .fail {color: #000000; background-color: #ff9c9c;}</style>'
    
    # Helper function to determine text color
    def get_text_color(hex_color):
        rgb = tuple(int(hex_color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
        return "textwhite" if relative_luminance(rgb) < 0.5 else "textblack"

    # Helper function to determine text-shadow color
    def get_text_shadow(hex_color):
        rgb = tuple(int(hex_color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
        return "bgwhite" if relative_luminance(rgb) >= 0.5 else "bgblack"

    # Store passing combinations for summary table
    summary_data = {bg[1]: {"name": bg[0], "AAA": [], "AA": [], "AALarge": []} for bg in colors}

    # Top row
    html += '<h2>Main Contrast Table</h2>'
    html += '<table border="0" style="border-collapse: collapse;">'
    html += '<tr><th style="color: black;">BG &#8594;<br>FG &#8595;</th>'
    for bg in colors:
        text_color = get_text_color(bg[1])
        html += f'<th class="{text_color}" style="background-color: {bg[1]};">{bg[0]}<br>{bg[1]}</th>'
    html += '</tr>'
    
    # Rows for each foreground color
    for fg in colors:
        text_color = get_text_color(fg[1])
        html += f'<tr><td class="{text_color}" style="background-color: {fg[1]};">{fg[0]}<br>{fg[1]}</td>'
        for bg in colors:
            rgb_fg = tuple(int(fg[1].lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
            rgb_bg = tuple(int(bg[1].lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
            ratio = round(contrast_ratio(rgb_fg, rgb_bg), 3)

            # Compliance check and store in summary data
            if ratio >= 7:
                summary_data[bg[1]]["AAA"].append((fg[0], fg[1], ratio))
            elif ratio >= 4.5:
                summary_data[bg[1]]["AA"].append((fg[0], fg[1], ratio))
            elif ratio >= 3:
                summary_data[bg[1]]["AALarge"].append((fg[0], fg[1], ratio))
            
            compliance = (
                "Pass - 100% (AAA &ge; 7.0)" if ratio >= 7 else
                "Pass - 80% (AA Normal &ge; 4.5)" if ratio >= 4.5 else
                "Pass - 60% (AA Large &ge; 3.0)" if ratio >= 3 else
                "Fail - 0% (< 3.0)"
            )
            labelclass = (
                "pass100" if ratio >= 7 else
                "pass80" if ratio >= 4.5 else
                "pass60" if ratio >= 3 else
                "fail"
            )
            text_shadow = get_text_shadow(bg[1])
            html += (
                f'<td style="background-color: {bg[1]}; color: {fg[1]}; padding: 1em;">'
                f'<strong class="result {text_shadow}">{fg[0]} over {bg[0]}</strong>'
                f'<br><span class="label {labelclass}">{compliance} with ratio {ratio}</span>'
                f'</td>'
            )
        html += '</tr>'
    
    html += '</table>'
    
    # Generate summary table
    html += '<h2>Summary Table: Passing Combinations by Category</h2>'
    html += '<table border="1" style="border-collapse: collapse;">'
    html += '<tr><th style="color: black;">Background</th>'
    html += '<th>AAA &ge; 7.0</th><th>AA Normal &ge; 4.5</th><th>AA Large &ge; 3.0</th></tr>'

    for bg_hex, bg_data in summary_data.items():
        html += f'<tr><td style="background-color: {bg_hex}; padding: 1em;" class="{get_text_color(bg_hex)}">{bg_data["name"]}<br>{bg_hex}</td>'
        
        # AAA Column
        html += f'<td style="background-color: {bg_hex}">'
        for fg_name, fg_hex, ratio in sorted(bg_data["AAA"], key=lambda x: x[2], reverse=True):
            html += f'<div style="background-color: {bg_hex}; color: {fg_hex}; padding: 0.5em;">{fg_name} (Ratio: {ratio})</div>'
        html += '</td>'
        
        # AA Column
        html += f'<td style="background-color: {bg_hex}">'
        for fg_name, fg_hex, ratio in sorted(bg_data["AA"], key=lambda x: x[2], reverse=True):
            html += f'<div style="background-color: {bg_hex}; color: {fg_hex}; padding: 0.5em;">{fg_name} (Ratio: {ratio})</div>'
        html += '</td>'
        
        # AA Large Column
        html += f'<td style="background-color: {bg_hex}">'
        for fg_name, fg_hex, ratio in sorted(bg_data["AALarge"], key=lambda x: x[2], reverse=True):
            html += f'<div style="background-color: {bg_hex}; color: {fg_hex}; padding: 0.5em; font-size: 1.5em;">{fg_name} (Ratio: {ratio})</div>'
        html += '</td>'
        
        html += '</tr>'

    html += '</table>'
    return html

# Color palette with names and hex codes
colors = [
    ("Pure White", "#FFFFFF"),
    ("Pure Black", "#000000"),
    ("Maroon", "#8C2531"),
    ("Black", "#222222"),
    ("Dark Gray", "#404040"),
    ("Medium Gray", "#7F7F7F"),
    ("Light Gray", "#BFBFBF"),
    ("Lightest Gray", "#F6F6F6"),
    ("Tan", "#ADA28C"),
    ("Dark Blue", "#002A3B"),
    ("Dark Green", "#225A41"),
    ("Blue", "#005E84"),
    ("Green", "#7E9A4B"),
    ("Orange", "#B2461F"),
    ("Yellow", "#D88A00"),
    ("Purple", "#3C2D57")
]

html_output = generate_html(colors)

# Generate timestamp for filename
timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
filename = f"color_contrast_grid_{timestamp}.html"

# Write to an HTML file
with open(filename, "w") as file:
    file.write(html_output)

print(f"HTML file '{filename}' generated.")