package com.mouday.demo.controller;

import org.springframework.stereotype.Controller;
import org.springframework.util.ResourceUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 文件上传
 */
@Controller
public class FileController {
    // 获取上传测试页面
    @GetMapping("/upload")
    public String upload() {
        return "upload";
    }

    // 单文件上传
    @PostMapping("/upload")
    @ResponseBody
    public String upload(MultipartFile file) {
        if (file.isEmpty()) {
            return "file is Empty";
        }

        String filename = file.getOriginalFilename();

        try {
            File path = this.getUploadDirectory();
            File dest = new File(path, filename);
            System.out.println(dest.getPath());
            file.transferTo(dest);

            return new File("/upload", filename).toString();

        } catch (IOException e) {
            e.printStackTrace();
            return "error";
        }
    }

    // 多文件上传
    @PostMapping("/multiUpload")
    @ResponseBody
    public List<String> upload(@RequestParam("file") List<MultipartFile> files) {
        List<String> list = new ArrayList<>();

        for (MultipartFile file : files) {
            list.add(this.upload(file));
        }
        return list;
    }


    /**
     * 获取文件保存路径
     * 参考：https:///A/GBJrE67Wz0/
     *
     * @return
     * @throws FileNotFoundException
     */
    public File getUploadDirectory() throws FileNotFoundException {
        String pathName = ResourceUtils.getURL("classpath:").getPath();
        File path = new File(pathName, "/public/upload");

        if (!path.exists()) {
            path.mkdirs();
        }

        return path;
    }

}