"""Tests for the Twitch component."""

from collections.abc import AsyncIterator
from typing import Any, Generic, TypeVar

from twitchAPI.object.base import TwitchObject
from typing_extensions import AsyncGenerator

from homeassistant.components.twitch import DOMAIN
from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry, load_json_array_fixture


async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
    """Fixture for setting up the component."""
    config_entry.add_to_hass(hass)

    assert await hass.config_entries.async_setup(config_entry.entry_id)
    await hass.async_block_till_done()


TwitchType = TypeVar("TwitchType", bound=TwitchObject)


class TwitchIterObject(Generic[TwitchType]):
    """Twitch object iterator."""

    def __init__(self, fixture: str, target_type: type[TwitchType]) -> None:
        """Initialize object."""
        self.raw_data = load_json_array_fixture(fixture, DOMAIN)
        self.data = [target_type(**item) for item in self.raw_data]
        self.total = len(self.raw_data)
        self.target_type = target_type

    async def __aiter__(self) -> AsyncIterator[TwitchType]:
        """Return async iterator."""
        async for item in get_generator_from_data(self.raw_data, self.target_type):
            yield item


async def get_generator(
    fixture: str, target_type: type[TwitchType]
) -> AsyncGenerator[TwitchType]:
    """Return async generator."""
    data = load_json_array_fixture(fixture, DOMAIN)
    async for item in get_generator_from_data(data, target_type):
        yield item


async def get_generator_from_data(
    items: list[dict[str, Any]], target_type: type[TwitchType]
) -> AsyncGenerator[TwitchType]:
    """Return async generator."""
    for item in items:
        yield target_type(**item)