from __future__ import annotations

import asyncio
from typing import TYPE_CHECKING

from yuxi.utils.logging_config import logger

if TYPE_CHECKING:
    from yuxi.channels.services.runtime_state import RuntimeState


class StatsCollector:
    def __init__(self, state: RuntimeState):
        self._state = state

    async def run(self, interval: float = 60) -> None:
        logger.info("StatsCollector started")
        while True:
            await asyncio.sleep(interval)
            try:
                self._collect()
            except asyncio.CancelledError:
                break
            except Exception:
                logger.exception("StatsCollector error")

    def _collect(self) -> None:
        logger.debug(
            f"Stats: channels={self._state.active_channels}, "
            f"requests={self._state.request_count}, errors={self._state.error_count}"
        )