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

"""
FastAPI应用程序入口
为CryptoAI系统提供web API接口层
"""

import os
import logging
import uvicorn
import json
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import time
from typing import Dict, Any
from cryptoai.utils.db_manager import init_db

from cryptoai.routes.user import router as user_router
from cryptoai.routes.adata import router as adata_router
from cryptoai.routes.crypto import router as crypto_router
from cryptoai.routes.platform import router as platform_router
from cryptoai.routes.analysis import router as analysis_router
from cryptoai.routes.alltick import router as alltick_router
from cryptoai.routes.payment import router as payment_router
# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler("api_server.log"),
        logging.StreamHandler()
    ]
)
logger = logging.getLogger("fastapi")

# 配置选项
LOG_REQUEST_BODY = os.environ.get("LOG_REQUEST_BODY", "true").lower() == "true"

# 创建FastAPI应用
app = FastAPI(
    title="CryptoAI API",
    description="加密货币AI分析系统API接口",
    version="0.1.0"
)

# 添加CORS中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 可以设置为特定域名，如["http://localhost:3000"]
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 添加API路由
app.include_router(platform_router, prefix="/platform", tags=["平台信息"])
app.include_router(user_router, prefix="/user", tags=["用户管理"])
app.include_router(adata_router, prefix="/adata", tags=["A股数据"])
app.include_router(crypto_router, prefix="/crypto", tags=["加密货币数据"])
app.include_router(analysis_router, prefix="/analysis", tags=["分析历史"])
app.include_router(alltick_router, prefix="/alltick", tags=["AllTick数据"])
app.include_router(payment_router, prefix="/payment", tags=["支付"])

# 请求体日志中间件
@app.middleware("http")
async def log_request_body(request: Request, call_next):
    """
    记录请求体的中间件
    可通过环境变量 LOG_REQUEST_BODY=false 来禁用
    """
    # 如果禁用了请求体日志记录，则直接处理请求
    if not LOG_REQUEST_BODY:
        response = await call_next(request)
        return response
    
    # 获取请求的基本信息
    method = request.method
    url = str(request.url)
    client_ip = request.client.host if request.client else "unknown"
    
    # 初始化请求体内容
    body_content = ""
    
    # 只记录非 GET 请求的请求体
    if method in ["POST", "PUT", "PATCH", "DELETE"]:
        try:
            # 读取请求体
            body = await request.body()
            
            if body:
                # 尝试解析为 JSON
                try:
                    body_json = json.loads(body.decode('utf-8'))
                    # 敏感信息过滤 - 隐藏密码等敏感字段
                    sensitive_fields = ['password', 'token', 'secret', 'key', 'auth']
                    filtered_body = body_json.copy()
                    
                    def filter_sensitive_data(data, fields_to_hide):
                        if isinstance(data, dict):
                            for key in data:
                                if any(sensitive in key.lower() for sensitive in fields_to_hide):
                                    data[key] = "***HIDDEN***"
                                elif isinstance(data[key], dict):
                                    filter_sensitive_data(data[key], fields_to_hide)
                                elif isinstance(data[key], list):
                                    for item in data[key]:
                                        if isinstance(item, dict):
                                            filter_sensitive_data(item, fields_to_hide)
                    
                    filter_sensitive_data(filtered_body, sensitive_fields)
                    body_content = json.dumps(filtered_body, ensure_ascii=False, indent=2)
                except (json.JSONDecodeError, UnicodeDecodeError):
                    # 如果不是 JSON，则显示原始内容（限制长度）
                    body_content = body.decode('utf-8', errors='ignore')[:1000]
                    if len(body) > 1000:
                        body_content += "... (truncated)"
            else:
                body_content = "(empty)"
                
        except Exception as e:
            body_content = f"(error reading body: {str(e)})"
        
        # 记录请求信息
        logger.info(f"""
╭─── REQUEST LOG ───────────────────────────────────────────
│ Method: {method}
│ URL: {url}
│ Client IP: {client_ip}
│ Content-Type: {request.headers.get('content-type', 'N/A')}
│ User-Agent: {request.headers.get('user-agent', 'N/A')[:100]}
│ Body:
{body_content}
╰─────────────────────────────────────────────────────────
""")
    else:
        # GET 请求只记录基本信息
        query_params = str(request.query_params) if request.query_params else "(none)"
        logger.info(f"""
╭─── REQUEST LOG ───────────────────────────────────────────
│ Method: {method}
│ URL: {url}
│ Client IP: {client_ip}
│ Query Params: {query_params}
│ User-Agent: {request.headers.get('user-agent', 'N/A')[:100]}
╰─────────────────────────────────────────────────────────
""")
    
    # 处理请求
    response = await call_next(request)
    
    return response

# 请求计时中间件
@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
    start_time = time.time()
    response = await call_next(request)
    process_time = time.time() - start_time
    response.headers["X-Process-Time"] = str(process_time)
    return response

# 根路由
@app.get("/", tags=["信息"])
async def root() -> Dict[str, Any]:
    """
    API根路径，提供API基本信息
    """
    return {
        "name": "CryptoAI API",
        "version": "0.1.0",
        "description": "加密货币AI分析系统API接口",
        "documentation": "/docs",
        "status": "running"
    }

# 健康检查
@app.get("/health", tags=["信息"])
async def health_check() -> Dict[str, Any]:
    """
    API健康检查接口
    """
    return {
        "status": "healthy", 
        "timestamp": time.time()
    }

# 异常处理
@app.exception_handler(Exception)
async def global_exception_handler(request: Request, exc: Exception):
    logger.error(f"全局异常: {str(exc)}", exc_info=True)
    return JSONResponse(
        status_code=500,
        content={"detail": f"服务器内部错误: {str(exc)}"}
    )

def start():
    """
    启动FastAPI服务器
    """
    # 获取环境变量或使用默认值
    host = os.environ.get("API_HOST", "127.0.0.1")
    port = int(os.environ.get("API_PORT", 8000))
    
    # 启动服务器
    uvicorn.run(
        "cryptoai.routes.fastapi_app:app",
        host=host,
        port=port,
        reload=True  # 生产环境设为False
    )

if __name__ == "__main__":
    init_db()
    start()