/*
 *    Copyright (c) 2018-2025, ztt 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 pig4cloud.com 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: ztt
 */
package com.chinaztt.mes.quality.service.impl;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.basic.entity.Warehouse;
import com.chinaztt.mes.common.numgen.NumberGenerator;
import com.chinaztt.mes.quality.dto.QualityExceptionDTO;
import com.chinaztt.mes.quality.entity.ExceptionAttachment;
import com.chinaztt.mes.quality.entity.QualityException;
import com.chinaztt.mes.quality.entity.ReportSampleAttachment;
import com.chinaztt.mes.quality.mapper.ExceptionAttachmentMapper;
import com.chinaztt.mes.quality.mapper.QualityExceptionMapper;
import com.chinaztt.mes.quality.service.QualityExceptionService;
import com.chinaztt.ztt.common.core.util.R;
import com.chinaztt.ztt.common.oss.OssProperties;
import com.chinaztt.ztt.common.oss.service.OssTemplate;
import com.sun.org.apache.bcel.internal.generic.IF_ACMPEQ;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.util.List;

/**
 * 生产异常处理
 *
 * @author sunxl
 * @date 2021-07-01 14:48:35
 */
@Slf4j
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class QualityExceptionServiceImpl extends ServiceImpl<QualityExceptionMapper, QualityException> implements QualityExceptionService {
	private NumberGenerator<QualityException> exceptionNumberGenerator;
	private final OssProperties ossProperties;
	private final OssTemplate minioTemplate;
	private ExceptionAttachmentMapper exceptionAttachmentMapper;
	@Override
	public boolean fullSave(QualityException exception) {
		exception.setCode(exceptionNumberGenerator.generateNumberWithPrefix(QualityException.DIGIT, QualityException.PREFIX, QualityException::getCode));
		baseMapper.insert(exception);
		return true;
	}

	@Override
	public IPage<List<QualityExceptionDTO>> getExceptionPage(Page page, QueryWrapper<QualityExceptionDTO> gen) {
		return baseMapper.getExceptionPage(page,gen);
	}

	@Override
	public QualityExceptionDTO getExceptionPageById(Long id) {
		return baseMapper.getExceptionPageById(id);
	}

	@Override
	public R uploadFile(MultipartFile file, Long exceptionId) {
		String fileName = IdUtil.simpleUUID() + StrUtil.DOT + FileUtil.extName(file.getOriginalFilename());
		String url = String.format("/mes/exception/%s/%s", ossProperties.getBucketName(), fileName);
		try {
			minioTemplate.putObject(ossProperties.getBucketName(), fileName, file.getInputStream());

			ExceptionAttachment attachment = new ExceptionAttachment();
			attachment.setOriginal(file.getOriginalFilename());
			attachment.setUrl(url);
			attachment.setFileName(fileName);
			attachment.setExceptionId(exceptionId);
			exceptionAttachmentMapper.insert(attachment);

		} catch (Exception e) {
			log.error("上传失败", e);
			TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
			return R.failed(e.getLocalizedMessage());
		}
		return R.ok();
	}

	@Override
	public List<ExceptionAttachment> getExceptionAttachment(Long id) {
		return exceptionAttachmentMapper.selectList(Wrappers.<ExceptionAttachment>lambdaQuery().eq(ExceptionAttachment::getExceptionId,id));
	}

	@Override
	public R deleteFile(String fileName) {
		ExceptionAttachment exceptionAttachment = exceptionAttachmentMapper.selectOne(Wrappers.<ExceptionAttachment>lambdaQuery().eq(ExceptionAttachment::getFileName,fileName));
		try {
			minioTemplate.removeObject(ossProperties.getBucketName(), exceptionAttachment.getFileName());
		} catch (Exception e) {
			e.printStackTrace();
		}
		exceptionAttachmentMapper.deleteById(exceptionAttachment.getId());
		return R.ok();
	}

	@Override
	public void getFile(String bucket, String fileName, HttpServletResponse response) {
		try (InputStream inputStream = minioTemplate.getObject(bucket, fileName)) {
			response.setContentType("application/octet-stream; charset=UTF-8");
			IoUtil.copy(inputStream, response.getOutputStream());
		} catch (Exception e) {
			log.error("文件读取异常: {}", e.getLocalizedMessage());
		}
	}

	@Override
	public IPage<QualityExceptionDTO> getExceptionFilePage(Page page, QueryWrapper<QualityExceptionDTO> gen) {
		IPage<QualityExceptionDTO> iPage = baseMapper.getExceptionFilePage(page,gen);
		if (CollectionUtil.isNotEmpty(iPage.getRecords())){
			for (QualityExceptionDTO qualityExceptionDTO:iPage.getRecords()){
				qualityExceptionDTO.setUrlList(exceptionAttachmentMapper.getUrl(qualityExceptionDTO.getId()));
			}
		}
		return iPage;
	}

	@Override
	public QualityException addQualityException(QualityException qualityException, List<MultipartFile> file) {
		qualityException.setCode(exceptionNumberGenerator.generateNumberWithPrefix(QualityException.DIGIT, QualityException.PREFIX, QualityException::getCode));
		baseMapper.insert(qualityException);
		for (MultipartFile multipartFile:file){
			uploadFile(multipartFile,qualityException.getId());
		}
		return qualityException;
	}

	@Override
	public QualityException editQualityException(QualityExceptionDTO qualityException, List<MultipartFile> file) {
		baseMapper.updateById(qualityException);
		List<ExceptionAttachment> exceptionAttachmentList = exceptionAttachmentMapper.getUrl(qualityException.getId());
		if (CollectionUtil.isNotEmpty(exceptionAttachmentList)){
			A:for (ExceptionAttachment exceptionAttachment:exceptionAttachmentList){
				if (CollectionUtil.isNotEmpty(qualityException.getUrlList())){
					for (ExceptionAttachment newExceptionAttachment:qualityException.getUrlList()){
						if (exceptionAttachment.getUrl().equals(newExceptionAttachment.getUrl())){
							continue A;
						}
					}
					exceptionAttachmentMapper.deleteById(exceptionAttachment.getId());
				}
			}
		}
		if (CollectionUtil.isNotEmpty(file)){
			for (MultipartFile multipartFile:file){
				uploadFile(multipartFile,qualityException.getId());
			}
		}
		return qualityException;
	}
}