import threading
from datetime import datetime

from . import LOGGER
from .bilibiliupload import core, VideoPart


IS_LIVING = threading.Event()
IS_ENCODING = threading.Event()
IS_UPLOADING = threading.Event()


class Bilibili:
    def __init__(self):
        self.access_token = ""
        self.session_id = ""
        self.user_id = ""
        self.expires = 0
        self.login_time = None
        self.parts = []

    def login(self):
        with open("access_token", "r") as f:
            self.access_token = f.read(64).strip()
        self.session_id, self.user_id, self.expires = core.login_by_access_token(self.access_token)
        self.login_time = datetime.now()
        LOGGER.info("B站登录，UID【{}】，过期时间【{}】".format(self.user_id, self.expires))

    def upload(self,
               parts,
               title,
               tid,
               tag,
               desc,
               source='',
               cover='',
               no_reprint=1,
               ):
        """

        :param parts: e.g. VideoPart('part path', 'part title', 'part desc'), or [VideoPart(...), VideoPart(...)]
        :type parts: VideoPart or list<VideoPart>
        :param title: video's title
        :type title: str
        :param tid: video type, see: https://member.bilibili.com/x/web/archive/pre
                                  or https://github.com/uupers/BiliSpider/wiki/%E8%A7%86%E9%A2%91%E5%88%86%E5%8C%BA%E5%AF%B9%E5%BA%94%E8%A1%A8
        :type tid: int
        :param tag: video's tag
        :type tag: list<str>
        :param desc: video's description
        :type desc: str
        :param source: (optional) 转载地址
        :type source: str
        :param cover: (optional) cover's URL, use method *cover_up* to get
        :type cover: str
        :param no_reprint: (optional) 0=可以转载, 1=禁止转载(default)
        :type no_reprint: int
        """
        self.pre_upload(parts)
        self.finish_upload(title, tid, tag, desc, source, cover, no_reprint)
        self.clear()

    def pre_upload(self, parts: "VideoPart", max_retry=5):
        """
        :param max_retry:
        :param parts: e.g. VideoPart('part path', 'part title', 'part desc'), or [VideoPart(...), VideoPart(...)]
        :type parts: VideoPart or list<VideoPart>
        """
        if not isinstance(parts, list):
            parts = [parts]

        IS_UPLOADING.set()
        for part in parts:
            if isinstance(part, str):
                part = VideoPart(part)
            LOGGER.info("Start Uploading >{}<".format(part.path))
            status = core.upload_video_part(self.access_token, self.session_id, self.user_id, part, max_retry)
            if status:
                # 上传完毕
                LOGGER.info("Upload >{}< Finished；【{}】".format(part.path, part.server_file_name))
                self.parts.append(part)
            else:
                LOGGER.warn("Upload >{}< Failed".format(part.path))
        IS_UPLOADING.clear()

    def finish_upload(self,
                     title,
                     tid,
                     tag,
                     desc,
                     source='',
                     cover='',
                     no_reprint=1,
                     ):
        """
        :param title: video's title
        :type title: str
        :param tid: video type, see: https://member.bilibili.com/x/web/archive/pre
                                  or https://github.com/uupers/BiliSpider/wiki/%E8%A7%86%E9%A2%91%E5%88%86%E5%8C%BA%E5%AF%B9%E5%BA%94%E8%A1%A8
        :type tid: int
        :param tag: video's tag
        :type tag: str
        :param desc: video's description
        :type desc: str
        :param source: (optional) 转载地址
        :type source: str
        :param cover: (optional) cover's URL, use method *cover_up* to get
        :type cover: str
        :param no_reprint: (optional) 0=可以转载, 1=禁止转载(default)
        :type no_reprint: int
        :param copyright: (optional) 0=转载的, 1=自制的(default)
        :type copyright: int
        """
        if len(self.parts) == 0:
            return
        if IS_ENCODING.is_set():
            LOGGER.info("[{}]仍在压制，取消发布".format(title))
            return
        if IS_LIVING.is_set():
            LOGGER.info("[{}]仍在直播，取消发布".format(title))
            return
        if IS_UPLOADING.is_set():
            LOGGER.info("[{}]仍在上传，取消发布".format(title))
            return
        LOGGER.info("[{}]投稿中，请稍后".format(title))
        copyright = 2 if source else 1
        try:
            avid, bvid = core.upload(self.access_token, self.session_id, self.user_id, self.parts, copyright,
                        title=title, tid=tid, tag=tag, desc=desc, source=source, cover=cover, no_reprint=no_reprint)
            LOGGER.info("[{}]投稿成功；AVID【{}】，BVID【{}】".format(title, avid, bvid))
            self.clear()
        except Exception as e:
            LOGGER.error("[{}]投稿失败".format(title), exc_info=e)


    def reloadFromPrevious(self):
        ...

    def clear(self):
        self.parts = []


INSTANCE = Bilibili()