label_backend/src/main/java/com/label/controller/SourceController.java

package com.label.controller;

import com.label.common.result.PageResult;
import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal;
import com.label.dto.SourceResponse;
import com.label.service.SourceService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

/**
 * 原始资料管理接口。
 *
 * 权限设计：
 * - 上传 / 列表 / 详情：UPLOADER 及以上角色（含 ANNOTATOR、REVIEWER、ADMIN）
 * - 删除：仅 ADMIN
 */
@Tag(name = "资料管理", description = "原始资料上传、查询和删除")
@RestController
@RequestMapping("/api/source")
@RequiredArgsConstructor
public class SourceController {

    private final SourceService sourceService;

    /**
     * 上传文件（multipart/form-data）。
     * 返回 201 Created + 资料摘要。
     */
    @Operation(summary = "上传原始资料", description = "dataType: text,image, video")
    @PostMapping("/upload")
    @RequiresRoles("UPLOADER")
    @ResponseStatus(HttpStatus.CREATED)
    public Result<SourceResponse> upload(
            @RequestParam("file") MultipartFile file,
            @RequestParam("dataType") String dataType,
            HttpServletRequest request) {
        TokenPrincipal principal = (TokenPrincipal) request.getAttribute("__token_principal__");
        return Result.success(sourceService.upload(file, dataType, principal));
    }

    /**
     * 分页查询资料列表。
     * UPLOADER 只见自己的资料；ADMIN 见全公司资料。
     */
    @Operation(summary = "分页查询资料列表")
    @GetMapping("/list")
    @RequiresRoles("UPLOADER")
    public Result<PageResult<SourceResponse>> list(
            @RequestParam(defaultValue = "1") int page,
            @RequestParam(defaultValue = "20") int pageSize,
            @RequestParam(required = false) String dataType,
            @RequestParam(required = false) String status,
            HttpServletRequest request) {
        TokenPrincipal principal = (TokenPrincipal) request.getAttribute("__token_principal__");
        return Result.success(sourceService.list(page, pageSize, dataType, status, principal));
    }

    /**
     * 查询资料详情（含 15 分钟预签名下载链接）。
     */
    @Operation(summary = "查询资料详情")
    @GetMapping("/{id}")
    @RequiresRoles("UPLOADER")
    public Result<SourceResponse> findById(@PathVariable Long id) {
        return Result.success(sourceService.findById(id));
    }

    /**
     * 删除资料（仅 PENDING 状态可删）。
     * 同步删除 RustFS 文件及 DB 记录。
     */
    @Operation(summary = "删除资料")
    @DeleteMapping("/{id}")
    @RequiresRoles("ADMIN")
    public Result<Void> delete(@PathVariable Long id, HttpServletRequest request) {
        TokenPrincipal principal = (TokenPrincipal) request.getAttribute("__token_principal__");
        sourceService.delete(id, principal.getCompanyId());
        return Result.success(null);
    }
}