package cn.xcy.demo.controller;

import cn.hutool.core.io.FileUtil;
import cn.xcy.demo.FlieUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;

@Slf4j
@RestController
@RequestMapping("/file")
public class FileUploadController {
    /**
     *  上傳文件
     *  1、判斷文件是否為空
     *  2、獲取文件名、文件類(lèi)型、文件大小(可以限定文件類(lèi)型、以及文件大小)
     *  3、設(shè)置文件上傳路徑
     *  4、以流的方式將文件輸出
     *  5、返回上傳成功
     * @param file
     * @return
     * @throws Exception
     */
    @PostMapping("/upload")
    public String uploadImage(@RequestParam("file") MultipartFile file) throws Exception{
        // 判斷文件是否為空
        if (file.isEmpty()) {
            return "上傳失敗";
        }
        //文件類(lèi)型為
        String contentType = file.getContentType();
        log.info("文件類(lèi)型為：" + contentType);
        if (contentType.equals("image/jpeg")){
            return "請(qǐng)上傳正確格式的文件";
        }
        // 獲取原始文件名
        String fileName = file.getOriginalFilename();
        log.info("上傳的文件名為：" + fileName);
        try {
            // 將文件保存到指定位置
            byte[] bytes = file.getBytes();
            log.info("文件大小為：" + bytes.length);
            // TODO: 根據(jù)業(yè)務(wù)需求選擇合適的存儲(chǔ)方式，比如保存到本地磁盤(pán)或云存儲(chǔ)
//            獲取生成后的文件名
            String s = FlieUtils.generateRandomFileName(fileName);
//            將文件保存到指定位置
            FlieUtils.upload(file,"C:\\Users\\liguoming\\Desktop",s);
            return "上傳成功";
        } catch (Exception e) {
            e.printStackTrace();
            return "上傳失敗";
        }
    }

}

package cn.xcy.demo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.util.UUID;

@Slf4j
public class FlieUtils {
    /**
     * @param file     文件
     * @param fileName 新的隨機(jī)文件名
     */
    public static void upload(MultipartFile file, String destPath, String fileName) {
        File dest = new File(destPath + File.separator + fileName);
        //判斷文件父目錄是否存在
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        try {
            //保存文件
            file.transferTo(dest);
        } catch (Exception e) {
            log.info("Save file exception. {}", e.getMessage());
        }

    }

    /**
     * 輸出文件的后綴名
     * @param fileName
     * @return
     */
    public static String getSuffix(String fileName) {
//        fileName.substring(fileName.lastIndexOf("."))
//        是C#中的字符串操作，它的作用是從文件名字符串中截取最后一個(gè)點(diǎn)（.）及其后面的所有字符。這個(gè)操作可以用來(lái)獲取文件的擴(kuò)展名。
        return fileName.substring(fileName.lastIndexOf("."));
    }

    /**
     * 輸出文件的新名稱(chēng) (名字+后綴名)
     * @param fileName
     * @return
     */
    public static String generateRandomFileName(String fileName) {
        return UUID.randomUUID() + getSuffix(fileName);
    }

    /**
     * @param name
     * @Description 設(shè)置響應(yīng)頭部信息
     * @Throws
     * @Return java.lang.String
     * @Date 2023-08-02 13:39:15
     * @Author lgm
     */
    public static String fileContentType(String name) {
        String result = "";
        String fileType = name.toLowerCase();
        if (fileType.endsWith(".png")) {
            result = "image/png";
        } else if (fileType.endsWith(".gif")) {
            result = "image/gif";
        } else if (fileType.endsWith(".jpg") || fileType.endsWith(".jpeg")) {
            result = "image/jpeg";
        } else if (fileType.endsWith(".svg")) {
            result = "image/svg+xml";
        } else if (fileType.endsWith(".doc")) {
            result = "application/msword";
        } else if (fileType.endsWith(".xls")) {
            result = "application/x-excel";
        } else if (fileType.endsWith(".zip")) {
            result = "application/zip";
        } else if (fileType.endsWith(".pdf")) {
            result = "application/pdf";
        } else if (fileType.endsWith(".mpeg")) { //MP3
            result = "audio/mpeg";
        } else if (fileType.endsWith(".mp4")) {
            result = "video/mp4";
        } else if (fileType.endsWith(".plain")) {
            result = "text/plain";
        } else if (fileType.endsWith(".html")) {
            result = "text/html";
        } else if (fileType.endsWith(".json")) {
            result = "application/json";
        } else{
            result = "application/octet-stream";
        }
        return result;
    }
}

/**
     *  下載文件
     *  1、需要傳入文件名稱(chēng)
     *  2、先將文件的存放路徑+名稱(chēng)獲取到
     *  3、設(shè)置響應(yīng)頭，告訴瀏覽器需要下載文件夾
     *  4、設(shè)置響應(yīng)體，將文件讀取出來(lái)，寫(xiě)入到響應(yīng)體中
     *  5、response.getOutputStream()  輸出字節(jié)流， 客戶(hù)端會(huì)下載字節(jié)流并生成文件保存本地
     * @param filename
     * @param response
     * @return
     * @throws IOException
     */
    @RequestMapping("/fileload")
    public String fileload(@RequestParam("filename")String filename, HttpServletResponse response) throws IOException {
        // 文件的存放位置
        String filePath = "C:\\Users\\liguoming\\Desktop\\" + filename;
        File file = new File(filePath);
        // 設(shè)置響應(yīng)頭，告訴瀏覽器要下載文件
        // response.setContentType()是設(shè)置響應(yīng)類(lèi)型的 瀏覽器讀取的是什么文件 依據(jù)文件類(lèi)型，告訴瀏覽器接下來(lái)要做的是什么
        // application/octet-stream  告訴瀏覽器需要下載二進(jìn)制流數(shù)據(jù)（如常見(jiàn)的文件下載）
        response.setContentType("application/octet-stream");
        /**
         * response.setHeader("Content-Disposition", "attachment; filename=" +new String( filename.getBytes("gb2312"), "ISO8859-1" ));
         * 在確保附件文件名都是簡(jiǎn)體中文字的情況下，那么這個(gè)辦法確實(shí)是最有效的，不用讓客戶(hù)逐個(gè)的升級(jí)IE。
         * 如果臺(tái)灣同胞用，把gb2312改成big5就行。但現(xiàn)在的系統(tǒng)通常都加入了 國(guó)際化的支持，普遍使用UTF-8。
         * 如果文件名中又有簡(jiǎn)體中文字，又有繁體中文，還有日文。那么亂碼便產(chǎn)生了。另外，在上Firefox (v1.0-en)下載也是亂碼。
         */
        response.setHeader("Content-Disposition", "attachment; filename=" +new String( filename.getBytes("gb2312"), "ISO8859-1" ));
        //response.getOutputStream()  輸出字節(jié)流， 客戶(hù)端會(huì)下載字節(jié)流并生成文件保存本地
        ServletOutputStream outputStream = response.getOutputStream();
        outputStream.write(FileUtil.readBytes(file));
        outputStream.close();
        return "success";
    }