#   <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

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

uiLoader = QUiLoader()


class UpdateProcess(QThread):

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

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

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

    def run(self):

        # 下载
        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}")
        self.accomplish.emit()


class Updater(QObject):

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

        self.ui = uiLoader.load(app_path + "/gui/ui/updater.ui")
        self.ui.setWindowTitle("AUTO_MAA更新器")
        self.ui.setWindowIcon(QIcon(app_path + "/res/AUTO_MAA.ico"))

        self.info = self.ui.findChild(QLabel, "label")
        self.info.setText("正在初始化")

        self.progress = self.ui.findChild(QProgressBar, "progressBar")
        self.progress.setRange(0, 0)

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

        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):
        super().__init__()

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


if __name__ == "__main__":
    # 获取软件自身的路径
    app_path = os.path.dirname(os.path.realpath(sys.argv[0])).replace("\\", "/")
    # 从本地版本信息文件获取当前版本信息
    with open(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(".")))
    # 从远程服务器获取最新版本信息
    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"]
        )
        sys.exit(app.exec())