from __future__ import annotations

import argparse
import json
import sys
from pathlib import Path

from rich import print

PROJECT_ROOT = Path(__file__).resolve().parents[2]

if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_ROOT))


from tools.rag.common import DB_FILE
from tools.rag.search_utils import search_database
from tools.retrieval.retrieval_contract import error_result


def print_results(
    query: str,
    response: dict,
) -> None:
    results = response["results"]

    print(f"[bold]Dopyt:[/bold] " f"{query}")

    print(f"[bold]Vyhľadávač:[/bold] " f"{response['engine']}")

    print(f"[bold]Stratégie:[/bold] " f"{', '.join(response['strategies']) or 'žiadna'}")

    print(f"[bold]Počet výsledkov:[/bold] " f"{len(results)}")

    print("\n[bold]Top výsledky:[/bold]\n")

    for rank, item in enumerate(
        results,
        start=1,
    ):
        print(
            f"[cyan]{rank}. " f"Skóre: {item['score']} " f"(BM25: {item['bm25_score']})" f"[/cyan]"
        )

        print(f"[bold]Názov:[/bold] " f"{item['title']}")

        print(f"[bold]Cesta:[/bold] " f"{item['document_path']}")

        print(f"[bold]URL:[/bold] " f"{item['source_url']}")

        print(f"[bold]Chunk:[/bold] " f"{item['chunk_index']}")

        print(f"[bold]Zhoda:[/bold] " f"{item['match_strategy']}")

        print(f"[bold]Kategórie:[/bold] " f"{item['categories']}")

        print(f"[bold]Tagy:[/bold] " f"{item['tags']}")

        print(f"[bold]Autor:[/bold] " f"{item['author']}")

        print(f"[bold]Ukážka:[/bold] " f"{item['snippet']}")

        print("[bold]Text:[/bold]")

        print((item["text"] or "")[:700])

        print("-" * 80)


def main() -> None:
    parser = argparse.ArgumentParser(description=("FTS5 vyhľadávanie " "v SQLite indexe zpwiki."))

    parser.add_argument(
        "query",
        nargs="+",
        help="Text, ktorý sa má vyhľadať.",
    )

    parser.add_argument(
        "--limit",
        type=int,
        default=10,
        help="Počet výsledkov.",
    )

    parser.add_argument(
        "--published-only",
        action="store_true",
        default=True,
        help=("Vylúči explicitné published=false (predvolené)."),
    )

    parser.add_argument(
        "--max-per-document",
        type=int,
        default=3,
        help=("Maximálny počet chunkov " "z jedného dokumentu. " "Hodnota 0 vypne limit."),
    )

    parser.add_argument(
        "--include-unpublished",
        action="store_false",
        dest="published_only",
        help="Výslovne zahrnie aj nepublikované dokumenty.",
    )
    parser.add_argument(
        "--json", action="store_true", help="Strojový výstup so spoločným kontraktom."
    )
    args = parser.parse_args()
    query = " ".join(args.query)

    try:
        response = search_database(
            DB_FILE,
            query,
            args.limit,
            published_only=args.published_only,
            max_per_document=args.max_per_document,
        )

    except Exception as error:
        if args.json:
            sys.stdout.write(json.dumps(error_result(error), ensure_ascii=False, indent=2) + "\n")
            raise SystemExit(1) from error
        raise SystemExit(str(error)) from error

    if args.json:
        sys.stdout.write(json.dumps(response, ensure_ascii=False, indent=2) + "\n")
        return

    print_results(
        query,
        response,
    )


if __name__ == "__main__":
    main()