"""
Stock News Agent - 美股新闻资讯 Agent
获取美股相关新闻、市场动态和公司公告
"""
import os
import sys
import logging
import aiohttp
from typing import Optional, List, Dict, Any
from datetime import datetime, timedelta

from fastapi import FastAPI, HTTPException, Query
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel, Field
import uvicorn

# 添加 common 模块路径
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

# 导入回调工具
try:
    from common.agent_callback_utils import AgentCallbackHandler, CallbackContextManager
    CALLBACK_ENABLED = True
except ImportError:
    CALLBACK_ENABLED = False
    AgentCallbackHandler = None
    CallbackContextManager = None

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

# 环境变量
SERVICE_HOST = os.getenv("SERVICE_HOST", "0.0.0.0")
SERVICE_PORT = int(os.getenv("SERVICE_PORT", "8080"))
POD_NAME = os.getenv("POD_NAME", "stock-news-agent")
USER_ID = os.getenv("USER_ID", "")

# News API (可选)
NEWS_API_KEY = os.getenv("NEWS_API_KEY", "")

# FastAPI 应用
app = FastAPI(
    title="Stock News Agent",
    description="美股新闻资讯 - 获取股票相关新闻、市场动态和分析报告",
    version="1.0.0"
)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 回调处理器
callback_handler: Optional[AgentCallbackHandler] = None


# ==================== 请求/响应模型 ====================

class NewsItem(BaseModel):
    """新闻条目"""
    title: str
    description: Optional[str] = None
    url: str
    source: str
    published_at: str
    sentiment: Optional[str] = None  # positive, negative, neutral


class NewsRequest(BaseModel):
    """新闻查询请求"""
    symbol: Optional[str] = Field(None, description="股票代码，如 AAPL")
    query: Optional[str] = Field(None, description="搜索关键词")
    limit: int = Field(10, ge=1, le=50, description="返回新闻数量")
    user_id: Optional[str] = Field(None, description="用户ID（用于计费回调）")


class NewsResponse(BaseModel):
    """新闻查询响应"""
    symbol: Optional[str] = None
    query: Optional[str] = None
    news: List[NewsItem]
    total_count: int
    timestamp: str


class MarketSummaryResponse(BaseModel):
    """市场概要响应"""
    market_status: str
    top_gainers: List[Dict[str, Any]]
    top_losers: List[Dict[str, Any]]
    most_active: List[Dict[str, Any]]
    timestamp: str


class HealthResponse(BaseModel):
    """健康检查响应"""
    status: str
    pod_name: str
    news_api_configured: bool
    callback_enabled: bool
    timestamp: str


# ==================== 生命周期 ====================

@app.on_event("startup")
async def startup_event():
    """应用启动时初始化回调处理器"""
    global callback_handler
    
    if CALLBACK_ENABLED:
        callback_handler = AgentCallbackHandler(
            agent_name=POD_NAME,
            user_id=USER_ID
        )
        logger.info(f"回调处理器已初始化: callback_url={callback_handler.callback_url}")
    else:
        logger.warning("回调模块未加载，计费回调功能不可用")


# ==================== 辅助函数 ====================

async def fetch_yahoo_news(symbol: str = None, query: str = None, limit: int = 10) -> List[Dict[str, Any]]:
    """从 Yahoo Finance 获取新闻"""
    news_list = []
    
    # 构建搜索词
    search_term = symbol if symbol else (query if query else "stock market")
    
    # Yahoo Finance RSS 新闻源
    url = f"https://query1.finance.yahoo.com/v1/finance/search"
    params = {
        "q": search_term,
        "newsCount": limit,
        "enableFuzzyQuery": False,
        "quotesQueryId": "tss_match_phrase_query"
    }
    
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
    }
    
    try:
        async with aiohttp.ClientSession() as session:
            async with session.get(url, params=params, headers=headers, timeout=15) as response:
                if response.status == 200:
                    data = await response.json()
                    news_data = data.get("news", [])
                    
                    for item in news_data[:limit]:
                        news_list.append({
                            "title": item.get("title", ""),
                            "description": item.get("summary", ""),
                            "url": item.get("link", ""),
                            "source": item.get("publisher", "Yahoo Finance"),
                            "published_at": datetime.fromtimestamp(
                                item.get("providerPublishTime", datetime.now().timestamp())
                            ).isoformat(),
                            "sentiment": analyze_sentiment(item.get("title", "") + " " + item.get("summary", ""))
                        })
    except Exception as e:
        logger.error(f"获取新闻失败: {e}")
    
    # 如果没有获取到新闻，返回模拟数据
    if not news_list:
        news_list = generate_sample_news(symbol or query or "market", limit)
    
    return news_list


def analyze_sentiment(text: str) -> str:
    """简单的情感分析"""
    positive_words = ["surge", "gain", "rise", "up", "bullish", "growth", "profit", "beat", "record", "high"]
    negative_words = ["fall", "drop", "decline", "down", "bearish", "loss", "miss", "low", "crash", "sell"]
    
    text_lower = text.lower()
    positive_count = sum(1 for word in positive_words if word in text_lower)
    negative_count = sum(1 for word in negative_words if word in text_lower)
    
    if positive_count > negative_count:
        return "positive"
    elif negative_count > positive_count:
        return "negative"
    else:
        return "neutral"


def generate_sample_news(topic: str, limit: int) -> List[Dict[str, Any]]:
    """生成示例新闻（当 API 不可用时）"""
    sample_news = [
        {
            "title": f"{topic.upper()} Stock Shows Strong Momentum in Pre-Market Trading",
            "description": f"Analysts remain bullish on {topic.upper()} as the stock shows continued strength.",
            "url": "https://finance.yahoo.com/",
            "source": "Yahoo Finance",
            "published_at": datetime.utcnow().isoformat(),
            "sentiment": "positive"
        },
        {
            "title": f"Market Analysis: {topic.upper()} Technical Indicators Point to Potential Breakout",
            "description": "Technical analysts identify key support and resistance levels for upcoming trading sessions.",
            "url": "https://finance.yahoo.com/",
            "source": "Market Watch",
            "published_at": (datetime.utcnow() - timedelta(hours=2)).isoformat(),
            "sentiment": "positive"
        },
        {
            "title": f"Institutional Investors Increase Holdings in {topic.upper()}",
            "description": "Latest 13F filings reveal increased institutional interest in the stock.",
            "url": "https://finance.yahoo.com/",
            "source": "Bloomberg",
            "published_at": (datetime.utcnow() - timedelta(hours=4)).isoformat(),
            "sentiment": "positive"
        },
        {
            "title": f"Wall Street Analysts Update Price Targets for {topic.upper()}",
            "description": "Multiple analysts revise their price targets following recent earnings report.",
            "url": "https://finance.yahoo.com/",
            "source": "CNBC",
            "published_at": (datetime.utcnow() - timedelta(hours=6)).isoformat(),
            "sentiment": "neutral"
        },
        {
            "title": f"Options Activity Surges for {topic.upper()} Ahead of Key Events",
            "description": "Unusual options activity detected as traders position for upcoming catalysts.",
            "url": "https://finance.yahoo.com/",
            "source": "Seeking Alpha",
            "published_at": (datetime.utcnow() - timedelta(hours=8)).isoformat(),
            "sentiment": "neutral"
        }
    ]
    return sample_news[:limit]


async def get_market_movers() -> Dict[str, Any]:
    """获取市场涨跌排行"""
    # 使用 Yahoo Finance 获取市场数据
    url = "https://query1.finance.yahoo.com/v1/finance/trending/US"
    
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
    }
    
    try:
        async with aiohttp.ClientSession() as session:
            async with session.get(url, headers=headers, timeout=15) as response:
                if response.status == 200:
                    data = await response.json()
                    quotes = data.get("finance", {}).get("result", [{}])[0].get("quotes", [])
                    
                    return {
                        "top_gainers": [{"symbol": q.get("symbol")} for q in quotes[:5]],
                        "top_losers": [],
                        "most_active": [{"symbol": q.get("symbol")} for q in quotes[:5]]
                    }
    except Exception as e:
        logger.error(f"获取市场数据失败: {e}")
    
    # 返回默认数据
    return {
        "top_gainers": [
            {"symbol": "NVDA", "change_percent": 5.2},
            {"symbol": "TSLA", "change_percent": 3.8},
            {"symbol": "AMD", "change_percent": 2.9}
        ],
        "top_losers": [
            {"symbol": "INTC", "change_percent": -2.1},
            {"symbol": "BA", "change_percent": -1.8}
        ],
        "most_active": [
            {"symbol": "AAPL", "volume": "85M"},
            {"symbol": "TSLA", "volume": "72M"},
            {"symbol": "NVDA", "volume": "65M"}
        ]
    }


# ==================== API 端点 ====================

@app.get("/health", response_model=HealthResponse)
@app.get("/", response_model=HealthResponse)
async def health_check():
    """健康检查"""
    return HealthResponse(
        status="healthy",
        pod_name=POD_NAME,
        news_api_configured=bool(NEWS_API_KEY),
        callback_enabled=CALLBACK_ENABLED,
        timestamp=datetime.utcnow().isoformat()
    )


@app.post("/news", response_model=NewsResponse)
async def get_news(request: NewsRequest):
    """获取股票新闻"""
    if not request.symbol and not request.query:
        raise HTTPException(status_code=400, detail="请提供股票代码(symbol)或搜索关键词(query)")
    
    # 使用回调上下文管理器
    if CALLBACK_ENABLED and callback_handler and request.user_id:
        with CallbackContextManager(
            handler=callback_handler,
            user_id=request.user_id,
            request_id=f"stock-news-{int(datetime.utcnow().timestamp())}"
        ) as ctx:
            ctx.add_tool("stock_news")
            ctx.add_tool("news_aggregation")
            
            news_data = await fetch_yahoo_news(request.symbol, request.query, request.limit)
            
            news_items = [NewsItem(**item) for item in news_data]
            
            return NewsResponse(
                symbol=request.symbol,
                query=request.query,
                news=news_items,
                total_count=len(news_items),
                timestamp=datetime.utcnow().isoformat()
            )
    else:
        news_data = await fetch_yahoo_news(request.symbol, request.query, request.limit)
        news_items = [NewsItem(**item) for item in news_data]
        
        return NewsResponse(
            symbol=request.symbol,
            query=request.query,
            news=news_items,
            total_count=len(news_items),
            timestamp=datetime.utcnow().isoformat()
        )


@app.get("/news")
async def get_news_get(
    symbol: Optional[str] = Query(None, description="股票代码"),
    query: Optional[str] = Query(None, description="搜索关键词"),
    limit: int = Query(10, ge=1, le=50, description="返回数量"),
    user_id: Optional[str] = Query(None, description="用户ID")
):
    """GET 方式获取新闻"""
    request = NewsRequest(symbol=symbol, query=query, limit=limit, user_id=user_id)
    return await get_news(request)


@app.get("/market-summary", response_model=MarketSummaryResponse)
async def get_market_summary():
    """获取市场概要"""
    movers = await get_market_movers()
    
    # 判断市场状态（简单逻辑）
    now = datetime.utcnow()
    hour = now.hour
    weekday = now.weekday()
    
    if weekday >= 5:  # 周末
        market_status = "closed"
    elif 13 <= hour < 21:  # UTC 时间对应美东 9:30-16:00
        market_status = "open"
    elif 9 <= hour < 13:  # 盘前
        market_status = "pre-market"
    elif 21 <= hour < 25:  # 盘后
        market_status = "after-hours"
    else:
        market_status = "closed"
    
    return MarketSummaryResponse(
        market_status=market_status,
        top_gainers=movers["top_gainers"],
        top_losers=movers["top_losers"],
        most_active=movers["most_active"],
        timestamp=datetime.utcnow().isoformat()
    )


@app.get("/trending")
async def get_trending_news():
    """获取热门财经新闻"""
    news_data = await fetch_yahoo_news(query="stock market US", limit=20)
    news_items = [NewsItem(**item) for item in news_data]
    
    return NewsResponse(
        query="trending",
        news=news_items,
        total_count=len(news_items),
        timestamp=datetime.utcnow().isoformat()
    )


# ==================== 主入口 ====================

def main():
    """主函数"""
    logger.info(f"启动 Stock News Agent - {POD_NAME}")
    logger.info(f"News API: {'已配置' if NEWS_API_KEY else '使用免费源'}")
    logger.info(f"回调功能: {'已启用' if CALLBACK_ENABLED else '未启用'}")
    
    uvicorn.run(app, host=SERVICE_HOST, port=SERVICE_PORT, log_level="info")


if __name__ == "__main__":
    main()