from json import dump, load
from pathlib import Path
from platform import system
from shutil import move
from .static import ROOT, USERAGENT

__all__ = ["Settings"]


class Settings:
    # 默认配置参数
    default = {
        "mapping_data": {},  # 账号备注映射数据
        "work_path": "",  # 工作目录路径
        "folder_name": "Download",  # 下载文件夹名称
        "name_format": "发布时间 作者昵称 作品标题",  # 文件命名格式
        "user_agent": USERAGENT,  # 请求头
        # "a_user_agent": USERAGENT,  # 请求头
        # "b_user_agent": USERAGENT,  # 请求头
        "cookie": "",  # Cookie
        "proxy": None,  # 代理设置
        "timeout": 10,  # 超时时间(秒)
        "chunk": 1024 * 1024 * 2,  # 下载块大小(字节)
        "max_retry": 5,  # 最大重试次数
        "record_data": False,  # 是否记录作品数据
        "image_format": "JPEG",  # 图文作品格式
        "image_download": True,  # 是否下载图文
        "video_download": True,  # 是否下载视频
        "live_download": False,  # 是否下载动图
        "video_preference": "resolution",  # 视频文件偏好
        "folder_mode": False,  # 文件夹归档模式
        "download_record": True,  # 是否记录下载历史
        "author_archive": False,  # 是否按作者归档
        "write_mtime": False,  # 是否写入修改时间
        "language": "zh_CN",  # 语言设置
        "script_server": False,  # 是否启用脚本服务器
    }
    # 根据操作系统设置编码格式
    encode = "UTF-8-SIG" if system() == "Windows" else "UTF-8"

    def __init__(self, root: Path = ROOT):
        """初始化Settings类

        Args:
            root: 设置文件的根目录路径,默认为ROOT
        """
        # 设置文件路径
        self.name = "settings.json"
        self.root = root
        self.path = root.joinpath(self.name)

    def run(self):
        """运行设置管理

        Returns:
            dict: 设置参数字典
        """
        self.migration_file()
        # 如果文件存在则读取,否则创建新文件
        return self.read() if self.path.is_file() else self.create()

    def read(self) -> dict:
        """读取设置文件

        Returns:
            dict: 读取的设置参数字典
        """
        # 读取设置文件
        with self.path.open("r", encoding=self.encode) as f:
            return self.compatible(load(f))

    def create(self) -> dict:
        """创建新的设置文件

        Returns:
            dict: 默认设置参数字典
        """
        # 创建新的设置文件
        with self.path.open("w", encoding=self.encode) as f:
            dump(self.default, f, indent=4, ensure_ascii=False)
            return self.default

    def update(self, data: dict):
        """更新设置文件内容

        Args:
            data: 要更新的设置参数字典
        """
        # 更新设置文件
        with self.path.open("w", encoding=self.encode) as f:
            dump(data, f, indent=4, ensure_ascii=False)

    def compatible(
        self,
        data: dict,
    ) -> dict:
        """兼容性检查,确保所有默认配置都存在

        Args:
            data: 要检查的设置参数字典

        Returns:
            dict: 经过兼容性检查后的设置参数字典
        """
        # 兼容性检查: 确保所有默认配置都存在
        update = False
        for i, j in self.default.items():
            if i not in data:
                data[i] = j
                update = True
        if update:
            self.update(data)
        return data

    def migration_file(self):
        """迁移设置文件

        如果旧的设置文件存在且新路径下不存在,则移动旧文件到新路径
        """
        if (
            old := self.root.parent.joinpath(self.name)
        ).exists() and not self.path.exists():
            move(old, self.path)
