package com.ruoyi.common.easyExcel;

import com.alibaba.excel.write.handler.AbstractSheetWriteHandler;
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
import org.apache.poi.ss.usermodel.DataValidation;
import org.apache.poi.ss.usermodel.DataValidationConstraint;
import org.apache.poi.ss.usermodel.DataValidationHelper;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.util.CellRangeAddressList;

import java.util.Map;

public class MultiDropdownWriteHandler extends AbstractSheetWriteHandler {

    // 使用 Map 存储列索引和对应的下拉选项
    private final Map<Integer, String[]> dropdownOptionsMap;

    public MultiDropdownWriteHandler(Map<Integer, String[]> dropdownOptionsMap) {
        this.dropdownOptionsMap = dropdownOptionsMap;
    }

    @Override
    public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
        Sheet sheet = writeSheetHolder.getSheet();
        DataValidationHelper helper = sheet.getDataValidationHelper();

        for (Map.Entry<Integer, String[]> entry : dropdownOptionsMap.entrySet()) {
            int columnIndex = entry.getKey();
            String[] options = entry.getValue();

            // 定义下拉框应用的单元格范围（这里假设从第1行到第100行）
//            CellRangeAddressList addressList = new CellRangeAddressList(1, 100, columnIndex, columnIndex);
            CellRangeAddressList addressList = new CellRangeAddressList(0, 100, columnIndex, columnIndex);

            // 创建下拉选项
            DataValidationConstraint constraint = helper.createExplicitListConstraint(options);
            DataValidation dataValidation = helper.createValidation(constraint, addressList);

            // 设置校验
            dataValidation.setSuppressDropDownArrow(true);
            dataValidation.setShowErrorBox(true);

            // 将校验应用到 sheet
            sheet.addValidationData(dataValidation);
        }
    }
}