package com.estate.expert.admin.io.processor;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.xssf.usermodel.XSSFSheet;

import java.util.ArrayList;
import java.util.List;

/**
 * 行数据处理器基础类
 * 
 * 用于处理Excel文件的读取和写入
 */
public abstract class RowDataProcessorBase<T> extends DataProcessorBase<List<T>> {

    protected abstract List<String> buildRowData(T data);

    /**
     * 构建行数据（支持不同类型）
     * 子类可以重写此方法以返回包含不同类型数据的列表（如数字、字符串等）
     * 如果子类重写了此方法，writeSheet 将使用此方法而不是 buildRowData
     *
     * @param data 数据对象
     * @return 行数据列表，可以包含 String、Number 等类型
     */
    protected List<Object> buildRowDataObjects(T data) {
        return null;
    }

    protected T parseRowData(int rowIndex, List<String> rowData) {
        return null;
    }

    @Override
    protected List<T> parseSheet(XSSFSheet sheet) {
        var result = new ArrayList<T>();

        int columnCount = getHeaders().length;

        for(int i = 1; i < sheet.getPhysicalNumberOfRows(); i++) {
            var row = sheet.getRow(i);
            // 跳过空行
            if(row == null) {
                continue;
            }
            var data = new ArrayList<String>();
            for(int j = 0; j < columnCount; j++) {
                var cell = row.getCell(j);
                if(cell == null) {
                    data.add("");
                } else {
                    data.add(getCellStringValue(cell));
                }
            }
            var rowData = parseRowData(i, data);
            if(rowData != null) {
                result.add(rowData);
            }
        }
        return result;
    }

    @Override
    protected void writeSheet(XSSFSheet sheet, List<T> data) {
        var headers = getHeaders();

        int index = 1;
        for (var d : data) {
            // 优先使用 buildRowDataObjects，如果未实现则使用 buildRowData
            List<Object> rowDataObjects = buildRowDataObjects(d);
            List<?> rowData;

            if (rowDataObjects != null) {
                rowData = rowDataObjects;
            } else {
                rowData = buildRowData(d);
            }

            if (rowData == null)
                continue;

            if (rowData.size() != headers.length) {
                throw new IllegalArgumentException("数据列数与表头列数不一致");
            }

            var row = sheet.createRow(index++);
            for (int i = 0; i < rowData.size(); i++) {
                var cell = row.createCell(i);
                var value = rowData.get(i);

                // 根据数据类型设置单元格值
                if (value == null || value.equals("-")) {
                    cell.setCellValue("-");
                } else if (value instanceof Number) {
                    cell.setCellValue(((Number) value).doubleValue());
                } else if (value instanceof Boolean) {
                    cell.setCellValue((Boolean) value);
                } else {
                    cell.setCellValue(value.toString());
                }
            }
        }
    }

    /**
     * 获取单元格的字符串值
     * @param cell 单元格
     * @return 字符串值
     */
    private String getCellStringValue(Cell cell) {
        if(cell == null) {
            return "";
        }
        
        CellType cellType = cell.getCellType();
        if(cellType == CellType.STRING) {
            return cell.getStringCellValue();
        } else if(cellType == CellType.NUMERIC) {
            // 处理数字类型，避免科学计数法
            double numericValue = cell.getNumericCellValue();
            // 如果是整数，返回整数格式；否则返回小数格式
            if(numericValue == (long) numericValue) {
                return String.valueOf((long) numericValue);
            } else {
                return String.valueOf(numericValue);
            }
        } else if(cellType == CellType.BOOLEAN) {
            return String.valueOf(cell.getBooleanCellValue());
        } else if(cellType == CellType.FORMULA) {
            // 公式单元格，获取计算后的值
            try {
                CellType cachedType = cell.getCachedFormulaResultType();
                if(cachedType == CellType.STRING) {
                    return cell.getStringCellValue();
                } else if(cachedType == CellType.NUMERIC) {
                    double numericValue = cell.getNumericCellValue();
                    if(numericValue == (long) numericValue) {
                        return String.valueOf((long) numericValue);
                    } else {
                        return String.valueOf(numericValue);
                    }
                } else if(cachedType == CellType.BOOLEAN) {
                    return String.valueOf(cell.getBooleanCellValue());
                } else {
                    return cell.getCellFormula();
                }
            } catch (Exception e) {
                return cell.getCellFormula();
            }
        } else if(cellType == CellType.BLANK) {
            return "";
        } else {
            return "";
        }
    }

}