import json
import locale
import os

current_dir = os.path.dirname(os.path.abspath(__file__))
# 修正路径：从 app/server/user_init 目录向上3级到达项目根目录
init_file_path = os.path.abspath(os.path.join(current_dir, "../../../init.json"))

# 检测系统语言
localLan = locale.getdefaultlocale()[0]
localLan = "zh_CN"
if localLan != "zh_CN":
    localLan = "en_US"

default_settings = {
    "user_lang": localLan,
    "select_openai": 0,
    "openai_settings": [
        {"api_key": "", "base_url": "https://api.openai.com/v1", "model": "gpt-4o"},
        {
            "api_key": "",
            "base_url": "https://api.openai.com/v1",
            "model": "gpt-4o-mini",
        },
        {"api_key": "", "base_url": "https://api.openai.com/v1", "model": "o1"},
        {"api_key": "", "base_url": "https://api.openai.com/v1", "model": "go1-mini"},
        {
            "api_key": "",
            "base_url": "https://api.openai.com/v1",
            "model": "gpt-3.5-turbo",
        },
        {
            "api_key": "",
            "base_url": "https://api.deepseek.com/v1",
            "model": "deepseek-chat",
        },
    ],
    "translate_setting": "translater",
    "translate_service": "alibaba",
    "translate_source_lang": "en",
    "translate_target_lang": "zh",
    "show_auto_limit": 25,
    "random_template": "",
    "ai_info_setting": {
        "api_key": "",
        "base_url": "https://api.siliconflow.cn/v1",
        "model": "THUDM/glm-4-9b-chat",
        "temperature": 0,
        "top_p": 0.7,
        "max_tokens": 4096,
    },
}


def read_init_file():
    """读取 init.json 文件中的所有设置"""
    if not os.path.exists(init_file_path):
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(default_settings, f, ensure_ascii=False, indent=4)
        return default_settings

    with open(init_file_path, encoding="utf-8") as f:
        data = json.load(f)
        return data


def set_user_lang(lang):
    from ..dao.dao import set_language

    """修改 init.json 文件中的 user_lang 设置"""
    data = read_init_file() or {}

    data["user_lang"] = lang

    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)

    # 修改 DAO 中的语言设置
    set_language(lang)


def get_translate_setting():
    """获取translate_setting参数，如果不存在则添加默认值"""
    data = read_init_file() or {}

    # 如果不存在translate_setting参数，则添加默认值
    if "translate_setting" not in data:
        data["translate_setting"] = "translater"
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)

    return data["translate_setting"]


def update_translate_setting(new_setting):
    """更新translate_setting参数"""
    data = read_init_file() or {}
    data["translate_setting"] = new_setting
    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)
    return True


def get_translate_settings():
    """获取翻译相关参数，如果不存在则添加默认值"""
    data = read_init_file() or {}

    # 如果不存在相关参数，则添加默认值
    if "translate_service" not in data:
        data["translate_service"] = "alibaba"
    if "translate_source_lang" not in data:
        data["translate_source_lang"] = "en"
    if "translate_target_lang" not in data:
        data["translate_target_lang"] = "zh"

    # 如果有新增参数，则保存到文件
    if any(
        key not in data
        for key in [
            "translate_service",
            "translate_source_lang",
            "translate_target_lang",
        ]
    ):
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)

    return {
        "translate_service": data["translate_service"],
        "translate_source_lang": data["translate_source_lang"],
        "translate_target_lang": data["translate_target_lang"],
    }


def update_translate_settings(service=None, source_lang=None, target_lang=None):
    """更新翻译相关参数"""
    data = read_init_file() or {}

    if service is not None:
        data["translate_service"] = service
    if source_lang is not None:
        data["translate_source_lang"] = source_lang
    if target_lang is not None:
        data["translate_target_lang"] = target_lang

    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)
    return True


def get_auto_limit_setting():
    """获取show_auto_limit参数，如果不存在则添加默认值"""
    data = read_init_file() or {}

    # 如果不存在show_auto_limit参数，则添加默认值
    if "show_auto_limit" not in data:
        data["show_auto_limit"] = 25
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)

    return data["show_auto_limit"]


def update_auto_limit_setting(new_setting: int):
    """更新show_auto_limit参数"""
    data = read_init_file() or {}
    data["show_auto_limit"] = new_setting
    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)
    return True


def get_random_template_setting():
    """获取random_template参数，如果不存在则添加默认值"""
    data = read_init_file() or {}

    # 如果不存在random_template参数，则添加默认值
    if "random_template" not in data:
        data["random_template"] = ""
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)

    return data["random_template"]


def update_random_template_setting(new_setting):
    """更新random_template参数"""
    data = read_init_file() or {}
    data["random_template"] = new_setting
    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)
    return True


def get_ai_info_setting():
    """获取ai_info_setting参数，如果不存在则添加默认值"""
    data = read_init_file() or {}
    if "ai_info_setting" not in data:
        data["ai_info_setting"] = {
            "api_key": "",
            "base_url": "https://api.siliconflow.cn/v1",
            "model": "THUDM/glm-4-9b-chat",
            "temperature": 0,
            "top_p": 0.7,
            "max_tokens": 4096,
        }
        with open(init_file_path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)
    return data["ai_info_setting"]


def update_ai_info_setting(new_setting: dict):
    """更新ai_info_setting参数"""
    data = read_init_file() or {}
    data["ai_info_setting"] = new_setting
    with open(init_file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)
    return True