#!/usr/bin/env python3
import utime
from machine import WDT
from settings import Settings
from presets import Presets


def run_for(p, wdt, ms):
    """Helper: run current pattern for given ms using tick()."""
    start = utime.ticks_ms()
    while utime.ticks_diff(utime.ticks_ms(), start) < ms:
        wdt.feed()
        p.tick()
        utime.sleep_ms(10)


def main():
    s = Settings()
    pin = s.get("led_pin", 10)
    num = s.get("num_leds", 30)

    p = Presets(pin=pin, num_leds=num)
    wdt = WDT(timeout=10000)
    
    # Test 1: Basic rainbow with auto=True (continuous)
    print("Test 1: Basic rainbow (auto=True, n1=1)")
    p.edit("rainbow1", {
        "p": "rainbow",
        "b": 255,
        "d": 100,
        "n1": 1,
        "a": True,
    })
    p.select("rainbow1")
    run_for(p, wdt, 3000)

    # Test 2: Fast rainbow
    print("Test 2: Fast rainbow (low delay, n1=1)")
    p.edit("rainbow2", {
        "p": "rainbow",
        "d": 50,
        "n1": 1,
        "a": True,
    })
    p.select("rainbow2")
    run_for(p, wdt, 2000)

    # Test 3: Slow rainbow
    print("Test 3: Slow rainbow (high delay, n1=1)")
    p.edit("rainbow3", {
        "p": "rainbow",
        "d": 500,
        "n1": 1,
        "a": True,
    })
    p.select("rainbow3")
    run_for(p, wdt, 3000)

    # Test 4: Low brightness rainbow
    print("Test 4: Low brightness rainbow (n1=1)")
    p.edit("rainbow4", {
        "p": "rainbow",
        "b": 64,
        "d": 100,
        "n1": 1,
        "a": True,
    })
    p.select("rainbow4")
    run_for(p, wdt, 2000)

    # Test 5: Single-step rainbow (auto=False)
    print("Test 5: Single-step rainbow (auto=False, n1=1)")
    p.edit("rainbow5", {
        "p": "rainbow",
        "b": 255,
        "d": 100,
        "n1": 1,
        "a": False,
    })
    p.step = 0
    for i in range(10):
        p.select("rainbow5")
        # One tick advances the generator one frame when auto=False
        p.tick()
        utime.sleep_ms(100)
        wdt.feed()

    # Test 6: Verify step updates correctly
    print("Test 6: Verify step updates (auto=False, n1=1)")
    p.edit("rainbow6", {
        "p": "rainbow",
        "n1": 1,
        "a": False,
    })
    initial_step = p.step
    p.select("rainbow6")
    p.tick()
    final_step = p.step
    print(f"Step updated from {initial_step} to {final_step} (expected increment: 1)")

    # Test 7: Fast step increment (n1=5)
    print("Test 7: Fast rainbow (n1=5, auto=True)")
    p.edit("rainbow7", {
        "p": "rainbow",
        "b": 255,
        "d": 100,
        "n1": 5,
        "a": True,
    })
    p.select("rainbow7")
    run_for(p, wdt, 2000)

    # Test 8: Very fast step increment (n1=10)
    print("Test 8: Very fast rainbow (n1=10, auto=True)")
    p.edit("rainbow8", {
        "p": "rainbow",
        "n1": 10,
        "a": True,
    })
    p.select("rainbow8")
    run_for(p, wdt, 2000)

    # Test 9: Verify n1 controls step increment (auto=False)
    print("Test 9: Verify n1 step increment (auto=False, n1=5)")
    p.edit("rainbow9", {
        "p": "rainbow",
        "n1": 5,
        "a": False,
    })
    p.step = 0
    initial_step = p.step
    p.select("rainbow9")
    p.tick()
    final_step = p.step
    expected_step = (initial_step + 5) % 256
    print(f"Step updated from {initial_step} to {final_step} (expected: {expected_step})")
    if final_step == expected_step:
        print("✓ n1 step increment working correctly")
    else:
        print(f"✗ Step increment mismatch! Expected {expected_step}, got {final_step}")

    # Cleanup
    print("Test complete, turning off")
    p.edit("cleanup_off", {"p": "off"})
    p.select("cleanup_off")
    run_for(p, wdt, 100)


if __name__ == "__main__":
    main()