package com.ruoyi.system.service.impl;

import com.ruoyi.common.config.ImageUrlConfig;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.file.FileUploadUtils;
import com.ruoyi.common.utils.file.FileUtils;
import com.ruoyi.system.domain.SysTaskAttachment;
import com.ruoyi.system.mapper.SysTaskAttachmentMapper;
import com.ruoyi.system.service.ISysTaskAttachmentService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;

@Service
public class SysTaskAttachmentServiceImpl implements ISysTaskAttachmentService {

    @Autowired
    private SysTaskAttachmentMapper sysTaskAttachmentMapper;

    @Autowired
    private ImageUrlConfig imageUrlConfig;

    @Override
    @Transactional
    public Long uploadAttachment(Long taskId, MultipartFile file, String category) {
        try {
            String fileName = category + "_" + System.currentTimeMillis() + "_" + file.getOriginalFilename();
            String relativePath = saveLocalPath(fileName, file.getInputStream());

            SysTaskAttachment attachment = new SysTaskAttachment();
            attachment.setTaskId(taskId);
            attachment.setFileName(file.getOriginalFilename());
            attachment.setFilePath(relativePath);
            attachment.setFileSize(file.getSize());
            attachment.setFileType(getFileType(file.getOriginalFilename()));
            attachment.setAttachmentCategory(category);
            attachment.setUploadTime(DateUtils.getNowDate());
            attachment.setUploadBy(SecurityUtils.getUsername());

            Long result = sysTaskAttachmentMapper.insertSysTaskAttachment(attachment);
            return result > 0 ? attachment.getAttachmentId() : 0L;
        } catch (IOException e) {
            throw new RuntimeException("文件上传失败：" + e.getMessage());
        }
    }

    @Override
    @Transactional
    public Long uploadAttachmentFromWechat(Long taskId, String accessToken, String mediaId, String category) {
        try {
            String wechatUrl = String.format(
                    "https://api.weixin.qq.com/cgi-bin/media/get?access_token=%s&media_id=%s",
                    accessToken, mediaId
            );

            byte[] fileBytes = downloadFromUrl(wechatUrl);
            if (fileBytes == null || fileBytes.length == 0) {
                throw new RuntimeException("从微信下载文件失败");
            }

            String fileName = "wx_" + mediaId.substring(0, Math.min(20, mediaId.length())) + "_" + System.currentTimeMillis() + ".jpg";
            String relativeFilePath = saveLocalPath(fileName, fileBytes);

            SysTaskAttachment attachment = new SysTaskAttachment();
            attachment.setTaskId(taskId);
            attachment.setFileName(fileName);
            attachment.setFilePath(relativeFilePath);
            attachment.setFileSize((long) fileBytes.length);
            attachment.setFileType("jpg");
            attachment.setAttachmentCategory(category);
            attachment.setUploadTime(DateUtils.getNowDate());
            attachment.setUploadBy(SecurityUtils.getUsername());

            Long result = sysTaskAttachmentMapper.insertSysTaskAttachment(attachment);
            return result > 0 ? attachment.getAttachmentId() : 0L;
        } catch (Exception e) {
            throw new RuntimeException("从微信上传文件失败：" + e.getMessage());
        }
    }

    @Override
    @Transactional
    public int deleteAttachment(Long attachmentId) {
        SysTaskAttachment attachment = sysTaskAttachmentMapper.selectSysTaskAttachmentByAttachmentId(attachmentId);
        if (attachment == null) {
            throw new RuntimeException("附件不存在");
        }

        try {
            FileUtils.deleteFile(attachment.getFilePath());
        } catch (Exception e) {
            // ignore
        }

        return sysTaskAttachmentMapper.deleteSysTaskAttachmentByAttachmentId(attachmentId);
    }

    @Override
    public SysTaskAttachment getAttachmentById(Long attachmentId) {
        SysTaskAttachment attachment = sysTaskAttachmentMapper.selectSysTaskAttachmentByAttachmentId(attachmentId);
        if (attachment != null) {
            buildAttachmentUrl(attachment);
        }
        return attachment;
    }

    @Override
    public List<SysTaskAttachment> getAttachmentsByTaskId(Long taskId) {
        List<SysTaskAttachment> list = sysTaskAttachmentMapper.selectSysTaskAttachmentByTaskId(taskId);
        if (list != null && !list.isEmpty()) {
            list.forEach(this::buildAttachmentUrl);
        }
        return list;
    }

    private String saveLocalPath(String fileName, InputStream stream) throws IOException {
        String baseDir = FileUploadUtils.getDefaultBaseDir();
        String datePath = DateUtils.datePath();
        String uploadDir = baseDir + "/task/" + datePath;

        File uploadPath = new File(uploadDir);
        if (!uploadPath.exists()) {
            uploadPath.mkdirs();
        }

        String filePath = uploadDir + "/" + fileName;
        try (OutputStream os = new FileOutputStream(filePath)) {
            byte[] buffer = new byte[1024];
            int bytesRead;
            while ((bytesRead = stream.read(buffer)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
            os.flush();
        }

        return "/task/" + datePath + "/" + fileName;
    }

    private String saveLocalPath(String fileName, byte[] fileBytes) throws IOException {
        String baseDir = FileUploadUtils.getDefaultBaseDir();
        String datePath = DateUtils.datePath();
        String uploadDir = baseDir + "/task/" + datePath;

        File uploadPath = new File(uploadDir);
        if (!uploadPath.exists()) {
            uploadPath.mkdirs();
        }

        String filePath = uploadDir + "/" + fileName;
        File file = new File(filePath);
        try (FileOutputStream fos = new FileOutputStream(file)) {
            fos.write(fileBytes);
        }

        return "/task/" + datePath + "/" + fileName;
    }

    private byte[] downloadFromUrl(String fileUrl) throws IOException {
        URL url = new URL(fileUrl);
        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        connection.setRequestMethod("GET");
        connection.setConnectTimeout(10000);
        connection.setReadTimeout(30000);

        try (InputStream inputStream = connection.getInputStream()) {
            byte[] buffer = new byte[4096];
            int bytesRead;
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();

            while ((bytesRead = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }

            return outputStream.toByteArray();
        } finally {
            connection.disconnect();
        }
    }

    private String getFileType(String fileName) {
        if (StringUtils.isEmpty(fileName)) {
            return "";
        }
        int lastDotIndex = fileName.lastIndexOf(".");
        if (lastDotIndex > 0 && lastDotIndex < fileName.length() - 1) {
            return fileName.substring(lastDotIndex + 1).toLowerCase();
        }
        return "";
    }

    @Override
    public void buildAttachmentUrl(SysTaskAttachment attachment) {
        if (attachment == null || StringUtils.isEmpty(attachment.getFilePath())) {
            return;
        }
        String baseUrl = imageUrlConfig.getImageUrl();
        if (!attachment.getFilePath().startsWith("http")) {
            attachment.setFileUrl(baseUrl + attachment.getFilePath());
        } else {
            attachment.setFileUrl(attachment.getFilePath());
        }
    }

    @Override
    public String getCategoryDesc(String category) {
        if (category == null || category.isEmpty()) {
            return "未分类";
        }
        switch (category) {
            case "1": return "知情同意书";
            case "2": return "病人资料";
            case "3": return "操作记录";
            case "4": return "出车前";
            case "5": return "出车后";
            case "6": return "系安全带";
            default: return "其他";
        }
    }
}