#   <AUTO_MAA:A MAA Multi Account Management and Automation Tool>
#   Copyright © <2024> <DLmaster361>

#   This file is part of AUTO_MAA.

#   AUTO_MAA is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published
#   by the Free Software Foundation, either version 3 of the License,
#   or (at your option) any later version.

#   AUTO_MAA 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 General Public License for more details.

#   You should have received a copy of the GNU General Public License
#   along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.

#   DLmaster_361@163.com

"""
AUTO_MAA
AUTO_MAA更新器
v1.0
作者：DLmaster_361
"""

import os
import sys
import json
import zipfile
import requests
import subprocess

from PySide6.QtWidgets import (
    QApplication,
    QDialog,
    QVBoxLayout,
    QLabel,
    QProgressBar,
)
from PySide6.QtGui import QIcon
from PySide6.QtCore import QObject, QThread, Signal


class UpdateProcess(QThread):

    info = Signal(str)
    progress = Signal(int, int, int)
    accomplish = Signal()

    def __init__(self, app_path, name, download_url, version):
        super(UpdateProcess, self).__init__()

        self.app_path = app_path
        self.name = name
        self.download_url = download_url
        self.version = version
        self.download_path = f"{app_path}/AUTO_MAA_Update.zip"  #  临时下载文件的路径
        self.version_path = f"{app_path}/res/version.json"

    def run(self):

        # 清理可能存在的临时文件
        try:
            os.remove(self.download_path)
        except FileNotFoundError:
            pass

        # 下载
        try:
            response = requests.get(self.download_url, stream=True)
            file_size = response.headers.get("Content-Length")
            if file_size is None:
                file_size = 1
            else:
                file_size = int(file_size)
            with open(self.download_path, "wb") as f:
                downloaded_size = 0
                for chunk in response.iter_content(chunk_size=8192):
                    f.write(chunk)
                    downloaded_size += len(chunk)
                    self.info.emit(
                        f"正在下载：{self.name} 已下载: {downloaded_size / 1048576:.2f}/{file_size / 1048576:.2f} MB ({downloaded_size / file_size * 100:.2f}%)"
                    )
                    self.progress.emit(0, 100, int(downloaded_size / file_size * 100))
        except Exception as e:
            e = str(e)
            e = "\n".join([e[_ : _ + 75] for _ in range(0, len(e), 75)])
            self.info.emit(f"下载{self.name}时出错：\n{e}")
            return None

        # 解压
        try:
            self.info.emit("正在解压更新文件")
            self.progress.emit(0, 0, 0)
            with zipfile.ZipFile(self.download_path, "r") as zip_ref:
                zip_ref.extractall(self.app_path)

            self.info.emit("正在删除临时文件")
            self.progress.emit(0, 0, 0)
            os.remove(self.download_path)

            self.info.emit(f"{self.name}更新成功！")
            self.progress.emit(0, 100, 100)
        except Exception as e:
            e = str(e)
            e = "\n".join([e[_ : _ + 75] for _ in range(0, len(e), 75)])
            self.info.emit(f"解压更新时出错：\n{e}")
            return None

        # 更新version文件
        with open(self.version_path, "r", encoding="utf-8") as f:
            version_info = json.load(f)
        if self.name == "AUTO_MAA更新器":
            version_info["updater_version"] = self.version
        elif self.name == "AUTO_MAA主程序":
            version_info["main_version"] = self.version
        with open(self.version_path, "w", encoding="utf-8") as f:
            json.dump(version_info, f, indent=4)

        # 主程序更新完成后打开AUTO_MAA
        if self.name == "AUTO_MAA主程序":
            subprocess.Popen(
                f"{self.app_path}/AUTO_MAA.exe",
                shell=True,
                creationflags=subprocess.CREATE_NO_WINDOW,
            )

        self.accomplish.emit()


class Updater(QObject):

    def __init__(self, app_path, name, download_url, version):
        super().__init__()

        self.ui = QDialog()
        self.ui.setWindowTitle("AUTO_MAA更新器")
        self.ui.resize(500, 70)
        self.ui.setWindowIcon(QIcon(f"{app_path}/gui/ico/AUTO_MAA_Updater.ico"))

        # 创建垂直布局
        self.Layout_v = QVBoxLayout(self.ui)

        self.info = QLabel("正在初始化", self.ui)
        self.Layout_v.addWidget(self.info)

        self.progress = QProgressBar(self.ui)
        self.progress.setRange(0, 0)
        self.Layout_v.addWidget(self.progress)

        self.update_process = UpdateProcess(app_path, name, download_url, version)

        self.update_process.info.connect(self.update_info)
        self.update_process.progress.connect(self.update_progress)

        self.update_process.start()

    def update_info(self, text):
        self.info.setText(text)

    def update_progress(self, begin, end, current):
        self.progress.setRange(begin, end)
        self.progress.setValue(current)


class AUTO_MAA_Updater(QApplication):
    def __init__(self, app_path, name, download_url, version):
        super().__init__()

        self.main = Updater(app_path, name, download_url, version)
        self.main.ui.show()


if __name__ == "__main__":

    # 获取软件自身的路径
    app_path = os.path.dirname(os.path.realpath(sys.argv[0])).replace("\\", "/")

    # 从本地版本信息文件获取当前版本信息
    if os.path.exists(f"{app_path}/res/version.json"):
        with open(f"{app_path}/res/version.json", "r", encoding="utf-8") as f:
            version_current = json.load(f)
        main_version_current = list(
            map(int, version_current["main_version"].split("."))
        )
    else:
        main_version_current = [0, 0, 0, 0]

    # 从远程服务器获取最新版本信息
    response = requests.get(
        "https://ghp.ci/https://github.com/DLmaster361/AUTO_MAA/blob/main/res/version.json"
    )
    version_remote = response.json()
    main_version_remote = list(map(int, version_remote["main_version"].split(".")))

    # 启动更新线程
    if main_version_remote > main_version_current:
        app = AUTO_MAA_Updater(
            app_path,
            "AUTO_MAA主程序",
            version_remote["main_download_url"],
            version_remote["main_version"],
        )
        sys.exit(app.exec())