package com.xy.xyaicpzs.controller;

import com.xy.xyaicpzs.common.ErrorCode;
import com.xy.xyaicpzs.common.ResultUtils;
import com.xy.xyaicpzs.common.response.ApiResponse;
import com.xy.xyaicpzs.domain.entity.User;
import com.xy.xyaicpzs.service.OperationHistoryService;
import com.xy.xyaicpzs.service.UserService;
import com.xy.xyaicpzs.util.DltDataImporter;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;

/**
 * 大乐透数据导入控制器
 */
@Slf4j
@RestController
@RequestMapping("/dlt")
@Tag(name = "大乐透数据导入", description = "大乐透数据导入相关接口")
public class DltImportController {

    @Autowired
    private DltDataImporter dltDataImporter;

    @Autowired
    private UserService userService;

    @Autowired
    private OperationHistoryService operationHistoryService;

    /**
     * 上传Excel文件并完整导入大乐透数据（D3-D12工作表）
     */
    @PostMapping("/upload")
    @Operation(summary = "上传Excel文件完整导入大乐透数据", description = "上传包含D3-D12工作表的Excel文件，将前区历史数据、后区历史数据和系数数据导入到对应表中")
    public ApiResponse<String> uploadDltFile(
            @Parameter(description = "包含D3-D12工作表的Excel文件(.xlsx格式)", required = true)
            @RequestParam("file") MultipartFile file, HttpServletRequest httpServletRequest) {
        User loginUser = userService.getLoginUser(httpServletRequest);
        if (!userService.isAdmin(loginUser)){
            return ResultUtils.error(ErrorCode.NO_AUTH_ERROR, "无权限");
        }
        Long userId = loginUser.getId();
        String userName = loginUser.getUserName();
        
        String fileName = file.getOriginalFilename();
        log.info("接收到大乐透完整数据上传请求，文件名：{}", fileName);
        
        // 验证文件格式
        if (fileName == null || !fileName.toLowerCase().endsWith(".xlsx")) {
            return ResultUtils.error(ErrorCode.PARAMS_ERROR, "只支持.xlsx格式的Excel文件");
        }
        
        try {
            // 创建临时文件
            Path tempFile = Files.createTempFile("dlt_upload_", ".xlsx");
            Files.copy(file.getInputStream(), tempFile, StandardCopyOption.REPLACE_EXISTING);
            
            // 导入完整数据（D3-D12）
            dltDataImporter.importDltCompleteData(tempFile.toString());
            
            // 删除临时文件
            Files.deleteIfExists(tempFile);
            
            String message = "大乐透完整数据导入成功";
            
            // 记录操作历史 - 成功
            String resultMessage = String.format("%s成功上传并导入大乐透完整数据文件：%s", userName, fileName);
            operationHistoryService.recordOperation(userId, "大乐透完整数据导入", 1, "成功", resultMessage);
            
            return ResultUtils.success(message);
            
        } catch (Exception e) {
            log.error("大乐透完整数据导入失败，文件名：{}，错误：{}", fileName, e.getMessage(), e);
            
            // 记录操作历史 - 失败
            String resultMessage = String.format("%s大乐透完整数据导入失败，文件名：%s，错误原因：%s", userName, fileName, e.getMessage());
            operationHistoryService.recordOperation(userId, "大乐透完整数据导入", 1, "失败", resultMessage);
            
            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "大乐透完整数据导入失败：" + e.getMessage());
        }
    }

    /**
     * 上传Excel文件并导入大乐透开奖数据（覆盖）
     */
    @PostMapping("/upload-draw-data")
    @Operation(summary = "上传Excel文件导入大乐透开奖数据", description = "上传包含D1工作表的Excel文件，将大乐透开奖数据导入到dlt_draw_record表中（覆盖现有数据）")
    public ApiResponse<String> uploadDltDrawData(
            @Parameter(description = "包含D1工作表的Excel文件(.xlsx格式)", required = true)
            @RequestParam("file") MultipartFile file, HttpServletRequest httpServletRequest) {
        User loginUser = userService.getLoginUser(httpServletRequest);
        if (!userService.isAdmin(loginUser)){
            return ResultUtils.error(ErrorCode.NO_AUTH_ERROR, "无权限");
        }
        Long userId = loginUser.getId();
        String userName = loginUser.getUserName();
        
        String fileName = file.getOriginalFilename();
        log.info("接收到大乐透开奖数据上传请求，文件名：{}", fileName);
        
        // 验证文件格式
        if (fileName == null || !fileName.toLowerCase().endsWith(".xlsx")) {
            return ResultUtils.error(ErrorCode.PARAMS_ERROR, "只支持.xlsx格式的Excel文件");
        }
        
        try {
            // 创建临时文件
            Path tempFile = Files.createTempFile("dlt_draw_upload_", ".xlsx");
            Files.copy(file.getInputStream(), tempFile, StandardCopyOption.REPLACE_EXISTING);
            
            // 导入数据
            dltDataImporter.importDltDrawData(tempFile.toString());
            
            // 删除临时文件
            Files.deleteIfExists(tempFile);
            
            String message = "大乐透开奖数据导入成功";
            
            // 记录操作历史 - 成功
            String resultMessage = String.format("%s成功上传并导入大乐透开奖数据文件：%s", userName, fileName);
            operationHistoryService.recordOperation(userId, "大乐透开奖数据导入", 1, "成功", resultMessage);
            
            return ResultUtils.success(message);
        } catch (Exception e) {
            log.error("大乐透开奖数据导入失败，文件名：{}，错误：{}", fileName, e.getMessage(), e);
            
            // 记录操作历史 - 失败
            String resultMessage = String.format("%s大乐透开奖数据导入失败，文件名：%s，错误原因：%s", userName, fileName, e.getMessage());
            operationHistoryService.recordOperation(userId, "大乐透开奖数据导入", 1, "失败", resultMessage);
            
            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "大乐透开奖数据导入失败：" + e.getMessage());
        }
    }

    /**
     * 上传Excel文件并追加导入大乐透开奖数据
     */
    @PostMapping("/append-draw-data")
    @Operation(summary = "追加导入大乐透开奖数据", description = "上传包含D1工作表的Excel文件，将大乐透开奖数据追加导入到dlt_draw_record表中（不清空现有数据，跳过重复记录）")
    public ApiResponse<String> appendDltDrawData(
            @Parameter(description = "包含D1工作表的Excel文件(.xlsx格式)", required = true)
            @RequestParam("file") MultipartFile file, HttpServletRequest httpServletRequest) {
        User loginUser = userService.getLoginUser(httpServletRequest);
        if (!userService.isAdmin(loginUser)){
            return ResultUtils.error(ErrorCode.NO_AUTH_ERROR, "无权限");
        }
        Long userId = loginUser.getId();
        String userName = loginUser.getUserName();
        
        String fileName = file.getOriginalFilename();
        log.info("接收到大乐透开奖数据追加导入请求，文件名：{}", fileName);
        
        // 验证文件格式
        if (fileName == null || !fileName.toLowerCase().endsWith(".xlsx")) {
            return ResultUtils.error(ErrorCode.PARAMS_ERROR, "只支持.xlsx格式的Excel文件");
        }
        
        try {
            // 创建临时文件
            Path tempFile = Files.createTempFile("dlt_append_", ".xlsx");
            Files.copy(file.getInputStream(), tempFile, StandardCopyOption.REPLACE_EXISTING);
            
            // 追加导入数据
            dltDataImporter.appendDltDrawData(tempFile.toString());
            
            // 删除临时文件
            Files.deleteIfExists(tempFile);
            
            String message = "大乐透开奖数据追加导入成功";
            
            // 记录操作历史 - 成功
            String resultMessage = String.format("%s成功追加导入大乐透开奖数据文件：%s", userName, fileName);
            operationHistoryService.recordOperation(userId, "大乐透开奖数据追加导入", 1, "成功", resultMessage);
            
            return ResultUtils.success(message);
        } catch (Exception e) {
            log.error("大乐透开奖数据追加导入失败，文件名：{}，错误：{}", fileName, e.getMessage(), e);
            
            // 记录操作历史 - 失败
            String resultMessage = String.format("%s大乐透开奖数据追加导入失败，文件名：%s，错误原因：%s", userName, fileName, e.getMessage());
            operationHistoryService.recordOperation(userId, "大乐透开奖数据追加导入", 1, "失败", resultMessage);
            
            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "大乐透开奖数据追加导入失败：" + e.getMessage());
        }
    }
}