修复消息提醒红点提示

router_messages.py

@@ -210,7 +210,21 @@ async def get_chat_history(account: str, target_account: str):
 # 🔥 性能优化：先用只读方式检查是否有实际变更，避免无意义的写入和HF上传
 # ==========================================
 @router.get("/api/messages/{account}")
-async def get_messages(account: str):
+async def get_messages(account: str, count_only: bool = False, current_user: str = Depends(require_auth)):
+    # 🔥 count_only 模式：轻量级轮询，只返回未读数，不标记已读
+    if count_only:
+        messages_db = db.load_data("messages.json", default_data={})
+        user_msgs = messages_db.get(account, [])
+        
+        now = int(time.time())
+        seven_days = 7 * 24 * 3600
+        
+        # 只统计未过期的未读消息
+        unread = sum(1 for m in user_msgs 
+                     if not m.get("is_read") 
+                     and (now - m.get("created_at", 0) < seven_days))
+        return {"status": "success", "unread_count": unread}
+    
     # 公告是只读的，先加载
     announcements_db = db.load_data("announcements.json", default_data=[])
     
@@ -240,12 +254,17 @@ async def get_messages(account: str):
     
     needs_update = has_new_announcements or has_unread or has_expired
     
+    # 🔥 修复：在标记已读之前先计算真实的未读数
+    unread_before_mark = sum(1 for m in user_msgs 
+                              if not m.get("is_read") 
+                              and (now - m.get("created_at", 0) < seven_days))
+    
     if not needs_update:
         # 无变更，直接返回只读数据，不触发写入和HF上传
         return {
             "status": "success",
             "data": user_msgs,
-            "unread_count": 0  # 用户已查看消息列表，未读数设为0
+            "unread_count": unread_before_mark  # 🔥 修复：返回真实的未读数
         }
     
     # 有变更需要写入，使用 atomic_update 保证并发安全
@@ -295,7 +314,7 @@ async def get_messages(account: str):
         result_container[0] = {
             "status": "success",
             "data": valid,
-            "unread_count": 0  # 🔥 修复：用户已查看消息列表，未读数设为0
+            "unread_count": unread_before_mark  # 🔥 修复：返回标记已读前的真实未读数
         }
     
     db.atomic_update("messages.json", updater, default_data={})