Update

config/config_manager.py

@@ -0,0 +1,623 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+配置管理模块 - 负责读取和保存应用配置
+"""
+import json
+import os
+import sys
+from typing import Dict, List, Optional
+from .icon_config import IconConfigManager
+from .constants import DEFAULT_TASK_FOLDER_TEMPLATES
+
+
+class ConfigManager:
+    """管理NexusLauncher的配置文件"""
+    
+    def __init__(self, config_file: str = "config.json"):
+        self.config_file = config_file
+        self.config_data = self._load_config()
+        
+        # 如果配置文件不存在，保存默认配置
+        if not os.path.exists(self.config_file):
+            print("[INFO] Config file not found, creating default config.json")
+            self.save_config()
+        
+        # 创建图标配置管理器
+        self.icon_config = IconConfigManager(self.config_data, self._get_icons_dir)
+    
+    def _get_icons_dir(self) -> str:
+        """获取 icons 目录路径（避免循环导入）"""
+        if getattr(sys, 'frozen', False):
+            # 打包后的应用
+            return os.path.join(sys._MEIPASS, "icons")
+        else:
+            # 开发环境
+            config_dir = os.path.dirname(os.path.abspath(__file__))
+            project_root = os.path.dirname(config_dir)
+            return os.path.join(project_root, "icons")
+    
+    def _load_config(self) -> Dict:
+        """加载配置文件"""
+        if os.path.exists(self.config_file):
+            try:
+                with open(self.config_file, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            except Exception as e:
+                print(f"Failed to load configuration file: {e}")
+                return self._get_default_config()
+        else:
+            return self._get_default_config()
+    
+    def _get_default_config(self) -> Dict:
+        """获取默认配置"""
+        return {
+            "projects": {
+                "Project_01": {
+                    "icon": "NexusLauncher.ico",  # 默认项目图标
+                    "color": "",  # 项目背景颜色
+                    "apps": []
+                }
+            },
+            "current_project": "Project_01",
+            "window_size": {
+                "width": 425,
+                "height": 480
+            },
+            "icon_size": 80,  # 图标大小，默认80x80
+            "app_icons": {},  # 应用图标映射，格式：{"app_path": "icon_path"}
+            "app_colors": {},  # 应用按钮颜色映射，格式：{"app_path": "#RRGGBB"}
+            "task_folder_templates": self._get_default_task_templates(),  # 任务类型默认文件夹结构
+            "task_settings": {}  # 已废弃：旧的任务设置存储位置，现在存储在 projects.项目名.task_settings
+        }
+    
+    def save_config(self) -> bool:
+        """保存配置到文件"""
+        try:
+            # 重新排序项目字段：icon, color, apps, task_settings
+            if "projects" in self.config_data:
+                for project_name, project_data in self.config_data["projects"].items():
+                    ordered_project = {}
+                    # 按顺序添加字段
+                    if "icon" in project_data:
+                        ordered_project["icon"] = project_data["icon"]
+                    if "color" in project_data:
+                        ordered_project["color"] = project_data["color"]
+                    if "apps" in project_data:
+                        ordered_project["apps"] = project_data["apps"]
+                    if "task_settings" in project_data:
+                        ordered_project["task_settings"] = project_data["task_settings"]
+                    
+                    # 替换原有项目数据
+                    self.config_data["projects"][project_name] = ordered_project
+            
+            with open(self.config_file, 'w', encoding='utf-8') as f:
+                json.dump(self.config_data, f, ensure_ascii=False, indent=4)
+            return True
+        except Exception as e:
+            print(f"Failed to save configuration file: {e}")
+            return False
+    
+    def reload_config(self) -> bool:
+        """重新加载配置文件
+        
+        Returns:
+            是否加载成功
+        """
+        try:
+            self.config_data = self._load_config()
+            # 重新创建图标配置管理器以使用新的config_data引用
+            self.icon_config = IconConfigManager(self.config_data, self._get_icons_dir)
+            return True
+        except Exception as e:
+            print(f"Failed to reload config file: {e}")
+            return False
+    
+    def get_projects(self) -> List[str]:
+        """获取所有项目名称"""
+        return list(self.config_data.get("projects", {}).keys())
+    
+    def get_current_project(self) -> str:
+        """获取当前选中的项目"""
+        return self.config_data.get("current_project", "")
+    
+    def set_current_project(self, project_name: str):
+        """设置当前项目"""
+        if project_name in self.config_data.get("projects", {}):
+            self.config_data["current_project"] = project_name
+            self.save_config()
+    
+    def get_apps(self, project_name: Optional[str] = None) -> List[Dict]:
+        """获取指定项目的应用列表"""
+        if project_name is None:
+            project_name = self.get_current_project()
+        
+        projects = self.config_data.get("projects", {})
+        if project_name in projects:
+            return projects[project_name].get("apps", [])
+        return []
+    
+    def add_project(self, project_name: str, default_icon: str = None) -> bool:
+        """添加新项目
+        
+        Args:
+            project_name: 项目名称
+            default_icon: 默认图标路径（可选）
+        """
+        if "projects" not in self.config_data:
+            self.config_data["projects"] = {}
+        
+        if project_name in self.config_data["projects"]:
+            return False
+        
+        self.config_data["projects"][project_name] = {
+            "icon": default_icon if default_icon else "",
+            "color": "",
+            "apps": []
+        }
+        return self.save_config()
+    
+    def delete_project(self, project_name: str) -> bool:
+        """删除项目"""
+        if project_name in self.config_data.get("projects", {}):
+            # 删除项目配置（包含图标和颜色）
+            del self.config_data["projects"][project_name]
+            
+            # 如果删除的是当前项目，切换到第一个项目
+            if self.config_data.get("current_project") == project_name:
+                projects = self.get_projects()
+                if projects:
+                    self.config_data["current_project"] = projects[0]
+                else:
+                    self.config_data["current_project"] = ""
+            
+            return self.save_config()
+        return False
+    
+    def rename_project(self, old_name: str, new_name: str) -> bool:
+        """重命名项目"""
+        # 检查旧项目是否存在
+        if old_name not in self.config_data.get("projects", {}):
+            return False
+        
+        # 检查新名称是否已存在
+        if new_name in self.config_data.get("projects", {}):
+            return False
+        
+        # 重命名项目
+        self.config_data["projects"][new_name] = self.config_data["projects"].pop(old_name)
+        
+        # 如果重命名的是当前项目，更新当前项目名称
+        if self.config_data.get("current_project") == old_name:
+            self.config_data["current_project"] = new_name
+        
+        return self.save_config()
+    
+    def add_app(self, project_name: str, app_name: str, app_path: str, version: str) -> bool:
+        """添加应用到指定项目"""
+        if project_name not in self.config_data.get("projects", {}):
+            return False
+        
+        app_data = {
+            "name": app_name,
+            "path": app_path,
+            "version": version
+        }
+        
+        self.config_data["projects"][project_name]["apps"].append(app_data)
+        return self.save_config()
+    
+    def update_app(self, project_name: str, app_index: int, app_name: str, app_path: str, version: str) -> bool:
+        """更新应用信息"""
+        if project_name not in self.config_data.get("projects", {}):
+            return False
+        
+        apps = self.config_data["projects"][project_name]["apps"]
+        if 0 <= app_index < len(apps):
+            apps[app_index] = {
+                "name": app_name,
+                "path": app_path,
+                "version": version
+            }
+            return self.save_config()
+        return False
+    
+    def delete_app(self, project_name: str, app_index: int) -> bool:
+        """删除应用"""
+        if project_name not in self.config_data.get("projects", {}):
+            return False
+        
+        apps = self.config_data["projects"][project_name]["apps"]
+        if 0 <= app_index < len(apps):
+            apps.pop(app_index)
+            return self.save_config()
+        return False
+    
+    def reorder_apps(self, project_name: str, from_index: int, to_index: int) -> bool:
+        """重新排序应用，支持拖到列表末尾"""
+        if project_name not in self.config_data.get("projects", {}):
+            return False
+        
+        apps = self.config_data["projects"][project_name]["apps"]
+        app_count = len(apps)
+        # 允许目标索引等于列表长度（表示插入到末尾）
+        if not (0 <= from_index < app_count and 0 <= to_index <= app_count):
+            return False
+        
+        app = apps.pop(from_index)
+        # 如果从上方拖到下方，移除后目标索引需要左移一位
+        if from_index < to_index:
+            to_index -= 1
+        # 再次夹紧，防止越界
+        to_index = max(0, min(to_index, len(apps)))
+        apps.insert(to_index, app)
+        return self.save_config()
+    
+    def get_window_size(self) -> tuple:
+        """获取窗口大小"""
+        size = self.config_data.get("window_size", {"width": 400, "height": 400})
+        return (size.get("width", 400), size.get("height", 400))
+    
+    def save_window_size(self, width: int, height: int):
+        """保存窗口大小"""
+        self.config_data["window_size"] = {"width": width, "height": height}
+        self.save_config()
+    
+    def get_app_icon(self, app_path: str) -> str:
+        """获取应用图标路径"""
+        return self.icon_config.get_app_icon(app_path)
+    
+    def set_app_icon(self, app_path: str, icon_path: str) -> bool:
+        """设置应用图标路径"""
+        result = self.icon_config.set_app_icon(app_path, icon_path)
+        if result:
+            return self.save_config()
+        return False
+    
+    def remove_app_icon(self, app_path: str) -> bool:
+        """移除应用图标设置"""
+        result = self.icon_config.remove_app_icon(app_path)
+        if result:
+            return self.save_config()
+        return False
+    
+    def get_all_app_icons(self) -> Dict[str, str]:
+        """获取所有应用图标映射"""
+        return self.config_data.get("app_icons", {})
+    
+    def get_app_color(self, app_path: str) -> str:
+        """获取应用按钮颜色"""
+        return self.icon_config.get_app_color(app_path)
+    
+    def set_app_color(self, app_path: str, color: str) -> bool:
+        """设置应用按钮颜色"""
+        result = self.icon_config.set_app_color(app_path, color)
+        if result:
+            return self.save_config()
+        return False
+    
+    def remove_app_color(self, app_path: str) -> bool:
+        """移除应用按钮颜色设置"""
+        result = self.icon_config.remove_app_color(app_path)
+        if result:
+            return self.save_config()
+        return False
+    
+    def get_all_app_colors(self) -> Dict[str, str]:
+        """获取所有应用按钮颜色映射"""
+        return self.config_data.get("app_colors", {})
+    
+    def get_icon_size(self) -> int:
+        """获取图标大小"""
+        return self.config_data.get("icon_size", 80)
+    
+    def save_icon_size(self, size: int) -> bool:
+        """保存图标大小"""
+        self.config_data["icon_size"] = size
+        return self.save_config()
+    
+    def get_project_icon(self, project_name: str) -> str:
+        """获取项目图标路径"""
+        return self.icon_config.get_project_icon(project_name)
+    
+    def set_project_icon(self, project_name: str, icon_path: str) -> bool:
+        """设置项目图标路径"""
+        result = self.icon_config.set_project_icon(project_name, icon_path)
+        if result:
+            return self.save_config()
+        return False
+    
+    def remove_project_icon(self, project_name: str) -> bool:
+        """移除项目图标设置"""
+        result = self.icon_config.remove_project_icon(project_name)
+        if result:
+            return self.save_config()
+        return False
+    
+    def get_project_color(self, project_name: str) -> str:
+        """获取项目背景颜色"""
+        return self.icon_config.get_project_color(project_name)
+    
+    def set_project_color(self, project_name: str, color: str) -> bool:
+        """设置项目背景颜色"""
+        result = self.icon_config.set_project_color(project_name, color)
+        if result:
+            return self.save_config()
+        return False
+    
+    def remove_project_color(self, project_name: str) -> bool:
+        """移除项目背景颜色设置"""
+        result = self.icon_config.remove_project_color(project_name)
+        if result:
+            return self.save_config()
+        return False
+    
+    def _get_default_task_templates(self) -> Dict[str, List[str]]:
+        """获取默认任务类型文件夹模板"""
+        return DEFAULT_TASK_FOLDER_TEMPLATES
+    
+    def get_task_folder_template(self, task_type: str) -> List[str]:
+        """获取指定任务类型的文件夹模板
+        
+        Args:
+            task_type: 任务类型名称
+            
+        Returns:
+            文件夹路径列表
+        """
+        templates = self.config_data.get("task_folder_templates", {})
+        if task_type in templates:
+            return templates[task_type]
+        # 如果配置中没有，返回默认模板
+        default_templates = self._get_default_task_templates()
+        return default_templates.get(task_type, [])
+    
+    def get_task_types(self) -> List[str]:
+        """获取所有可用的任务类型名称列表
+        
+        Returns:
+            任务类型名称列表
+        """
+        templates = self.config_data.get("task_folder_templates", {})
+        if not templates:
+            # 如果配置中没有，使用默认模板
+            templates = self._get_default_task_templates()
+        return sorted(list(templates.keys()))
+    
+    def get_all_task_folder_templates(self) -> Dict[str, List[str]]:
+        """获取所有任务类型的文件夹模板
+        
+        Returns:
+            任务类型到文件夹列表的映射
+        """
+        templates = self.config_data.get("task_folder_templates", {})
+        if not templates:
+            # 如果配置中没有，初始化默认模板并保存
+            templates = self._get_default_task_templates()
+            self.config_data["task_folder_templates"] = templates
+            self.save_config()
+        return templates
+    
+    def set_task_folder_template(self, task_type: str, folders: List[str]) -> bool:
+        """设置指定任务类型的文件夹模板
+        
+        Args:
+            task_type: 任务类型名称
+            folders: 文件夹路径列表
+            
+        Returns:
+            是否保存成功
+        """
+        if "task_folder_templates" not in self.config_data:
+            self.config_data["task_folder_templates"] = {}
+        
+        # 统一路径格式为正斜杠（JSON 存储格式）
+        normalized_folders = [folder.replace("\\", "/") for folder in folders]
+        self.config_data["task_folder_templates"][task_type] = normalized_folders
+        return self.save_config()
+    
+    def update_all_task_folder_templates(self, templates: Dict[str, List[str]]) -> bool:
+        """更新所有任务类型的文件夹模板
+        
+        Args:
+            templates: 任务类型到文件夹列表的映射
+            
+        Returns:
+            是否保存成功
+        """
+        # 统一所有模板的路径格式为正斜杠（JSON 存储格式）
+        normalized_templates = {}
+        for task_type, folders in templates.items():
+            normalized_templates[task_type] = [folder.replace("\\", "/") for folder in folders]
+        
+        self.config_data["task_folder_templates"] = normalized_templates
+        return self.save_config()
+    
+    def get_task_settings(self, project_name: str) -> Dict:
+        """获取指定项目的 Task 面板设置
+        
+        Args:
+            project_name: 项目名称
+            
+        Returns:
+            Task 设置字典，包含 workspace, task_type, use_type_hierarchy
+        """
+        # 确保 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] = {}
+        
+        # 从 projects.项目名.task_settings 读取
+        project_data = self.config_data["projects"][project_name]
+        settings = project_data.get("task_settings", {
+            "workspace": "D:\\Workspace",
+            "task_type": "Character",
+            "use_type_hierarchy": False
+        })
+        
+        # 标准化路径格式（JSON中存储正斜杠，转换为反斜杠供UI显示）
+        if "workspace" in settings and settings["workspace"]:
+            settings["workspace"] = settings["workspace"].replace("/", "\\")
+        if "maya_plugin_path" in settings and settings["maya_plugin_path"]:
+            settings["maya_plugin_path"] = settings["maya_plugin_path"].replace("/", "\\")
+        if "sp_shelf_path" in settings and settings["sp_shelf_path"]:
+            settings["sp_shelf_path"] = settings["sp_shelf_path"].replace("/", "\\")
+        
+        return settings
+    
+    def set_task_settings(self, project_name: str, workspace: str = None, 
+                         task_type: str = None, use_type_hierarchy: bool = None,
+                         maya_plugin_path: str = None, sp_shelf_path: str = None) -> bool:
+        """设置指定项目的 Task 面板设置
+        
+        Args:
+            project_name: 项目名称
+            workspace: 工作空间路径
+            task_type: 任务类型
+            use_type_hierarchy: 是否使用类型层级
+            maya_plugin_path: Maya 插件路径
+            sp_shelf_path: Substance Painter 架子路径
+            
+        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] = {}
+        
+        # 确保 task_settings 字段存在
+        if "task_settings" not in self.config_data["projects"][project_name]:
+            self.config_data["projects"][project_name]["task_settings"] = {}
+        
+        # 保存到 projects.项目名.task_settings
+        settings = self.config_data["projects"][project_name]["task_settings"]
+        
+        # 保存 SubFolders（如果存在，先临时保存）
+        subfolders = settings.get("SubFolders")
+        
+        # 按顺序更新字段
+        if workspace is not None:
+            settings["workspace"] = workspace.replace("\\", "/")
+        if task_type is not None:
+            settings["task_type"] = task_type
+        if use_type_hierarchy is not None:
+            settings["use_type_hierarchy"] = use_type_hierarchy
+        if maya_plugin_path is not None:
+            settings["maya_plugin_path"] = maya_plugin_path.replace("\\", "/")
+        if sp_shelf_path is not None:
+            settings["sp_shelf_path"] = sp_shelf_path.replace("\\", "/")
+        
+        # 重新构建有序字典，确保字段顺序：workspace, maya_plugin_path, sp_shelf_path, task_type, use_type_hierarchy, SubFolders
+        ordered_settings = {}
+        for key in ["workspace", "maya_plugin_path", "sp_shelf_path", "task_type", "use_type_hierarchy"]:
+            if key in settings:
+                ordered_settings[key] = settings[key]
+        
+        # 将 SubFolders 放在最后
+        if subfolders is not None:
+            ordered_settings["SubFolders"] = subfolders
+        
+        # 替换原有的 task_settings
+        self.config_data["projects"][project_name]["task_settings"] = ordered_settings
+        
+        print(f"[OK] Save the task settings to projects.{project_name}.task_settings")
+        return self.save_config()
+    
+    def copy_apps_to_config(self, apps_data: List[Dict]) -> bool:
+        """将复制的应用数据保存到配置文件"""
+        try:
+            if "clipboard" not in self.config_data:
+                self.config_data["clipboard"] = {}
+            
+            self.config_data["clipboard"]["apps"] = apps_data
+            print(f"[DEBUG] Saved {len(apps_data)} apps to config clipboard")
+            return self.save_config()
+        except Exception as e:
+            print(f"[ERROR] Failed to save apps to config clipboard: {e}")
+            return False
+    
+    def get_clipboard_apps(self) -> List[Dict]:
+        """从配置文件获取复制的应用数据"""
+        try:
+            if "clipboard" in self.config_data and "apps" in self.config_data["clipboard"]:
+                apps = self.config_data["clipboard"]["apps"]
+                print(f"[DEBUG] Retrieved {len(apps)} apps from config clipboard")
+                return apps
+            else:
+                print("[DEBUG] No apps in config clipboard")
+                return []
+        except Exception as e:
+            print(f"[ERROR] Failed to get apps from config clipboard: {e}")
+            return []
+    
+    def clear_clipboard_apps(self) -> bool:
+        """清空配置文件中的应用剪贴板"""
+        try:
+            if "clipboard" in self.config_data:
+                self.config_data["clipboard"]["apps"] = []
+                print("[DEBUG] Cleared config clipboard")
+                return self.save_config()
+            return True
+        except Exception as e:
+            print(f"[ERROR] Failed to clear config clipboard: {e}")
+            return False
+    
+    def save_selection_state(self, project_name: str, selected_indices: List[int]) -> bool:
+        """保存项目的选择状态到配置文件"""
+        try:
+            if "selection_states" not in self.config_data:
+                self.config_data["selection_states"] = {}
+            
+            self.config_data["selection_states"][project_name] = selected_indices
+            print(f"[DEBUG] Saved selection state for {project_name}: {selected_indices}")
+            return self.save_config()
+        except Exception as e:
+            print(f"[ERROR] Failed to save selection state: {e}")
+            return False
+    
+    def get_selection_state(self, project_name: str) -> List[int]:
+        """从配置文件获取项目的选择状态"""
+        try:
+            if ("selection_states" in self.config_data and 
+                project_name in self.config_data["selection_states"]):
+                indices = self.config_data["selection_states"][project_name]
+                print(f"[DEBUG] Retrieved selection state for {project_name}: {indices}")
+                return indices
+            else:
+                print(f"[DEBUG] No selection state found for {project_name}")
+                return []
+        except Exception as e:
+            print(f"[ERROR] Failed to get selection state: {e}")
+            return []
+    
+    def select_all_apps(self, project_name: str) -> List[int]:
+        """选择项目的所有应用并保存状态"""
+        try:
+            apps = self.get_apps(project_name)
+            all_indices = list(range(len(apps)))
+            self.save_selection_state(project_name, all_indices)
+            print(f"[DEBUG] Selected all {len(all_indices)} apps in {project_name}")
+            return all_indices
+        except Exception as e:
+            print(f"[ERROR] Failed to select all apps: {e}")
+            return []
+    
+    def clear_selection_state(self, project_name: str) -> bool:
+        """清空项目的选择状态"""
+        try:
+            if ("selection_states" in self.config_data and 
+                project_name in self.config_data["selection_states"]):
+                self.config_data["selection_states"][project_name] = []
+                print(f"[DEBUG] Cleared selection state for {project_name}")
+                return self.save_config()
+            return True
+        except Exception as e:
+            print(f"[ERROR] Failed to clear selection state: {e}")
+            return False