/**
 * 定義Excel列下拉列表屬性的注解。
 */
@Documented
@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface ExcelSelected {
    /**
     * 方式一：固定的下拉選項(xiàng)
     */
    String[] source() default {};

    /**
     * 方式二：提供動(dòng)態(tài)下拉選項(xiàng)的類
     */
    Class<? extends ExcelDynamicSelect>[] sourceClass() default {};

    /**
     * 方式三：基于碼值從數(shù)據(jù)庫查詢數(shù)據(jù)
     */
    String codeField() default "";

    /**
     * 下拉列表的起始行（默認(rèn)從第二行開始）。
     */
    int firstRow() default 1;

    /**
     * 下拉列表的結(jié)束行（默認(rèn)到第65536行）。
     */
    int lastRow() default 65536;
}

/**
 * 動(dòng)態(tài)下拉列表數(shù)據(jù)提供者接口。
 */
public interface ExcelDynamicSelect {
    /**
     * 獲取動(dòng)態(tài)生成的下拉列表選項(xiàng)。
     * 
     * @return 下拉選項(xiàng)數(shù)組。
     */
    String[] getSource();
}

/**
 * 根據(jù) ExcelSelected 注解解析下拉列表數(shù)據(jù)源。
 */
@Data
@Slf4j
public class ExcelSelectedResolve {
    /**
     * 下拉選項(xiàng)數(shù)組。
     */
    private String[] source;

    /**
     * 下拉列表的起始行。
     */
    private int firstRow;

    /**
     * 下拉列表的結(jié)束行。
     */
    private int lastRow;

    /**
     * 解析下拉列表數(shù)據(jù)來源
     *
     * @param excelSelected 下拉框注解對(duì)象
     * @return 下拉框選項(xiàng)數(shù)組
     */
    public String[] resolveSelectedSource(ExcelSelected excelSelected) {
        if (excelSelected == null) {
            return null;
        }

        // 方式一：獲取固定下拉框的內(nèi)容
        String[] source = excelSelected.source();
        if (source.length > 0) {
            return source;
        }

        // 方式二：獲取動(dòng)態(tài)下拉框的內(nèi)容
        Class<? extends ExcelDynamicSelect>[] classes = excelSelected.sourceClass();
        if (classes.length > 0) {
            try {
                ExcelDynamicSelect excelDynamicSelect = classes[0].newInstance();
                String[] dynamicSelectSource = excelDynamicSelect.getSource();
                if (dynamicSelectSource != null && dynamicSelectSource.length > 0) {
                    return dynamicSelectSource;
                }
            } catch (InstantiationException | IllegalAccessException e) {
                log.error("解析動(dòng)態(tài)下拉框數(shù)據(jù)異常", e);
            }
        }

        // 方式三：獲取碼值下拉數(shù)據(jù)（動(dòng)態(tài)下拉）
        String codeField = excelSelected.codeField();
        if (ObjectUtils.isNotEmpty(codeField)) {
            try {
                // 這里就是通過碼值查詢碼值表，寫死了，每次傳碼值查詢即可
                String[] codeFieldSource = SpringUtil.getBean(xxxService.class)
                        .selectByCode(codeField);
                if (ObjectUtils.isNotEmpty(codeFieldSource)) {
                    return codeFieldSource;
                }
            } catch (Exception e) {
                log.error("解析動(dòng)態(tài)下拉框(碼值)數(shù)據(jù)異常", e);
            }
        }
        
        return null;
    }
}

/**
 * 處理Excel下拉列表的SheetWriteHandler實(shí)現(xiàn)類。
 */
@Slf4j
@Data
public class SelectedSheetWriteHandler implements SheetWriteHandler {
    // 存儲(chǔ)列索引與對(duì)應(yīng)下拉列表解析器的映射
    private Map<Integer, ExcelSelectedResolve> selectedMap = new HashMap<>();

    /**
     * 構(gòu)造方法，解析表頭類中的下拉列表注解信息。
     *
     * @param head 表頭類。
     */
    public SelectedSheetWriteHandler(Class<?> head) {
        // 獲取所有聲明的字段
        Field[] fields = head.getDeclaredFields();
        for (int i = 0; i < fields.length; i++) {
            Field field = fields[i];
            // 獲取 ExcelSelected 注解
            ExcelSelected selected = field.getAnnotation(ExcelSelected.class);
            ExcelProperty property = field.getAnnotation(ExcelProperty.class);
            if (selected != null) {
                ExcelSelectedResolve resolve = new ExcelSelectedResolve();
                // 解析下拉列表數(shù)據(jù)源
                String[] source = resolve.resolveSelectedSource(selected);
                if (source != null && source.length > 0) {
                    resolve.setSource(source);
                    resolve.setFirstRow(selected.firstRow());
                    resolve.setLastRow(selected.lastRow());
                    // 使用注解中的索引或字段順序作為列索引
                    if (property != null && property.index() >= 0) {
                        selectedMap.put(property.index(), resolve);
                    } else {
                        selectedMap.put(i, resolve);
                    }
                }
            }
        }
    }

    /**
     * 在創(chuàng)建Sheet之前調(diào)用的方法。
     */
    @Override
    public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
        // 此處無需操作，保持空實(shí)現(xiàn)
    }

    /**
     * 在Sheet創(chuàng)建后調(diào)用的方法，用于設(shè)置Excel下拉列表。
     *
     * @param writeWorkbookHolder 寫入的工作簿持有者。
     * @param writeSheetHolder    寫入的Sheet持有者。
     */
    @Override
    public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
        Sheet sheet = writeSheetHolder.getSheet();
        Workbook workbook = sheet.getWorkbook();
        // SXSSFWorkbook 是 Apache POI 庫中用于處理大文件的一種特殊工作簿類型
        SXSSFWorkbook sw = (SXSSFWorkbook) workbook;
        // 1.創(chuàng)建一個(gè)隱藏的sheet，名稱為hidden，用于存儲(chǔ)下拉列表選項(xiàng)
        String hiddenName = "hidden";
        XSSFSheet hiddenSheet = sw.getXSSFWorkbook().createSheet(hiddenName);
        // 將隱藏的sheet設(shè)置為不可見
        workbook.setSheetHidden(workbook.getSheetIndex(hiddenName), true);
        // 創(chuàng)建數(shù)據(jù)驗(yàn)證輔助器
        DataValidationHelper helper = sheet.getDataValidationHelper();
        // 為每個(gè)需要下拉列表的列創(chuàng)建數(shù)據(jù)驗(yàn)證
        selectedMap.forEach((index, selectedResolve) -> {
            // 設(shè)置下拉列表的范圍：起始行，結(jié)束行，起始列，結(jié)束列
            CellRangeAddressList rangeList = new CellRangeAddressList(
                    selectedResolve.getFirstRow(),
                    selectedResolve.getLastRow(),
                    index,
                    index
            );
            // 在隱藏的sheet中生成下拉列表選項(xiàng)值
            String[] values = selectedResolve.getSource();
            generateSelectValue(hiddenSheet, index, values);
            // 獲取Excel列標(biāo)，例如A, B, AA
            String excelLine = getExcelLine(index);
            // 引用隱藏sheet中的單元格區(qū)域，例如hidden!$H$1:$H$50
            String refers = hiddenName + "!$" + excelLine + "$1:$" + excelLine + "$" + values.length;
            // 使用引用的內(nèi)容作為下拉列表的值
            DataValidationConstraint constraint = helper.createFormulaListConstraint(refers);
            DataValidation validation = helper.createValidation(constraint, rangeList);

            // 設(shè)置驗(yàn)證屬性，阻止輸入非下拉選項(xiàng)的值
            validation.setErrorStyle(DataValidation.ErrorStyle.STOP);
            validation.setShowErrorBox(true);
            validation.setSuppressDropDownArrow(true);
            validation.createErrorBox("提示", "請(qǐng)輸入下拉選項(xiàng)中的內(nèi)容");
            // 將驗(yàn)證添加到當(dāng)前的sheet中
            sheet.addValidationData(validation);
        });
    }

    /**
     * 獲取Excel列標(biāo)（例如：A-Z, AA-ZZ）。
     *
     * @param num 列索引，從0開始。
     * @return Excel列標(biāo)字符串。
     */
    public static String getExcelLine(int num) {
        StringBuilder line = new StringBuilder();
        // 計(jì)算列標(biāo)，使用字母表示，例如 A, B, ..., Z, AA, AB, ...
        int first = num / 26;
        int second = num % 26;
        if (first > 0) {
            line.append((char) ('A' + first - 1));
        }
        line.append((char) ('A' + second));
        return line.toString();
    }

    /**
     * 在隱藏的sheet中生成下拉列表選項(xiàng)值。
     *
     * @param sheet  隱藏的sheet對(duì)象。
     * @param col    列索引。
     * @param values 下拉列表選項(xiàng)值數(shù)組。
     */
    private void generateSelectValue(Sheet sheet, int col, String[] values) {
        // 將下拉列表選項(xiàng)值寫入隱藏的sheet中，每個(gè)選項(xiàng)值占用一行
        for (int i = 0, length = values.length; i < length; i++) {
            Row row = sheet.getRow(i);
            if (row == null) {
                row = sheet.createRow(i);
            }
            // 在指定列中創(chuàng)建單元格并設(shè)置下拉列表選項(xiàng)值
            row.createCell(col).setCellValue(values[i]);
        }
    }
}

@Data
public class Employee {

    @ExcelProperty(value = "用戶編號(hào)")
    private Integer id;

    @ExcelProperty(value = "姓名")
    private String name;

    @ExcelProperty(value = "性別")
    @ExcelSelected(source = {"男", "女"})
    private String gender;

    @ExcelProperty(value = "職位")
    @ExcelSelected(sourceClass = {PositionDynamicSelect.class})
    private String position;

    @ExcelProperty(value = "國家")
    @ExcelSelected(codeField = "country_code")
    private String country;
}

public class PositionDynamicSelect implements ExcelDynamicSelect {
    @Override
    public String[] getSource() {
        // 動(dòng)態(tài)生成職位列表
        return new String[]{"軟件工程師", "項(xiàng)目經(jīng)理", "人事專員", "財(cái)務(wù)分析師"};
    }
}

public class EmployeeExcelTest {
    public static void main(String[] args) {
        String fileName = "/Users/xuchang/Documents/employee.xlsx";
        EasyExcel.write(fileName, Employee.class)
                .registerWriteHandler(new SelectedSheetWriteHandler(Employee.class))
                .sheet().doWrite((Collection<?>) null);
    }
}