package com.sinata.appointment.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import com.sinata.appointment.common.Result;
import com.sinata.appointment.common.ResultUtil;
import com.sinata.appointment.dto.ExportTaskStatusVo;
import com.sinata.appointment.service.ExportApplicationService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.HashMap;
import java.util.Map;
import lombok.RequiredArgsConstructor;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@Tag(name = "导出")
@RestController
@RequestMapping("/api/exports")
@RequiredArgsConstructor
public class ExportController {

    private final ExportApplicationService exportApplicationService;

    @Operation(summary = "异步导出任务入队")
    @PostMapping("/enqueue")
    @SaCheckPermission("exports:enqueue")
    public Result<Map<String, String>> enqueue(@RequestBody Map<String, Object> body) {
        String type = body.get("type") != null ? body.get("type").toString() : "appointment";
        @SuppressWarnings("unchecked")
        Map<String, Object> criteria =
                body.get("criteria") instanceof Map ? (Map<String, Object>) body.get("criteria") : Map.of();
        String taskId = exportApplicationService.enqueueExport(type, criteria);
        Map<String, String> res = new HashMap<>();
        res.put("taskId", taskId);
        return ResultUtil.data(res);
    }

    @Operation(summary = "查询导出任务状态")
    @GetMapping("/{taskId}")
    @SaCheckPermission("exports:enqueue")
    public Result<ExportTaskStatusVo> status(@PathVariable String taskId) {
        ExportTaskStatusVo status = exportApplicationService.getTaskStatus(taskId);
        if (status == null) {
            return ResultUtil.error(404, "导出任务不存在");
        }
        return ResultUtil.data(status);
    }

    @Operation(summary = "下载导出文件")
    @GetMapping("/{taskId}/download")
    @SaCheckPermission("exports:enqueue")
    public ResponseEntity<byte[]> download(@PathVariable String taskId) throws IOException {
        Path filePath = exportApplicationService.getExportFilePath(taskId);
        if (filePath == null) {
            return ResponseEntity.status(HttpStatus.NOT_FOUND).body(new byte[0]);
        }
        return ResponseEntity.ok()
                .contentType(
                        MediaType.parseMediaType(
                                "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"))
                .header(
                        HttpHeaders.CONTENT_DISPOSITION,
                        "attachment; filename=\"" + filePath.getFileName() + "\"")
                .body(Files.readAllBytes(filePath));
    }
}