更新

backend/app/api/publish.py

@@ -56,4 +56,36 @@ async def list_accounts():
 
 @router.post("/login/{platform}")
 async def login_platform(platform: str):
-    return await publish_service.login(platform)
+    result = await publish_service.login(platform)
+    if result.get("success"):
+        return result
+    else:
+        raise HTTPException(status_code=400, detail=result.get("message"))
+
+@router.get("/login/status/{platform}")
+async def get_login_status(platform: str):
+    """检查登录状态"""
+    # 这里简化处理，实际应该维护一个登录会话字典
+    cookie_file = publish_service.cookies_dir / f"{platform}_cookies.json"
+    
+    if cookie_file.exists():
+        return {"success": True, "message": "已登录"}
+    else:
+        return {"success": False, "message": "未登录"}
+
+@router.post("/cookies/save/{platform}")
+async def save_platform_cookie(platform: str, cookie_data: dict):
+    """
+    保存从客户端浏览器提取的Cookie
+    
+    Args:
+        platform: 平台ID
+        cookie_data: {"cookie_string": "document.cookie的内容"}
+    """
+    cookie_string = cookie_data.get("cookie_string", "")
+    result = await publish_service.save_cookie_string(platform, cookie_string)
+    
+    if result.get("success"):
+        return result
+    else:
+        raise HTTPException(status_code=400, detail=result.get("message"))