package com.ruoyi.web.controller.api;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.deepoove.poi.XWPFTemplate;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.system.model.TPartyBuilding;
import com.ruoyi.system.query.PartyBuildingListQuery;
import com.ruoyi.system.service.TPartyBuildingService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.*;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.core.io.ClassPathResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.*;

import static com.ruoyi.web.controller.common.ExcelExportExample.example1;

@Slf4j
@RestController
@RequestMapping("/partyBuilding")
@Api(tags = "党建引领")
public class PartyBuildingController {

    private final TPartyBuildingService partyBuildingService;

    public PartyBuildingController(TPartyBuildingService partyBuildingService) {
        this.partyBuildingService = partyBuildingService;
    }

    @ApiOperation(value = "党建引领--列表",response = TPartyBuilding.class)
    @PostMapping(value = "/getPartyBuildingList")
    public R<Page<TPartyBuilding>> getPartyBuildingList(@RequestBody PartyBuildingListQuery query) {
        Page<TPartyBuilding> page = new Page<>(query.getPageNum(), query.getPageSize());
        Page<TPartyBuilding> pageTPartyBuilding = partyBuildingService.page(page,
                Wrappers.<TPartyBuilding>lambdaQuery()
                        .eq(TPartyBuilding::getPartyType, query.getPartyType())
                        .orderByDesc(TPartyBuilding::getCreateTime));

        return R.ok(pageTPartyBuilding);
    }

    @ApiOperation(value = "党建引领--详情",response = TPartyBuilding.class)
    @PostMapping(value = "/getPartyBuildingDetail/{id}")
    public R<TPartyBuilding> getPartyBuildingDetail(@PathVariable String id) {
        TPartyBuilding partyBuilding = partyBuildingService.getById(id);
        partyBuilding.setReadCount(partyBuilding.getReadCount() + 1);
        partyBuildingService.updateById(partyBuilding);
        return R.ok(partyBuilding);
    }
    // 默认模板路径 - 根据实际情况修改
    private static final String DEFAULT_TEMPLATE_PATH = "F:\\code\\mobile_community\\ruoyi-applet\\src\\main\\resources\\11.xlsx";

    /**
     * 导出Excel文件
     */
    @PostMapping("/excel")
    public ResponseEntity<?> exportExcel() {
        try {
            // 使用提供的模板路径或默认路径
            String templatePath =  DEFAULT_TEMPLATE_PATH;

            // 生成Excel文件字节数据
            byte[] excelData = createExcelFromTemplate(templatePath, generateTestData());

            // 创建资源对象
            ByteArrayResource resource = new ByteArrayResource(excelData);

            // 设置文件名
            String filename =  "export_data.xlsx";

            HttpHeaders headers = new HttpHeaders();
            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + URLEncoder.encode(filename, "UTF-8"));
            return ResponseEntity.ok()
                    .headers(headers)
                    .contentType(MediaType.APPLICATION_OCTET_STREAM)
                    .body(resource);


        } catch (Exception e) {
            throw new RuntimeException("导出Excel文件失败: " + e.getMessage(), e);
        }
    }

    /**
     * 基于模板创建Excel文件
     */
    private byte[] createExcelFromTemplate(String templatePath, List<List<Object>> dataList)
            throws IOException {

        try (FileInputStream templateStream = new FileInputStream(templatePath);
             Workbook workbook = WorkbookFactory.create(templateStream);
             ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {

            Sheet sheet = workbook.getSheetAt(0); // 获取第一个工作表

            // 从第四行开始填充数据（行索引从0开始，第四行索引为3）
            fillDataToSheet(sheet, dataList, 3);

            // 写入输出流
            workbook.write(outputStream);
            return outputStream.toByteArray();
        }
    }

    /**
     * 填充数据到工作表
     */
    private void fillDataToSheet(Sheet sheet, List<List<Object>> dataList, int startRowIndex) {
        if (dataList == null || dataList.isEmpty()) {
            return;
        }

        for (int i = 0; i < dataList.size(); i++) {
            Row row = sheet.createRow(startRowIndex + i);
            List<Object> rowData = dataList.get(i);

            // 确保每行有22列数据
            for (int j = 0; j < 22; j++) {
                Cell cell = row.createCell(j);
                Object value = j < rowData.size() ? rowData.get(j) : "";
                setCellValue(cell, value);
            }
        }
    }

    /**
     * 设置单元格值
     */
    private void setCellValue(Cell cell, Object value) {
        if (value == null) {
            cell.setCellValue("");
        } else if (value instanceof String) {
            cell.setCellValue((String) value);
        } else if (value instanceof Integer) {
            cell.setCellValue((Integer) value);
        } else if (value instanceof Double) {
            cell.setCellValue((Double) value);
        } else if (value instanceof Boolean) {
            cell.setCellValue((Boolean) value);
        } else if (value instanceof Long) {
            cell.setCellValue((Long) value);
        } else {
            cell.setCellValue(value.toString());
        }
    }

    /**
     * 生成测试数据
     */
    private List<List<Object>> generateTestData() {
        List<List<Object>> dataList = new java.util.ArrayList<>();

        // 生成5行测试数据
        for (int i = 1; i <= 5; i++) {
            List<Object> row = new java.util.ArrayList<>();

            // 22列测试数据
            row.add("数据" + i + "-1");
            row.add("数据" + i + "-2");
            row.add(i * 100);
            row.add(i * 50.5);
            row.add(i % 2 == 0);

            // 填充剩余列
            for (int j = 5; j < 22; j++) {
                row.add("值" + i + "-" + (j + 1));
            }

            dataList.add(row);
        }

        return dataList;
    }
}