#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
图标和颜色配置管理器
负责管理应用和项目的图标、颜色配置
"""
import os
from typing import Dict, Optional


class IconConfigManager:
    """图标和颜色配置管理器"""
    
    def __init__(self, config_data: Dict, get_icons_dir_func):
        """
        初始化图标配置管理器
        
        Args:
            config_data: 配置数据字典的引用
            get_icons_dir_func: 获取图标目录的函数
        """
        self.config_data = config_data
        self._get_icons_dir = get_icons_dir_func
    
    # ==================== 应用图标管理 ====================
    
    def get_app_icon(self, app_path: str) -> str:
        """
        获取应用图标路径
        
        如果配置中保存的是相对路径（只有文件名），则自动拼接 icons 目录
        
        Args:
            app_path: 应用路径
            
        Returns:
            图标的完整路径，如果不存在则返回空字符串
        """
        icon_value = self.config_data.get("app_icons", {}).get(app_path, "")
        
        if not icon_value:
            return ""
        
        # 如果是绝对路径且存在，直接返回
        if os.path.isabs(icon_value) and os.path.exists(icon_value):
            return icon_value
        
        # 如果是相对路径（只有文件名），拼接 icons 目录
        if not os.path.isabs(icon_value):
            icons_dir = self._get_icons_dir()
            full_path = os.path.join(icons_dir, icon_value)
            if os.path.exists(full_path):
                return full_path
        
        # 如果都不存在，返回原值（可能是旧的绝对路径）
        return icon_value
    
    def set_app_icon(self, app_path: str, icon_path: str) -> bool:
        """
        设置应用图标路径
        
        如果图标在 icons 目录下，只保存文件名（相对路径）
        否则保存完整路径
        
        Args:
            app_path: 应用路径
            icon_path: 图标路径
            
        Returns:
            是否设置成功
        """
        if "app_icons" not in self.config_data:
            self.config_data["app_icons"] = {}
        
        # 获取 icons 目录
        icons_dir = self._get_icons_dir()
        
        # 如果图标在 icons 目录下，只保存文件名
        try:
            # 标准化路径以便比较
            icon_path_normalized = os.path.normpath(icon_path)
            icons_dir_normalized = os.path.normpath(icons_dir)
            
            # 检查是否在 icons 目录下
            if icon_path_normalized.startswith(icons_dir_normalized):
                # 只保存文件名
                icon_filename = os.path.basename(icon_path)
                self.config_data["app_icons"][app_path] = icon_filename
                print(f"[OK] Saved icon as relative path: {icon_filename}")
            else:
                # 保存完整路径
                self.config_data["app_icons"][app_path] = icon_path
                print(f"[OK] Saved icon as absolute path: {icon_path}")
            
            return True
        except Exception as e:
            print(f"[WARNING] Error processing icon path: {e}")
            # 出错时保存完整路径
            self.config_data["app_icons"][app_path] = icon_path
            return False
    
    def remove_app_icon(self, app_path: str) -> bool:
        """
        移除应用图标设置
        
        Args:
            app_path: 应用路径
            
        Returns:
            是否移除成功
        """
        if app_path in self.config_data.get("app_icons", {}):
            del self.config_data["app_icons"][app_path]
            return True
        return False
    
    # ==================== 项目图标管理 ====================
    
    def get_project_icon(self, project_name: str) -> str:
        """
        获取项目图标路径
        
        如果配置中保存的是相对路径（只有文件名），则自动拼接 icons 目录
        
        Args:
            project_name: 项目名称
            
        Returns:
            图标的完整路径，如果不存在则返回空字符串
        """
        # 从项目配置中获取图标路径
        projects = self.config_data.get("projects", {})
        project_config = projects.get(project_name, {})
        icon_value = project_config.get("icon", "")
        
        if not icon_value:
            return ""
        
        # 如果是绝对路径且存在，直接返回
        if os.path.isabs(icon_value) and os.path.exists(icon_value):
            return icon_value
        
        # 如果是相对路径（只有文件名），拼接 icons 目录
        if not os.path.isabs(icon_value):
            icons_dir = self._get_icons_dir()
            full_path = os.path.join(icons_dir, icon_value)
            if os.path.exists(full_path):
                return full_path
        
        # 如果都不存在，返回原值（可能是旧的绝对路径）
        return icon_value
    
    def set_project_icon(self, project_name: str, icon_path: str) -> bool:
        """
        设置项目图标路径
        
        如果图标在 icons 目录下，只保存文件名（相对路径）
        否则保存完整路径
        
        Args:
            project_name: 项目名称
            icon_path: 图标路径
            
        Returns:
            是否设置成功
        """
        # 确保projects配置存在
        if "projects" not in self.config_data:
            self.config_data["projects"] = {}
        
        # 确保项目配置存在
        if project_name not in self.config_data["projects"]:
            self.config_data["projects"][project_name] = {
                "apps": [],
                "icon": "",
                "color": ""
            }
        
        # 获取 icons 目录
        icons_dir = self._get_icons_dir()
        
        # 如果图标在 icons 目录下，只保存文件名
        try:
            # 标准化路径以便比较
            icon_path_normalized = os.path.normpath(icon_path)
            icons_dir_normalized = os.path.normpath(icons_dir)
            
            # 检查是否在 icons 目录下
            if icon_path_normalized.startswith(icons_dir_normalized):
                # 只保存文件名
                icon_filename = os.path.basename(icon_path)
                self.config_data["projects"][project_name]["icon"] = icon_filename
                print(f"[OK] Saved project icon as relative path: {icon_filename}")
            else:
                # 保存完整路径
                self.config_data["projects"][project_name]["icon"] = icon_path
                print(f"[OK] Saved project icon as absolute path: {icon_path}")
            
            return True
        except Exception as e:
            print(f"[WARNING] Error processing project icon path: {e}")
            # 出错时保存完整路径
            self.config_data["projects"][project_name]["icon"] = icon_path
            return False
    
    def remove_project_icon(self, project_name: str) -> bool:
        """
        移除项目图标设置
        
        Args:
            project_name: 项目名称
            
        Returns:
            是否移除成功
        """
        projects = self.config_data.get("projects", {})
        if project_name in projects and "icon" in projects[project_name]:
            projects[project_name]["icon"] = ""
            return True
        return False
    
    # ==================== 应用颜色管理 ====================
    
    def get_app_color(self, app_path: str) -> str:
        """
        获取应用按钮颜色
        
        Args:
            app_path: 应用路径
            
        Returns:
            颜色值（十六进制），如果未设置则返回空字符串
        """
        return self.config_data.get("app_colors", {}).get(app_path, "")
    
    def set_app_color(self, app_path: str, color: str) -> bool:
        """
        设置应用按钮颜色
        
        Args:
            app_path: 应用路径
            color: 颜色值（十六进制）
            
        Returns:
            是否设置成功
        """
        if "app_colors" not in self.config_data:
            self.config_data["app_colors"] = {}
        
        self.config_data["app_colors"][app_path] = color
        return True
    
    def remove_app_color(self, app_path: str) -> bool:
        """
        移除应用按钮颜色设置
        
        Args:
            app_path: 应用路径
            
        Returns:
            是否移除成功
        """
        if app_path in self.config_data.get("app_colors", {}):
            del self.config_data["app_colors"][app_path]
            return True
        return False
    
    # ==================== 项目颜色管理 ====================
    
    def get_project_color(self, project_name: str) -> str:
        """
        获取项目背景颜色
        
        Args:
            project_name: 项目名称
            
        Returns:
            颜色值（十六进制），默认为 #2b4c6f
        """
        return self.config_data.get("project_colors", {}).get(project_name, "#2b4c6f")
    
    def set_project_color(self, project_name: str, color: str) -> bool:
        """
        设置项目背景颜色
        
        Args:
            project_name: 项目名称
            color: 颜色值（十六进制）
            
        Returns:
            是否设置成功
        """
        if "project_colors" not in self.config_data:
            self.config_data["project_colors"] = {}
        
        self.config_data["project_colors"][project_name] = color
        return True
    
    def remove_project_color(self, project_name: str) -> bool:
        """
        移除项目背景颜色设置
        
        Args:
            project_name: 项目名称
            
        Returns:
            是否移除成功
        """
        if project_name in self.config_data.get("project_colors", {}):
            del self.config_data["project_colors"][project_name]
            return True
        return False