1

zj

2025-08-21 01bd03652fef2399c2acfb1930d044b106393f85

 ruoyi-admin/src/main/java/com/ruoyi/im/ImApiController.java

@@ -2,15 +2,29 @@

import com.ruoyi.common.core.domain.R;
import com.ruoyi.im.comm.Result;
import com.ruoyi.im.dto.UpdateUserBusinessDto;
import com.ruoyi.im.service.ImApiServcie;
import com.ruoyi.im.dto.RegisterDto;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.PosixFilePermission;
import java.nio.file.attribute.PosixFilePermissions;
import java.util.Map;
import java.util.Set;
import java.util.UUID;

@RestController
@RequestMapping("/im/api")
@@ -19,6 +33,14 @@

    @Autowired
    private ImApiServcie imApiServcie;


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

    @Value("${file.upload-prefix}")
    private String prefix;


    /**
     * 获取参数配置列表
@@ -32,4 +54,88 @@
            return Result.error("注册失败，请稍后再试！");
        }
    }


    @PostMapping("/upload")
    public Result uploadFile(@RequestParam("file") MultipartFile file,@RequestParam("accountId") String accountId) {
        try {

            // 1. 验证文件类型
            String contentType = file.getContentType();
            if (contentType == null ||
                    (!contentType.equals("image/jpeg") &&
                            !contentType.equals("image/png") &&
                            !contentType.equals("image/gif"))) {

                return Result.error("只支持JPEG、PNG和GIF格式的图片!");
            }

            // 确保上传目录存在
            File directory = new File(uploadDir);
            if (!directory.exists()) {
                directory.mkdirs();
                // 设置目录权限为755 (rwxr-xr-x)
                setPermissions(directory, "rwxr-xr-x");
            }

            // 生成唯一文件名
            String fileName = UUID.randomUUID().toString() + "_" + file.getOriginalFilename();
            Path filePath = Paths.get(uploadDir, fileName);

            // 保存文件
            Files.write(filePath, file.getBytes());

            // 设置文件权限为644 (rw-r--r--)
            setPermissions(filePath.toFile(), "rw-r--r--");

            // 5. 调用网易云信API更新头像
            fileName = prefix+"/"+fileName;
            UpdateUserBusinessDto dto = new UpdateUserBusinessDto();
            dto.setAvatar(fileName);
            Map<String, Object> result = imApiServcie.updateUserAvatar(accountId,dto);
            if ((Boolean) result.get("success")) {
                return Result.success("文件上传成功");
            } else {
                return Result.error("上传失败！");
            }
        } catch (IOException e) {
            return Result.error("文件上传失败");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("设置文件权限失败！");
        }
        return Result.success();
    }

    /**
     * 设置文件/目录权限
     * @param file 文件或目录对象
     * @param perm 权限字符串，如 "rw-r--r--"
     */
    private void setPermissions(File file, String perm) throws IOException {
        // 获取当前操作系统
        String os = System.getProperty("os.name").toLowerCase();

        // 只有在Unix/Linux系统上才能设置POSIX权限
        if (os.contains("nix") || os.contains("nux") || os.contains("mac")) {
            Set<PosixFilePermission> permissions = PosixFilePermissions.fromString(perm);
            Files.setPosixFilePermissions(file.toPath(), permissions);
        } else {
            // Windows系统不支持POSIX权限，可以设置基本权限
            file.setReadable(true, false);    // 所有用户可读
            if (perm.startsWith("rw")) {
                file.setWritable(true, false); // 所有用户可写（如果权限字符串以rw开头）
            } else {
                file.setWritable(false, false);
            }

            // 如果是目录且包含执行权限，设置可执行
            if (file.isDirectory() && perm.contains("x")) {
                file.setExecutable(true, false);
            }
        }
    }



}