"""ChatToolContext -- a chat tool-ok closure-ben kapott állapot-handle-je.

A tool-ok NEM tartanak párhuzamos referenciát a feltöltött dokumentumokra —
mindig a HybridStore-ból + a "documents" listából (in-memory snapshot)
olvasnak. Ez biztosítja a friss-adat működést.

Egyszerűsített Fázis 5 design: a context-ben egy in-memory snapshot van a
ProcessedDocument-ekről + a HybridStore singleton. A Fázis 7-ben (UI) ezt
SqliteSaver-rel váltjuk fel a teljes thread_id alapú perzisztenciára.
"""

from __future__ import annotations

from dataclasses import dataclass, field

from graph.states.pipeline_state import ProcessedDocument
from store import HybridStore


@dataclass
class ChatToolContext:
    """A chat tool-ok osztott állapot-handle-je."""

    store: HybridStore
    """A hibrid store -- a search_documents tool használja."""

    documents: dict[str, ProcessedDocument] = field(default_factory=dict)
    """file_name → ProcessedDocument map. A Streamlit UI a feltöltés után
    populates-eli a pipeline-eredményből."""

    def add_document(self, doc: ProcessedDocument) -> None:
        if doc.ingested:
            self.documents[doc.ingested.file_name] = doc

    def get_document(self, filename: str) -> ProcessedDocument | None:
        return self.documents.get(filename)

    def list_filenames(self) -> list[str]:
        return list(self.documents.keys())