package com.skua.common.system.controller;

import com.skua.aop.annotation.CustomExceptionAnno;
import com.skua.common.util.FileUtil;
import com.skua.core.api.vo.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.util.AntPathMatcher;
import org.springframework.util.FileCopyUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.HandlerMapping;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

/**
 * 用户表 前端控制器
 */
@Slf4j
@RestController
@RequestMapping("/sys/common")
public class CommonController {

	@Value(value = "${skua.path.upload}")
	private String uploadpath;

	@Value(value = "${skua.path.chunk}")
	private String chunkPath;

	/**
	 * @Author 政辉
	 * @return
	 */
	@GetMapping("/403")
	public Result<?> noauth()  {
		return Result.error("没有权限，请联系管理员授权");
	}

	@PostMapping(value = "/upload")
	public Result<Object> upload(HttpServletRequest request, HttpServletResponse response) {
		Result<Object> result = new Result<>();
		try {
			String ctxPath = uploadpath;
			String fileName = null;
			String bizPath = "files";
			String nowday = new SimpleDateFormat("yyyyMMdd").format(new Date());
			File file = new File(ctxPath + File.separator + bizPath + File.separator + nowday);
			if (!file.exists()) {
				file.mkdirs();// 创建文件根目录
			}
			MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
			MultipartFile mf = multipartRequest.getFile("file");// 获取上传文件对象
			String orgName = mf.getOriginalFilename();// 获取文件名
			fileName = orgName.substring(0, orgName.lastIndexOf(".")) + "_" + System.currentTimeMillis() + orgName.substring(orgName.indexOf("."));
			String savePath = file.getPath() + File.separator + fileName;
			File savefile = new File(savePath);
			FileCopyUtils.copy(mf.getBytes(), savefile);
			String dbpath = bizPath + File.separator + nowday + File.separator + fileName;
			if (dbpath.contains("\\")) {
				dbpath = dbpath.replace("\\", "/");
			}
			result.setResult(dbpath);
			result.setMessage(dbpath);
			result.setSuccess(true);
		} catch (IOException e) {
			result.setSuccess(false);
			result.setMessage(e.getMessage());
			log.error(e.getMessage(), e);
		}
		return result;
	}

	@PostMapping(value = "/uploadFile")
	public Result<String> uploadFile(HttpServletRequest request, HttpServletResponse response) {
		Result<String> result = new Result<>();
		try {
			String ctxPath = uploadpath;
			String fileName = null;
			String bizPath = "files";
			String nowday = new SimpleDateFormat("yyyyMMdd").format(new Date());
			File file = new File(ctxPath + File.separator + bizPath + File.separator + nowday);
			if (!file.exists()) {
				file.mkdirs();// 创建文件根目录
			}
			MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
			MultipartFile mf = multipartRequest.getFile("file");// 获取上传文件对象
			String orgName = mf.getOriginalFilename();// 获取文件名
			fileName = orgName.substring(0, orgName.lastIndexOf(".")) + "_" + System.currentTimeMillis() + orgName.substring(orgName.indexOf("."));
			String savePath = file.getPath() + File.separator + fileName;
			File savefile = new File(savePath);
			FileCopyUtils.copy(mf.getBytes(), savefile);
			String dbpath = bizPath + File.separator + nowday + File.separator + fileName;
			if (dbpath.contains("\\")) {
				dbpath = dbpath.replace("\\", "/");
			}
			result.setResult(dbpath);
			result.setSuccess(true);
		} catch (IOException e) {
			result.setSuccess(false);
			result.setMessage(e.getMessage());
			log.error(e.getMessage(), e);
		}
		return result;
	}

	/**
	 * 预览图片
	 * 请求地址：http://localhost:8080/common/view/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
	 *
	 * @param request
	 * @param response
	 */
	@GetMapping(value = "/view/**")
	public void view(HttpServletRequest request, HttpServletResponse response) {
		// ISO-8859-1 ==> UTF-8 进行编码转换
		String imgPath = extractPathFromPattern(request);
		// 其余处理略
		InputStream inputStream = null;
		OutputStream outputStream = null;
		try {
			imgPath = imgPath.replace("..", "");
			if (imgPath.endsWith(",")) {
				imgPath = imgPath.substring(0, imgPath.length() - 1);
			}
			response.setContentType("image/jpeg;charset=utf-8");
			String localPath = uploadpath;
			String imgurl = localPath + File.separator + imgPath;
			inputStream = new BufferedInputStream(new FileInputStream(imgurl));
			outputStream = response.getOutputStream();
			byte[] buf = new byte[1024];
			int len;
			while ((len = inputStream.read(buf)) > 0) {
				outputStream.write(buf, 0, len);
			}
			response.flushBuffer();
		} catch (IOException e) {
			log.error("预览图片失败" + e.getMessage());
			// e.printStackTrace();
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
			if (outputStream != null) {
				try {
					outputStream.close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
		}

	}

	/**
	 * 下载文件
	 * 请求地址：http://localhost:8080/common/download/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
	 *
	 * @param request
	 * @param response
	 * @throws Exception
	 */
	@GetMapping(value = "/download/**")
	public void download(HttpServletRequest request, HttpServletResponse response) throws Exception {
		// ISO-8859-1 ==> UTF-8 进行编码转换
		String filePath = extractPathFromPattern(request);
		// 其余处理略
		InputStream inputStream = null;
		OutputStream outputStream = null;
		try {
			filePath = filePath.replace("..", "");
			if (filePath.endsWith(",")) {
				filePath = filePath.substring(0, filePath.length() - 1);
			}
			String localPath = uploadpath;
			String downloadFilePath = localPath + File.separator + filePath;
			File file = new File(downloadFilePath);
	         if (file.exists()) {
	        	 response.setContentType("application/force-download");// 设置强制下载不打开            
	 			response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
				 response.setContentLengthLong(file.length());
	 			inputStream = new BufferedInputStream(new FileInputStream(file));
	 			outputStream = response.getOutputStream();
	 			byte[] buf = new byte[1024];
	 			int len;
	 			while ((len = inputStream.read(buf)) > 0) {
	 				outputStream.write(buf, 0, len);
	 			}
	 			response.flushBuffer();
	         }

		} catch (Exception e) {
			log.info("文件下载失败" + e.getMessage());
			// e.printStackTrace();
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
			if (outputStream != null) {
				try {
					outputStream.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}

	}
	/**
	 * @功能：pdf预览Iframe
	 * @param modelAndView
	 * @return
	 */
	@RequestMapping("/pdf/pdfPreviewIframe")
	public ModelAndView pdfPreviewIframe(ModelAndView modelAndView) {
		modelAndView.setViewName("pdfPreviewIframe");
		return modelAndView;
	}

	/**
	  *  把指定URL后的字符串全部截断当成参数
	  *  这么做是为了防止URL中包含中文或者特殊字符（/等）时，匹配不了的问题
	 * @param request
	 * @return
	 */
	private static String extractPathFromPattern(final HttpServletRequest request) {
		String path = (String) request.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE);
		String bestMatchPattern = (String) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE);
		return new AntPathMatcher().extractPathWithinPattern(bestMatchPattern, path);
	}



	@PostMapping("/shardUpload")
	@CustomExceptionAnno(description = "断点续传-上传分片文件")
	public Result<?> shardUploads(@RequestParam("file") MultipartFile multipartFile,
								  @RequestParam("fileType") String fileType,
								  @RequestParam("md5") String md5,
								  @RequestParam("fileName") String fileName,
								  @RequestParam(name = "currentChunk", defaultValue = "-1") Integer currentChunk) throws Exception {
		Result<?> result = new Result<>();
		String dirPath = chunkPath + File.separator + fileType + File.separator + md5;
		String fileName0 = fileName.split("\\.")[0];
		if (FileUtil.mkDirs(dirPath)) {
			File tmpFolder = new File(dirPath + File.separator + fileName0 + ".chunk");
			if (!tmpFolder.exists()) {
				tmpFolder.mkdirs();
			}
			File chunkFile = new File(dirPath + File.separator + fileName0 + ".chunk" + File.separator + currentChunk);
			if (!chunkFile.exists()) {
				multipartFile.transferTo(chunkFile);
			}
		}
		return result;
	}

	@CustomExceptionAnno(description = "断点续传-根据文件名判断是否存在")
	@GetMapping("/isExist")
	public Result<Object> isExistCtrl(@RequestParam("fileType") String fileType, @RequestParam("md5") String md5, @RequestParam("fileName") String fileName) {
		Result<Object> result = new Result<>();
		File dirFile = new File(chunkPath + File.separator + fileType + File.separator + md5);
		List<Integer> list = new ArrayList<>();
		if (dirFile.exists()) {
			File[] folders = dirFile.listFiles();
			if (folders != null && folders.length != 0) {
				for (File folder : folders) {
					// 判断是否存在文件夹，即已经上传的部分
					if (folder.isDirectory()) {
						File[] chunkFiles = folder.listFiles();
						if (chunkFiles != null && chunkFiles.length != 0) {
							for (File file : chunkFiles) {
								list.add(Integer.parseInt(file.getName()));
							}
						}
					}
				}
			}
			if (list.size() == 0) {
				result.setMessage("文件已上传");
				result.setResult(File.separator + fileType + File.separator + md5 + File.separator + fileName);
			} else {
				result.setResult(list);
				result.setMessage("文件仅上传部分");
			}
		} else {
			result.setCode(201);
			result.setMessage("文件未上传");
		}
		return result;
	}

	@GetMapping("/shardMerge")
	@CustomExceptionAnno(description = "断点续传-分片合并")
	public Result<String> shardMergeCtrl(@RequestParam("fileType") String fileType, @RequestParam("md5") String md5, @RequestParam("fileName") String fileName) {
		Result<String> result = new Result<>();
		String dirPath = chunkPath + File.separator + fileType + File.separator + md5;
		String fileName0 = fileName.split("\\.")[0];
		String chunkFolder = dirPath + File.separator + fileName0 + ".chunk";
		File chunkFolderFile = new File(chunkFolder);
		File[] files = chunkFolderFile.listFiles();
		File mergeFile = new File(dirPath + File.separator + fileName);
		if (files != null) {
			List<File> fileList = Arrays.asList(files);
			FileUtil.shardMerge(fileList, mergeFile);
			FileUtil.delFile(chunkFolderFile);
			result.success("分片合并成功");
		} else {
			result.error500("分片合并失败");
		}
		result.setResult(File.separator + fileType + File.separator + md5 + File.separator + fileName);
		return result;
	}


}