from __future__ import annotations

import re
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[3]
BACKEND = ROOT / "backend"
sys.path.insert(0, str(BACKEND))

from app.persistence.job_queue_sql import (  # noqa: E402
    CLAIM_JOB_SQL,
    COMPLETE_JOB_SQL,
    FAIL_OR_RETRY_JOB_SQL,
    HEARTBEAT_JOB_SQL,
    REAP_EXPIRED_JOBS_SQL,
)

COMPOSE = (ROOT / "compose.yaml").read_text(encoding="utf-8")
BOOTSTRAP = (ROOT / "ops/postgres/init/10-bootstrap-rag.sh").read_text(encoding="utf-8")
MIGRATION = (BACKEND / "migrations/versions/0001_initial_schema.py").read_text(encoding="utf-8")


def _service_block(name: str) -> str:
    pattern = re.compile(rf"(?ms)^  {re.escape(name)}:\n(.*?)(?=^  [a-zA-Z0-9_-]+:\n|\Z)")
    match = pattern.search(COMPOSE)
    assert match is not None, f"missing Compose service: {name}"
    return match.group(1)


def test_compose_isolates_database_credentials_and_networks() -> None:
    db = _service_block("db")
    migrate = _service_block("migrate")
    provider_smoke = _service_block("provider-smoke")
    seed_demo = _service_block("seed-demo")
    seed_demo_offline = _service_block("seed-demo-offline")

    assert "postgres_bootstrap_password" in db
    assert "postgres_migrator_password" in db
    assert "postgres_app_password" in db

    assert "postgres_migrator_password" in migrate
    assert "postgres_bootstrap_password" not in migrate
    assert "postgres_app_password" not in migrate

    assert "bailian_api_key" in provider_smoke
    assert "postgres_" not in provider_smoke

    assert "postgres_app_password" in seed_demo
    assert "postgres_bootstrap_password" not in seed_demo
    assert "postgres_migrator_password" not in seed_demo
    assert "bailian_api_key" in seed_demo
    assert "./data/samples/public:/demo:ro" in seed_demo

    assert "postgres_app_password" in seed_demo_offline
    assert "bailian_api_key" not in seed_demo_offline
    assert "DEMO_PROVIDER_MODE: fake" in seed_demo_offline
    assert "./data/samples/public:/demo:ro" in seed_demo_offline

    assert re.search(r"(?ms)^  data:\n.*?^    internal: true$", COMPOSE)
    assert re.search(r"(?m)^  edge:$", COMPOSE)
    assert re.search(r"(?m)^  egress:$", COMPOSE)


def test_database_health_requires_tcp_and_atomic_bootstrap_sentinel() -> None:
    db = _service_block("db")
    assert "pg_isready -h 127.0.0.1 -p 5432" in db
    assert ".rag-bootstrap-complete" in db

    assert "CREATE EXTENSION IF NOT EXISTS vector" in BOOTSTRAP
    assert 'CREATE ROLE :"migrator_user"' in BOOTSTRAP
    assert 'CREATE ROLE :"app_user"' in BOOTSTRAP
    assert "NOSUPERUSER" in BOOTSTRAP
    assert "CREATE SCHEMA rag AUTHORIZATION" in BOOTSTRAP
    assert "ALTER DEFAULT PRIVILEGES" in BOOTSTRAP
    assert BOOTSTRAP.index("COMMIT;") < BOOTSTRAP.index("mv -f")
    assert "\\getenv migrator_password RAG_MIGRATOR_PASSWORD" in BOOTSTRAP
    assert "--set=migrator_password" not in BOOTSTRAP
    assert "--set=app_password" not in BOOTSTRAP
    assert "set -x" not in BOOTSTRAP
    assert "official postgres entrypoint resolves POSTGRES_PASSWORD_FILE" in BOOTSTRAP
    assert 'bootstrap_password="${POSTGRES_PASSWORD:-}"' in BOOTSTRAP


def test_initial_migration_has_vector_and_activation_contracts() -> None:
    normalized = " ".join(MIGRATION.lower().split())

    assert "embedding vector(1024)" in normalized
    assert "using hnsw (embedding vector_cosine_ops)" in normalized
    assert "where searchable" in normalized
    assert "chunks_searchable_requires_ready_approved_embedding" in normalized
    assert "check (embedding_dimension = 1024)" in normalized
    assert "active_version_id uuid" in normalized
    assert "documents_active_version_fk" in normalized
    assert "deferrable initially deferred" in normalized
    assert "foreign key (id, active_version_id)" in normalized
    assert "references rag.document_versions (document_id, id)" in normalized
    assert "local_parsed_pending_cloud_review" in normalized
    assert "cloud_approved" in normalized
    assert "outbound_manifest_sha256" in normalized
    assert "cloud_processing_allowed" not in normalized
    assert "document_versions_cloud_approval_bound" in normalized
    assert "chunks_approval_binding_fk" in normalized
    assert "create table rag.outbound_manifest_items" in normalized
    assert "chunks_manifest_item_binding_fk" in normalized
    assert "outbound_manifest_items_immutable_after_approval" in normalized
    assert "chunks_guard_approved_mutation" in normalized
    assert "chunks_guard_ready_vector_update" in normalized
    assert "new.access_scope_id is distinct from old.access_scope_id" in normalized
    assert "new.document_version_id is distinct from old.document_version_id" in normalized
    assert "cloud_text text not null" in normalized
    assert "cloud_text_sha256 char(64) not null" in normalized
    assert "embedding_text = embedding_prefix || cloud_text" in normalized
    assert "sha256(convert_to(cloud_text, 'utf8'))" in normalized
    assert "sha256(convert_to(embedding_text, 'utf8'))" in normalized
    assert "approval_status = 'cloud_approved'" in normalized
    assert "document_versions_revoke_cloud_approval" in normalized
    assert "new.review_state is distinct from old.review_state" in normalized
    assert "set searchable = false" in normalized
    assert "approval_status = revoked_state" in normalized
    assert "embedding = null" in normalized
    assert "embedding_profile_hash = null" in normalized
    assert "where document_version_id = old.id" in normalized
    assert "chunks_enforce_active_search_projection" in normalized
    assert "document.active_version_id = new.document_version_id" in normalized
    assert "version.status = 'ready'" in normalized
    assert "documents_enforce_activation" in normalized


def test_downgrade_drops_manifest_items_before_document_versions() -> None:
    normalized = " ".join(MIGRATION.lower().split())

    manifest_drop = normalized.index("drop table if exists rag.outbound_manifest_items")
    version_drop = normalized.index("drop table if exists rag.document_versions")
    assert manifest_drop < version_drop


def test_initial_migration_has_fenced_job_schema_and_indexes() -> None:
    normalized = " ".join(MIGRATION.lower().split())

    assert "create table rag.background_jobs" in normalized
    assert "lease_owner text" in normalized
    assert "lease_token uuid" in normalized
    assert "lease_until timestamptz" in normalized
    assert "attempt <= max_attempts" in normalized
    assert "background_jobs_lease_consistent" in normalized
    assert "unique (job_type, idempotency_key)" in normalized
    assert "background_jobs_claim_queued" in normalized
    assert "background_jobs_reap_expired" in normalized


def test_job_claim_and_terminal_updates_are_fenced() -> None:
    claim = " ".join(CLAIM_JOB_SQL.lower().split())
    heartbeat = " ".join(HEARTBEAT_JOB_SQL.lower().split())
    complete = " ".join(COMPLETE_JOB_SQL.lower().split())
    failure = " ".join(FAIL_OR_RETRY_JOB_SQL.lower().split())

    assert "for update skip locked" in claim
    assert "lease_token = gen_random_uuid()" in claim
    assert "attempt = job.attempt + 1" in claim

    for statement in (heartbeat, complete, failure):
        assert "job.status = 'running'" in statement
        assert "job.lease_owner = :worker_id" in statement
        assert "job.lease_token = :lease_token" in statement
        assert "returning" in statement

    assert "when job.attempt < job.max_attempts then 'queued'" in failure
    assert "else 'failed'" in failure


def test_reaper_uses_advisory_lock_and_handles_exhausted_attempts() -> None:
    reaper = " ".join(REAP_EXPIRED_JOBS_SQL.lower().split())

    assert "pg_try_advisory_xact_lock" in reaper
    assert "job.status = 'running'" in reaper
    assert "job.lease_until < now()" in reaper
    assert "for update of job skip locked" in reaper
    assert "when job.attempt < job.max_attempts then 'queued'" in reaper
    assert "else 'failed'" in reaper
    assert "lease_owner = null" in reaper
    assert "lease_token = null" in reaper
    assert "lease_until = null" in reaper