mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-05-11 18:10:10 +08:00
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from fastapi import APIRouter, Depends, Query
|
|
|
|
from api.response import success
|
|
from domain.auth import User, get_current_active_user
|
|
from domain.permission.service import PermissionService
|
|
from domain.permission.types import PathAction
|
|
from .search_service import VirtualFSSearchService
|
|
|
|
router = APIRouter(prefix="/api/fs/search", tags=["search"])
|
|
|
|
|
|
@router.get("")
|
|
async def search_files(
|
|
q: str = Query(..., description="搜索查询"),
|
|
top_k: int = Query(10, description="返回结果数量"),
|
|
mode: str = Query("vector", description="搜索模式: 'vector' 或 'filename'"),
|
|
page: int = Query(1, description="分页页码，仅在文件名搜索模式下生效"),
|
|
page_size: int = Query(10, description="分页大小，仅在文件名搜索模式下生效"),
|
|
user: User = Depends(get_current_active_user),
|
|
):
|
|
if not q.strip():
|
|
return success({"items": [], "query": q, "mode": mode})
|
|
|
|
top_k = max(top_k, 1)
|
|
page = max(page, 1)
|
|
page_size = max(min(page_size, 100), 1)
|
|
|
|
data = await VirtualFSSearchService.search(q, top_k, mode, page, page_size)
|
|
items = data.get("items") if isinstance(data, dict) else None
|
|
if isinstance(items, list) and items:
|
|
filtered = []
|
|
for item in items:
|
|
path = getattr(item, "path", None)
|
|
if not path:
|
|
continue
|
|
if await PermissionService.check_path_permission(user.id, str(path), PathAction.READ):
|
|
filtered.append(item)
|
|
data["items"] = filtered
|
|
return success(data)
|