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


def run_for(p, wdt, ms):
    start = utime.ticks_ms()
    while utime.ticks_diff(utime.ticks_ms(), start) < ms:
        wdt.feed()
        run_tick(p)
        utime.sleep_ms(10)


def main():
    s = Settings()
    p = Presets(pin=s.get("led_pin", 10), num_leds=s.get("num_leds", 30))
    wdt = WDT(timeout=10000)

    p.edit("test_comet_dual", {
        "p": "comet_dual",
        "b": 200,
        "d": 60,
        "c": [(255, 0, 0), (0, 0, 255), (0, 255, 0)],
        "n1": 4,
        "n2": 2,
        "n3": 120,
        "a": True,
    })
    p.select("test_comet_dual")
    run_for(p, wdt, 3000)

    p.edit("cleanup_off", {"p": "off"})
    p.select("cleanup_off")
    run_for(p, wdt, 100)


if __name__ == "__main__":
    main()