MuMuAINovel/backend/app/services/workshop_client.py

"""云端提示词工坊 API 客户端（client 模式使用）"""
import httpx
from typing import Optional, Dict, Any
from app.config import settings, INSTANCE_ID
from app.logger import get_logger

logger = get_logger(__name__)


class WorkshopClientError(Exception):
    """工坊客户端错误"""
    pass


class WorkshopClient:
    """云端 API 客户端"""
    
    def __init__(self):
        self.base_url = settings.WORKSHOP_CLOUD_URL
        self.timeout = settings.WORKSHOP_API_TIMEOUT
    
    async def _request(
        self,
        method: str,
        path: str,
        params: Optional[Dict] = None,
        json: Optional[Dict] = None,
        user_identifier: Optional[str] = None
    ) -> Dict[str, Any]:
        """发送请求到云端"""
        headers = {
            "X-Instance-ID": INSTANCE_ID,
            "Content-Type": "application/json"
        }
        if user_identifier:
            headers["X-User-ID"] = user_identifier
        
        url = f"{self.base_url}/api/prompt-workshop{path}"
        
        try:
            async with httpx.AsyncClient(timeout=self.timeout, verify=False) as client:
                response = await client.request(
                    method=method,
                    url=url,
                    params=params,
                    json=json,
                    headers=headers
                )
                response.raise_for_status()
                return response.json()
        except httpx.ConnectError as e:
            logger.error(f"无法连接到云端服务: {self.base_url}, 错误: {e}")
            raise WorkshopClientError("无法连接到云端服务，请检查网络连接")
        except httpx.TimeoutException:
            logger.error(f"云端服务请求超时: {url}")
            raise WorkshopClientError("云端服务请求超时，请稍后重试")
        except httpx.HTTPStatusError as e:
            logger.error(f"云端服务返回错误: {e.response.status_code}, {e.response.text}")
            raise WorkshopClientError(f"云端服务错误: {e.response.status_code}")
        except Exception as e:
            logger.error(f"请求云端服务异常: {e}")
            raise WorkshopClientError(f"请求云端服务失败: {str(e)}")
    
    async def check_connection(self) -> bool:
        """检查云端连接状态"""
        try:
            await self._request("GET", "/status")
            return True
        except Exception as e:
            logger.warning(f"云端连接检查失败: {e}")
            return False
    
    async def get_items(
        self,
        category: Optional[str] = None,
        search: Optional[str] = None,
        tags: Optional[str] = None,
        sort: str = "newest",
        page: int = 1,
        limit: int = 20,
        user_identifier: Optional[str] = None
    ) -> Dict:
        """获取提示词列表"""
        params = {
            "sort": sort,
            "page": page,
            "limit": limit
        }
        if category:
            params["category"] = category
        if search:
            params["search"] = search
        if tags:
            params["tags"] = tags
        
        return await self._request(
            "GET", "/items",
            params=params,
            user_identifier=user_identifier
        )
    
    async def get_item(self, item_id: str) -> Dict:
        """获取单个提示词详情"""
        return await self._request("GET", f"/items/{item_id}")
    
    async def record_download(self, item_id: str, user_identifier: str) -> Dict:
        """记录下载"""
        return await self._request(
            "POST",
            f"/items/{item_id}/download",
            json={
                "instance_id": INSTANCE_ID,
                "user_identifier": user_identifier
            }
        )
    
    async def toggle_like(self, item_id: str, user_identifier: str) -> Dict:
        """点赞/取消点赞"""
        return await self._request(
            "POST",
            f"/items/{item_id}/like",
            user_identifier=user_identifier
        )
    
    async def submit(
        self,
        user_identifier: str,
        submitter_name: str,
        data: Dict
    ) -> Dict:
        """提交提示词"""
        payload = {
            "instance_id": INSTANCE_ID,
            "submitter_id": user_identifier,
            "submitter_name": submitter_name,
            **data
        }
        # 注意：必须传递 user_identifier 以设置 X-User-ID Header
        return await self._request("POST", "/submit", json=payload, user_identifier=user_identifier)
    
    async def get_submissions(
        self,
        user_identifier: str,
        status: Optional[str] = None
    ) -> Dict:
        """获取用户的提交记录"""
        params = {}
        if status:
            params["status"] = status
        return await self._request(
            "GET", "/my-submissions",
            params=params,
            user_identifier=user_identifier
        )
    
    async def withdraw_submission(
        self,
        submission_id: str,
        user_identifier: str
    ) -> Dict:
        """撤回提交"""
        return await self._request(
            "DELETE",
            f"/submissions/{submission_id}",
            user_identifier=user_identifier
        )


# 全局客户端实例
workshop_client = WorkshopClient()