from __future__ import annotations

import argparse
import sys
from pathlib import Path

from rich import print


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

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


from scripts.common import DB_FILE
from scripts.search_utils import search_database


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",
        help=(
            "Vyhľadáva iba v dokumentoch "
            "s published: true."
        ),
    )

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

    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 (
        FileNotFoundError,
        RuntimeError,
    ) as error:
        raise SystemExit(
            str(error)
        ) from error

    print_results(
        query,
        response,
    )


if __name__ == "__main__":
    main()