Update node display names to English

2025-10-20 12:35:39 +08:00
parent ebc5bc66f2
commit 02bcc439be
1 changed files with 366 additions and 0 deletions
@@ -0,0 +1,366 @@
+import os
+import shutil
+import json
+import zipfile
+import requests
+import hashlib
+from datetime import datetime
+from urllib.parse import urlparse
+
+__version__ = "2.0"
+
+class CCXManagerNode:
+    """CCX管理器核心类，负责配置加载、保存和自动运行检查"""
+    # 修复1: 添加配置文件名参数
+    def __init__(self, config_filename="config.json"):
+        self.config_path = os.path.join(os.path.dirname(__file__), config_filename)
+        self.config = self.load_config()
+        self.status = "未运行"
+        self.temp_dir = os.path.join(os.path.dirname(__file__), "temp")
+        # 创建临时目录
+        os.makedirs(self.temp_dir, exist_ok=True)
+
+    def load_config(self):
+        """加载配置文件，如果不存在则创建默认配置"""
+        default_config = {
+            "source_path": "",
+            "target_path": "",
+            "ccx_filename": "",
+            "auto_run_on_restart": False,
+            "last_run_time": "",
+            "github_repo_url": "https://github.com/zombieyang/sd-ppp.git",
+            "last_commit_hash": "",
+            "version": __version__
+        }
+        try:
+            if os.path.exists(self.config_path):
+                with open(self.config_path, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            else:
+                with open(self.config_path, 'w', encoding='utf-8') as f:
+                    json.dump(default_config, f, indent=4, ensure_ascii=False)
+                return default_config
+        except Exception as e:
+            self.status = f"加载配置失败: {str(e)}"
+            print(f"[CCXManager] 加载配置失败: {str(e)}")
+            return default_config
+
+    def save_config(self):
+        """保存配置文件"""
+        try:
+            with open(self.config_path, 'w', encoding='utf-8') as f:
+                json.dump(self.config, f, indent=4, ensure_ascii=False)
+        except Exception as e:
+            self.status = f"保存配置失败: {str(e)}"
+            print(f"[CCXManager] 保存配置失败: {str(e)}")
+
+    def download_from_url(self, url):
+        """从URL下载CCX文件并返回本地临时路径"""
+        try:
+            # 添加时间戳参数
+            timestamp = int(datetime.now().timestamp() * 1000)
+            if '?' in url:
+                url += f"&_={timestamp}"
+            else:
+                url += f"?_={timestamp}"
+              
+            # 发送请求
+            print(f"[CCXManager] 开始下载: {url}")
+            response = requests.get(url, stream=True, timeout=30)
+            response.raise_for_status()
+              
+            # 获取文件名
+            filename = os.path.basename(urlparse(url).path)
+            if not filename.endswith('.ccx'):
+                self.status = "URL不是有效的CCX文件"
+                return None, None
+              
+            temp_path = os.path.join(self.temp_dir, filename)
+              
+            # 保存文件
+            with open(temp_path, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+              
+            print(f"[CCXManager] 下载完成: {temp_path}")
+            return temp_path, filename
+        except Exception as e:
+            self.status = f"下载失败: {str(e)}"
+            print(f"[CCXManager] 下载失败: {str(e)}")
+            return None, None
+
+    def unzip_ccx(self, source_path, target_path):
+        """解压CCX文件到目标路径"""
+        try:
+            with zipfile.ZipFile(source_path, 'r') as zip_ref:
+                zip_ref.extractall(target_path)
+            return True
+        except Exception as e:
+            self.status = f"解压失败: {str(e)}"
+            print(f"[CCXManager] 解压失败: {str(e)}")
+            return False
+
+    def clean_temp_files(self):
+        """清理临时文件"""
+        try:
+            if os.path.exists(self.temp_dir):
+                shutil.rmtree(self.temp_dir)
+                os.makedirs(self.temp_dir, exist_ok=True)
+        except Exception as e:
+            print(f"[CCXManager] 清理临时文件失败: {str(e)}")
+
+    def get_github_latest_commit(self, repo_url):
+        """获取GitHub仓库最新commit哈希值，支持多种URL格式和分支"""
+        try:
+            # 解析仓库路径和分支
+            branch = "main"  # 默认分支
+
+            # 处理不同格式的URL
+            if repo_url.startswith('git@github.com:'):
+                # SSH格式: git@github.com:owner/repo.git
+                path = repo_url.replace('git@github.com:', '')
+                if ':' in path:
+                    # 包含分支: git@github.com:owner/repo.git:branch
+                    path, branch = path.split(':', 1)
+                owner, repo = path.split('/', 1)
+                repo = repo.replace('.git', '')
+            elif repo_url.startswith(('http://', 'https://')):
+                # HTTP/HTTPS格式
+                parsed_url = urlparse(repo_url)
+                path_parts = parsed_url.path.strip('/').split('/')
+                if len(path_parts) < 2:
+                    return None
+                owner, repo = path_parts[0], path_parts[1].replace('.git', '')
+                # 检查URL中是否包含分支信息
+                if len(path_parts) >= 4 and path_parts[2] == 'tree':
+                    branch = path_parts[3]
+            else:
+                # 不支持的URL格式
+                print(f"[CCXManager] 不支持的GitHub URL格式: {repo_url}")
+                return None
+
+            # 构建API URL
+            api_url = f"https://api.github.com/repos/{owner}/{repo}/commits/{branch}"
+            print(f"[CCXManager] 获取GitHub最新commit: {api_url}")
+
+            # 发送请求
+            response = requests.get(api_url, timeout=10)
+            response.raise_for_status()
+            commit_data = response.json()
+            return commit_data['sha']
+        except Exception as e:
+            print(f"[CCXManager] 获取GitHub最新commit失败: {str(e)}")
+            return None
+
+    def check_github_update(self):
+        """检查GitHub仓库是否有更新"""
+        repo_url = self.config.get("github_repo_url", "")
+        if not repo_url:
+            return False
+
+        current_hash = self.config.get("last_commit_hash", "")
+        latest_hash = self.get_github_latest_commit(repo_url)
+
+        if not latest_hash:
+            return False
+
+        if current_hash != latest_hash:
+            self.config["last_commit_hash"] = latest_hash
+            self.save_config()
+            return True
+        return False
+
+    def run(self, source_path, target_path, auto_run=False):
+        """执行CCX文件处理流程"""
+        self.status = "处理中"
+        self.config["source_path"] = source_path
+        self.config["target_path"] = target_path
+        self.config["auto_run_on_restart"] = auto_run
+        self.config["last_run_time"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        self.save_config()
+
+        # 验证目标路径
+        if not target_path:
+            self.status = "错误: 目标路径未设置"
+            return self.status
+        os.makedirs(target_path, exist_ok=True)
+
+        # 清空目标文件夹内容
+        try:
+            if os.path.exists(target_path):
+                for item in os.listdir(target_path):
+                    item_path = os.path.join(target_path, item)
+                    if os.path.isfile(item_path) or os.path.islink(item_path):
+                        os.unlink(item_path)
+                    elif os.path.isdir(item_path):
+                        shutil.rmtree(item_path)
+            print(f"[CCXManager] 目标文件夹已清空: {target_path}")
+        except Exception as e:
+            self.status = f"清理目标文件夹失败: {str(e)}"
+            print(f"[CCXManager] 清理目标文件夹失败: {str(e)}")
+            return self.status
+
+        # 处理URL下载
+        local_path = source_path
+        ccx_filename = None
+        is_url = source_path.startswith(('http://', 'https://'))
+
+        if is_url:
+            local_path, ccx_filename = self.download_from_url(source_path)
+            if not local_path:
+                return self.status
+        else:
+            # 验证本地文件
+            if not os.path.exists(local_path):
+                self.status = "错误: 源文件不存在"
+                return self.status
+            ccx_filename = os.path.basename(local_path)
+
+        # 解压文件
+        if self.unzip_ccx(local_path, target_path):
+            self.status = f"成功: {ccx_filename} 已解压到 {target_path}"
+        else:
+            self.status = f"失败: 无法解压 {ccx_filename}"
+
+        # 清理临时文件
+        if is_url:
+            self.clean_temp_files()
+
+        print(f"[CCXManager] {self.status}")
+        return self.status
+
+    def auto_run(self):
+        """重启后自动运行"""
+        # 检查GitHub是否有更新
+        has_update = self.check_github_update()
+        auto_run_enabled = self.config.get("auto_run_on_restart", False)
+
+        # 只有在检测到更新或启用了自动运行时才执行
+        if has_update or auto_run_enabled:
+            print(f"[CCXManager] 检测到{'更新' if has_update else '自动运行配置'}，正在执行CCX更新...")
+            if all([self.config.get("source_path"), self.config.get("target_path")]):
+                # 运行后禁用自动运行，除非是由更新触发的
+                auto_run_after = has_update  # 如果是更新触发的，则保持启用状态
+                self.run(self.config["source_path"], self.config["target_path"], auto_run=auto_run_after)
+            else:
+                print(f"[CCXManager] 自动运行失败: 配置不完整")
+        else:
+            print(f"[CCXManager] 没有检测到更新且自动运行已禁用")
+
+# 主节点
+class CCXManager:
+    @classmethod
+    def INPUT_TYPES(s):
+        # 加载配置以获取保存的target_path
+        config_path = os.path.join(os.path.dirname(__file__), "config.json")
+        default_target_path = ""
+        default_github_url = "https://github.com/zombieyang/sd-ppp.git"
+        try:
+            if os.path.exists(config_path):
+                with open(config_path, 'r', encoding='utf-8') as f:
+                    config = json.load(f)
+                    default_target_path = config.get("target_path", "")
+                    default_github_url = config.get("github_repo_url", default_github_url)
+        except Exception as e:
+            print(f"[CCXManager] 加载配置失败: {str(e)}")
+
+        return {
+            "required": {
+                "source_type": (["local_path", "url"], {"default": "url"}),
+                "source_path": ("STRING", {"default": "https://gitee.com/zombieyang/sd-ppp/raw/main/static/sd-ppp2_PS.ccx", "placeholder": "本地文件路径或URL"}),
+                "target_path": ("STRING", {"default": default_target_path, "placeholder": "目标解压路径，例如：i:/311/ComfyUI/custom_nodes"}),
+                "github_repo_url": ("STRING", {"default": default_github_url, "placeholder": "GitHub仓库URL，留空禁用更新检测"}),
+                "auto_run_on_restart": (["enable", "disable"], {"default": "enable"}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("运行状态",)
+    FUNCTION = "process_ccx"
+    CATEGORY = "SD-PPP插件更新"
+    TITLE = "SDPPP插件更新助手2.0最新"
+
+    def __init__(self):
+        self.manager = CCXManagerNode()
+
+    def process_ccx(self, source_type, source_path, target_path, github_repo_url, auto_run_on_restart):
+        auto_run = auto_run_on_restart == "enable"
+        # 更新GitHub仓库URL配置
+        self.manager.config["github_repo_url"] = github_repo_url
+        try:
+            self.manager.save_config()
+            if source_type == "url" and not source_path.startswith(("http://", "https://")):
+                return ("错误: URL格式不正确",)
+            status = self.manager.run(source_path, target_path, auto_run)
+            return (status,)
+        except Exception as e:
+            error_msg = f"处理失败: {str(e)}"
+            print(f"[CCXManager] {error_msg}")
+            return (error_msg,)
+
+# 分身节点
+class CCXManagerCopy:
+    @classmethod
+    def INPUT_TYPES(s):
+        # 加载配置以获取保存的target_path
+        config_path = os.path.join(os.path.dirname(__file__), "config_copy.json")
+        default_target_path = ""
+        default_github_url = "https://github.com/zombieyang/sd-ppp.git"
+        try:
+            if os.path.exists(config_path):
+                with open(config_path, 'r', encoding='utf-8') as f:
+                    config = json.load(f)
+                    default_target_path = config.get("target_path", "")
+                    default_github_url = config.get("github_repo_url", default_github_url)
+        except Exception as e:
+            print(f"[CCXManagerCopy] 加载配置失败: {str(e)}")
+
+        return {
+            "required": {
+                "source_type": (["local_path", "url"], {"default": "url"}),
+                "source_path": ("STRING", {"default": "https://gitee.com/zombieyang/sd-ppp/raw/main/static/sd-ppp_PS.ccx", "placeholder": "本地文件路径或URL"}),
+                "target_path": ("STRING", {"default": default_target_path, "placeholder": "目标解压路径，例如：i:/311/ComfyUI/custom_nodes"}),
+                "github_repo_url": ("STRING", {"default": default_github_url, "placeholder": "GitHub仓库URL，留空禁用更新检测"}),
+                "auto_run_on_restart": (["enable", "disable"], {"default": "enable"}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("运行状态",)
+    FUNCTION = "process_ccx"
+    CATEGORY = "SD-PPP插件更新"
+    TITLE = "SDPPP插件更新助手1.0最新"
+
+    def __init__(self):
+        # 使用独立配置文件
+        self.manager = CCXManagerNode(config_filename="config_copy.json")
+
+    def process_ccx(self, source_type, source_path, target_path, github_repo_url, auto_run_on_restart):
+        auto_run = auto_run_on_restart == "enable"
+        # 更新GitHub仓库URL配置
+        self.manager.config["github_repo_url"] = github_repo_url
+        self.manager.save_config()
+        if source_type == "url" and not source_path.startswith(("http://", "https://")):
+            return ("错误: URL格式不正确",)
+        status = self.manager.run(source_path, target_path, auto_run)
+        return (status,)
+
+# 节点注册
+NODE_CLASS_MAPPINGS = {
+    "CCXManager": CCXManager,
+    "CCXManagerCopy": CCXManagerCopy
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "CCXManager": "Photoshop side automatic update SDPPP2.0",
+    "CCXManagerCopy": "Photoshop side automatic update SDPPP1.0"
+}
+
+# 程序启动时自动检查并运行
+print(f"[CCXManager] SDPPP插件更新助手节点已加载 (版本: {__version__})")
+# 主节点自动运行检查
+manager = CCXManagerNode()
+manager.auto_run()
+# 克隆节点自动运行检查
+manager_copy = CCXManagerNode(config_filename="config_copy.json")
+manager_copy.auto_run()