mirror of
https://github.com/DrizzleTime/Foxel.git
synced 2026-05-16 05:47:36 +08:00
40 lines
1.6 KiB
Python
40 lines
1.6 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:
|
|
path_pairs = [(str(item.path), item) for item in items if getattr(item, "path", None)]
|
|
allowed_paths = await PermissionService.filter_paths_by_permission(
|
|
user.id,
|
|
[path for path, _ in path_pairs],
|
|
PathAction.READ,
|
|
)
|
|
allowed_set = set(allowed_paths)
|
|
data["items"] = [item for path, item in path_pairs if path in allowed_set]
|
|
return success(data)
|