# ==========================================
# 2. 字体工具函数 (Font Utilities)
# ==========================================
import os
import platform
import shutil
import time
from pathlib import Path

from loguru import logger

from config.config import load_config

config = load_config("config.toml")


def get_system_fonts():
    """获取系统中可用的字体列表"""
    fonts = set()
    if platform.system() == "Windows":
        try:
            system_fonts_dir = Path(os.environ['WINDIR']) / 'Fonts'
            user_fonts_dir = Path.home() / 'AppData' / 'Local' / 'Microsoft' / 'Windows' / 'Fonts'

            for folder in [system_fonts_dir, user_fonts_dir]:
                if folder.exists():
                    for ext in ['*.ttf', '*.ttc', '*.otf']:
                        for font_file in folder.glob(ext):
                            fonts.add(font_file.stem)
        except Exception as e:
            logger.error(f"读取系统字体时出错: {e}")
            fonts = {"微软雅黑", "宋体", "黑体", "Arial", "Microsoft YaHei"}
    return fonts


def is_font_available(font_name):
    """检查字体是否在系统中可用"""
    system_fonts = get_system_fonts()
    check_names = [font_name, font_name.replace(" ", ""), font_name.replace("-", ""),
                   font_name.lower(), font_name.upper()]
    for name in check_names:
        if name in system_fonts:
            return True
    return False


def install_fonts_from_directory(fonts_dir="fonts"):
    """从指定目录安装字体到系统"""
    if platform.system() != "Windows":
        logger.success("字体安装功能目前仅支持Windows系统")
        return False

    target_font_dir = Path.home() / 'AppData' / 'Local' / 'Microsoft' / 'Windows' / 'Fonts'
    target_font_dir.mkdir(parents=True, exist_ok=True)

    if not os.path.exists(fonts_dir):
        logger.error(f"字体目录 {fonts_dir} 不存在，请创建该目录并将字体文件放入")
        return False

    font_files = []
    for ext in ['.ttf', '.ttc', '.otf', '.fon']:
        font_files.extend(Path(fonts_dir).glob(f'*{ext}'))

    if not font_files:
        logger.error(f"在 {fonts_dir} 目录中未找到字体文件")
        return False

    installed_count = 0
    for font_file in font_files:
        try:
            target_path = target_font_dir / font_file.name
            if not target_path.exists():
                shutil.copy2(font_file, target_path)
                logger.success(f"已安装字体: {font_file.name}")
                installed_count += 1
        except Exception as e:
            logger.error(f"安装字体 {font_file.name} 时出错: {str(e)}")

    if installed_count > 0:
        logger.success(f"共安装了 {installed_count} 个新字体文件，建议重启Python环境")
        return True
    return False