aegon-gen 1.0.0

package/src/components/Export/excelExport.js

@@ -0,0 +1,547 @@
+// excelExport.js
+import ExcelJS from "exceljs";
+
+/**
+ * 工具函数：安全获取单元格值的长度
+ * @param {any} cellValue Excel单元格值
+ * @returns {number} 单元格值的字符长度（数字转字符串，空值返回0）
+ */
+const getCellValueLength = (cellValue) => {
+  if (cellValue === null || cellValue === undefined) return 0;
+  // 处理不同类型的CellValue，统一转为字符串后计算长度
+  let strValue = "";
+  if (typeof cellValue === "string") {
+    strValue = cellValue;
+  } else if (typeof cellValue === "number" || typeof cellValue === "boolean") {
+    strValue = cellValue.toString();
+  } else if (cellValue instanceof Date) {
+    strValue = cellValue.toISOString();
+  } else {
+    // 处理富文本、公式等特殊类型，转为字符串
+    strValue = String(cellValue);
+  }
+  return strValue.length;
+};
+
+/**
+ * 判断是否为数字的辅助函数
+ * @param {any} value 单元格值
+ * @returns {boolean} 是否为数字（包括带小数点的数字字符串）
+ */
+const isNumberValue = (value) => {
+  if (value === null || value === undefined) return false;
+  if (typeof value === "number") return true;
+  if (typeof value !== "string") return false;
+
+  // 移除千分位逗号和货币符号
+  const cleanedValue = value.replace(/[,$]/g, "").replace(/\s/g, "");
+
+  // 检查是否为数字（包括负数、小数、带单位的数字）
+  return /^-?\d+(\.\d+)?/.test(cleanedValue);
+};
+
+/**
+ * 按日期分组数据
+ * @param {Array<Array<string>>} data 原始数据（包含表头）
+ * @param {string} dateColumnName 日期列名，默认为"會計日"
+ * @returns {Object<string, Array<Array<string>>>} 按日期分组的对象，键为日期，值为该日期的数据数组
+ */
+const groupDataByDate = (data, dateColumnName = "會計日") => {
+  if (data.length <= 1) return {}; // 只有表头或空数据
+
+  const header = data[0] || [];
+  const dateGroups = {};
+
+  // 安全检查：确保header存在
+  if (!header || !Array.isArray(header)) {
+    dateGroups["全部数据"] = data;
+    return dateGroups;
+  }
+
+  const dateColumnIndex = header.indexOf(dateColumnName);
+
+  if (dateColumnIndex === -1) {
+    // 如果没有找到日期列，将所有数据放在一个分组中
+    dateGroups["全部数据"] = data;
+    return dateGroups;
+  }
+
+  // 遍历数据行（跳过表头）
+  for (let i = 1; i < data.length; i++) {
+    const row = data[i] || [];
+    const date = row[dateColumnIndex];
+
+    if (!date) continue;
+
+    if (!dateGroups[date]) {
+      // 为新日期创建分组，包含表头
+      dateGroups[date] = [header];
+    }
+
+    // 添加数据行到对应日期的分组
+    dateGroups[date].push(row);
+  }
+
+  return dateGroups;
+};
+
+/**
+ * 配置工作表的样式
+ * @param {Worksheet} worksheet Excel工作表对象
+ * @param {Array<string>} header 表头数组
+ * @param {Object} options 配置选项
+ */
+const configureWorksheetStyle = (worksheet, header, options = {}) => {
+  const {
+    rightAlignedColumns = [],
+    centerAlignedColumns = [],
+    headerStyle = {},
+    dataRowStyle = {},
+    columnWidthLimit = 50,
+  } = options;
+
+  if (!header || !Array.isArray(header)) {
+    return;
+  }
+
+  // 1. 配置表头样式
+  const headerRow = worksheet.getRow(1);
+  if (headerRow) {
+    headerRow.eachCell((cell) => {
+      // 默认表头样式
+      const defaultHeaderStyle = {
+        name: "Calibri",
+        size: 11,
+        bold: true,
+        color: { argb: "FF000000" }, // 黑色文本
+      };
+
+      cell.font = { ...defaultHeaderStyle, ...headerStyle.font };
+      cell.fill = {
+        type: "pattern",
+        pattern: "solid",
+        fgColor: { argb: "FFFFFFFF" }, // 白色背景
+        ...headerStyle.fill,
+      };
+      cell.alignment = {
+        vertical: "middle",
+        horizontal: "left", // 左对齐
+        wrapText: true,
+        ...headerStyle.alignment,
+      };
+      cell.border = {
+        top: { style: "thin", color: { argb: "FF000000" } },
+        left: { style: "thin", color: { argb: "FF000000" } },
+        bottom: { style: "thin", color: { argb: "FF000000" } },
+        right: { style: "thin", color: { argb: "FF000000" } },
+        ...headerStyle.border,
+      };
+    });
+  }
+
+  // 2. 配置数据行样式
+  worksheet.eachRow((row, rowIndex) => {
+    // 跳过表头行
+    if (rowIndex === 1) return;
+
+    row.eachCell((cell, cellNumber) => {
+      // 基础字体设置
+      const defaultFontStyle = {
+        name: "Calibri",
+        size: 11,
+        color: { argb: "FF000000" }, // 黑色文本
+      };
+
+      cell.font = { ...defaultFontStyle, ...dataRowStyle.font };
+
+      // 获取列名
+      const columnIndex = cellNumber - 1;
+      const columnName = columnIndex < header.length ? header[columnIndex] : "";
+
+      // 特殊列处理：垂直水平居中
+      if (columnName && centerAlignedColumns.includes(columnName)) {
+        cell.alignment = {
+          vertical: "middle",
+          horizontal: "center", // 水平居中
+          wrapText: true,
+          ...dataRowStyle.alignment,
+        };
+      }
+      // 数字列处理：右对齐
+      else if (
+        (columnName && rightAlignedColumns.includes(columnName)) ||
+        isNumberValue(cell.value)
+      ) {
+        cell.alignment = {
+          vertical: "middle",
+          horizontal: "right", // 右对齐
+          wrapText: true,
+          ...dataRowStyle.alignment,
+        };
+      }
+      // 其他列：左对齐
+      else {
+        cell.alignment = {
+          vertical: "middle",
+          horizontal: "left", // 左对齐
+          wrapText: true,
+          ...dataRowStyle.alignment,
+        };
+      }
+
+      // 数据行边框
+      cell.border = {
+        top: { style: "thin", color: { argb: "FF000000" } },
+        left: { style: "thin", color: { argb: "FF000000" } },
+        bottom: { style: "thin", color: { argb: "FF000000" } },
+        right: { style: "thin", color: { argb: "FF000000" } },
+        ...dataRowStyle.border,
+      };
+    });
+  });
+
+  // 3. 列宽自适应
+  worksheet.columns?.forEach((column) => {
+    // 初始化默认宽度
+    let maxWidth = 10;
+
+    // 安全处理column.values（空值判断）
+    if (column.values && column.values.length > 0) {
+      // 明确指定reduce的返回类型为number
+      maxWidth = column.values.reduce((w, cellValue) => {
+        // 安全获取单元格值长度（处理所有CellValue类型）
+        const cellLength = getCellValueLength(cellValue);
+        return Math.max(w, cellLength);
+      }, 10); // 初始值设为数字10
+    }
+
+    // 确保maxWidth是数字后再计算
+    column.width = Math.min(maxWidth + 2, columnWidthLimit);
+  }); 
+};
+
+/**
+ * 数据分组函数：根据档案编号分组数据
+ * @param {Array<Array<any>>} data 原始数据数组（不包含表头)
+ * @param {string} groupColumnName 分组列名，默认为第一列
+ * @returns {Array} 分组后的数据，包含每个组的起始行和结束行索引
+ */
+const groupDataByColumn = (data, groupColumnName = "") => {
+  const groups = [];
+
+  let currentGroup = null;
+  const groupColumnIndex = groupColumnName
+    ? data[0]?.indexOf(groupColumnName)
+    : 0;
+
+  data.forEach((row, index) => {
+    const groupValue = row[groupColumnIndex || 0]; // 分组值
+
+    if (!currentGroup || currentGroup.groupValue !== groupValue) {
+      // 开始新的分组
+      if (currentGroup) {
+        currentGroup.endRow = index - 1;
+        groups.push(currentGroup);
+      }
+      currentGroup = {
+        groupValue,
+        startRow: index,
+        endRow: index,
+        rows: [row],
+      };
+    } else {
+      // 继续当前分组
+      currentGroup.endRow = index;
+      currentGroup.rows.push(row);
+    }
+  });
+
+  // 添加最后一个分组
+  if (currentGroup) {
+    groups.push(currentGroup);
+  }
+
+  return groups;
+};
+
+/**
+ * 合并相同分组的数据
+ * @param {Worksheet} worksheet Excel工作表对象
+ * @param {Array<Array<any>>} data 原始数据（不包含表头）
+ * @param {Array<string>} columnsToMerge 需要合并的列名
+ * @param {Array<string>} headers 表头数组
+ */
+const mergeCellsByGroup = (worksheet, data, columnsToMerge, headers) => {
+  if (
+    !data ||
+    data.length === 0 ||
+    !columnsToMerge ||
+    columnsToMerge.length === 0
+  ) {
+    return;
+  }
+
+  const groups = groupDataByColumn(data);
+  const mergedRanges = new Set();
+
+  groups.forEach((group) => {
+    // 如果组内只有一行数据，不需要合并
+    if (group.startRow === group.endRow) {
+      return;
+    }
+
+    columnsToMerge.forEach((columnName) => {
+      // 获取该列名的所有索引（处理重复列名）
+      const columnIndexes = [];
+      headers.forEach((header, index) => {
+        if (header === columnName) {
+          columnIndexes.push(index);
+        }
+      });
+
+      columnIndexes.forEach((columnIndex) => {
+        if (columnIndex === -1) return;
+
+        // Excel行号从1开始，加上表头行
+        const startRow = group.startRow + 2; // +2是因为Excel行从1开始，表头占第1行
+        const endRow = group.endRow + 2;
+
+        // 创建唯一标识符来避免重复合并
+        const rangeKey = `${startRow}-${endRow}-${columnIndex + 1}`;
+
+        // 检查是否已经合并过这个范围
+        if (mergedRanges.has(rangeKey)) {
+          return;
+        }
+
+        try {
+          // 合并单元格
+          worksheet.mergeCells(
+            startRow,
+            columnIndex + 1,
+            endRow,
+            columnIndex + 1
+          );
+          mergedRanges.add(rangeKey);
+
+          // 获取合并后的单元格值（取第一个非空值）
+          let mergedValue = "";
+
+          // 修复：确保rows存在且非空
+          if (group.rows && group.rows.length > 0) {
+            for (let i = 0; i < group.rows.length; i++) {
+              const rowData = group.rows[i];
+              if (rowData && rowData[columnIndex] !== undefined) {
+                const value = rowData[columnIndex];
+                if (
+                  value !== null &&
+                  value !== undefined &&
+                  String(value).trim() !== ""
+                ) {
+                  mergedValue = String(value);
+                  break;
+                }
+              }
+            }
+          }
+
+          // 设置合并后单元格的值
+          const cell = worksheet.getCell(startRow, columnIndex + 1);
+          cell.value = mergedValue;
+        } catch (error) {
+          console.warn(`合并单元格失败: ${error.message}`, {
+            startRow,
+            columnIndex: columnIndex + 1,
+            endRow,
+            columnName,
+            groupValue: group.groupValue,
+          });
+        }
+      });
+    });
+  });
+};
+
+/**
+ * 获取日期范围字符串
+ * @param {Array<Array<string>>} data 数据数组
+ * @param {string} dateColumnName 日期列名，默认为"會計日"
+ * @returns {string} 日期范围字符串，如 "2025-12-10~2025-12-12"
+ */
+const getDateRangeString = (data, dateColumnName = "會計日") => {
+  if (data.length <= 1) return "";
+
+  const header = data[0] || [];
+  const dateColumnIndex = header.indexOf(dateColumnName);
+
+  if (dateColumnIndex === -1) return "";
+
+  const dates = [];
+
+  // 收集所有日期
+  for (let i = 1; i < data.length; i++) {
+    const row = data[i] || [];
+    const date = row[dateColumnIndex];
+    if (date && !dates.includes(date)) {
+      dates.push(date);
+    }
+  }
+
+  if (dates.length === 0) return "";
+
+  // 排序日期
+  dates.sort();
+
+  if (dates.length === 1) {
+    return dates[0];
+  } else {
+    return `${dates[0]}~${dates[dates.length - 1]}`;
+  }
+};
+
+/**
+ * 统一的Excel导出函数
+ * @param {Array<Array<string>>} data 要导出的数据（二维数组，包含表头）
+ * @param {Object} options 配置选项
+ * @returns {Promise<void>}
+ */
+export const guiExportToExcel = async (data, options = {}) => {
+  const {
+    excelName = "导出数据",
+    sheetName = "数据",
+    dateColumnName = "會計日",
+    groupColumnName = "",
+    columnsToMerge = [],
+    rightAlignedColumns = [],
+    centerAlignedColumns = [],
+    exportType = "single", // 'single' 或 'multi-sheet'
+    headerStyle = {},
+    dataRowStyle = {},
+    columnWidthLimit = 50,
+    customFileName = "",
+  } = options;
+
+  const workbook = new ExcelJS.Workbook();
+
+  if (exportType === "multi-sheet" && dateColumnName) {
+    // 1. 按日期分组数据（多工作表模式）
+    const dateGroups = groupDataByDate(data, dateColumnName);
+
+    if (Object.keys(dateGroups).length === 0) {
+      // 如果没有数据，只创建一个空的工作表
+      const worksheet = workbook.addWorksheet(sheetName);
+      const header = data[0] || [];
+      worksheet.addRows(data.length > 0 ? data : [header]);
+      configureWorksheetStyle(worksheet, header, {
+        rightAlignedColumns,
+        centerAlignedColumns,
+        headerStyle,
+        dataRowStyle,
+        columnWidthLimit,
+      });
+    } else {
+      // 为每个日期创建独立的工作表
+      Object.keys(dateGroups).forEach((date) => {
+        const dateData = dateGroups[date] || [];
+
+        // 创建工作表，工作表名称为日期
+        let sheetName = date;
+        if (sheetName === "全部数据") {
+          sheetName = "汇总";
+        } else {
+          // 移除可能影响工作表名称的字符
+          sheetName = sheetName.replace(/[\/\\\*\?\[\]:]/g, "-");
+          // 限制工作表名称长度（Excel限制31个字符）
+          if (sheetName.length > 31) {
+            sheetName = sheetName.substring(0, 31);
+          }
+        }
+
+        const worksheet = workbook.addWorksheet(sheetName);
+
+        // 写入数据
+        if (dateData.length > 0) {
+          worksheet.addRows(dateData);
+
+          // 配置工作表样式
+          const header = dateData[0] || [];
+          configureWorksheetStyle(worksheet, header, {
+            rightAlignedColumns,
+            centerAlignedColumns,
+            headerStyle,
+            dataRowStyle,
+            columnWidthLimit,
+          });
+
+          // 如果需要合并单元格
+          if (columnsToMerge.length > 0 && groupColumnName) {
+            const dataWithoutHeader = dateData.slice(1);
+            mergeCellsByGroup(
+              worksheet,
+              dataWithoutHeader,
+              columnsToMerge,
+              header
+            );
+          }
+        }
+      });
+    }
+  } else {
+    // 2. 单工作表模式
+    const worksheet = workbook.addWorksheet(sheetName);
+    const header = data[0] || [];
+
+    // 写入数据
+    worksheet.addRows(data);
+
+    // 配置工作表样式
+    configureWorksheetStyle(worksheet, header, {
+      rightAlignedColumns,
+      centerAlignedColumns,
+      headerStyle,
+      dataRowStyle,
+      columnWidthLimit,
+    });
+
+    // 如果需要合并单元格
+    if (columnsToMerge.length > 0 && groupColumnName) {
+      const dataWithoutHeader = data.slice(1);
+      mergeCellsByGroup(worksheet, dataWithoutHeader, columnsToMerge, header);
+    }
+  }
+
+  // 3. 导出 Excel
+  try {
+    const buffer = await workbook.xlsx.writeBuffer();
+    const blob = new Blob([buffer], {
+      type: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+    });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement("a");
+
+    // 构建文件名
+    let fileName = customFileName || excelName;
+    if (exportType === "multi-sheet" && dateColumnName) {
+      const dateRange = getDateRangeString(data, dateColumnName);
+      if (dateRange) {
+        fileName = `${excelName}_${dateRange}`;
+      }
+    }
+
+    a.href = url;
+    a.download = `${fileName}.xlsx`;
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+    URL.revokeObjectURL(url);
+    return Promise.resolve();
+  } catch (error) {
+    return Promise.reject(error);
+  }
+};
+
+export default {
+  guiExportToExcel,
+  getDateRangeString,
+  groupDataByDate,
+};