package com.ruoyi.basic.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.basic.dto.WorkShopDto;
import com.ruoyi.basic.mapper.WorkShopFileMapper;
import com.ruoyi.basic.mapper.WorkShopMapper;
import com.ruoyi.basic.pojo.WorkShop;
import com.ruoyi.basic.pojo.WorkShopFile;
import com.ruoyi.basic.service.WorkShopService;
import com.ruoyi.common.config.MinioConfig;
import com.ruoyi.common.core.domain.MinioResult;
import com.ruoyi.common.utils.QueryWrappers;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.file.MinioUtils;
import com.ruoyi.framework.exception.ErrorException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;

/**
 * @description 针对表【work_shop(车间名称)】的数据库操作Service实现
 * @createDate 2024-03-12 16:17:55
 */
@Service
public class WorkShopServiceImpl extends ServiceImpl<WorkShopMapper, WorkShop> implements WorkShopService {

    @Resource
    private WorkShopMapper workShopMapper;
    @Resource
    private WorkShopFileMapper workShopFileMapper;

    @Resource
    private MinioUtils minioUtils;

    @Autowired
    private MinioConfig minioConfig;


    @Override
    public IPage<WorkShopDto> selectWorkShop(Page page, WorkShopDto workShopDto) {
        return workShopMapper.selectPage(page, QueryWrappers.queryWrappers(workShopDto));
    }

    @Override
    public int addWorkShop(WorkShopDto workShopDto) {
        if (workShopDto.getId() == null) {
            return workShopMapper.insert(workShopDto);
        } else {
            return workShopMapper.updateById(workShopDto);
        }
    }

    @Override
    public int delWorkShop(Integer id) {
        return workShopMapper.deleteById(id);
    }

    @Override
    public int uploadFile(Integer id, MultipartFile file) throws Exception {
        // 基础校验
        if (file == null || file.isEmpty()) {
            throw new ErrorException("上传文件为空");
        }

        WorkShopFile workShopFile = new WorkShopFile();
        workShopFile.setWorkShopId(id);
        workShopFile.setFileName(file.getOriginalFilename());
        workShopFile.setName(SecurityUtils.getUsername());

        try {
            String contentType = file.getContentType();
            String category = contentType != null && contentType.startsWith("image/") ? "images" : "docs";
            MinioResult upload = minioUtils.upload(minioConfig.getBucketName(), file, true);
            // 记录存储路径
            workShopFile.setFileUrl(upload.getBucketFileName());
            workShopFile.setFileMinioUrl(upload.getPreviewExpiry());
            workShopFile.setType(category.equals("images") ? 1 : 2);
            // 数据库操作
            int insertResult = workShopFileMapper.insert(workShopFile);
            if (insertResult <= 0) {
                throw new ErrorException("数据库插入失败");
            }
            return insertResult;
        } catch (Exception e) {
            throw new Exception("系统异常: ", e);
        }
    }
}