Update main.py

main.py

@@ -10,6 +10,7 @@ from pikpakapi import PikPakApi
 from typing import Union, Any, Dict, List, Optional
 from fastapi import (
     FastAPI,
+    APIRouter,
     Depends,
     Request,
     Body,
@@ -70,9 +71,9 @@ PROXY_URL = os.getenv("PROXY_URL")
 async def verify_token(
     request: Request, credentials: HTTPAuthorizationCredentials = Depends(security)
 ):
-    excluded_paths = ["/"]  # 需要排除的路径列表
-    if request.url.path in excluded_paths:
-        return  # 直接跳过验证
+    # excluded_paths = ["/"]  # 需要排除的路径列表
+    # if request.url.path in excluded_paths:
+    #     return  # 直接跳过验证
 
     # 验证Bearer格式
     if credentials.scheme != "Bearer":
@@ -88,7 +89,7 @@ async def verify_token(
         )
 
 
-app = FastAPI(dependencies=[Depends(verify_token)])
+app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -97,6 +98,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+api_router = APIRouter(dependencies=[Depends(verify_token)])
+front_router = APIRouter()
+
 templates = Jinja2Templates(
     directory="templates", variable_start_string="{[", variable_end_string="]}"
 )
@@ -138,30 +142,34 @@ async def init_client():
             )
 
 
-@app.get("/", response_class=HTMLResponse)
+@front_router.get("/", response_class=HTMLResponse)
 async def home(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 
-@app.post("/files")
+@api_router.post("/files")
 async def get_files(item: FileRequest):
     return await THUNDERX_CLIENT.file_list(
         item.size, item.parent_id, item.next_page_token, item.additional_filters
     )
 
 
-@app.get("/files/{file_id}")
+@api_router.get("/files/{file_id}")
 async def get_file_info(file_id: str):
     return await THUNDERX_CLIENT.get_download_url(file_id)
 
 
-@app.post("/offline")
+@api_router.post("/offline")
 async def offline(item: OfflineRequest):
     return await THUNDERX_CLIENT.offline_download(
         item.file_url, item.parent_id, item.name
     )
 
 
-@app.get("/userinfo")
+@api_router.get("/userinfo")
 async def userinfo():
     return THUNDERX_CLIENT.get_user_info()
+
+
+app.include_router(front_router)
+app.include_router(api_router)