package com.example.springbootdemo.exception;

import com.example.springbootdemo.common.ResultCode;
import lombok.Getter;

@Getter
public class BusinessException extends RuntimeException {

    private final ResultCode resultCode;

    public BusinessException(ResultCode resultCode) {
        super(resultCode.getMessage());
        this.resultCode = resultCode;
    }
}

// ... imports ...
import com.example.springbootdemo.common.ResultCode;
import com.example.springbootdemo.exception.BusinessException;
import com.example.springbootdemo.entity.User;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


@Service
@RequiredArgsConstructor
public class UserServiceImpl implements UserService {

    private final UserMapper userMapper;

    @Override
    public Long saveUser(UserEditDTO dto) {
        // 业务校验：检查用户名是否已存在
        User existingUser = userMapper.selectOne(new QueryWrapper<User>().lambda().eq(User::getUsername, dto.getUsername()));
        if (existingUser != null) {
            // 如果用户已存在，抛出自定义业务异常
            throw new BusinessException(ResultCode.UserAlreadyExists);
        }
        // ... 后续代码不变 ...
    }
    
    
    @Override
    public void updateUser(UserEditDTO dto) {
        // 1. 业务校验：断言要修改的用户必须存在
        User user = userMapper.selectById(dto.getId());
        if (user == null) {
            throw new BusinessException(ResultCode.UserNotFound);
        }

    }

    
    // ... 其他方法 ...
}

package com.example.springbootdemo.advice;

import com.example.springbootdemo.common.Result;
import com.example.springbootdemo.exception.BusinessException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

@Slf4j
@RestControllerAdvice(basePackages = "com.example.springbootdemo.controller")
public class GlobalExceptionHandler {

    /**
     * 专门处理我们自定义的 BusinessException
     */
    @ExceptionHandler(BusinessException.class)
    public ResponseEntity<Result<Void>> handleBusinessException(BusinessException ex) {
        log.error("业务异常: {}", ex.getMessage(), ex);
        // 对于业务异常，我们通常认为这是客户端的“错误”请求，所以返回 400 Bad Request
        // 响应体中包含我们预定义的业务错误码和消息
        return ResponseEntity
                .status(HttpStatus.BAD_REQUEST)
                .body(Result.error(ex.getResultCode()));
    }
}

// ... imports ...
import org.springframework.web.bind.MethodArgumentNotValidException;
import java.util.stream.Collectors;


@Slf4j
@RestControllerAdvice(basePackages = "com.example.springbootdemo.controller")
public class GlobalExceptionHandler {

    // ... 已有的 handleBusinessException 方法 ...

    /**
     * 专门处理由 @Validated 注解触发的参数校验异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResponseEntity<Result<Void>> handleValidationException(MethodArgumentNotValidException ex) {
        // 从异常中获取所有校验失败的信息，并拼接成一个字符串
        String message = ex.getBindingResult().getFieldErrors().stream()
                .map(fieldError -> fieldError.getField() + ": " + fieldError.getDefaultMessage())
                .collect(Collectors.joining("; "));
        
        log.warn("参数校验失败: {}", message);
        
        return ResponseEntity
                .status(HttpStatus.BAD_REQUEST)
                .body(Result.error(message));
    }
}

// ... imports ...

@Slf4j
@RestControllerAdvice(basePackages = "com.example.springbootdemo.controller")
public class GlobalExceptionHandler {

    // ... 已有的两个 ExceptionHandler ...

    /**
     * 兜底处理所有其他未被捕获的异常
     */
    @ExceptionHandler(Exception.class)
    public ResponseEntity<Result<Void>> handleUnknownException(Exception ex) {
        log.error("系统未知异常: {}", ex.getMessage(), ex);
        // 对于未知的系统异常，我们返回 500 Internal Server Error
        // 并使用一个通用的、模糊的错误提示，避免泄露服务器内部细节
        return ResponseEntity
                .status(HttpStatus.INTERNAL_SERVER_ERROR)
                .body(Result.error("系统异常，请联系管理员"));
    }
}

package com.example.springbootdemo.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.CorsRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addCorsMappings(CorsRegistry registry) {
        registry.addMapping("/**") // 对所有请求路径生效
                .allowedOrigins("http://localhost:5173") // 允许来自指定源的请求
                .allowedMethods("GET", "POST", "PUT", "DELETE", "OPTIONS") // 允许的方法
                .allowCredentials(true) // 允许发送 Cookie
                .maxAge(3600); // 预检请求的有效期，单位为秒
    }
}

package com.example.springbootdemo.config;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.CorsRegistry;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Value("${app.upload.dir}")
    private String uploadDir;

    @Override
    public void addCorsMappings(CorsRegistry registry) {
        // ... 已有的 CORS 配置 ...
    }

    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        /**
         * 配置静态资源映射
         * addResourceHandler("/springboot-uploads/**"): 对外暴露的访问路径
         * addResourceLocations("file:" + uploadDir): 文件存放的物理路径
         */
        registry.addResourceHandler("/springboot-uploads/**")
                .addResourceLocations("file:" + uploadDir);
    }
}

package com.example.springbootdemo.controller;


import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import com.example.springbootdemo.common.Result;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

@Tag(name = "文件管理", description = "提供文件上传下载接口")
@RestController
@RequestMapping("/files")
public class FileController {
    @Value("${app.upload.dir}")
    private String uploadDir;

    @Operation(summary = "文件上传")
    @PostMapping("/upload")
    public Result<String> upload(@RequestParam("file") MultipartFile file) throws IOException {
        if (file.isEmpty()) {
            return Result.error("文件名不能为空");
        }
        // 1.获取文件的原始文件名
        String originalFilename = file.getOriginalFilename();
        // 2.获取文件的扩展名 例如:.jpg
        String extName = FileUtil.extName(originalFilename);
        // 3.生成一个随机ID
        String uniqueId = IdUtil.fastSimpleUUID();
        // 4.拼接新文件名 id.jpg
        String newFileName = uniqueId + "." + extName;
        // 5.创建一个文件对象 读取到我们预设置的文件夹中
        File uploadPath = new File(uploadDir);
        if (!uploadPath.exists()) {
            // 如果文件夹不存在，创建文件夹
            uploadPath.mkdir();
        }
        // 6.将我们的文件对象写入到我们预设置的文件夹中
        File destFile = new File(uploadPath, newFileName);
        // 7.将文件转移到我们预设置的文件夹中
        file.transferTo(destFile);
        return Result.success(newFileName);
    }

    @Operation(summary = "获取文件列表")
    @GetMapping("/list")
    public Result<List<String>> listFiles() {
        File uploadDirFile = new File(uploadDir);
        File[] files = uploadDirFile.listFiles();
        if (files == null) {
            return Result.success(Collections.emptyList());
        }
        List<String> filenames = Arrays.stream(files).map(File::getName).toList();
        return Result.success(filenames);
    }
    @Operation(summary = "文件下载（强制附件）")
    @GetMapping("/download/{filename}")
    public ResponseEntity<Resource> download(@PathVariable String filename) throws IOException {
        // 构建文件路径
        File file = new File(uploadDir, filename);

        // 检查文件是否存在
        if (!file.exists()) {
            return ResponseEntity.notFound().build();
        }

        // 创建文件资源
        Resource resource = new FileSystemResource(file);

        // 设置响应头
        HttpHeaders headers = new HttpHeaders();
        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + filename + "\"");
        headers.add(HttpHeaders.CONTENT_TYPE, "application/octet-stream");

        return ResponseEntity.ok()
                .headers(headers)
                .body(resource);
    }

}