# 03.07.24

import sys
import logging


# External libraries
import httpx
from bs4 import BeautifulSoup
from unidecode import unidecode


# Internal utilities
from Src.Util.headers import get_headers
from Src.Util.table import TVShowManager
from ..Template import search_domain, get_select_title


# Logic class
from .Core.Class.SearchType import MediaManager


# Variable
from .costant import SITE_NAME
media_search_manager = MediaManager()
table_show_manager = TVShowManager()



def title_search(word_to_search: str) -> int:
    """
    Search for titles based on a search query.

    Args:
        - title_search (str): The title to search for.

    Returns:
        int: The number of titles found.
    """
    try:

        # Find new domain if prev dont work
        domain_to_use, _ = search_domain(SITE_NAME, '<meta property="og:site_name" content="CB01">', f"https://{SITE_NAME}")

        # Send request to search for titles
        response = httpx.get(f"https://{SITE_NAME}.{domain_to_use}/?s={unidecode(word_to_search)}", headers={'user-agent': get_headers()}, follow_redirects=True)
        response.raise_for_status()

        # Create soup and find table
        soup = BeautifulSoup(response.text, "html.parser")

        for div_title in soup.find_all("div", class_ = "card"):

            url = div_title.find("h3").find("a").get("href")
            title = div_title.find("h3").find("a").get_text(strip=True)
            desc = div_title.find("span").find("strong").get_text(strip=True)

            title_info = {
                'url': url,
                'name': title,
                'desc': desc
            }

            media_search_manager.add_media(title_info)

    except Exception as err:
        logging.error(f"An error occurred: {err}")

    # Return the number of titles found
    return media_search_manager.get_length()


def run_get_select_title():
    """
    Display a selection of titles and prompt the user to choose one.
    """
    return get_select_title(table_show_manager, media_search_manager)