#!/usr/bin/env python3

import logging
import os
import pathlib
import sys

import youtube_dl

from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
from bulkredditdownloader.errors import FileAlreadyExistsError
from bulkredditdownloader.utils import GLOBAL

logger = logging.getLogger(__name__)


class Youtube(BaseDownloader):
    def __init__(self, directory: pathlib.Path, post: dict):
        super().__init__(directory, post)
        self.download()

    def download(self):
        self.directory.mkdir(exist_ok=True)

        filename = GLOBAL.config['filename'].format(**self.post)
        logger.info(filename)

        self._download_video(filename, self.directory, self.post['CONTENTURL'])

    def _download_video(self, filename: str, directory: pathlib.Path, url: str):
        ydl_opts = {
            "format": "best",
            "outtmpl": str(directory / (filename + ".%(ext)s")),
            "progress_hooks": [self._hook],
            "playlistend": 1,
            "nooverwrites": True,
            "quiet": True
        }
        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            ydl.download([url])

        location = directory / (filename + ".mp4")

        with open(location, 'rb') as file:
            content = file.read()

        if GLOBAL.arguments.no_dupes:
            try:
                file_hash = self._create_hash(content)
            except FileNotFoundError:
                return None
            if file_hash in GLOBAL.downloadedPosts():
                os.remove(location)
                raise FileAlreadyExistsError
            GLOBAL.downloadedPosts.add(file_hash)

    @staticmethod
    def _hook(d):
        if d['status'] == 'finished':
            return logger.info("Downloaded")
        downloaded_mbs = int(d['downloaded_bytes'] * (10**(-6)))
        file_size = int(d['total_bytes'] * (10**(-6)))
        sys.stdout.write("{}Mb/{}Mb\r".format(downloaded_mbs, file_size))
        sys.stdout.flush()