/**
 * Copyright (c) 2018 人人开源 All rights reserved.
 * <p>
 * https://www.renren.io
 * <p>
 * 版权所有，侵权必究！
 */

package com.zt.modules.oss.cloud;

import com.aspose.words.Document;
import com.aspose.words.SaveFormat;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.spire.xls.Workbook;
import com.zt.common.exception.ErrorCode;
import com.zt.common.exception.RenException;
import com.zt.core.oss.encry.IOssEncryptService;
import com.zt.modules.oss.enums.CloudChannel;
import com.zt.modules.oss.model.QdSysOss;
import com.zt.modules.oss.model.SysOss;
import org.apache.commons.io.FileUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.util.FileCopyUtils;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

/**
 * 本地上传
 *
 * @author hehz
 */

public class LocalStorageService extends AbstractStorageService {

    private IOssEncryptService ossEncryptService;

    public LocalStorageService(StorageConfig config, QdStorageConfig qdConfig, IOssEncryptService ossEncryptService) {
        this.config = config;
        this.qdConfig = qdConfig;
        this.ossEncryptService = ossEncryptService;
    }

    @Override
    public SysOss upload(byte[] data, String path) {
        return upload(new ByteArrayInputStream(data), path);
    }

    @Override
    public SysOss upload(InputStream inputStream, String path) {
        File file = new File(config.getLocalPath() + File.separator + path);
        try {
            FileUtils.copyToFile(ossEncryptService.encrypt(inputStream), file);
        } catch (IOException e) {
            throw new RenException(ErrorCode.OSS_UPLOAD_FILE_ERROR.getCode(), e, "");
        }

        SysOss oss = new SysOss();
        oss.setId(IdWorker.getId());
        oss.setPath(path);
        oss.setUrl(config.getLocalDomain() + "/sys/oss/content");
        return oss;
    }

    @Override
    public SysOss uploadSuffix(byte[] data, String suffix) {
        return upload(data, getPath(config.getLocalPrefix(), suffix));
    }

    @Override
    public SysOss uploadSuffix(InputStream inputStream, String suffix) {
        return upload(inputStream, getPath(config.getLocalPrefix(), suffix));
    }

    @Override
    public void delete(SysOss oss) {
        if (oss != null) {
            File file = new File(config.getLocalPath() + File.separator + oss.getPath());
            if (file.exists()) {
                File dir = file.getParentFile();
                try {
                    if (dir.listFiles().length == 1) {
                        FileUtils.forceDelete(dir);
                    } else {
                        FileUtils.forceDelete(file);
                    }
                } catch (IOException e) {
                    throw new RenException("本地文件删除失败！");
                }
            }
        }
    }

    public void content(SysOss oss, HttpServletResponse response) throws Exception {
        if (oss == null) {
            throw new IOException("文件不存在！");
        }
        if (CloudChannel.LOCAL.getValue().equals(oss.getChannel())) { // 本地
            String filename = oss.getName();
            if (",gif,jpg,jpeg,png,bmp,".contains("," + oss.getType() + ",")) {// 浏览图片
                response.setContentType("image/jpeg");
            } else {
                response.setContentType("application/octet-stream");
//                response.setHeader("Content-disposition",
//                        "attachment;filename=" + new String(filename.getBytes("ISO8859-1"), "UTF-8"));
                response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
            }
            File file = new File(config.getLocalPath() + File.separator + oss.getPath());
            System.out.println(config.getLocalPath());
            System.out.println(File.separator);
            System.out.println(oss.getPath());
            // 创建文件输入流
            InputStream is = ossEncryptService.decrypt(file);
            // 响应输出流
            ServletOutputStream out = response.getOutputStream();
            // 创建缓冲区
            byte[] buffer = new byte[1024];
            int len = 0;
            while ((len = is.read(buffer)) != -1) {
                out.write(buffer, 0, len);
            }
            is.close();
            out.flush();
            out.close();
        }
    }

    public ResponseEntity<byte[]> content2(SysOss oss) throws Exception {
        if (oss == null) {
            throw new IOException("文件不存在！");
        }
        String errorMessage="";
        if (CloudChannel.LOCAL.getValue().equals(oss.getChannel())) { // 本地
            String filename = oss.getName();
            File file = new File(config.getLocalPath() + File.separator + oss.getPath());
            if (file.exists()) {
                try (InputStream in = ossEncryptService.decrypt(file)) {
                    // 从文件输入流中读取Word文档的二进制数据并存储在documentBytes数组中
                    byte[] documentBytes = new byte[0];
                    // 创建HTTP头部，设置内容类型为二进制流，并指定文件名
                    HttpHeaders headers = new HttpHeaders();
                    // 返回包含二进制数据流的HTTP响应，状态码为200
                    if (filename.endsWith(".xls")) {
                        // 如果是xls文件，将其转换为xlsx格式
                        Workbook workbook = new Workbook();
                        workbook.loadFromStream(in); // 使用loadFromStream方法加载xls文件
                        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
                        workbook.saveToStream(outputStream, com.spire.xls.FileFormat.Version2013); // 指定文件格式
                        documentBytes = outputStream.toByteArray();
                        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
                    }

                    if (filename.endsWith(".doc")) {
                        // 使用 Aspose.Words 进行文档转换
                        Document doc = new Document(in);
                        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
                        doc.save(outputStream, SaveFormat.DOCX);

                        documentBytes = outputStream.toByteArray();
                        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
                        headers.setContentDispositionFormData("attachment", filename.replace(".doc", ".docx"));
                    } else if (filename.endsWith(".docx") || filename.endsWith(".xlsx")) {
                        documentBytes = FileCopyUtils.copyToByteArray(in);
                        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
                        headers.setContentDispositionFormData("attachment", filename);
                    } else if (filename.endsWith(".pdf")) {
                        documentBytes = FileCopyUtils.copyToByteArray(in);
                        headers.setContentType(MediaType.APPLICATION_PDF);
                    } else if (filename.endsWith(".jpg") || filename.endsWith(".jpeg")) {
                        documentBytes = FileCopyUtils.copyToByteArray(in);
                        headers.setContentType(MediaType.IMAGE_JPEG);
                    } else if (filename.endsWith(".png")) {
                        documentBytes = FileCopyUtils.copyToByteArray(in);
                        headers.setContentType(MediaType.IMAGE_PNG);
                    }
                    return new ResponseEntity<>(documentBytes, headers, HttpStatus.OK);
                }
            } else {
                 errorMessage = "文件不存在";
                return new ResponseEntity<>(errorMessage.getBytes(), HttpStatus.NOT_FOUND);
            }
        }
        errorMessage = "文件不存在";
        return new ResponseEntity<>(errorMessage.getBytes(), HttpStatus.NOT_FOUND);
    }

    public void getCommFile(String fileFlag, HttpServletResponse response) throws Exception {
        if (fileFlag == null) {
            throw new IOException("图片不存在！");
        }
        if ("logo".contains(fileFlag)) {// 浏览图片
            response.setContentType("image/jpeg");
        } else {
            response.setContentType("application/octet-stream");
//                response.setHeader("Content-disposition",
//                        "attachment;filename=" + new String(filename.getBytes("ISO8859-1"), "UTF-8"));
            response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(fileFlag, "UTF-8"));
        }
        String url = config.getLocalPath() + File.separator + "comm-file" + File.separator + fileFlag;
        File file = new File(url);
        System.out.println(url);
        // 创建文件输入流
        InputStream is = ossEncryptService.decrypt(file);
        // 响应输出流
        ServletOutputStream out = response.getOutputStream();
        // 创建缓冲区
        byte[] buffer = new byte[1024];
        int len = 0;
        while ((len = is.read(buffer)) != -1) {
            out.write(buffer, 0, len);
        }
        out.flush();
        out.close();

    }

    public void qdContent(QdSysOss qdSysOss, HttpServletResponse response) throws Exception {
        if (qdSysOss == null) {
            throw new IOException("文件不存在！");
        }
        if (CloudChannel.LOCAL.getValue().equals(qdSysOss.getChannel())) { // 本地
            String filename = qdSysOss.getName();
            if (",gif,jpg,jpeg,png,bmp,".contains("," + qdSysOss.getType() + ",")) {// 浏览图片
                response.setContentType("image/jpeg");
            } else {
                response.setContentType("application/octet-stream");
//                response.setHeader("Content-disposition",
//                        "attachment;filename=" + new String(filename.getBytes("ISO8859-1"), "UTF-8"));
                response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
            }
            File file = new File(qdConfig.getQdLocalPath() + File.separator + qdSysOss.getPath());
            System.out.println("qd：" + qdConfig.getQdLocalPath());
            System.out.println("qd：" + File.separator);
            System.out.println("qd：" + qdSysOss.getPath());
            // 创建文件输入流
            InputStream is = ossEncryptService.decrypt(file);
            // 响应输出流
            ServletOutputStream out = response.getOutputStream();
            // 创建缓冲区
            byte[] buffer = new byte[1024];
            int len = 0;
            while ((len = is.read(buffer)) != -1) {
                out.write(buffer, 0, len);
            }
            is.close();
            out.flush();
            out.close();
        }
    }
}