from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from fastapi.middleware.cors import CORSMiddleware
from starlette.middleware.sessions import SessionMiddleware
from database import init_db
from routes import api
import os

def create_app():
    app = FastAPI(
        title="WorkList API",
        description="任务管理和时间追踪API",
        version="1.0.0"
    )

    # 配置 CORS
    app.add_middleware(
        CORSMiddleware,
        allow_origins=["*"],  # 生产环境应该设置具体的域名
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

    # 配置 Session
    app.add_middleware(
        SessionMiddleware,
        secret_key='your-secret-key-here-change-in-production',
        max_age=86400,  # 24小时
        same_site='lax',
        https_only=False  # 生产环境应设置为True
    )

    # 初始化数据库
    init_db()

    # 注册路由
    app.include_router(api, prefix='/api')

    # 静态文件服务（用于前端）
    frontend_path = os.path.join(os.path.dirname(__file__), '../frontend')
    if os.path.exists(frontend_path):
        @app.get("/")
        async def index():
            return FileResponse(os.path.join(frontend_path, 'index.html'))

        app.mount("/", StaticFiles(directory=frontend_path, html=True), name="static")

    return app

app = create_app()

if __name__ == '__main__':
    import uvicorn
    uvicorn.run("app:app", host='0.0.0.0', port=5000, reload=True)