package com.lf.server.controller.all;

import com.lf.server.aspect.SysLog;
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.helper.FileHelper;
import com.lf.server.service.all.FileService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.net.URLEncoder;
import java.util.Iterator;

/**
 * 文件控制器
 * @author WWW
 */
@Api(tags = "运维管理\\文件管理")
@RestController
@RequestMapping("/file")
public class FileController extends BaseController {
    @Autowired
    private FileService fileService;

    @SysLog()
    @ApiOperation(value = "文件上传")
    @PostMapping(value = "/upload")
    public ResponseMsg<String> upload(@RequestBody MultipartFile[] files, String tab, HttpServletRequest req, HttpServletResponse res) {
        try {
            MultipartHttpServletRequest mreq = (MultipartHttpServletRequest) req;
            Iterator<String> fileNames = mreq.getFileNames();

            // 获取原文件名
            // noinspection AlibabaLowerCamelCaseVariableNaming
            String OriginalFileName = files[0].getOriginalFilename();
            // 获取文件后缀名
            String suffixName = OriginalFileName.substring(OriginalFileName.lastIndexOf("."));

            String fileName = null;
            MultipartFile file = null;
            while (fileNames.hasNext()) {
                fileName = (String) fileNames.next();
                file = mreq.getFile(fileName);
            }

            // kb
            long size = file.getSize();
            file.transferTo(new File(""));

            return success("");
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }
    }

    @SysLog()
    @ApiOperation(value = "下载文件")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "guid", value = "文件GUID", dataType = "String", paramType = "query", example = "")
    })
    @GetMapping("/download")
    public void download(String guid, HttpServletRequest req, HttpServletResponse res) {
        try {
            String filePath = "";

            // 创建文件对象
            File file = new File(filePath);
            if (!file.exists()) {
                return;
            }

            String fileName = URLEncoder.encode("学生名册.pdf", "UTF-8");

            // 设置响应头中文件的下载方式为附件方式，以及设置文件名
            res.setHeader("Content-Disposition", "attachment; filename=" + fileName);

            // 定义输入流，通过输入流读取文件内容
            FileInputStream fileInputStream = new FileInputStream(file);

            // 通过response对象，获取到输出流
            ServletOutputStream outputStream = res.getOutputStream();

            // 设置响应头的编码格式为UTF-8
            res.setCharacterEncoding("UTF-8");

            // 通过response对象设置响应数据格式("text/plain; charset=utf-8")
            String ext = FileHelper.getExtension(file);
            String mime = FileHelper.getMime(ext);
            res.setContentType(mime);

            // 记录下载记录...

            int len = 0;
            byte[] bytes = new byte[1024];
            while ((len = fileInputStream.read(bytes)) != -1) {
                // 通过输入流读取文件数据，然后通过上述的输出流写回浏览器
                outputStream.write(bytes, 0, len);
                // 刷新
                outputStream.flush();
            }

            // 关闭资源
            outputStream.close();
            fileInputStream.close();
        } catch (Exception ex) {
            log.error(ex.getMessage() + ex.getStackTrace() + "\n");
        }
    }
}