/*
 *    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.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.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.quality.entity.ApplyPart;
import com.chinaztt.mes.quality.entity.ApplyPartFile;
import com.chinaztt.mes.quality.mapper.ApplyPartFileMapper;
import com.chinaztt.mes.quality.mapper.ApplyPartMapper;
import com.chinaztt.mes.quality.service.ApplyPartFileService;
import com.chinaztt.ztt.common.core.util.R;
import com.chinaztt.ztt.common.oss.OssProperties;
import com.chinaztt.ztt.common.oss.service.OssTemplate;
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 shz
 * @date 2023-03-28 14:29:21
 */
@Slf4j
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class ApplyPartFileServiceImpl extends ServiceImpl<ApplyPartFileMapper, ApplyPartFile> implements ApplyPartFileService {

	private final OssProperties ossProperties;
	private final OssTemplate minioTemplate;
	private final ApplyPartMapper applyPartMapper;

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

			ApplyPartFile applyPartFile = new ApplyPartFile();
			applyPartFile.setOriginal(file.getOriginalFilename());
			applyPartFile.setPath(url);
			applyPartFile.setFileName(fileName);
			applyPartFile.setApplyPartId(applyPartId);
			this.baseMapper.insert(applyPartFile);
		} catch (Exception e) {
			log.error("上传失败", e);
			TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
			return R.failed(e.getLocalizedMessage());
		}
		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 List<ApplyPartFile> getListByApplyPartId(Long applyPartId) {
		return this.list(Wrappers.<ApplyPartFile>query().lambda().eq(ApplyPartFile::getApplyPartId, applyPartId));
    }
}