import os
import tempfile
from pathlib import Path
from typing import Callable
from typing import Dict
from unittest.mock import patch

import pytest

from ytdl_sub.config.config_file import ConfigFile
from ytdl_sub.downloaders.downloader import Downloader
from ytdl_sub.entries.variables.kwargs import EPOCH
from ytdl_sub.entries.variables.kwargs import EXT
from ytdl_sub.entries.variables.kwargs import EXTRACTOR
from ytdl_sub.entries.variables.kwargs import PLAYLIST_ENTRY
from ytdl_sub.entries.variables.kwargs import PLAYLIST_INDEX
from ytdl_sub.entries.variables.kwargs import TITLE
from ytdl_sub.entries.variables.kwargs import UID
from ytdl_sub.entries.variables.kwargs import UPLOAD_DATE
from ytdl_sub.entries.variables.kwargs import WEBPAGE_URL


@pytest.fixture
def working_directory() -> str:
    with tempfile.TemporaryDirectory() as temp_dir:
        yield temp_dir


@pytest.fixture
def subscription_name(working_directory) -> str:
    name = "subscription_test"
    os.makedirs(Path(working_directory) / name, exist_ok=True)
    return name


@pytest.fixture
def output_directory() -> str:
    with tempfile.TemporaryDirectory() as temp_dir:
        yield temp_dir


@pytest.fixture
def config(working_directory) -> ConfigFile:
    return ConfigFile(
        name="config",
        value={"configuration": {"working_directory": working_directory}, "presets": {}},
    )

@pytest.fixture
def mock_file_factory(working_directory: str, subscription_name: str):
    def _mock_file_factory(file_name: str):
        with open(Path(working_directory) / subscription_name / file_name, "w"):
            pass

    return _mock_file_factory


@pytest.fixture
def mock_entry_dict_factory(mock_file_factory) -> Callable:
    def _mock_entry_dict_factory(uid: int, upload_date: str, playlist_index: int = 1) -> Dict:
        entry_dict = {
            UID: uid,
            EPOCH: 1596878400,
            PLAYLIST_INDEX: playlist_index,
            EXTRACTOR: "mock-entry-dict",
            TITLE: f"Mock Entry {uid}",
            EXT: "mp4",
            UPLOAD_DATE: upload_date,
            WEBPAGE_URL: f"https://{uid}.com",
            PLAYLIST_ENTRY: {
                "thumbnails": [
                    {
                        "id": "avatar_uncropped",
                        "url": "https://avatar_uncropped.com",
                    },
                    {
                        "id": "banner_uncropped",
                        "url": "https://banner_uncropped.com",
                    },
                ]
            },
        }

        # Create mock video file
        mock_file_factory(file_name=f"{uid}.mp4")
        mock_file_factory(file_name=f"{uid}.jpg")
        return entry_dict

    return _mock_entry_dict_factory


@pytest.fixture
def mock_download_collection_thumbnail(mock_file_factory):
    def _mock_download_thumbnail(output_path: str):
        mock_file_factory(file_name=output_path.split("/")[-1])
        pass

    with patch.object(
        Downloader,
        "_download_thumbnail",
        new=lambda _, thumbnail_url, output_thumbnail_path: _mock_download_thumbnail(
            output_thumbnail_path
        ),
    ):
        yield  # TODO: create file here


@pytest.fixture
def mock_download_collection_entries(
    mock_download_collection_thumbnail, mock_entry_dict_factory: Callable, working_directory: str
):
    collection_1_entry_dicts = [
        mock_entry_dict_factory(uid=1, upload_date="20200808", playlist_index=1),
        mock_entry_dict_factory(uid=2, upload_date="20200808", playlist_index=2),
        mock_entry_dict_factory(uid=3, upload_date="20210808", playlist_index=3),
    ]
    collection_2_entry_dicts = [
        mock_entry_dict_factory(uid=4, upload_date="20200808", playlist_index=1),
        mock_entry_dict_factory(uid=5, upload_date="20200808", playlist_index=2),
        mock_entry_dict_factory(uid=6, upload_date="20210808", playlist_index=3),
    ]

    with patch.object(
        Downloader, "extract_info_via_info_json"
    ) as mock_download_metadata, patch.object(
        Downloader, "_download_entry", new=lambda _, entry: entry
    ):
        # Stub out metadata. TODO: update this if we do metadata plugins
        mock_download_metadata.side_effect = [collection_1_entry_dicts, collection_2_entry_dicts]
        yield