"""
StorageConsumer: persists events to SQLite + JSONL log.
"""
from __future__ import annotations

import logging
from pathlib import Path

from app.config import settings
from app.domain.events import (
    BookL2Event,
    Event,
    EventType,
    QuoteEvent,
    TradeEvent,
)
from app.db import repo

logger = logging.getLogger(__name__)


class StorageConsumer:
    """
    Writes every event to:
    1. SQLite (via async repo) — structured, queryable.
    2. JSONL file — append-only event log for replay/audit.
    """

    def __init__(self, jsonl_path: str | None = None) -> None:
        self._jsonl_path = Path(jsonl_path or settings.jsonl_path)
        self._jsonl_file = None
        self._count = 0

    async def start(self) -> None:
        """Open JSONL file for appending."""
        self._jsonl_file = open(self._jsonl_path, "a", buffering=1)  # line-buffered
        logger.info("storage.started", extra={"jsonl": str(self._jsonl_path)})

    async def handle(self, event: Event) -> None:
        """Persist one event."""
        # 1. JSONL log (always)
        if self._jsonl_file:
            line = event.model_dump_json()
            self._jsonl_file.write(line + "\n")

        # 2. SQLite (by type)
        if event.event_type == EventType.TRADE:
            assert isinstance(event, TradeEvent)
            await repo.save_trade(event)
        elif event.event_type == EventType.QUOTE:
            assert isinstance(event, QuoteEvent)
            await repo.save_quote(event)
        elif event.event_type == EventType.BOOK_L2:
            assert isinstance(event, BookL2Event)
            await repo.save_book_snapshot(event)
        # Heartbeats → only JSONL, not SQLite

        self._count += 1

    async def stop(self) -> None:
        if self._jsonl_file:
            self._jsonl_file.close()
        logger.info("storage.stopped", extra={"events_written": self._count})