package org.springblade.mdm.machinefile.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
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.HttpServletResponse;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.UrlUtil;
import org.springblade.mdm.commons.service.OssFileCommonService;
import org.springblade.mdm.machinefile.entity.FileSendRecord;
import org.springblade.mdm.machinefile.entity.MachineAcceptedFile;
import org.springblade.mdm.machinefile.service.MachineFileService;
import org.springblade.mdm.machinefile.service.MachineAcceptedFileService;
import org.springblade.mdm.machinefile.service.NcProgramExportInnerService;
import org.springblade.mdm.machinefile.vo.MachineBackFileHandleExcelVO;
import org.springblade.mdm.machinefile.vo.MachineAcceptedFileHandleQueryVO;
import org.springblade.mdm.machinefile.vo.MachineAcceptedFileVO;
import org.springblade.mdm.program.service.programannotation.AnnotationUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

@Slf4j
@RestController
@AllArgsConstructor
@RequestMapping("/machinefile/filehandle")
@Tag(name = "机床回传程序处理", description = "机床回传程序处理")
public class MachineAcceptedFileHandleController {
	private final OssFileCommonService ossFileCommonService;
	private final MachineAcceptedFileService acceptedFileService;
	private final NcProgramExportInnerService ncProgramExportInnerService;
	/**
	 * 分页
	 */
	@Operation(summary = "机床回传文件处理分页查询", description = "名称或编码,已接受的文件列表")
	@GetMapping("/page")
	public R<IPage<MachineAcceptedFileVO>> page(MachineAcceptedFileHandleQueryVO query) {
		IPage<MachineAcceptedFileVO> pages = acceptedFileService.handlePageQuery(query);
		return R.data(pages);
	}

	@GetMapping("/download-by-id")
	@Operation(summary = "下发文件记录列表", description = "下发文件记录列表")
	public void downloadByRecordId(Long id,HttpServletResponse response) throws IOException {
		MachineAcceptedFile acceptedFile = acceptedFileService.getById(id);
		ossFileCommonService.download(acceptedFile.getOssName(),acceptedFile.getName(),response);
	}

	@GetMapping("export-to-inner")
	@ApiOperationSupport(order = 13)
	@Operation(summary = "导出到涉密网", description = "导出到涉密网")
	public void exportToInner(@Parameter(description = "机床接收文件id数组") String ids, HttpServletResponse response) {

		if(Func.isBlank(ids)) {
			throw new ServiceException("未选择文件导出");
		}
		try {
			String prefix = UrlUtil.encode(AuthUtil.getNickName()+"-工控网导出");
			String filename = prefix+"-"+ DateUtil.format(DateUtil.now(), "yyyyMMddHHmm")+".zip";

			response.setHeader("Content-Disposition", "attachment; filename="+filename);
			response.setContentType("application/octet-stream");
			ncProgramExportInnerService.exportToInner(Func.toLongList(ids),response.getOutputStream());
		} catch (IOException e) {
			log.error("工控网导出回传文件异常", e);
			throw new RuntimeException(e);
		}

	}

	@Operation(summary = "回传程序处理导出Excel", description = "回传程序处理导出Excel")
	@GetMapping("/export-excel")
	public void exportExcel(MachineAcceptedFileHandleQueryVO query, HttpServletResponse response) {
		query.setCurrent(1);
		query.setSize(Integer.MAX_VALUE);

		//IPage<MachineAcceptedFileVO> pages = machineFileService.handlePageQuery(query);
		IPage<MachineAcceptedFileVO> pages = acceptedFileService.handlePageQuery(query);
		List<MachineBackFileHandleExcelVO> list = new ArrayList<>();
		pages.getRecords().forEach(vo ->{
			MachineBackFileHandleExcelVO excelVO = new MachineBackFileHandleExcelVO();
			BeanUtil.copyProperties(vo, excelVO);
			excelVO.setProgramStatusName(parseProgramStatusName(vo.getProgramStatus()));

			list.add(excelVO);
		});
		ExcelUtil.export(response, "机床回传程序" + DateUtil.time(), "机床回传程序表", list, MachineBackFileHandleExcelVO.class);
	}

	/**
	 * 转换状态为显示文本
	 * @param programStatus 程序状态编码
	 * @return 状态名称
	 */
	String parseProgramStatusName(String programStatus){
		if(programStatus == null){
			return "未知";
		}
		return switch (programStatus) {
			case AnnotationUtil.SQ -> "试切";
			case AnnotationUtil.GH -> "固化";
			case AnnotationUtil.LG -> "临时更改单";
			default -> "未知";
		};
	}
}