# 29.03.24

import sys
import logging
from urllib.parse import urlparse, urljoin


class M3U8_UrlFix:
    def __init__(self) -> None:

        # Initialize the class with an empty playlist URL
        self.url_playlist: str = ""

    def set_playlist(self, url: str) -> None:
        """
        Set the M3U8 playlist URL.

        Parameters:
        - url (str): The M3U8 playlist URL.
        """
        self.url_playlist = url

    def generate_full_url(self, url_resource: str) -> str:
        """
        Generate a full URL for a given resource using the base URL from the playlist.

        Parameters:
        - url_resource (str): The relative URL of the resource within the playlist.

        Returns:
        - str: The full URL for the specified resource.
        """

        # Check if m3u8 url playlist is present
        if self.url_playlist == None:
            logging.error("[M3U8_UrlFix] Can't generate full url, playlist not present")
            sys.exit(0)

        # Parse the playlist URL to extract the base URL components
        parsed_playlist_url = urlparse(self.url_playlist)
        
        # Construct the base URL using the scheme, netloc, and path from the playlist URL
        base_url = f"{parsed_playlist_url.scheme}://{parsed_playlist_url.netloc}{parsed_playlist_url.path}"

        # Join the base URL with the relative resource URL to get the full URL
        full_url = urljoin(base_url, url_resource)

        return full_url