# ie.py - Omni Media Player infomation extractor
# This file is part of comp
#
# comp is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# comp program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with comp.  If not, see <http://www.gnu.org/licenses/>.
#
# Copyright (C) 2017  Nguyễn Gia Phong <vn.mcsinyx@gmail.com>

import json
from os.path import abspath, expanduser, expandvars, isfile
from time import gmtime, sleep, strftime
from urllib.request import urlretrieve

from youtube_dl import YoutubeDL
from mpv import MPV


DEFAULT_ENTRY = {'filename': '', 'title': '', 'duration': '00:00:00',
                 'error': False, 'playing': False, 'selected': False}
JSON_KEYS = 'filename', 'title', 'duration', 'error', 'selected'


class YoutubeDLLogger:
    def debug(self, msg): pass
    def warning(self, msg): pass
    def error(self, msg): pass


def json_extract_info(filename):
    """Return list of entries extracted from a file using json. If an
    error occur during the extraction, return None.
    """
    try:
        if not isfile(filename):
            filename = urlretrieve(filename)[0]
        with open(filename) as f: raw_info, info = json.load(f), []
        for i in raw_info:
            e = DEFAULT_ENTRY.copy()
            for k in e:
                if k in i and isinstance(i[k], type(e[k])): e[k] = i[k]
            info.append(e)
    except:
        return None
    else:
        return info


def mpv_extract_info(filename):
    """Return list of entries extracted from a path or URL using mpv. If
    an error occur during the extraction, return None.
    """
    mp = MPV(ytdl=True, vid=False)
    mp.play(filename)
    while mp.duration is None:
        sleep(0.25)
        if mp.playback_abort: return None
    info = {'filename': filename, 'title': mp.media_title.decode(),
            'duration': mp.osd.duration, 'error': False, 'playing': False,
            'selected': False}
    mp.quit()
    return [info]


def ytdl_extract_info(filename):
    """Return list of entries extracted from a path or URL using
    youtube-dl. If an error occur during the extraction, return None.
    """
    ytdl_opts = {'logger': YoutubeDLLogger(), 'default_search': 'ytsearch',
                 'extract_flat': 'in_playlist'}
    with YoutubeDL(ytdl_opts) as ytdl:
        try:
            raw_info = ytdl.extract_info(filename, download=False)
        except:
            return None
        info = raw_info.get('entries', [raw_info])
        for i in info:
            if 'webpage_url' in i:
                i['filename'] = i['webpage_url']
            elif (i.get('ie_key') == 'Youtube'
                  or i.get('extractor') == 'youtube'):
                i['filename'] = 'https://youtu.be/' + i['id']
            else:
                i['filename'] = i['url']
            if 'title' not in i:
                try:
                    i['title'] = ytdl.extract_info(i['filename'],
                                                   download=False)['title']
                except:
                    return None
            if 'duration' not in i:
                i['duration'] = '00:00:00'
            elif isinstance(i['duration'], int):
                i['duration'] = strftime('%H:%M:%S', gmtime(i['duration']))
            for k in 'error', 'playing', 'selected': i.setdefault(k, False)
            for k in i.copy():
                if k not in DEFAULT_ENTRY: i.pop(k)
    return info


def extract_info(filename, extractor='youtube-dl'):
    """Return list of entries extracted from a path or URL using
    specified extractor. If an error occur during the extraction,
    return None.

    The extractor could be either 'json', 'mpv' or 'youtube-dl' and
    fallback to 'youtube-dl'.
    """
    if isfile(expanduser(expandvars(filename))):
        filename = abspath(expanduser(expandvars(filename)))
    if extractor == 'json':
        return json_extract_info(filename)
    elif extractor == 'mpv':
        return mpv_extract_info(filename)
    else:
        return ytdl_extract_info(filename)