Upload web.py

web.py

@@ -0,0 +1,179 @@
+"""
+Main Web Integration - Integrates all routers and modules
+集合router并开启主服务
+"""
+import asyncio
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI, Response
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+
+# Import all routers
+from src.openai_router import router as openai_router
+from src.gemini_router import router as gemini_router
+from src.web_routes import router as web_router
+
+# Import managers and utilities
+from src.credential_manager import CredentialManager
+from src.task_manager import shutdown_all_tasks
+from config import get_server_host, get_server_port
+from log import log
+
+# 全局凭证管理器
+global_credential_manager = None
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """应用生命周期管理"""
+    global global_credential_manager
+    
+    log.info("启动 GCLI2API 主服务")
+    
+    # 初始化全局凭证管理器
+    try:
+        global_credential_manager = CredentialManager()
+        await global_credential_manager.initialize()
+        log.info("凭证管理器初始化成功")
+    except Exception as e:
+        log.error(f"凭证管理器初始化失败: {e}")
+        global_credential_manager = None
+    
+    # 自动从环境变量加载凭证（异步执行）
+    try:
+        from src.auth import auto_load_env_credentials_on_startup
+        import asyncio
+        
+        # 在后台任务中执行异步加载
+        async def load_env_creds():
+            try:
+                await auto_load_env_credentials_on_startup()
+            except Exception as e:
+                log.error(f"自动加载环境变量凭证失败: {e}")
+        
+        # 创建后台任务
+        asyncio.create_task(load_env_creds())
+    except Exception as e:
+        log.error(f"创建自动加载环境变量凭证任务失败: {e}")
+    
+    # OAuth回调服务器将在需要时按需启动
+    
+    yield
+    
+    # 清理资源
+    log.info("开始关闭 GCLI2API 主服务")
+    
+    # 首先关闭所有异步任务
+    try:
+        await shutdown_all_tasks(timeout=10.0)
+        log.info("所有异步任务已关闭")
+    except Exception as e:
+        log.error(f"关闭异步任务时出错: {e}")
+    
+    # 然后关闭凭证管理器
+    if global_credential_manager:
+        try:
+            await global_credential_manager.close()
+            log.info("凭证管理器已关闭")
+        except Exception as e:
+            log.error(f"关闭凭证管理器时出错: {e}")
+    
+    log.info("GCLI2API 主服务已停止")
+
+# 创建FastAPI应用
+app = FastAPI(
+    title="GCLI2API",
+    description="Gemini API proxy with OpenAI compatibility",
+    version="2.0.0",
+    lifespan=lifespan
+)
+
+# CORS中间件
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 挂载路由器
+# OpenAI兼容路由 - 处理OpenAI格式请求
+app.include_router(
+    openai_router,
+    prefix="",
+    tags=["OpenAI Compatible API"]
+)
+
+# Gemini原生路由 - 处理Gemini格式请求
+app.include_router(
+    gemini_router,
+    prefix="",
+    tags=["Gemini Native API"]
+)
+
+# Web路由 - 包含认证、凭证管理和控制面板功能
+app.include_router(
+    web_router,
+    prefix="",
+    tags=["Web Interface"]
+)
+
+# 静态文件路由 - 服务docs目录下的文件（如捐赠图片）
+app.mount("/docs", StaticFiles(directory="docs"), name="docs")
+
+# 保活接口（仅响应 HEAD）
+@app.head("/keepalive")
+async def keepalive() -> Response:
+    return Response(status_code=200)
+
+def get_credential_manager():
+    """获取全局凭证管理器实例"""
+    return global_credential_manager
+
+# 导出给其他模块使用
+__all__ = ['app', 'get_credential_manager']
+
+async def main():
+    """异步主启动函数"""
+    from hypercorn.asyncio import serve
+    from hypercorn.config import Config
+    
+    # 日志系统现在直接使用环境变量，无需初始化
+    
+    # 从环境变量或配置获取端口和主机
+    port = await get_server_port()
+    host = await get_server_host()
+    
+    log.info("=" * 60)
+    log.info("启动 GCLI2API")
+    log.info("=" * 60)
+    log.info(f"控制面板: http://127.0.0.1:{port}")
+    log.info("=" * 60)
+    log.info("API端点:")
+    log.info(f"   OpenAI兼容: http://127.0.0.1:{port}/v1")
+    log.info(f"   Gemini原生: http://127.0.0.1:{port}")
+
+    # 配置hypercorn
+    config = Config()
+    config.bind = [f"{host}:{port}"]
+    config.accesslog = "-"
+    config.errorlog = "-"
+    config.loglevel = "INFO"
+    config.use_colors = True
+    
+    # 设置请求体大小限制为100MB
+    config.max_request_body_size = 100 * 1024 * 1024
+    
+    # 设置连接超时
+    config.keep_alive_timeout = 300  # 5分钟
+    config.read_timeout = 300  # 5分钟读取超时
+    config.write_timeout = 300  # 5分钟写入超时
+    
+    # 增加启���超时时间以支持大量凭证的场景
+    config.startup_timeout = 120  # 2分钟启动超时
+
+    await serve(app, config)
+
+if __name__ == "__main__":
+    asyncio.run(main())