#!/usr/bin/env python3
import utime
from machine import WDT
from settings import Settings
from patterns import Patterns


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 = Patterns(pin=pin, num_leds=num)
    wdt = WDT(timeout=10000)

    # Test 1: Basic chase (n1=5, n2=5, n3=1, n4=1)
    print("Test 1: Basic chase (n1=5, n2=5, n3=1, n4=1)")
    p.set_param("br", 255)
    p.set_param("dl", 200)
    p.set_param("n1", 5)
    p.set_param("n2", 5)
    p.set_param("n3", 1)
    p.set_param("n4", 1)
    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])
    p.select("chase")
    run_for(p, wdt, 3000)

    # Test 2: Forward and backward (n3=2, n4=-1)
    print("Test 2: Forward and backward (n3=2, n4=-1)")
    p.set_param("n1", 3)
    p.set_param("n2", 3)
    p.set_param("n3", 2)
    p.set_param("n4", -1)
    p.set_param("dl", 150)
    p.set_param("cl", [(0, 0, 255), (255, 255, 0)])
    p.select("chase")
    run_for(p, wdt, 3000)

    # Test 3: Large segments (n1=10, n2=5)
    print("Test 3: Large segments (n1=10, n2=5, n3=3, n4=3)")
    p.set_param("n1", 10)
    p.set_param("n2", 5)
    p.set_param("n3", 3)
    p.set_param("n4", 3)
    p.set_param("dl", 200)
    p.set_param("cl", [(255, 128, 0), (128, 0, 255)])
    p.select("chase")
    run_for(p, wdt, 3000)

    # Test 4: Fast movement (n3=5, n4=5)
    print("Test 4: Fast movement (n3=5, n4=5)")
    p.set_param("n1", 4)
    p.set_param("n2", 4)
    p.set_param("n3", 5)
    p.set_param("n4", 5)
    p.set_param("dl", 100)
    p.set_param("cl", [(255, 0, 255), (0, 255, 255)])
    p.select("chase")
    run_for(p, wdt, 2000)

    # Test 5: Backward movement (n3=-2, n4=-2)
    print("Test 5: Backward movement (n3=-2, n4=-2)")
    p.set_param("n1", 6)
    p.set_param("n2", 4)
    p.set_param("n3", -2)
    p.set_param("n4", -2)
    p.set_param("dl", 200)
    p.set_param("cl", [(255, 255, 255), (0, 0, 0)])
    p.select("chase")
    run_for(p, wdt, 3000)

    # Test 6: Alternating forward/backward (n3=3, n4=-2)
    print("Test 6: Alternating forward/backward (n3=3, n4=-2)")
    p.set_param("n1", 5)
    p.set_param("n2", 5)
    p.set_param("n3", 3)
    p.set_param("n4", -2)
    p.set_param("dl", 250)
    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])
    p.select("chase")
    run_for(p, wdt, 4000)

    # Cleanup
    print("Test complete, turning off")
    p.select("off")
    run_for(p, wdt, 100)


if __name__ == "__main__":
    main()