#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
AI Agent信息流API路由模块，提供信息流的增删改查功能
"""

import logging
from fastapi import APIRouter, HTTPException, status, Query
from pydantic import BaseModel
from typing import Dict, Any, List, Optional
from datetime import datetime

from cryptoai.utils.db_manager import get_db_manager

# 配置日志
logger = logging.getLogger("feed_router")

# 创建路由
router = APIRouter()

# 请求模型
class AgentFeedCreate(BaseModel):
    """创建信息流请求模型"""
    agent_name: str
    content: str
    avatar_url: Optional[str] = None

# 响应模型
class AgentFeedResponse(BaseModel):
    """信息流响应模型"""
    id: int
    agent_name: str
    avatar_url: Optional[str] = None
    content: str
    create_time: datetime

@router.post("", response_model=Dict[str, Any], status_code=status.HTTP_201_CREATED)
async def create_feed(feed: AgentFeedCreate) -> Dict[str, Any]:
    """
    创建新的AI Agent信息流
    
    Args:
        feed: 信息流创建请求
        
    Returns:
        创建成功的状态信息
    """
    try:
        # 获取数据库管理器
        db_manager = get_db_manager()
        
        # 保存信息流
        success = db_manager.save_agent_feed(
            agent_name=feed.agent_name,
            content=feed.content,
            avatar_url=feed.avatar_url
        )
        
        if not success:
            raise HTTPException(
                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
                detail="保存信息流失败"
            )
        
        return {
            "status": "success",
            "message": "信息流创建成功"
        }
        
    except Exception as e:
        logger.error(f"创建信息流失败: {str(e)}")
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"创建信息流失败: {str(e)}"
        )

@router.get("", response_model=List[AgentFeedResponse])
async def get_feeds(
    agent_name: Optional[str] = Query(None, description="AI Agent名称，可选"),
    limit: int = Query(20, description="返回的最大记录数，默认20条"),
    skip: int = Query(0, description="跳过的记录数，默认0条")
) -> List[AgentFeedResponse]:
    """
    获取AI Agent信息流列表
    
    Args:
        agent_name: 可选，指定获取特定Agent的信息流
        limit: 返回的最大记录数，默认20条
        
    Returns:
        信息流列表
    """
    try:
        # 获取数据库管理器
        db_manager = get_db_manager()
        
        # 获取信息流
        feeds = db_manager.get_agent_feeds(agent_name=agent_name, limit=limit, skip=skip)
        
        return feeds
        
    except Exception as e:
        logger.error(f"获取信息流失败: {str(e)}")
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"获取信息流失败: {str(e)}"
        )