package cc.mrbird.febs.ai.controller.fileUpload;

import cc.mrbird.febs.common.controller.BaseController;
import cc.mrbird.febs.common.entity.FebsResponse;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.UUID;

/**
 * @author Administrator
 */
@Slf4j
@RestController
@RequiredArgsConstructor
@RequestMapping(value = "/fileUpload")
public class FileUploadController extends BaseController {

    // 基础上传目录
    public static String baseUploadDir = "/home/javaweb/webresource/ai/file";

    // 分片存储目录
    public static String baseChunkDir = "/home/javaweb/webresource/ai/file/chunks";

    /**
     * 上传文件分片
     */
    @PostMapping("/uploadChunk")
    public FebsResponse uploadChunk(@RequestParam("file") MultipartFile file,
                                    @RequestParam("fileName") String fileName,
                                    @RequestParam("chunk") int chunk,
                                    @RequestParam("chunks") int chunks,
                                    @RequestParam("fileMd5") String fileMd5) {
        try {
            String companyId = getCurrentUserCompanyId();
            // 构建公司专属分片目录
            String chunkDir = baseChunkDir + "/" + companyId;
            // 确保分片目录存在
            Path chunkPath = Paths.get(chunkDir, fileMd5);
            if (!Files.exists(chunkPath)) {
                Files.createDirectories(chunkPath);
            }

            // 保存分片文件
            Path chunkFilePath = chunkPath.resolve(chunk + ".part");
            Files.write(chunkFilePath, file.getBytes());

            return new FebsResponse().success().message("分片上传成功");
        } catch (Exception e) {
            e.printStackTrace();
            return new FebsResponse().fail().message("分片上传失败: " + e.getMessage());
        }
    }

    /**
     * 合并文件分片
     */
    @PostMapping("/mergeChunks")
    public FebsResponse mergeChunks(@RequestParam("fileName") String fileName,
                                    @RequestParam("fileMd5") String fileMd5,
                                    @RequestParam("chunks") int chunks) {
        try {

            String companyId = getCurrentUserCompanyId();
            // 构建公司专属上传目录
            String uploadDir = baseUploadDir + "/" + companyId;
            // 确保上传目录存在
            Path uploadPath = Paths.get(uploadDir);
            if (!Files.exists(uploadPath)) {
                Files.createDirectories(uploadPath);
            }

            // 生成唯一文件名
            String uniqueFileName = fileName+ "_" + UUID.randomUUID().toString() ;
            Path targetFilePath = uploadPath.resolve(uniqueFileName);

            // 构建公司专属分片目录
            String chunkDir = baseChunkDir + "/" + companyId;
            // 确保分片目录存在
            Path chunkPath = Paths.get(chunkDir, fileMd5);
            if (!Files.exists(chunkPath)) {
                return new FebsResponse().fail().message("分片目录不存在");
            }

            // 合并分片
            try (FileOutputStream outputStream = new FileOutputStream(targetFilePath.toFile())) {
                for (int i = 0; i < chunks; i++) {
                    Path chunkFilePath = chunkPath.resolve(i + ".part");
                    if (!Files.exists(chunkFilePath)) {
                        return new FebsResponse().fail().message("分片文件不存在: " + chunkFilePath);
                    }
                    byte[] chunkBytes = Files.readAllBytes(chunkFilePath);
                    outputStream.write(chunkBytes);
                    // 删除已合并的分片
                    Files.deleteIfExists(chunkFilePath);
                }
            }

            // 删除分片目录
            Files.deleteIfExists(chunkPath);

            // 保存文件信息到数据库或文件系统（这里简化处理）
            saveFileInfo(uniqueFileName, targetFilePath.toFile().length());

            return new FebsResponse().success().message("文件上传成功").data(uniqueFileName);
        } catch (Exception e) {
            e.printStackTrace();
            return new FebsResponse().fail().message("文件合并失败: " + e.getMessage());
        }
    }

    /**
     * 播放视频文件
     */
    @GetMapping("/play/{fileName}")
    public void playVideo(@PathVariable("fileName") String fileName, HttpServletRequest request, HttpServletResponse response) {
        try {
            String companyId = getCurrentUserCompanyId();
            String uploadDir = baseUploadDir + "/" + companyId;
            Path filePath = Paths.get(uploadDir, fileName);
            if (!Files.exists(filePath)) {
                response.setStatus(HttpStatus.NOT_FOUND.value());
                return;
            }

            // 获取文件大小
            long fileSize = Files.size(filePath);
            
            // 检查是否有Range请求头
            String rangeHeader = request.getHeader("Range");
            if (rangeHeader != null) {
                // 解析Range头，格式为"bytes=start-end"
                String[] ranges = rangeHeader.replace("bytes=", "").split("-");
                long start = 0, end = fileSize - 1;
                
                if (!ranges[0].isEmpty()) {
                    start = Long.parseLong(ranges[0]);
                }
                if (ranges.length > 1 && !ranges[1].isEmpty()) {
                    end = Long.parseLong(ranges[1]);
                } else {
                    end = fileSize - 1;
                }
                
                // 确保范围有效
                if (start > end || start >= fileSize) {
                    response.setStatus(HttpStatus.REQUESTED_RANGE_NOT_SATISFIABLE.value());
                    return;
                }
                
                // 限制结束位置不超过文件大小
                if (end >= fileSize) {
                    end = fileSize - 1;
                }
                
                // 计算内容长度
                long contentLength = end - start + 1;
                
                // 设置响应状态和头部信息
                response.setStatus(HttpStatus.PARTIAL_CONTENT.value());
                response.setContentType(getContentTypeByExtension(fileName));
                response.setContentLengthLong(contentLength);
                response.setHeader("Content-Range", "bytes " + start + "-" + end + "/" + fileSize);
                response.setHeader("Accept-Ranges", "bytes");
                response.setHeader("Content-Disposition", "inline; filename=\"" + fileName + "\"");
                
                // 从指定位置开始读取文件
                try (RandomAccessFile randomAccessFile = new RandomAccessFile(filePath.toFile(), "r");
                     OutputStream outputStream = response.getOutputStream()) {
                    
                    randomAccessFile.seek(start);
                    byte[] buffer = new byte[8192];
                    long remaining = contentLength;
                    int bytesRead;
                    
                    while (remaining > 0 && (bytesRead = randomAccessFile.read(buffer, 0, (int) Math.min(buffer.length, remaining))) != -1) {
                        outputStream.write(buffer, 0, bytesRead);
                        remaining -= bytesRead;
                    }
                }
            } else {
                // 没有Range头，返回完整文件
                response.setStatus(HttpStatus.OK.value());
                response.setContentType(getContentTypeByExtension(fileName));
                response.setContentLengthLong(fileSize);
                response.setHeader("Accept-Ranges", "bytes");
                response.setHeader("Content-Disposition", "inline; filename=\"" + fileName + "\"");

                // 流式传输文件
                try (InputStream inputStream = Files.newInputStream(filePath);
                     OutputStream outputStream = response.getOutputStream()) {
                    byte[] buffer = new byte[8192];
                    int bytesRead;
                    while ((bytesRead = inputStream.read(buffer)) != -1) {
                        outputStream.write(buffer, 0, bytesRead);
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
        }
    }

    /**
     * 根据文件扩展名获取MIME类型
     */
    private String getContentTypeByExtension(String fileName) {
        String extension = "";
        int lastDotIndex = fileName.lastIndexOf('.');
        if (lastDotIndex > 0) {
            extension = fileName.substring(lastDotIndex).toLowerCase();
        }
        
        switch (extension) {
            case ".mp4":
                return "video/mp4";
            case ".avi":
                return "video/x-msvideo";
            case ".mov":
                return "video/quicktime";
            case ".wmv":
                return "video/x-ms-wmv";
            case ".flv":
                return "video/x-flv";
            case ".webm":
                return "video/webm";
            case ".mkv":
                return "video/x-matroska";
            case ".m3u8":
                return "application/vnd.apple.mpegurl";
            case ".ts":
                return "video/MP2T";
            default:
                return "application/octet-stream";
        }
    }

    /**
     * 获取文件列表
     */
    @GetMapping("/list")
    public FebsResponse getFileList() {
        try {

            String companyId = getCurrentUserCompanyId();
            String uploadDir = baseUploadDir + "/" + companyId;
            Path uploadPath = Paths.get(uploadDir);
            if (!Files.exists(uploadPath)) {
                return new FebsResponse().data(new ArrayList<>());
            }

            List<FileInfo> fileList = new ArrayList<>();
            Files.list(uploadPath).forEach(path -> {
                if (Files.isRegularFile(path)) {
                    try {
                        FileInfo fileInfo = new FileInfo();
                        fileInfo.setFileName(path.getFileName().toString());
                        fileInfo.setFileSize(Files.size(path));
                        fileInfo.setUploadTime(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date(Files.getLastModifiedTime(path).toMillis())));
                        fileList.add(fileInfo);
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            });

            return new FebsResponse().success().data(fileList);
        } catch (Exception e) {
            e.printStackTrace();
            return new FebsResponse().fail().message("获取文件列表失败: " + e.getMessage());
        }
    }

    /**
     * 删除文件
     */
    @PostMapping("/delete")
    public FebsResponse deleteFile(@RequestParam("fileName") String fileName) {
        try {

            String companyId = getCurrentUserCompanyId();
            String uploadDir = baseUploadDir + "/" + companyId;
            Path filePath = Paths.get(uploadDir, fileName);
            if (Files.exists(filePath)) {
                Files.delete(filePath);
            }
            return new FebsResponse().success().message("文件删除成功");
        } catch (Exception e) {
            e.printStackTrace();
            return new FebsResponse().fail().message("文件删除失败: " + e.getMessage());
        }
    }

    /**
     * 保存文件信息
     */
    private void saveFileInfo(String fileName, long fileSize) {
        // 这里可以实现保存文件信息到数据库的逻辑
        // 简化处理，暂时不做数据库操作
    }

    /**
     * 文件信息实体类
     */
    public static class FileInfo {
        private String fileName;
        private long fileSize;
        private String uploadTime;

        public String getFileName() {
            return fileName;
        }

        public void setFileName(String fileName) {
            this.fileName = fileName;
        }

        public long getFileSize() {
            return fileSize;
        }

        public void setFileSize(long fileSize) {
            this.fileSize = fileSize;
        }

        public String getUploadTime() {
            return uploadTime;
        }

        public void setUploadTime(String uploadTime) {
            this.uploadTime = uploadTime;
        }
    }
}