from __future__ import annotations

import tempfile
from pathlib import Path
from typing import TYPE_CHECKING

from reader import Feed
from reader import Reader
from reader import make_reader

from discord_rss_bot.search import create_search_context

if TYPE_CHECKING:
    from collections.abc import Iterable


def test_create_search_context() -> None:
    """Test create_search_context."""
    # Create a reader.
    with tempfile.TemporaryDirectory() as temp_dir:
        # Create the temp directory.
        Path.mkdir(Path(temp_dir), exist_ok=True)
        assert Path.exists(Path(temp_dir)), f"The directory '{temp_dir}' should exist."

        # Create a temporary reader.
        reader: Reader = make_reader(url=str(Path(temp_dir, "test_db.sqlite")))
        assert reader is not None, "The reader should not be None."

        # Add a feed to the reader.
        reader.add_feed("https://lovinator.space/rss_test.xml", exist_ok=True)

        # Check that the feed was added.
        feeds: Iterable[Feed] = reader.get_feeds()
        assert feeds is not None, f"The feeds should not be None. Got: {feeds}"
        assert len(list(feeds)) == 1, f"The number of feeds should be 1. Got: {len(list(feeds))}"

        # Update the feed to get the entries.
        reader.update_feeds()

        # Get the feed.
        feed: Feed = reader.get_feed("https://lovinator.space/rss_test.xml")
        assert feed is not None, f"The feed should not be None. Got: {feed}"

        # Update the search index.
        reader.enable_search()
        reader.update_search()

        # Create the search context.
        context: dict = create_search_context("test", reader=reader)
        assert context is not None, f"The context should not be None. Got: {context}"

        # Close the reader, so we can delete the directory.
        reader.close()