SD-SafeAI
/
shudao-main


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339
							from fastapi import FastAPI, Request
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import HTMLResponse
from utils.config import settings
from utils.auth_middleware import auth_middleware
from utils.logger import logger
from routers import api_router
import uvicorn
import time
from pathlib import Path

# 创建FastAPI应用
app = FastAPI(
    title=settings.app.name,
    debug=settings.app.debug
)

# 配置CORS（必须先配置）
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
    allow_headers=["Origin", "Authorization", "Access-Control-Allow-Origin", 
                   "Access-Control-Allow-Headers", "Content-Type", "token"],
    expose_headers=["Content-Length", "Access-Control-Allow-Origin", 
                    "Access-Control-Allow-Headers", "Content-Type"]
)


# 添加请求日志和认证中间件
@app.middleware("http")
async def combined_middleware(request: Request, call_next):
    """组合中间件：日志 + 认证"""
    from fastapi.responses import JSONResponse
    from utils.token import verify_token
    
    start_time = time.time()
    path = request.url.path
    
    # 先打印，确认中间件被执行
    print(f"[DEBUG] 中间件执行 - 路径: {path}")
    logger.info(f"[中间件] 开始处理请求: {path}")
    
    # 白名单路径（不需要认证）
    whitelist_paths = ["/health", "/docs", "/redoc", "/openapi.json", "/static/", "/assets/", "/apiv1/auth/local_login", "/apiv1/auth/register"]
    
    # 检查是否在白名单中（精确匹配或以/结尾的前缀匹配）
    is_whitelist = path == "/" or any(path.startswith(wp) for wp in whitelist_paths)
    
    print(f"[DEBUG] 是否白名单: {is_whitelist}")
    
    if is_whitelist:
        print(f"[DEBUG] 白名单路径，跳过认证")
        request.state.user = None
        response = await call_next(request)
    else:
        # 获取Token
        token = request.headers.get("token") or request.headers.get("Authorization", "").replace("Bearer ", "")
        
        print(f"[DEBUG] Token: {token[:20] if token else 'None'}...")
        logger.info(f"认证中间件 - 路径: {path}")
        logger.info(f"认证中间件 - Token (前20字符): {token[:20] if token else 'None'}...")
        
        if not token:
            print(f"[DEBUG] 未提供Token")
            logger.warning("认证中间件 - 未提供Token")
            response = JSONResponse(
                status_code=401,
                content={"statusCode": 401, "msg": "未提供认证Token"}
            )
        else:
            # 验证Token
            print(f"[DEBUG] 开始验证Token")
            logger.info("认证中间件 - 开始验证Token")
            user_info = await verify_token(token)
            
            print(f"[DEBUG] 验证结果: {user_info}")
            
            if not user_info:
                print(f"[DEBUG] Token验证失败")
                logger.error("认证中间件 - Token验证失败，返回401")
                response = JSONResponse(
                    status_code=401,
                    content={"statusCode": 401, "msg": "Token验证失败"}
                )
            else:
                print(f"[DEBUG] Token验证成功: {user_info.username}")
                logger.info(f"认证中间件 - Token验证成功，用户: {user_info.username} ({user_info.account})")
                request.state.user = user_info
                response = await call_next(request)
    
    # 记录日志
    process_time = time.time() - start_time
    print(f"[DEBUG] 请求完成 - 状态码: {response.status_code}")
    logger.info(f"请求完成: {request.method} {path} - 状态码: {response.status_code} - 耗时: {process_time:.3f}s")
    
    return response

# 注册路由
app.include_router(api_router)

# 单独注册报告兼容路由（避免双重前缀）
from routers.report_compat import router as report_compat_router
app.include_router(report_compat_router)

# 创建静态文件目录
Path("static").mkdir(exist_ok=True)
Path("assets").mkdir(exist_ok=True)

# 挂载静态文件
app.mount("/static", StaticFiles(directory="static"), name="static")
app.mount("/assets", StaticFiles(directory="assets"), name="assets")


@app.get("/", response_class=HTMLResponse)
async def root():
    """根路径 - 欢迎页面"""
    html_content = """
    <!DOCTYPE html>
    <html lang="zh-CN">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        <title>Shudao Chat API</title>
        <style>
            * { margin: 0; padding: 0; box-sizing: border-box; }
            body {
                font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
                min-height: 100vh;
                display: flex;
                align-items: center;
                justify-content: center;
                padding: 20px;
            }
            .container {
                background: white;
                border-radius: 20px;
                box-shadow: 0 20px 60px rgba(0,0,0,0.3);
                padding: 60px 40px;
                max-width: 800px;
                width: 100%;
            }
            h1 {
                color: #667eea;
                font-size: 3em;
                margin-bottom: 20px;
                text-align: center;
            }
            .subtitle {
                color: #666;
                font-size: 1.2em;
                text-align: center;
                margin-bottom: 40px;
            }
            .info-grid {
                display: grid;
                grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
                gap: 20px;
                margin-bottom: 40px;
            }
            .info-card {
                background: #f8f9fa;
                padding: 20px;
                border-radius: 10px;
                text-align: center;
            }
            .info-card h3 {
                color: #667eea;
                font-size: 1.1em;
                margin-bottom: 10px;
            }
            .info-card p {
                color: #666;
                font-size: 0.95em;
            }
            .links {
                display: flex;
                gap: 15px;
                justify-content: center;
                flex-wrap: wrap;
            }
            .btn {
                display: inline-block;
                padding: 12px 30px;
                background: #667eea;
                color: white;
                text-decoration: none;
                border-radius: 8px;
                font-weight: 500;
                transition: all 0.3s;
            }
            .btn:hover {
                background: #764ba2;
                transform: translateY(-2px);
                box-shadow: 0 5px 15px rgba(102, 126, 234, 0.4);
            }
            .btn-secondary {
                background: #48bb78;
            }
            .btn-secondary:hover {
                background: #38a169;
            }
            .status {
                display: inline-block;
                padding: 5px 15px;
                background: #48bb78;
                color: white;
                border-radius: 20px;
                font-size: 0.9em;
                margin-bottom: 20px;
            }
            .features {
                margin-top: 40px;
                padding-top: 40px;
                border-top: 2px solid #f0f0f0;
            }
            .features h2 {
                color: #333;
                margin-bottom: 20px;
                text-align: center;
            }
            .feature-list {
                display: grid;
                grid-template-columns: repeat(auto-fit, minmax(250px, 1fr));
                gap: 15px;
            }
            .feature-item {
                display: flex;
                align-items: center;
                padding: 15px;
                background: #f8f9fa;
                border-radius: 8px;
            }
            .feature-icon {
                font-size: 1.5em;
                margin-right: 15px;
            }
            .feature-text {
                color: #666;
                font-size: 0.95em;
            }
        </style>
    </head>
    <body>
        <div class="container">
            <div style="text-align: center;">
                <span class="status">🟢 服务运行中</span>
            </div>
            <h1>🚀 Shudao Chat API</h1>
            <p class="subtitle">基于 FastAPI 的现代化 AI 聊天服务</p>
            
            <div class="info-grid">
                <div class="info-card">
                    <h3>📦 版本</h3>
                    <p>v1.0.0</p>
                </div>
                <div class="info-card">
                    <h3>⚡ 框架</h3>
                    <p>FastAPI</p>
                </div>
                <div class="info-card">
                    <h3>🗄️ 数据库</h3>
                    <p>MySQL + SQLAlchemy</p>
                </div>
                <div class="info-card">
                    <h3>🔐 认证</h3>
                    <p>Token Based</p>
                </div>
            </div>
            
            <div class="links">
                <a href="/docs" class="btn">📚 API 文档 (Swagger)</a>
                <a href="/redoc" class="btn btn-secondary">📖 API 文档 (ReDoc)</a>
                <a href="/health" class="btn">💚 健康检查</a>
            </div>
            
            <div class="features">
                <h2>✨ 核心功能</h2>
                <div class="feature-list">
                    <div class="feature-item">
                        <span class="feature-icon">💬</span>
                        <span class="feature-text">AI 智能对话</span>
                    </div>
                    <div class="feature-item">
                        <span class="feature-icon">📝</span>
                        <span class="feature-text">历史记录管理</span>
                    </div>
                    <div class="feature-item">
                        <span class="feature-icon">🎯</span>
                        <span class="feature-text">场景识别</span>
                    </div>
                    <div class="feature-item">
                        <span class="feature-icon">📊</span>
                        <span class="feature-text">埋点统计</span>
                    </div>
                    <div class="feature-item">
                        <span class="feature-icon">🔒</span>
                        <span class="feature-text">安全认证</span>
                    </div>
                    <div class="feature-item">
                        <span class="feature-icon">🌐</span>
                        <span class="feature-text">CORS 支持</span>
                    </div>
                </div>
            </div>
        </div>
    </body>
    </html>
    """
    return HTMLResponse(content=html_content)


@app.get("/health")
async def health_check():
    """健康检查"""
    return {"status": "ok"}


if __name__ == "__main__":
    logger.info("=" * 60)
    logger.info("🚀 Shudao Chat API 启动中...")
    logger.info(f"📍 服务地址: http://{settings.app.host}:{settings.app.port}")
    logger.info(f"📚 API 文档: http://{settings.app.host}:{settings.app.port}/docs")
    logger.info(f"🗄️ 数据库: {settings.database.host}:{settings.database.port}/{settings.database.database}")
    logger.info(f"🔧 调试模式: {'开启' if settings.app.debug else '关闭'}")
    logger.info("=" * 60)
    
    uvicorn.run(
        "main:app",
        host=settings.app.host,
        port=settings.app.port,
        reload=settings.app.debug,
        log_level="info"
    )