package org.sxkj.resource.feign;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.oss.model.OssFile;
import org.springblade.core.tenant.annotation.NonDS;
import org.springblade.core.tool.utils.FileUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.scheduling.annotation.Async;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.sxkj.common.utils.HeaderUtils;
import org.sxkj.common.utils.ffmpeg.FfmpegConverterUtils;
import org.sxkj.resource.builder.OssBuilder;
import org.sxkj.resource.entity.Attach;
import org.sxkj.resource.model.MinioPojo;
import org.sxkj.resource.service.IAttachService;
import org.sxkj.resource.vo.AttachTypeStatisticsVO;
import org.sxkj.resource.vo.AttachVO;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.*;

/**
 * 附件远程调用服务
 *
 * @author Chill
 */
@NonDS
@RestController
@AllArgsConstructor
@Slf4j
public class AttachClient implements IAttachClient {

	@Autowired
	private IAttachService attachService;

	@Autowired
	private OssBuilder ossBuilder;

	@Autowired
	private MinioPojo pojo;

	/**
	 * 保存附件表信息
	 *
	 * @param attach 附件表信息
	 * @return
	 */
	@Override
	public Boolean saveAttachInfo(Attach attach) {
		return attachService.saveAttachInfo(attach);
	}

	/**
	 * 获取任务下的图片信息
	 * @param jobs
	 * @return
	 */
	public List<AttachVO> getAttachData(@RequestParam String jobs) {
		AttachVO attach = new AttachVO();
		attach.setWayLineJobId(jobs);
		attach.setOrderByCreateTime(true);
		attach.setAreaCode(HeaderUtils.getAreaCode());

		List<String> integers = Arrays.asList("0", "2", "4", "5");
		attach.setResultTypes(integers);
		List<String> resultTypes = attach.getResultTypes();
		String wordOrderType = attach.getWordOrderType();
		settingResultType(attach, wordOrderType, resultTypes);

		if (Objects.nonNull(attach.getAiStatus()) && Objects.nonNull(attach.getResultTypes())
			&& attach.getResultTypes().contains(Attach.RESULT_TYPE_AI)) {
			attach.setResultTypes(Arrays.asList(Attach.RESULT_TYPE_AI));
		}

		List<String> aiEventMd5s = attachService.findAiEventMd5s(attach.getEventRecordIds());
		attach.setMd5s(aiEventMd5s);

		if (!CollectionUtils.isEmpty(attach.getEventRecordIds())) {
			attach.setResultTypes(Arrays.asList(Attach.RESULT_TYPE_AI, Attach.RESULT_TYPE_IMG));
		}

		// 使用大分页获取所有数据
		Page<AttachVO> page = new Page<>(1, 2000); // 或者 Integer.MAX_VALUE
		List<AttachVO> aiAttachImages = attachService.findAiAttachImages(page, attach);

		return aiAttachImages;
	}

	/**
	 * 设置结果类型
	 * @param attach
	 * @param wordOrderType
	 * @param resultTypes
	 */
	private static void settingResultType(AttachVO attach, String wordOrderType, List<String> resultTypes) {
		if (!StringUtils.isEmpty(wordOrderType) && !CollectionUtils.isEmpty(resultTypes)) {
			if ("-1".equals(wordOrderType)) {
				attach.setWordOrderType(null);
				if (resultTypes.contains(Attach.RESULT_TYPE_IMG)) {
					resultTypes = Arrays.asList(Attach.RESULT_TYPE_IMG);
				}
			} else {
				if (resultTypes.contains(Attach.RESULT_TYPE_AI)) {
					resultTypes = Arrays.asList(Attach.RESULT_TYPE_AI);
				}
			}
			attach.setResultTypes(resultTypes);
		}
	}
	@Override
	public BladeFile saveAttachFile(MultipartFile file, String fileName) throws IOException {
		BladeFile bladeFile = ossBuilder.template().putFile(pojo.getBucket(),fileName, file.getInputStream());
		Long attachId = buildAttach(fileName, file.getSize(), bladeFile, null);
		bladeFile.setAttachId(attachId);
		return bladeFile;
	}

	@Override
	public BladeFile saveAttachFile(MultipartFile file, String fileName, String type) throws IOException {
		BladeFile bladeFile = ossBuilder.template().putFile(pojo.getBucket(),fileName, file.getInputStream());
		Long attachId = buildAttach(fileName, file.getSize(), bladeFile, type);
		bladeFile.setAttachId(attachId);
		return bladeFile;
	}

	/**
	 * 附件上传
	 * @param file 文件
	 * @param fileName 文件名称
	 * @return
	 * @throws IOException
	 */
	@Override
	public BladeFile putFile(MultipartFile file, String fileName) throws IOException {
		BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
		return bladeFile;
	}

	private Long buildAttach(String fileName, Long fileSize, BladeFile bladeFile, String type) {
		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);
		if (null != type) {
			attach.setResultType(type);
		}
		attachService.save(attach);
		return attach.getId();
	}

	/**
	 * 根据附件ID集合获取去除 cloud-bucket 前缀后的 name 列表
	 *
	 * @param ids 附件ID集合
	 * @return 处理后的 name 列表
	 */
	@Override
	public List<String> getAttachNames(List<Long> ids) {
		return attachService.getAttachNames(ids);
	}

	@Override
	public Map<String, Object> findMetaDataByName(List<String> names) {
		return attachService.findMetaDataByName(names);
	}


	/**
	 * 查询成果数量
	 * @param jobId 任务id
	 * @return 成果数量
	 */
	@Override
	public Long findResultNumByJobId(@RequestParam  String jobId){
		return attachService.findResultNumByJobId(jobId);
	}

	/**
	 * 通过md5查询附件信息
	 * @param md5
	 * @return
	 */
	@Override
	public Attach getAttachByMd5(@RequestParam String md5) {
		List<Attach> list = attachService.list(new LambdaQueryWrapper<Attach>().eq(Attach::getMd5, md5));
		return (list != null && !list.isEmpty()) ? list.get(0) : null;
	}

	@Override
	public Attach getAttachById(String id) {
		List<Attach> list = attachService.list(new LambdaQueryWrapper<Attach>().eq(Attach::getId, id));
		return (list != null && !list.isEmpty()) ? list.get(0) : null;
	}

	@SneakyThrows
	@Override
	@Async
	public void processingYesterdayVideo(Date startTime, Date endTime) {
		// 查询昨日视频
		List<Attach> list = attachService.findAttachInfoByYesterday(1, startTime, endTime);
		log.info("昨日视频数量：" + list.size());
//
		for (int i = 0; i < list.size(); i++) {
			log.info("正在处理第" + (i + 1) + "个视频...");
			Attach attach = list.get(i);
			// 获取视频流
			OssFile ossFile = ossBuilder.template("minio").statFile(pojo.getBucket(), attach.getName().replace(pojo.getBucket() + "/", ""));

			String minioPath = pojo.getServerAddress();
			String name = addSmallSuffix(attach.getName(), "_show");
			String path = ossFile.getName().replace(pojo.getBucket() + "/", "");

			String mp4TempPath = minioPath + "/temp/" + UUID.randomUUID().toString().toLowerCase();
			// 开始压缩视频
			log.info("原始视频地址：" + minioPath +  path);
			File mediaFile = FfmpegConverterUtils.compressVideo(new File(minioPath +  path), 5, mp4TempPath);
			log.info("第" + (i + 1) + "个视频处理完成...{}", mediaFile.getName());

			saveAttachFile(new MockMultipartFile( mediaFile.getName(),
				mediaFile.getName(),
				"application/octet-stream",
				new FileInputStream(mediaFile)), name, Attach.RESULT_TYPE_VEDIO_SHOW);
			log.info("第{}个视频保存入库完成...", (i + 1));

		}
		log.debug("昨日视频处理完成...");
	}

	private static String addSmallSuffix(String fileName, String suffix) {
		int lastDotIndex = fileName.lastIndexOf(".");
		if (lastDotIndex != -1) {
			// 将文件名分成名称和扩展名两部分
			String name = fileName.substring(0, lastDotIndex);
			String extension = fileName.substring(lastDotIndex);
			// 在名称后面加上_small
			return name + suffix + extension;
		}
		// 如果没有扩展名，直接在末尾加上_small
		return fileName + suffix;
	}

	/**
	 *
	 * @param deviceSn 设备sn
	 * @param timestamp 时间戳
	 * @param operator 操作人
	 * @param workspaceId 工作空间id
	 * @return
	 */
	@Override
	public Boolean deleteAttach(String deviceSn, long timestamp, Long operator, String workspaceId) {
		return attachService.deleteAttach(deviceSn, timestamp, operator, workspaceId);
 	}

	/**
	 *
	 * @param deviceSn
	 * @param startDate
	 * @param endDate
	 * @param resultTypes
	 * @param areaCode
	 * @return
	 */
	@Override
	public List<AttachTypeStatisticsVO> calculateTheThreeDimensionalArea(String deviceSn, String startDate, String endDate, String resultTypes, String areaCode) {
		List<AttachTypeStatisticsVO> attachTypeStatisticsVO = attachService.calculateTheThreeDimensionalArea(deviceSn, startDate, endDate, resultTypes, areaCode);
		return attachTypeStatisticsVO;
	}
}