/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package cn.gistack.resource.endpoint;

import cn.gistack.resource.builder.oss.OssBuilder;
import cn.gistack.resource.entity.TextualResearchAttach;
import cn.gistack.resource.panoramaUtils.Img4JavaUtil;
import cn.gistack.resource.panoramaUtils.QietuResult;
import cn.gistack.resource.service.IAttachService;
import cn.gistack.resource.service.TextualResearchAttachService;
import io.swagger.annotations.Api;
import lombok.AllArgsConstructor;
import lombok.SneakyThrows;
import org.apache.commons.io.FileUtils;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.oss.model.OssFile;
import org.springblade.core.secure.annotation.PreAuth;
import org.springblade.core.tenant.annotation.NonDS;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.constant.RoleConstant;
import org.springblade.core.tool.utils.FileUtil;
import org.springblade.core.tool.utils.Func;
import cn.gistack.resource.entity.Attach;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileInputStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.*;

/**
 * 对象存储端点
 *
 * @author Chill
 */
@NonDS
@RestController
@AllArgsConstructor
@RequestMapping("/oss/endpoint")
@Api(value = "对象存储端点", tags = "对象存储端点")
public class OssEndpoint {

	/**
	 * 对象存储构建类
	 */
	private final OssBuilder ossBuilder;

	/**
	 * 附件表服务
	 */
	private final IAttachService attachService;

	/**
	 * 附件表服务
	 */
	private final TextualResearchAttachService textualResearchAttachService;

	/**
	 * 创建存储桶
	 *
	 * @param bucketName 存储桶名称
	 * @return Bucket
	 */
	@SneakyThrows
	@PostMapping("/make-bucket")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R makeBucket(@RequestParam String bucketName) {
		ossBuilder.template().makeBucket(bucketName);
		return R.success("创建成功");
	}

	/**
	 * 创建存储桶
	 *
	 * @param bucketName 存储桶名称
	 * @return R
	 */
	@SneakyThrows
	@PostMapping("/remove-bucket")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R removeBucket(@RequestParam String bucketName) {
		ossBuilder.template().removeBucket(bucketName);
		return R.success("删除成功");
	}

	/**
	 * 拷贝文件
	 *
	 * @param fileName       存储桶对象名称
	 * @param destBucketName 目标存储桶名称
	 * @param destFileName   目标存储桶对象名称
	 * @return R
	 */
	@SneakyThrows
	@PostMapping("/copy-file")
	public R copyFile(@RequestParam String fileName, @RequestParam String destBucketName, String destFileName) {
		ossBuilder.template().copyFile(fileName, destBucketName, destFileName);
		return R.success("操作成功");
	}

	/**
	 * 获取文件信息
	 *
	 * @param fileName 存储桶对象名称
	 * @return InputStream
	 */
	@SneakyThrows
	@GetMapping("/stat-file")
	public R<OssFile> statFile(@RequestParam String fileName) {
		return R.data(ossBuilder.template().statFile(fileName));
	}

	/**
	 * 获取文件相对路径
	 *
	 * @param fileName 存储桶对象名称
	 * @return String
	 */
	@SneakyThrows
	@GetMapping("/file-path")
	public R<String> filePath(@RequestParam String fileName) {
		return R.data(ossBuilder.template().filePath(fileName));
	}


	/**
	 * 获取文件外链
	 *
	 * @param fileName 存储桶对象名称
	 * @return String
	 */
	@SneakyThrows
	@GetMapping("/file-link")
	public R<String> fileLink(@RequestParam String fileName) {
		return R.data(ossBuilder.template().fileLink(fileName));
	}

	/**
	 * 上传文件
	 *
	 * @param file 文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/put-file")
	public R<BladeFile> putFile(@RequestParam MultipartFile file) {
		BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
		return R.data(bladeFile);
	}

	/**
	 * 自定义前缀上传文件
	 *
	 * @param file 文件
	 * @param prefixPath 文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/put-file-by-prefix-path")
	public R<BladeFile> putFileByPrefixPath(@RequestParam MultipartFile file,@RequestParam String prefixPath) {
		BladeFile bladeFile = ossBuilder.templateByPrefixPath(prefixPath).putFile(file.getOriginalFilename(), file.getInputStream());
		return R.data(bladeFile);
	}

	/**
	 * 批量自定义前缀上传文件
	 *
	 * @param files 文件
	 * @param prefixPath 文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/batch-put-file-by-prefix-path")
	public R batchPutFileByPrefixPath(@RequestParam MultipartFile[] files,@RequestParam String prefixPath) {
		List<BladeFile> list = new ArrayList<>();
		for (MultipartFile file : files) {
			BladeFile bladeFile = ossBuilder.templateByPrefixPath(prefixPath).putFile(file.getOriginalFilename(), file.getInputStream());
			list.add(bladeFile);
		}
		return R.data(list);
	}

	/**
	 * 考证附件上传并保存
	 * @param file 文件
	 * @param resGuid 水库编号
	 * @return
	 */
	@SneakyThrows
	@PostMapping("/put-file-by-textualResearchAttach")
	public R putFileByPrefixTextualResearchAttach(@RequestParam MultipartFile file,String resGuid,Integer type) {
		BladeFile bladeFile = ossBuilder.templateByPrefixPath("textualResearchAttach").putFile(file.getOriginalFilename(), file.getInputStream());
		buildTextualResearchAttach(resGuid,bladeFile,type);
		return R.data(bladeFile);
	}

	/**
	 * 全景图片上传，并切图
	 * @param multipartFile 文件
	 * @param resGuid 水库编号
	 * @return
	 */
	@SneakyThrows
	@PostMapping("/put-file-by-panorama")
	public R putFileByPanorama(@RequestParam MultipartFile multipartFile, String resGuid) {

		// 临时文件路径
		String  linShiBaseUrl = "/data/panorama/qietu/";
//		String fileName = multipartFile.getOriginalFilename();
		// 获取文件名
		String picName = multipartFile.getOriginalFilename();// 原始全景图名称
		String dirPath = linShiBaseUrl + resGuid + "/"; //临时文件夹路径

		File dir = new File(dirPath);
		if (dir.isDirectory()) {
			// 清除该目录下的文件及子目录文件而不删除该目录文件夹。该目录不存在会报错
			FileUtils.cleanDirectory(dir);
		} else {
			dir.mkdirs();
		}
		String picUrl = dirPath + picName; // 原始全景图文件路径
		File file = new File(picUrl); // 文件上传的原始全景图，存入临时路径
		multipartFile.transferTo(file);
		FileInputStream ysfileInputStream = new FileInputStream(file);
		BladeFile qjbladeFile = ossBuilder.templateByPrefixPathPanorama("panorama", resGuid).putFile(picName, ysfileInputStream);
		String yuanshiQJurl = qjbladeFile.getLink();
		ysfileInputStream.close();

		/**
		 * 这里开始切图
		 */
		HashMap<String,String> qietuHashMap = new HashMap<String,String>();
		int col = 8;
		int row = 4;
		String[] split = picName.split("\\.");
		String picHeadName = split[0];
		String picLastName = split[1];

		// 获取原始全景图图片信息
		HashMap<String, Integer> imgInfoRT = Img4JavaUtil.getImgInfoRT(file);
		Integer width = imgInfoRT.get("width");
		Integer height = imgInfoRT.get("height");
		// 计算切成小图的长和宽
		Integer biliWidth = width/col;
		Integer biliHeight = height/row;

		for (int i=0 ; i < col ; i++) {
			// 计算图片的横坐标（长边）
			int x = i * biliWidth;
			for (int j=0 ; j<row ; j++){
				// 计算图片的纵坐标（宽边）
				int  y = j * biliHeight;
				// 切图名称
				String outpicName = picHeadName + "_" + i + "_" +j + "." + picLastName;
				// 切图后临时图片的的路径
				String lastiUrl = dirPath + outpicName;
				Img4JavaUtil.cropImg(picUrl ,lastiUrl ,biliWidth,biliHeight,x,y);// 切图存放进临时图片中
				qietuHashMap.put(outpicName,lastiUrl);
			}
		}

		// 切图完成后删除原始全景图
		if (file.exists()) {
			// 删除指定文件，不存在报异常
			FileUtils.forceDelete(file);
		}
		/************************************切图结束********************************************/


		List<String> qieTuUrlList = new ArrayList<String>();
		String urlHead = null;
		Set<String> mapKey = qietuHashMap.keySet();
		if(qietuHashMap != null && qietuHashMap.size()>0) {
			for (String picQName : mapKey) {
				String url = qietuHashMap.get(picQName);
				Path qietuPath = Paths.get(url);
				File qietufile = qietuPath.toFile();
				FileInputStream fileInputStream = new FileInputStream(qietufile);
				BladeFile bladeFile = ossBuilder.templateByPrefixPathPanorama("panorama", resGuid).putFile(picQName, fileInputStream);
				fileInputStream.close(); // 关闭文件流

				// 切图上传完成后删除
				if (qietufile.exists()) {
					// 删除指定文件，不存在报异常
					FileUtils.forceDelete(qietufile);
				}
				String link = bladeFile.getLink();
				qieTuUrlList.add(link);// 上传后的图片路径
				urlHead = link.replace(picQName,"");
			}
		}
		QietuResult qietuResult = new QietuResult();
		qietuResult.setQieTuUrlList(qieTuUrlList);
		qietuResult.setUrlHead(urlHead);
		qietuResult.setRow(row);
		qietuResult.setCol(col);
		qietuResult.setQuanJingWidth(width);
		qietuResult.setQuanJingHeight(height);
		qietuResult.setQjName(picName);
		qietuResult.setQjUrl(yuanshiQJurl);


		return R.data(qietuResult) ;
	}

	@SneakyThrows
	@PostMapping("/put-file-by-panorama2")
	public R putFileByPanorama2(@RequestParam MultipartFile multipartFile, String resGuid){

		return null;
	}


	/**
	 * 保存考证附件上传数据
	 * @param bladeFile
	 */
	private void buildTextualResearchAttach(String resGuid,BladeFile bladeFile,Integer type) {
		TextualResearchAttach attach = new TextualResearchAttach();
		attach.setResGuid(resGuid);
		attach.setType(type);
		attach.setPath(bladeFile.getName());
		attach.setName(bladeFile.getOriginalName());
		attach.setCreateTime(new Date());
		// 保存
		textualResearchAttachService.save(attach);
	}

	/**
	 * 上传文件
	 *
	 * @param fileName 存储桶对象名称
	 * @param file     文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/put-file-by-name")
	public R<BladeFile> putFile(@RequestParam String fileName, @RequestParam MultipartFile file) {
		BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
		return R.data(bladeFile);
	}

	/**
	 * 上传文件并保存至附件表
	 *
	 * @param file 文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/put-file-attach")
	public R<BladeFile> putFileAttach(@RequestParam MultipartFile file) {
		String fileName = file.getOriginalFilename();
		BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
		Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
		bladeFile.setAttachId(attachId);
		return R.data(bladeFile);
	}



	/**
	 * 上传文件并保存至附件表
	 *
	 * @param fileName 存储桶对象名称
	 * @param file     文件
	 * @return ObjectStat
	 */
	@SneakyThrows
	@PostMapping("/put-file-attach-by-name")
	public R<BladeFile> putFileAttach(@RequestParam String fileName, @RequestParam MultipartFile file) {
		BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
		Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
		bladeFile.setAttachId(attachId);
		return R.data(bladeFile);
	}

	/**
	 * 构建附件表
	 *
	 * @param fileName  文件名
	 * @param fileSize  文件大小
	 * @param bladeFile 对象存储文件
	 * @return attachId
	 */
	private Long buildAttach(String fileName, Long fileSize, BladeFile bladeFile) {
		String fileExtension = FileUtil.getFileExtension(fileName);
		Attach attach = new Attach();
		attach.setDomainUrl(bladeFile.getDomain());
		attach.setLink(bladeFile.getLink());
		attach.setName(bladeFile.getName());
		attach.setOriginalName(bladeFile.getOriginalName());
		attach.setAttachSize(fileSize);
		attach.setExtension(fileExtension);
		attachService.save(attach);
		return attach.getId();
	}

	/**
	 * 删除文件
	 *
	 * @param fileName 存储桶对象名称
	 * @return R
	 */
	@SneakyThrows
	@PostMapping("/remove-file")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R removeFile(@RequestParam String fileName) {
		ossBuilder.template().removeFile(fileName);
		return R.success("操作成功");
	}

	/**
	 * 批量删除文件
	 *
	 * @param fileNames 存储桶对象名称集合
	 * @return R
	 */
	@SneakyThrows
	@PostMapping("/remove-files")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R removeFiles(@RequestParam String fileNames) {
		ossBuilder.template().removeFiles(Func.toStrList(fileNames));
		return R.success("操作成功");
	}

}