package com.ruoyi.web.controller.api;


import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.ExcelImportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.ImportParams;
import cn.hutool.json.JSONObject;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.exception.ServiceException;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.WebUtils;
import com.ruoyi.system.export.CleanToiletSanitizationExport;
import com.ruoyi.system.model.CleanSludge;
import com.ruoyi.system.model.CleanToiletSanitization;
import com.ruoyi.system.model.DrDisplacement;
import com.ruoyi.system.query.CleanSludgeListQuery;
import com.ruoyi.system.query.DrDisplacementListQuery;
import com.ruoyi.system.service.CleanSludgeService;
import com.ruoyi.system.service.CleanToiletSanitizationService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * <p>
 * 厕所消杀记录 前端控制器
 * </p>
 *
 * @author mitao
 * @since 2025-10-17
 */
@RestController
@Api(tags = "厕所消杀记录")
@Slf4j
@RequestMapping("/clean-toilet-sanitization")
public class CleanToiletSanitizationController {
    @Resource
    private CleanToiletSanitizationService cleanToiletSanitizationService;

    @ApiOperation(value = "厕所消杀记录分页列表")
    @PostMapping(value = "/pageList")
    public R<PageInfo<CleanToiletSanitization>> pageList(@RequestBody CleanSludgeListQuery query) {
        PageInfo<CleanToiletSanitization> cleanToiletSanitizationPageInfo = cleanToiletSanitizationService.pageList(query);
        return R.ok(cleanToiletSanitizationPageInfo);
    }

    @Log(title = "厕所消杀记录导入", businessType = BusinessType.INSERT)
    @ApiOperation(value = "厕所消杀记录导入")
    @Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)
    @PostMapping("/import")
    public R<String> importCleaner(@RequestPart("file") MultipartFile file) {
        ImportParams params = new ImportParams();
        params.setHeadRows(1); //表头行数
        InputStream inputStream = null;
        List<CleanToiletSanitizationExport> locationExcelList;
        try {
            inputStream = file.getInputStream();
            locationExcelList = ExcelImportUtil.importExcel(inputStream,
                    CleanToiletSanitizationExport.class, params);
        } catch (Exception e) {
            e.printStackTrace();
            log.error("厕所消杀记录导入失败:{}", e.getMessage());
            throw new ServiceException("厕所消杀记录导入失败!");
        } finally {
            try {
                inputStream.close();
            } catch (IOException e) {
                throw new ServiceException(e.getMessage());
            }
        }
        if (CollectionUtils.isEmpty(locationExcelList)) {
            throw new ServiceException("厕所消杀记录数据为空!");
        }
        List<CleanToiletSanitization> cleanToiletSanitizations = new ArrayList<>();
        for (CleanToiletSanitizationExport locationExcel : locationExcelList) {
            CleanToiletSanitization cleanToiletSanitization = new CleanToiletSanitization();
            String recordDate = locationExcel.getRecordDate();
            LocalDate localDate = LocalDate.parse(recordDate);
            cleanToiletSanitization.setRecordDate(localDate);
            cleanToiletSanitization.setIsCleaned(StringUtils.hasLength(locationExcel.getIsCleaned()) && locationExcel.getIsCleaned().equals("是"));
            cleanToiletSanitization.setIsDisinfected(StringUtils.hasLength(locationExcel.getIsDisinfected()) && locationExcel.getIsDisinfected().equals("是"));
            cleanToiletSanitization.setCleaner(locationExcel.getCleaner());
            cleanToiletSanitization.setInspectionTime(locationExcel.getInspectionTime());
            cleanToiletSanitization.setInspector(locationExcel.getInspector());
            cleanToiletSanitization.setInspectionSituation(locationExcel.getInspectionSituation());
            cleanToiletSanitizations.add(cleanToiletSanitization);
        }
        cleanToiletSanitizationService.saveBatch(cleanToiletSanitizations);
        return R.ok();
    }

    @ApiOperation(value = "厕所消杀记录导出")
    @Log(title = "厕所消杀记录导出", businessType = BusinessType.OTHER)
    @PostMapping(value = "/export")
    public void export(@RequestBody CleanSludgeListQuery query) {
        List<CleanToiletSanitization> list = cleanToiletSanitizationService.lambdaQuery()
                .like(StringUtils.hasLength(query.getRecordDate()), CleanToiletSanitization::getRecordDate, query.getRecordDate())
                .list();

        List<CleanToiletSanitizationExport> cleanToiletSanitizationExports = new ArrayList<>();

        int i = 1;
        for (CleanToiletSanitization cleanToiletSanitization : list) {
            CleanToiletSanitizationExport cleanToiletSanitizationExport = new CleanToiletSanitizationExport();
            cleanToiletSanitizationExport.setNumber(i);
            LocalDate recordDate = cleanToiletSanitization.getRecordDate();
            cleanToiletSanitizationExport.setRecordDate(recordDate.toString());
            cleanToiletSanitizationExport.setIsCleaned(cleanToiletSanitization.getIsCleaned() ? "是" : "否");
            cleanToiletSanitizationExport.setIsDisinfected(cleanToiletSanitization.getIsDisinfected() ? "是" : "否");
            cleanToiletSanitizationExport.setCleaner(cleanToiletSanitization.getCleaner());
            cleanToiletSanitizationExport.setInspectionTime(cleanToiletSanitization.getInspectionTime());
            cleanToiletSanitizationExport.setInspector(cleanToiletSanitization.getInspector());
            cleanToiletSanitizationExport.setInspectionSituation(cleanToiletSanitization.getInspectionSituation());
            cleanToiletSanitizationExports.add(cleanToiletSanitizationExport);
            i++;
        }
        //1.获取excel模板
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), CleanToiletSanitizationExport.class, cleanToiletSanitizationExports);
        HttpServletResponse response = WebUtils.response();
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        ServletOutputStream outputStream = null;
        try {
            String fileName = URLEncoder.encode("厕所消杀记录.xls", "utf-8");
            response.setHeader("Content-dispodition", "attachment;filename=" + fileName);
            outputStream = response.getOutputStream();
            workbook.write(outputStream);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    @ApiOperation(value = "厕所消杀记录导入模板下载")
    @GetMapping("/download")
    public void download() {
        List<CleanToiletSanitizationExport> locationImportExcels = new ArrayList<>();
        CleanToiletSanitizationExport tLocationImportExcel = new CleanToiletSanitizationExport();
        locationImportExcels.add(tLocationImportExcel);
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), CleanToiletSanitizationExport.class, locationImportExcels);
        HttpServletResponse response = WebUtils.response();
        ServletOutputStream outputStream = null;
        try {
            String fileName = URLEncoder.encode("厕所消杀记录导入模板.xls", "utf-8");
            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
            response.setHeader("content-Type", "application/vnd.ms-excel");
            response.setHeader("Pragma", "no-cache");
            response.setHeader("Cache-Control", "no-cache");
            outputStream = response.getOutputStream();
            workbook.write(outputStream);
        } catch (IOException e) {
            e.printStackTrace();
            System.out.println("厕所消杀记录导入模板下载失败!");
        } finally {
            try {
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    @ApiOperation(value = "添加")
    @Transactional
    @Log(title = "厕所消杀记录-添加", businessType = BusinessType.INSERT)
    @PostMapping(value = "/add")
    public R<Boolean> save(@RequestBody CleanToiletSanitization entity) {
        cleanToiletSanitizationService.save(entity);
        return R.ok();
    }

    @ApiOperation(value = "修改")
    @Transactional
    @Log(title = "厕所消杀记录-修改", businessType = BusinessType.UPDATE)
    @PostMapping(value = "/edit")
    public R<Boolean> edit(@RequestBody CleanToiletSanitization entity) {
        cleanToiletSanitizationService.updateById(entity);
        return R.ok();
    }

    @ApiOperation(value = "详情")
    @GetMapping(value = "/detail")
    public R<CleanToiletSanitization> detail(String id) {
        return R.ok(cleanToiletSanitizationService.getById(id));
    }

    @Log(title = "厕所消杀记录-删除", businessType = BusinessType.DELETE)
    @Transactional
    @ApiOperation(value = "厕所消杀记录-删除")
    @DeleteMapping(value = "/delete")
    public R delete(@RequestParam String ids) {
        cleanToiletSanitizationService.removeBatchByIds(Arrays.asList(ids.split(",")));
        return R.ok();
    }
}