shiyu
b23383a416
These endpoints fire on every visit or refresh and drown the audit log. Login, config writes, uploads, downloads, and other mutations stay audited.
105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Body, Depends, Request
|
|
|
|
from api.response import success
|
|
from domain.audit import AuditAction, audit
|
|
from domain.auth import User, get_current_active_user
|
|
from domain.permission import require_path_permission
|
|
from domain.permission import require_system_permission
|
|
from domain.permission.service import PermissionService
|
|
from domain.permission.types import PathAction
|
|
from domain.permission.types import SystemPermission
|
|
from domain.processors.registry import get_config_schema
|
|
from .service import ProcessorService
|
|
from .types import (
|
|
ProcessDirectoryRequest,
|
|
ProcessRequest,
|
|
UpdateSourceRequest,
|
|
)
|
|
|
|
router = APIRouter(prefix="/api/processors", tags=["processors"])
|
|
|
|
|
|
@router.get("")
|
|
async def list_processors(
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
data = ProcessorService.list_processors()
|
|
return success(data)
|
|
|
|
|
|
@router.post("/process")
|
|
@audit(
|
|
action=AuditAction.CREATE,
|
|
description="处理单个文件",
|
|
body_fields=["path", "processor_type", "save_to", "overwrite"],
|
|
)
|
|
@require_path_permission(PathAction.READ, "req.path")
|
|
async def process_file_with_processor(
|
|
request: Request,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
req: ProcessRequest = Body(...),
|
|
):
|
|
meta = get_config_schema(req.processor_type) or {}
|
|
if meta.get("produces_file"):
|
|
if req.overwrite:
|
|
await PermissionService.require_path_permission(current_user.id, req.path, PathAction.WRITE)
|
|
elif req.save_to:
|
|
await PermissionService.require_path_permission(current_user.id, req.save_to, PathAction.WRITE)
|
|
data = await ProcessorService.process_file(req)
|
|
return success(data)
|
|
|
|
|
|
@router.post("/process-directory")
|
|
@audit(
|
|
action=AuditAction.CREATE,
|
|
description="批量处理目录",
|
|
body_fields=["path", "processor_type", "overwrite", "max_depth", "suffix"],
|
|
)
|
|
@require_path_permission(PathAction.READ, "req.path")
|
|
async def process_directory_with_processor(
|
|
request: Request,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
req: ProcessDirectoryRequest = Body(...),
|
|
):
|
|
meta = get_config_schema(req.processor_type) or {}
|
|
if meta.get("produces_file"):
|
|
await PermissionService.require_path_permission(current_user.id, req.path, PathAction.WRITE)
|
|
data = await ProcessorService.process_directory(req)
|
|
return success(data)
|
|
|
|
|
|
@router.get("/source/{processor_type}")
|
|
@require_system_permission(SystemPermission.ROLE_MANAGE)
|
|
async def get_processor_source(
|
|
processor_type: str,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
data = await ProcessorService.get_source(processor_type)
|
|
return success(data)
|
|
|
|
|
|
@router.put("/source/{processor_type}")
|
|
@audit(action=AuditAction.UPDATE, description="更新处理器源码")
|
|
@require_system_permission(SystemPermission.ROLE_MANAGE)
|
|
async def update_processor_source(
|
|
request: Request,
|
|
processor_type: str,
|
|
req: UpdateSourceRequest,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
data = await ProcessorService.update_source(processor_type, req)
|
|
return success(data)
|
|
|
|
|
|
@router.post("/reload")
|
|
@audit(action=AuditAction.UPDATE, description="重载处理器模块")
|
|
@require_system_permission(SystemPermission.ROLE_MANAGE)
|
|
async def reload_processor_modules(
|
|
request: Request,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
):
|
|
data = ProcessorService.reload()
|
|
return success(data)
|