# 02.07.24

import os


# External libraries
import httpx
from bs4 import BeautifulSoup


# Internal utilities
from Src.Util.console import console
from Src.Util.os import os_manager
from Src.Util.message import start_message
from Src.Util.headers import get_headers
from Src.Lib.Downloader import TOR_downloader


# Logic class
from ..Template.Class.SearchType import MediaItem


# Config
from .costant import ROOT_PATH, DOMAIN_NOW, SITE_NAME, MOVIE_FOLDER


def download_title(select_title: MediaItem):
    """
    Downloads a media item and saves it as an MP4 file.

    Parameters:
        - select_title (MediaItem): The media item to be downloaded. This should be an instance of the MediaItem class, containing attributes like `name` and `url`.
    """

    start_message()
    console.print(f"[yellow]Download:  [red]{select_title.name} \n")
    print() 

    # Define output path
    title_name = os_manager.get_sanitize_file(select_title.name)
    mp4_path = os_manager.get_sanitize_path(
        os.path.join(ROOT_PATH, SITE_NAME, MOVIE_FOLDER, title_name.replace(".mp4", ""))
    )
    
    # Create output folder
    os_manager.create_path(mp4_path)                                                                    

    # Make request to page with magnet
    full_site_name = f"{SITE_NAME}.{DOMAIN_NOW}"
    response = httpx.get(
        url="https://" + full_site_name + select_title.url, 
        headers={
            'user-agent': get_headers()
        }, 
        follow_redirects=True
    )

    # Create soup and find table
    soup = BeautifulSoup(response.text, "html.parser")
    final_url = soup.find("a", class_="torrentdown1").get("href")

    # Tor manager
    manager = TOR_downloader()
    manager.add_magnet_link(final_url)
    manager.start_download()
    manager.move_downloaded_files(mp4_path)