增加插件更新提醒

router_items.py

@@ -379,6 +379,29 @@ async def check_github_updates():
         "updated_count": updated_count
     }
 
+@router.get("/api/items/{item_id}")
+async def get_item_by_id(item_id: str):
+    """根据ID获取单个资源的详细信息"""
+    items_db = db.load_data("items.json", default_data=[])
+    comments_db = db.load_data("comments.json", default_data={})
+    versions_db = db.load_data("versions.json", default_data={})
+    
+    for item in items_db:
+        if item["id"] == item_id:
+            # 添加关联数据
+            item["commentsData"] = comments_db.get(item_id, [])
+            item["comments"] = len(item["commentsData"])
+            item["latest_version"] = versions_db.get(item_id, "")
+            
+            # 🔴 【安全防线】：抹除敏感信息
+            item.pop("github_token", None)
+            item.pop("netdisk_password", None)
+            item.pop("viewed_by", None)
+            
+            return {"status": "success", "data": item}
+    
+    raise HTTPException(status_code=404, detail="资源不存在")
+
 @router.get("/api/item/{item_id}/version")
 async def get_item_version(item_id: str):
     """获取单个资源的最新版本号"""

云端_定时版本检测引擎.py

@@ -5,6 +5,9 @@ from datetime import date
 import httpx
 import logging
 import 数据库连接 as db
+from notifications import add_notification
+from database_sql import SessionLocal
+from models_sql import Ownership
 
 logger = logging.getLogger("ComfyUI-Ranking.VersionCheck")
 
@@ -45,6 +48,108 @@ async def fetch_latest_github_hash(repo_url, token):
         logger.error(f"检测版本失败 {repo_url}: {e}")
     return None
 
+async def trigger_update_notifications(updated_items: list):
+    """
+    向已购买/已安装插件的用户发送更新通知
+    
+    参数:
+        updated_items: 有更新的插件列表，每项包含 id, title, version_hash
+    """
+    if not updated_items:
+        return
+    
+    session = None
+    try:
+        # 加载已通知记录，防止重复通知
+        update_notifications_db = db.load_data("update_notifications.json", default_data={})
+        
+        # 构建插件ID到最新版本号的映射
+        updated_versions = {item["id"]: item["version_hash"] for item in updated_items}
+        
+        # 筛选出需要通知的插件（版本号变化才通知）
+        items_to_notify = []
+        for item in updated_items:
+            item_id = item["id"]
+            current_version = item["version_hash"]
+            last_notified_version = update_notifications_db.get(item_id)
+            
+            if last_notified_version != current_version:
+                items_to_notify.append(item)
+        
+        if not items_to_notify:
+            print("📢 所有插件更新已通知过，无需重复发送")
+            return
+        
+        # 从 SQL 数据库查询购买记录
+        session = SessionLocal()
+        
+        # 为每个需要通知的插件，查找已购买用户并发送通知
+        total_notified = 0
+        batch_size = 50  # 分批发送，每批50个用户
+        
+        for item in items_to_notify:
+            item_id = item["id"]
+            title = item.get("title", "未知插件")
+            version_hash = item["version_hash"]
+            
+            # 从 SQL ownerships 表查询已购买该插件的用户（排除已退款记录）
+            ownerships = session.query(Ownership).filter(
+                Ownership.item_id == item_id,
+                Ownership.is_refunded == False
+            ).all()
+            
+            # 使用 set 去重用户账号
+            target_users = list(set(record.account for record in ownerships if record.account))
+            
+            if not target_users:
+                # 无人购买，但仍记录已通知版本
+                update_notifications_db[item_id] = version_hash
+                continue
+            
+            # 准备通知数据
+            notif_data = {
+                "type": "plugin_update",
+                "from_user": "system",
+                "target_item_id": item_id,
+                "target_item_title": title,
+                "content": f"您已安装的插件 [{title}] 有新版本可用"
+            }
+            
+            # 分批发送通知
+            notified_count = 0
+            for i in range(0, len(target_users), batch_size):
+                batch = target_users[i:i + batch_size]
+                
+                for account in batch:
+                    try:
+                        add_notification(account, notif_data)
+                        notified_count += 1
+                    except Exception as e:
+                        logger.error(f"发送更新通知失败 (user={account}, item={item_id}): {e}")
+                        # 单个失败不影响其他用户
+                
+                # 每批之间短暂休眠，避免通知风暴
+                if i + batch_size < len(target_users):
+                    await asyncio.sleep(0.5)
+            
+            # 记录该插件已通知的版本
+            update_notifications_db[item_id] = version_hash
+            total_notified += notified_count
+            print(f"📢 插件 [{title}] 已向 {notified_count} 位用户发送更新通知")
+        
+        # 保存已通知记录
+        db.save_data("update_notifications.json", update_notifications_db)
+        print(f"✅ 更新通知发送完成，共通知 {total_notified} 位用户，涉及 {len(items_to_notify)} 个插件")
+        
+    except Exception as e:
+        logger.error(f"触发更新通知时发生错误: {e}")
+        # 不抛出异常，让主流程继续
+    finally:
+        # 确保 session 正确关闭
+        if session:
+            session.close()
+
+
 async def daily_version_check_task():
     """每日 02:00 定时执行的守护进程"""
     while True:
@@ -63,6 +168,8 @@ async def daily_version_check_task():
         versions_db = db.load_data("versions.json", default_data={})
         
         updated_count = 0
+        updated_items = []  # 记录有更新的插件信息
+        
         for item in items_db:
             link = item.get("link", "")
             if "github.com" in link:
@@ -71,10 +178,18 @@ async def daily_version_check_task():
                 if latest_hash and versions_db.get(item["id"]) != latest_hash:
                     versions_db[item["id"]] = latest_hash
                     updated_count += 1
+                    # 记录更新的插件信息
+                    updated_items.append({
+                        "id": item["id"],
+                        "title": item.get("title", "未知插件"),
+                        "version_hash": latest_hash
+                    })
         
         if updated_count > 0:
             db.save_data("versions.json", versions_db)
             print(f"✅ 版本检测任务完成，共发现并更新了 {updated_count} 个资源的新版本。")
+            # 触发更新通知
+            await trigger_update_notifications(updated_items)
         else:
             print("✅ 版本检测任务完成，暂无任何新版本发现。")