from types import SimpleNamespace

import pytest
import typer

from tht.cli import db_cmd
from tht.cli.lsh_cmd import _extract_lsh_values
from tht.mschema.models import Annotations, ColumnPhysical, PhysicalSchema, TablePhysical
from tht.ports.dwh import DistinctValues, DwhHealth
from tht.cli import memory_cmd
from tht.memory import MemoryRecord
from datetime import datetime


def _ping(monkeypatch, health, capsys):
    monkeypatch.setattr(db_cmd, "load_config", lambda path: SimpleNamespace(database=SimpleNamespace(user="u")))
    monkeypatch.setattr(db_cmd, "build_dwh", lambda cfg: SimpleNamespace(health=lambda: health))
    try:
        db_cmd.ping_cmd()
    except typer.Exit as exc:
        code = exc.exit_code
    else:
        code = 0
    return code, capsys.readouterr()


def test_db_ping_public_health_success(monkeypatch, capsys):
    code, output = _ping(monkeypatch, DwhHealth(ok=True, database="d", schema="s", read_only=True), capsys)
    assert code == 0
    assert "OK: connesso a d (schema s)" in output.out


def test_db_ping_rest_inaccessible_historical_wording(monkeypatch, capsys):
    code, output = _ping(monkeypatch, DwhHealth(ok=False, detail="{'db_connected': False}", error_kind="inaccessible"), capsys)
    assert code == 1
    assert "ERRORE: DWH non accessibile via REST (risposta: {'db_connected': False})." in output.err


def test_db_ping_direct_connection_historical_wording(monkeypatch, capsys):
    code, output = _ping(monkeypatch, DwhHealth(ok=False, detail="connection refused", error_kind="connection"), capsys)
    assert code == 1
    assert "ERRORE di connessione: connection refused" in output.err


@pytest.mark.parametrize(("limit", "truncated"), [(7, False), (1201, True)])
def test_lsh_extraction_honors_configured_limit(limit, truncated):
    physical = PhysicalSchema(database="d", schema="s", introspected_at=datetime(2026, 1, 1), tables={
        "t": TablePhysical(columns={"c": ColumnPhysical(type="text", eligible=True)})
    })
    calls = []
    class Dwh:
        def distinct_values(self, table, column, *, limit):
            calls.append(limit)
            return DistinctValues(values=list(range(limit)), truncated=truncated)
    values, _, reports = _extract_lsh_values(Dwh(), physical, Annotations(), limit)
    assert calls == [limit]
    assert len(values["t"]["c"]) == limit
    assert [report.indexed for report in reports] == ([limit] if truncated else [])


def test_memory_command_writes_through_factory_vector_store(monkeypatch):
    store = SimpleNamespace(existing_hashes=lambda *args: {}, upsert=lambda table, rows: 1)
    captured = []
    original_upsert = store.upsert
    store.upsert = lambda table, rows: captured.extend(rows) or original_upsert(table, rows)
    # Server deployments write directly to pgvector and intentionally do not
    # configure the workstation-only REST writer key.
    cfg = SimpleNamespace(profile="server", embeddings=object(), vector_write_rest=None)
    manifest = SimpleNamespace(id="s1")
    snapshot = SimpleNamespace(manifest=manifest, decisions=[], artifacts={})
    record = MemoryRecord(
        id="m1", ts=datetime(2026, 1, 1), session_id="s1",
        decision_seq=7, type="concept_clarified", subject="paziente attivo",
        detail="flag_attivo = TRUE", question_context="q",
    )
    monkeypatch.setattr(memory_cmd, "_load_config_or_exit", lambda path: cfg)
    monkeypatch.setattr(memory_cmd, "load_snapshot_or_exit", lambda cfg, session: snapshot)
    monkeypatch.setattr(memory_cmd, "registry_path", lambda cfg: None)
    monkeypatch.setattr("tht.adapters.factory.build_vector_store", lambda cfg, require_write: store)
    monkeypatch.setattr("tht.cli.vector_cmd.make_embedder",
                        lambda cfg: SimpleNamespace(embed_documents=lambda texts: [[0.1]]))
    monkeypatch.setattr("tht.memory.promote_snapshot", lambda *args, **kwargs: None)
    monkeypatch.setattr("tht.memory.load_registry", lambda path: [record])
    memory_cmd.save_one_cmd(session="s1", decision=7, json_out=True)
    from tht.ports.vector import VectorWriteRecord
    assert len(captured) == 1 and isinstance(captured[0], VectorWriteRecord)


def test_solved_index_writes_through_writer_only_factory_store(monkeypatch):
    writer_only_store = SimpleNamespace(
        capabilities=SimpleNamespace(search=False, upsert=True),
        existing_hashes=lambda *args: {},
        upsert=lambda table, rows: 1,
    )
    cfg = SimpleNamespace(embeddings=object(), vector_write_rest=object())
    manifest = SimpleNamespace(id="s1")
    solved_record = object()
    calls = []

    monkeypatch.setattr(memory_cmd, "has_vector_write_rest", lambda cfg: True)
    monkeypatch.setattr(memory_cmd, "load_snapshot_or_exit", lambda cfg, session: SimpleNamespace(manifest=manifest, decisions=[], artifacts={}))
    monkeypatch.setattr(
        "tht.adapters.factory.build_vector_store",
        lambda cfg, require_write: calls.append(require_write) or writer_only_store,
    )
    monkeypatch.setattr("tht.cli.sql_cmd.promoted_tables_for", lambda *args: [])
    monkeypatch.setattr("tht.solved.build_solved_snapshot", lambda *args: solved_record)
    monkeypatch.setattr(
        "tht.solved.save_solved_question",
        lambda record, *, store, embedder: int(
            record is solved_record and store is writer_only_store
        ),
    )
    monkeypatch.setattr("tht.cli.vector_cmd.make_embedder", lambda cfg: object())

    assert memory_cmd.index_solved_session(cfg, "s1") == 1
    assert calls == [True]