from __future__ import unicode_literals

from .common import InfoExtractor
from ..compat import compat_urlparse
from ..utils import (
    int_or_none,
    js_to_json,
    remove_end,
    unified_strdate,
)


class VidbitIE(InfoExtractor):
    _VALID_URL = (
        r"https?://(?:www\.)?vidbit\.co/(?:watch|embed)\?.*?\bv=(?P<id>[\da-zA-Z]+)"
    )
    _TESTS = [
        {
            "url": "http://www.vidbit.co/watch?v=jkL2yDOEq2",
            "md5": "1a34b7f14defe3b8fafca9796892924d",
            "info_dict": {
                "id": "jkL2yDOEq2",
                "ext": "mp4",
                "title": "Intro to VidBit",
                "description": "md5:5e0d6142eec00b766cbf114bfd3d16b7",
                "thumbnail": r"re:https?://.*\.jpg$",
                "upload_date": "20160618",
                "view_count": int,
                "comment_count": int,
            },
        },
        {
            "url": "http://www.vidbit.co/embed?v=jkL2yDOEq2&auto=0&water=0",
            "only_matching": True,
        },
    ]

    def _real_extract(self, url):
        video_id = self._match_id(url)

        webpage = self._download_webpage(
            compat_urlparse.urljoin(url, "/watch?v=%s" % video_id), video_id
        )

        video_url, title = [None] * 2

        config = self._parse_json(
            self._search_regex(
                r"(?s)\.setup\(({.+?})\);", webpage, "setup", default="{}"
            ),
            video_id,
            transform_source=js_to_json,
        )
        if config:
            if config.get("file"):
                video_url = compat_urlparse.urljoin(url, config["file"])
            title = config.get("title")

        if not video_url:
            video_url = compat_urlparse.urljoin(
                url,
                self._search_regex(
                    r'file\s*:\s*(["\'])(?P<url>(?:(?!\1).)+)\1',
                    webpage,
                    "video URL",
                    group="url",
                ),
            )

        if not title:
            title = remove_end(
                self._html_search_regex(
                    (r"<h1>(.+?)</h1>", r"<title>(.+?)</title>"),
                    webpage,
                    "title",
                    default=None,
                )
                or self._og_search_title(webpage),
                " - VidBit",
            )

        description = self._html_search_meta(
            ("description", "og:description", "twitter:description"),
            webpage,
            "description",
        )

        upload_date = unified_strdate(
            self._html_search_meta("datePublished", webpage, "upload date")
        )

        view_count = int_or_none(
            self._search_regex(
                r"<strong>(\d+)</strong> views", webpage, "view count", fatal=False
            )
        )
        comment_count = int_or_none(
            self._search_regex(
                r'id=["\']cmt_num["\'][^>]*>\((\d+)\)',
                webpage,
                "comment count",
                fatal=False,
            )
        )

        return {
            "id": video_id,
            "url": video_url,
            "title": title,
            "description": description,
            "thumbnail": self._og_search_thumbnail(webpage),
            "upload_date": upload_date,
            "view_count": view_count,
            "comment_count": comment_count,
        }
