"""Compatibility pattern helpers for NeoPixel demos.

This file mirrors `workspace/code/led_patterns.py` so imports like
`from led_patterns import ...` work even in older worker sessions that only
include `/workspace/lib` in `sys.path`.
"""

from __future__ import annotations

import random


Color = tuple[int, int, int]


def _clamp(channel: int) -> int:
    return max(0, min(255, int(channel)))


def wheel(pos: int) -> Color:
    pos = 255 - (pos & 255)
    if pos < 85:
        return (_clamp(255 - pos * 3), 0, _clamp(pos * 3))
    if pos < 170:
        pos -= 85
        return (0, _clamp(pos * 3), _clamp(255 - pos * 3))
    pos -= 170
    return (_clamp(pos * 3), _clamp(255 - pos * 3), 0)


def rainbow_frame(led_count: int, frame: int, step: int = 4) -> list[Color]:
    if led_count <= 0:
        return []
    return [wheel((i * 256 // led_count + frame * step) & 255) for i in range(led_count)]


def chase_frame(
    led_count: int,
    frame: int,
    color: Color = (255, 120, 0),
    tail: Color = (16, 0, 0),
) -> list[Color]:
    if led_count <= 0:
        return []
    out: list[Color] = [(0, 0, 0) for _ in range(led_count)]
    head = frame % led_count
    trail = (head - 1) % led_count
    out[trail] = tuple(_clamp(v) for v in tail)  # type: ignore[assignment]
    out[head] = tuple(_clamp(v) for v in color)  # type: ignore[assignment]
    return out


def twinkle_frame(
    led_count: int,
    frame: int,
    base: Color = (0, 0, 8),
    sparkle: Color = (255, 255, 180),
    sparkles: int = 3,
    seed: int = 1337,
) -> list[Color]:
    if led_count <= 0:
        return []
    out: list[Color] = [tuple(_clamp(v) for v in base) for _ in range(led_count)]  # type: ignore[list-item]
    rng = random.Random(seed + frame)
    for _ in range(min(max(0, sparkles), led_count)):
        idx = rng.randrange(led_count)
        out[idx] = tuple(_clamp(v) for v in sparkle)  # type: ignore[assignment]
    return out