# -*- coding: utf-8 -*-
# author:　cheneyjin@outlook.com
# update 20220212
import traceback
import _thread
import json
import logging
import os
import re
import sys
import uuid
import time
from subprocess import Popen, PIPE

from PyQt5 import QtCore, QtWidgets, QtGui
from PyQt5.QtGui import QIcon, QTextCursor
from PyQt5.QtWidgets import QMainWindow, QApplication
import cgitb

# import qdarkstyle

style_global = """
        *{
            font-family: "Microsoft YaHei";
        }
    """

path_app = os.path.split(os.path.abspath(sys.argv[0]))[0]

TAG_DUPLICATE = "名字重复!"


def validate(file_name_str):
    r_str = r"[\/\\\:\*\?\"\<\>\|]"
    new_title = re.sub(r_str, "_", file_name_str)
    return new_title


class Target(object):

    def __init__(self, bvid, name, video_path, audio_path, root) -> None:
        super().__init__()
        self.bvid = bvid
        self.name = name
        self.video_path = video_path
        self.audio_path = audio_path
        self.root = root

    def __str__(self) -> str:
        return "name:{}, video_path:{}, audio_path:{}, root:{}".format(
            self.bvid,
            self.name,
            self.video_path,
            self.audio_path,
            self.root
        )


class UiMainWindow(object):

    def __init__(self, logger) -> None:
        super().__init__()
        self.logger = logger
        self.task_id = None
        self.task_list = None
        self.directory = None
        self.target_video_name = "video.m4s"
        self.target_audio_name = "audio.m4s"
        self.target_file_info = "entry.json"

        self.merging = False
        self.searching = False
        self.font = QtGui.QFont()
        self.font.setPointSize(10)
        self.translate = QtCore.QCoreApplication.translate

        self.path_output = "{}/{}".format(path_app, self.translate("output", "输出目录"))
        if not os.path.exists(self.path_output):
            os.mkdir(self.path_output)

    def retranslate_ui(self, main_window):
        main_window.setWindowTitle(self.translate(
            "bili cache merging tool",
            "哔哩BCM-翻滚吧年糕君 ID:1489684 "
        ))
        main_window.setFixedSize(main_window.width(), main_window.height())
        main_window.setWindowIcon(QIcon('local/favicon.ico'))
        self.text_view.setText(self.translate(
            "copy cache folder from your phone local storage path :\n"
            "'/Android/data/tv.danmaku.bili/download' into your pc. \n"
            "note: do not change anything in cache folder",
            "从手机存储路径：'/Android/data/tv.danmaku.bili/download'"
            "拷贝缓存'完整文件夹'到电脑中，至任意目录内。"
        ))
        self.text_view.append("\r")
        self.text_view.append(self.translate(
            "if application gets some exception, please email me at 'cheneyjin@outlook.com' "
            "and attach the file named 'log.md' that in application folder.",
            "如果运行异常，请邮件至'cheneyjin@outlook.com'，"
            "邮件中请上传程序目录下的log.md文件， 内容能图文并茂就更好了~"
        ))
        self.text_view.append("<a href='https://gitee.com/Oops404/nas-guard/raw/master"
                              "/%E6%AC%A2%E8%BF%8E%E6%94%AF%E6%8C%81.jpg'>👉欢迎打赏👈o(￣▽￣)ブ</a>")
        self.path_select_btn.setText(self.translate("select cache path", "选择路径"))

        self.start_btn.setText(self.translate("start merge", "开始"))

    # noinspection PyAttributeOutsideInit,PyUnresolvedReferences
    def setup_ui(self, main_window):
        main_window.setObjectName("mainWindow")
        main_window.resize(600, 320)
        main_window.setStyleSheet(style_global)

        self.central_widget = QtWidgets.QWidget(main_window)
        self.central_widget.setObjectName("central_widget")

        self.vertical_LayoutWidget = QtWidgets.QWidget(self.central_widget)
        self.vertical_LayoutWidget.setGeometry(QtCore.QRect(0, 0, 600, 309))
        self.vertical_LayoutWidget.setObjectName("verticalLayoutWidget")
        self.vertical_LayoutWidget.setContentsMargins(1, 1, 1, 1)
        self.vertical_layout = QtWidgets.QVBoxLayout(self.vertical_LayoutWidget)
        self.vertical_layout.setObjectName("verticalLayout")

        self.path_select_btn = QtWidgets.QPushButton(self.vertical_LayoutWidget)
        self.path_select_btn.setObjectName("path_select_btn")
        self.path_select_btn.setFont(self.font)

        self.vertical_layout.addWidget(self.path_select_btn)
        self.text_view = QtWidgets.QTextBrowser(self.vertical_LayoutWidget)
        self.text_view.setObjectName("text_view")
        self.text_view.setFont(self.font)
        self.text_view.setOpenExternalLinks(True)

        self.vertical_layout.addWidget(self.text_view)
        self.start_btn = QtWidgets.QPushButton(self.vertical_LayoutWidget)
        self.start_btn.setObjectName("start_btn")
        self.start_btn.setContentsMargins(0, 10, 0, 10)
        self.start_btn.setFont(self.font)

        self.vertical_layout.addWidget(self.start_btn)
        main_window.setCentralWidget(self.central_widget)
        self.status_bar = QtWidgets.QStatusBar(main_window)
        self.status_bar.setObjectName("status_bar")
        main_window.setStatusBar(self.status_bar)

        self.path_select_btn.clicked.connect(self.select_path)
        self.start_btn.clicked.connect(self.start_merge)
        self.retranslate_ui(main_window)
        QtCore.QMetaObject.connectSlotsByName(main_window)

    def start_merge(self):
        try:
            if self.searching:
                self.text_view.append(self.translate(
                    "searching cache task is not over, do not click again",
                    "检测缓存数量任务还没结束，别点啦！"
                ))
                return
            if self.task_list is None or len(self.task_list) < 1:
                self.text_view.append(self.translate(
                    "no cache file to be searched",
                    "别点啦，没检测到文件"
                ))
                return
            if not self.merging:
                self.text_view.setText(self.translate(
                    "start merging,please wait a minute",
                    "开始合并，请等待..."
                ))
                self.merging = True
            else:
                self.text_view.append(self.translate(
                    "program is working,do not click again",
                    "正在工作，不要重复点击"
                ))
                return
            _thread.start_new_thread(self.task, ("bcm-task1", 2,))
        finally:
            self.text_view.moveCursor(QTextCursor.End)

    def task(self, arg1, arg2):
        try:
            self.task_id = uuid.uuid1()
            self.logger.info("\nTASK START {}".format(self.task_id))
            for index, f in enumerate(self.task_list):
                result_file = "{}/{}.mp4".format(f.root, f.name)
                if os.path.exists(result_file):
                    f.name = TAG_DUPLICATE + f.name + str(time.time())
                cmd = "\"{}/local/ffmpeg.exe\" -i \"{}\" -i \"{}\" -c:v copy -c:a copy \"{}/{}.mp4\"".format(
                    path_app,
                    f.video_path,
                    f.audio_path,
                    f.root,
                    f.name
                )
                self.logger.info("[{}]:{}".format(f.bvid, cmd))
                with Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE) as p:
                    p.communicate()
                    self.text_view.append("内容:{}，{}\n进度：{}%".format(
                        f.name,
                        "已合并",
                        str(round(((index + 1) / len(self.task_list)) * 100, 2)))
                    )
                    self.text_view.moveCursor(QTextCursor.End)
            if os.path.exists(self.path_output):
                self.text_view.append(self.translate(
                    "finished. check the output path in app dir.",
                    "运行结束，查看本应用路径下的\"输出目录\"吧。"
                ))
            else:
                self.text_view.append(self.translate(
                    "output path was not exist, please check result in cache path.",
                    "输出文件夹不存在，请在缓存目录下查找合并结果，或者使用Everything工具搜索缓存路径下MP4后缀的文件。"
                ))
        except Exception as e:
            self.logger.error(traceback.format_exc())
            self.text_view.append(str(e))
        finally:
            self.merging = False
            self.logger.info("TASK FINISHED {}".format(self.task_id))

    def select_path(self):
        self.task_list = list()
        self.directory = QtWidgets.QFileDialog.getExistingDirectory(
            None,
            self.translate("select cache path", "选取文件夹"),
            "D:/"
        )
        self.text_view.setText("{}: {}".format(self.translate("path", "路径"), self.directory))
        path_str = str(self.directory)
        if "" == path_str:
            self.text_view.append(self.translate("no path selected", "没选择任何路径。"))
        else:
            try:
                self.searching = True
                self.search_path(path_str)
                self.text_view.append("{}: {}{}".format(
                    self.translate("find", "识别到"),
                    len(self.task_list),
                    self.translate("cache can merged", "个可合并视频。"))
                )
            finally:
                self.searching = False

    def search_path(self, root_path):
        for l in os.listdir(root_path):
            path = os.path.join(root_path, l)
            if os.path.isdir(path):
                self.search_path(path)
            else:
                if path.__contains__(self.target_video_name):
                    self.load_file(path)

    def parse_video_name(self, file_info):
        _name = ""
        if "ep" in file_info.keys():
            _name = "({}){}".format(file_info["ep"]["index"], file_info["ep"]["index_title"])
        elif "page_data" in file_info.keys():
            if "download_subtitle" in (file_info["page_data"]).keys():
                _name = file_info["page_data"]["download_subtitle"]
            else:
                _name = file_info["title"]
        if len(_name) < 1:
            _name = file_info["title"]
        if len(_name) < 1:
            _name = "无法解析名称" + str(time.time())
        return _name

    def parse_bvid(self, file_info):
        _bvid = ""
        if "bvid" in file_info.keys():
            _bvid = file_info["bvid"]
        elif "ep" in file_info.keys() and "bvid" in (file_info["ep"]).keys():
            _bvid = file_info["ep"]["bvid"]
        if len(_bvid) < 1:
            _name = "无法解析bvid" + str(time.time())
        return _bvid

    def detect_loss(self, parent_path):
        video_path = "{}/{}".format(parent_path, self.target_video_name)
        audio_path = "{}/{}".format(parent_path, self.target_audio_name)
        video_exist = os.path.exists(video_path)
        audio_exist = os.path.exists(audio_path)
        if video_exist is False or audio_exist is False:
            self.logger.error("FILE PATH {}:{}".format(video_exist, video_path))
            self.logger.error("FILE PATH {}:{}".format(audio_exist, audio_path))
            self.text_view.append("{}{}".format(
                self.translate(
                    "missing file detected in cache path: ",
                    "检测到缓存存在缺失,路径："
                ),
                parent_path
            ))
            return False, video_path, audio_path
        return True, video_path, audio_path

    def load_file(self, path):
        parent_path = os.path.abspath(os.path.join(path, ".."))
        file_info_path = "{}/../{}".format(parent_path, self.target_file_info)
        name = "default"
        bvid = "default"
        try:
            with open(file_info_path, "r", encoding="utf-8") as f:
                file_info = json.load(f)
                name = self.parse_video_name(file_info)
                name = validate(name) if name is not None else "临时名称{}".format(uuid.uuid1())
                bvid = self.parse_bvid(file_info)
        except Exception as e:
            self.logger.error(traceback.format_exc())
        cache_intact, video_path, audio_path = self.detect_loss(parent_path)
        if not cache_intact:
            return
        if os.path.exists(self.path_output):
            parent_path = self.path_output
        self.task_list.append(
            Target(bvid, name, video_path, audio_path, parent_path)
        )