import logging

from .DownloadQueue import DownloadQueue
from .encoder import Encoder

LOG = logging.getLogger("common")


class common:
    """
    This class is used to share common methods between the socket and the API gateways.
    """

    queue: DownloadQueue
    encoder: Encoder

    def __init__(self, queue: DownloadQueue, encoder: Encoder):
        super().__init__()
        self.queue = queue
        self.encoder = encoder

    async def add(
        self, url: str, preset: str, folder: str, ytdlp_cookies: str, ytdlp_config: dict, output_template: str
    ) -> dict[str, str]:
        if ytdlp_config is None:
            ytdlp_config = {}

        if ytdlp_cookies and isinstance(ytdlp_cookies, dict):
            ytdlp_cookies = self.encoder.encode(ytdlp_cookies)

        status = await self.queue.add(
            url=url,
            preset=preset if preset else "default",
            folder=folder,
            ytdlp_cookies=ytdlp_cookies,
            ytdlp_config=ytdlp_config,
            output_template=output_template,
        )

        return status