@lambo-design/shared 1.0.0-beta.36 → 1.0.0-beta.361

package/utils/excel.js

@@ -1,9 +1,14 @@
 /* eslint-disable */
-import XLSX from 'xlsx';
-import * as cpexcel from 'xlsx/dist/cpexcel.full.mjs';
+import XLSX from '@lambo-design/xlsx';
+import * as cpexcel from '@lambo-design/xlsx/dist/cpexcel.full.mjs';
 XLSX.set_cptable(cpexcel);
 import {deepCopy} from "./assist";
+import XLSXStyle from '@lambo-design/xlsx-style'
 
+/**
+ * 表头内容转换规则
+ * @type {{'': string, '#': string, 序号: string, 操作: string, 编号: string}}
+ */
 const titleRules = {
   '': '',
   '#': '#',
@@ -11,14 +16,305 @@ const titleRules = {
   '编号': '编号',
   '操作': '操作',
 }
+/**
+ * 列类型转换规则
+ * @type {{'single-selection': string, select: string, selection: string, checkbox: string, index: string, radio: string}}
+ */
 const typeRules = {
   'index': 'index',
   'select': 'select',
   'selection': 'selection',
-  'single-selection': 'single-selection'
+  'single-selection': 'single-selection',
+  'checkbox': 'checkbox',
+  'radio': 'radio'
 }
 
-function auto_width(ws, data) {
+/**
+ * 标题样式
+ * @type {{border: {top: {color: {rgb: string}, style: string}, left: {color: {rgb: string}, style: string}, bottom: {color: {rgb: string}, style: string}, right: {color: {rgb: string}, style: string}}, alignment: {horizontal: string, vertical: string}, fill: {fgColor: {rgb: string}, bgColor: {indexed: number}}}}
+ */
+let titleStyle = {
+  border: {
+    left: {style: 'thin', color: {rgb: "000000"}},
+    top: {style: 'thin', color: {rgb: "000000"}},
+    right: {style: 'thin', color: {rgb: "000000"}},
+    bottom: {style: 'thin', color: {rgb: "000000"}},
+  },
+  font: { bold: true },
+  alignment: {horizontal: "center", vertical: "center"}
+}
+/**
+ * 表头样式
+ * @type {{border: {top: {color: {rgb: string}, style: string}, left: {color: {rgb: string}, style: string}, bottom: {color: {rgb: string}, style: string}, right: {color: {rgb: string}, style: string}}, alignment: {horizontal: string, vertical: string}, fill: {fgColor: {rgb: string}, bgColor: {indexed: number}}}}
+ */
+let headStyle = {
+  border: {
+    left: {style: 'thin', color: {rgb: "000000"}},
+    top: {style: 'thin', color: {rgb: "000000"}},
+    right: {style: 'thin', color: {rgb: "000000"}},
+    bottom: {style: 'thin', color: {rgb: "000000"}},
+  },
+  alignment: {horizontal: "center", vertical: "center"},
+  fill: {bgColor: {indexed: 64}, fgColor: {rgb: "5A9BD5"}}
+}
+/**
+ * 表格斑马纹样式
+ * @type {{border: {top: {color: {rgb: string}, style: string}, left: {color: {rgb: string}, style: string}, bottom: {color: {rgb: string}, style: string}, right: {color: {rgb: string}, style: string}}, alignment: {vertical: string}, fill: {fgColor: {rgb: string}, bgColor: {indexed: number}}}}
+ */
+let bodyPairStyle = {
+  border: {
+    left: {style: 'thin', color: {rgb: "000000"}},
+    top: {style: 'thin', color: {rgb: "000000"}},
+    right: {style: 'thin', color: {rgb: "000000"}},
+    bottom: {style: 'thin', color: {rgb: "000000"}},
+  },
+  alignment: {vertical: "center"},
+  fill: {bgColor: {indexed: 64}, fgColor: {rgb: "DDEBF7"}}
+}
+/**
+ * 表格普通样式
+ * @type {{border: {top: {color: {rgb: string}, style: string}, left: {color: {rgb: string}, style: string}, bottom: {color: {rgb: string}, style: string}, right: {color: {rgb: string}, style: string}}, alignment: {vertical: string}}}
+ */
+let bodyStyle = {
+  border: {
+    left: {style: 'thin', color: {rgb: "000000"}},
+    top: {style: 'thin', color: {rgb: "000000"}},
+    right: {style: 'thin', color: {rgb: "000000"}},
+    bottom: {style: 'thin', color: {rgb: "000000"}},
+  },
+  alignment: {vertical: "center"}
+}
+
+/**
+ * 表格底部扩展行样式
+ * @type {alignment: {vertical: string}}
+ */
+let footerStyle = {
+  alignment: {vertical: "center"}
+}
+
+/**
+ * 过滤掉filterData中titleRules、typeRules相关的列
+ * @param filterData
+ * @param key 包含typeRules内容的属性名
+ * @param name 包含titleRules内容的属性名
+ * @returns {*}
+ */
+function title_filter_base(filterData,key,name) {
+  for (var i = 0; i < filterData.length; i++) {
+    if ((filterData[i].hasOwnProperty(key) && filterData[i][key]
+            && filterData[i][key] === typeRules[filterData[i][key]])
+        || filterData[i][name] === titleRules[filterData[i][name]]) {
+      filterData.splice(i, 1)
+      i--
+    }
+  }
+  return filterData
+}
+
+function title_filter(data) {
+  let filterData = deepCopy(data);
+  return title_filter_base(filterData,'type','title')
+}
+
+function title_el_filter(data) {
+  let filterData = deepCopy(data);
+  return title_filter_base(filterData,'property','label')
+}
+
+/**
+ * 组织表格导出时需要的表头单元格内容
+ * @param treeData 列描述
+ * @param key 单元格显示内容对应列描述中的属性名
+ * @returns {*[][]}
+ */
+function title_transform_base(treeData,key) {
+  // 找到树的最大深度
+  const maxDepth = findMaxDepth(treeData);
+
+  // 初始化结果数组，每一层都是一个空数组
+  const result = Array.from({ length: maxDepth }, () => []);
+
+  // 辅助函数，用于找到树的最大深度
+  function findMaxDepth(nodes) {
+    let maxDepth = 0;
+
+    function traverse(node, depth = 1) {
+      maxDepth = Math.max(maxDepth, depth);
+      if (node.children) {
+        node.children.forEach(child => traverse(child, depth + 1));
+      }
+    }
+
+    nodes.forEach(node => traverse(node));
+
+    return maxDepth;
+  }
+
+  // 辅助函数，用于创建并返回从根节点到当前节点的路径
+  function getPath(node, path = [], key) {
+    path.unshift(node[key]); // 将当前节点的标题添加到路径的开头
+    if (node.parent) {
+      return getPath(node.parent, path, key); // 递归调用，直到根节点
+    }
+    return path;
+  }
+
+  // 递归函数，用于遍历树的每个节点并收集标题
+  function traverseTree(node, depth = 0, parent = null, key) {
+    // 给节点添加 parent 属性，方便后续创建路径
+    node.parent = parent;
+
+    // 确保当前深度的结果数组已初始化
+    if (result.length <= depth) {
+      result.push([]);
+    }
+
+    // 获取从根节点到当前节点的路径
+    const path = getPath(node,[],key);
+
+    // 当前层级和子层级都增加当前节点
+    for (let i = depth; i < maxDepth; i++) {
+      result[i].push(node[key])
+    }
+    // 当前层级的父层级增加对应的父节点
+    for (let j = 0; j < depth; j++) {
+      result[j].push(path[j])
+    }
+
+
+    // 如果还有子节点，递归遍历子节点
+    if (node.children) {
+      for (let i = 0; i < maxDepth; i++) {
+        result[i].pop()
+      }
+      node.children.forEach(child => {
+        traverseTree(child, depth + 1, node, key)
+      });
+    }
+  }
+
+  // 遍历所有根节点并收集标题
+  treeData.forEach(rootNode => traverseTree(rootNode,0,null,key));
+
+  // 去除每个节点上的 parent 属性（如果需要）
+  function removeParentProps(nodes) {
+    nodes.forEach(node => {
+      delete node.parent;
+      if (node.children) {
+        removeParentProps(node.children);
+      }
+    });
+  }
+  removeParentProps(treeData); // 清理 treeData 中的 parent 属性（可选）
+
+  return result;
+}
+
+export const title_transform = (paramData) => {
+  let data = title_filter(paramData)
+  return title_transform_base(data,'title')
+}
+export const title_el_transform = (paramData) => {
+  let data = title_el_filter(paramData)
+  return title_transform_base(data,'label')
+}
+
+/**
+ * 表格导出时数据区域内容对应的属性名列表
+ * @param data 列描述
+ * @param key 表格导出时数据区域内容对应列描述中的属性名
+ * @returns {*[]}
+ */
+export const key_transform_base = (data,key) => {
+  let result = []
+  function traverse(node,key) {
+    if (node.children) {
+      node.children.forEach(child => traverse(child,key));
+    }else{
+      result.push(node[key])
+    }
+  }
+
+  data.forEach(node => traverse(node,key));
+
+  return result;
+}
+
+export const key_transform = (paramData) => {
+  let data = title_filter(paramData)
+  // let data = paramData
+  return key_transform_base(data,'key')
+}
+
+export const key_el_transform = (paramData) => {
+  let data = title_el_filter(paramData)
+  // let data = paramData
+  return key_transform_base(data,'property')
+}
+
+export const key_vxe_transform = (paramData) => {
+  let data = title_filter(paramData)
+  // let data = paramData
+  return key_transform_base(data,'field')
+}
+
+/**
+ * 表格导出时，表头数据中存储的枚举信息汇总
+ * @param data 列描述
+ * @param key  表格数据内容对应列描述的key
+ * @returns {*[]}
+ */
+export const enums_transform_base = (data,key) => {
+  let result = []
+  function traverse(node,key) {
+    if (node.children) {
+      node.children.forEach(child => traverse(child,key));
+    }else if (node.enums){
+      result.push({key:node[key],enums:node.enums})
+    }
+  }
+
+  data.forEach(node => traverse(node,key));
+  return result
+}
+
+export const enums_transform = (paramData) => {
+  let result = []
+  let data = title_filter(paramData)
+  // let data = paramData
+  return enums_transform_base(data,'key')
+}
+
+export const enums_el_transform = (paramData) => {
+  let result = []
+  let data = title_el_filter(paramData)
+  // let data = paramData
+  return enums_transform_base(data,'property')
+}
+
+export const enums_vxe_transform = (paramData) => {
+  let result = []
+  let data = title_filter(paramData)
+  // let data = paramData
+  return enums_transform_base(data,'field')
+}
+
+
+export const enums_to_value = (enums,jsonData)=>{
+  if(enums&&enums.length>0){
+    return jsonData.map(v => enums.map(e =>{
+      if(e.enums[v[e.key]]){
+        v[e.key] = e.enums[v[e.key]]
+      }
+      return v
+    })[0])
+  }else{
+    return jsonData
+  }
+}
+
+function auto_width(ws, data, startRow=0) {
   /*set worksheet max width per col*/
   const colWidth = data.map(row => row.map(val => {
     /*if null/undefined*/
@@ -34,7 +330,12 @@ function auto_width(ws, data) {
   }))
   /*start in the first row*/
   let result = colWidth[0];
-  for (let i = 1; i < colWidth.length; i++) {
+  if (startRow > 0) {
+    // 排除header中title和search数据
+    result = colWidth[startRow];
+  }
+  // 计算列宽时，查询条件行数据不参与计算
+  for (let i = startRow > 0 ? startRow : 1; i < colWidth.length; i++) {
     for (let j = 0; j < colWidth[i].length; j++) {
       if (result[j]['wch'] < colWidth[i][j]['wch']) {
         result[j]['wch'] = colWidth[i][j]['wch'] > 7 ? colWidth[i][j]['wch'] : 7;
@@ -44,33 +345,80 @@ function auto_width(ws, data) {
   ws['!cols'] = result;
 }
 
-function merge_title(ws, title) {
-  let mergeData = []
-  let flag = false
-  let row = 1, col = 1
-  for (let i = 0; i < title.length; i++) {
-    for (let j = 0; j < title[i].length; j++) {
-      flag = false, row = 1, col = 1
-      let merge = {s: {r: i, c: j}, e: {r: i, c: j}}
-      while ((j + row) < title[i].length && title[i][j] === title[i][j + row]) {
-        merge.e.c = j + row
-        row++
-        flag = true
+
+
+function merge_cell({ws,data,startRow}) {
+  let mergeData = [];
+
+  for (let i = 0; i < data.length; i++) {
+    for (let j = 0; j < data[i].length; j++) {
+      let currentCellValue = data[i][j];
+      let merge = { s: { r: i + startRow, c: j }, e: { r: i + startRow, c: j } };
+
+      // Check horizontal direction for merges
+      while (
+          j + 1 < data[i].length &&
+          data[i][j + 1] === currentCellValue &&
+          !isCellMerged(mergeData, { r: i + startRow, c: j + 1 })
+          ) {
+        merge.e.c = j + 1;
+        j++;
       }
-      while ((i + col) < title.length && title[i][j] === title[i + col][j]) {
-        merge.e.r = i + col
-        col++
-        flag = true
+
+      // Check vertical direction for merges
+      let originalRow = i;
+      while (
+          i + 1 < data.length &&
+          data[i + 1][j] === currentCellValue &&
+          !isCellMerged(mergeData, { r: i + startRow + 1, c: j })
+          ) {
+        i++;
+
+        // Update the starting row if it's the first iteration of the vertical loop
+        // if (i === originalRow + 1) {
+        //   merge.s.r = i;
+        // }
+
+        // Expand the end row
+        merge.e.r = i + startRow;
       }
-      if (flag) {
-        mergeData.push(merge)
+
+      // Add the merge object to the array if it's not empty
+      if (merge.e.c > merge.s.c || merge.e.r > merge.s.r) {
+        mergeData.push(merge);
+
+        // Adjust the outer loop variables since we might have incremented i inside the loop
+        i = originalRow;
+        j = merge.e.c;
       }
     }
   }
-  ws['!merges'] = ws['!merges'].concat(mergeData)
+
+  // return mergeData
+  // Add the merge ranges to the worksheet
+  if (ws['!merges']) {
+    ws['!merges'] = ws['!merges'].concat(mergeData);
+  } else {
+    ws['!merges'] = mergeData;
+  }
+
+}
+// Helper function to check if a cell is already part of a merge range
+function isCellMerged(mergeData, cell) {
+  for (let merge of mergeData) {
+    if (
+        cell.r >= merge.s.r &&
+        cell.r <= merge.e.r &&
+        cell.c >= merge.s.c &&
+        cell.c <= merge.e.c
+    ) {
+      return true;
+    }
+  }
+  return false;
 }
 
-function merge_content(ws, tableData, spanColumnKeys, titleLen) {
+function merge_content({ws, tableData, spanColumnKeys, startRow}) {
   if (spanColumnKeys == null || spanColumnKeys.length === 0) {
     return null;
   }
@@ -85,7 +433,7 @@ function merge_content(ws, tableData, spanColumnKeys, titleLen) {
   }
   for (let i = 0; i < dLen; i++) {
     for (let j = 0; j < kLen; j++) {
-      let merge = {s: {r: i + titleLen, c: j}, e: {r: i + titleLen, c: j}}
+      let merge = {s: {r: i + startRow, c: j}, e: {r: i + startRow, c: j}}
       flag = false
       //如果单元格和右侧内容相同则合并
       let n = 1;
@@ -105,7 +453,7 @@ function merge_content(ws, tableData, spanColumnKeys, titleLen) {
           break
         }
         params[i + m][j].disable = true
-        merge.e.r = i + titleLen + m
+        merge.e.r = i + startRow + m
         flag = true
         m++;
       }
@@ -118,7 +466,7 @@ function merge_content(ws, tableData, spanColumnKeys, titleLen) {
   ws['!merges'] = ws['!merges'].concat(mergeData)
 }
 
-function add_style(ws, title) {
+function add_style({ws, title, startRow, footerRow}) {
 
   // 补充空白数据
   let rangs = ws['!ref'].split(":");
@@ -127,7 +475,7 @@ function add_style(ws, title) {
   Object.keys(ws).forEach(item => {
     if (/^[A-Z]+\d+$/.test(item)) {
       var alphabet = item.replace(/[^a-z]+/ig, "");
-      if (colhash.indexOf(alphabet) == -1) {
+      if (colhash.indexOf(alphabet) === -1) {
         colhash.push(alphabet)
         for (var i = 1; i <= rowSum; i++) {
           if (!ws.hasOwnProperty(alphabet + i)) {
@@ -141,55 +489,47 @@ function add_style(ws, title) {
     }
   })
 
-  let titlestyle = {
-    border: {
-      left: {style: 'thin', color: {rgb: "000000"}},
-      top: {style: 'thin', color: {rgb: "000000"}},
-      right: {style: 'thin', color: {rgb: "000000"}},
-      bottom: {style: 'thin', color: {rgb: "000000"}},
-    },
-    alignment: {horizontal: "center", vertical: "center"},
-    fill: {bgColor: {indexed: 64}, fgColor: {rgb: "5A9BD5"}}
-  }
-  let bodyPairStyle = {
-    border: {
-      left: {style: 'thin', color: {rgb: "000000"}},
-      top: {style: 'thin', color: {rgb: "000000"}},
-      right: {style: 'thin', color: {rgb: "000000"}},
-      bottom: {style: 'thin', color: {rgb: "000000"}},
-    },
-    alignment: {vertical: "center"},
-    fill: {bgColor: {indexed: 64}, fgColor: {rgb: "DDEBF7"}}
-  }
-  let bodyStyle = {
-    border: {
-      left: {style: 'thin', color: {rgb: "000000"}},
-      top: {style: 'thin', color: {rgb: "000000"}},
-      right: {style: 'thin', color: {rgb: "000000"}},
-      bottom: {style: 'thin', color: {rgb: "000000"}},
-    },
-    alignment: {vertical: "center"}
-  }
-  Object.keys(ws).forEach(item => {
+  let titleReg = new RegExp("^[A-Z]1$", "im");
+  let titleExtReg = new RegExp("^[A-Z][1-" + parseInt(startRow>0?startRow:2) + "]$", "im");
+  let headReg = new RegExp("^[A-Z]+[" + parseInt(startRow + 1) + "-" + parseInt(startRow + title.length) + "]$", "im");
+  let bodyReg = new RegExp("^[A-Z]+([" + parseInt(startRow + title.length) + "-9]|[1-9]\\d+)$", "im");
+  let footerExtReg = null
 
-    /**
-     * 加载表头样式：蓝色背景，水平垂直居中，边框
-     */
-    var reg = new RegExp("^[A-Z]+[1 -" + title.length + "]$", "gim");
-    if (reg.test(item)) {
-      ws[item].s = titlestyle
-    } else
+  if (footerRow === 1) {
+    // 底部扩展最后一行正则表达式 该行样式需要无边框
+    footerExtReg = new RegExp("^[A-Z]+" + rowSum + "$", "im");
+  }
 
+  Object.keys(ws).forEach(item => {
+    if(titleReg.test(item)&&startRow>0){
+      /**
+       * 加载标题样式：加粗，水平垂直居中，边框
+       */
+      ws[item].s = titleStyle
+    }else if(titleExtReg.test(item)&&startRow>0){
+      /**
+       * 加载标题扩展信息样式：加粗，水平垂直居中，边框
+       */
+      ws[item].s = bodyStyle
+    } else if (headReg.test(item)) {
+      /**
+       * 加载表头样式：蓝色背景，水平垂直居中，边框
+       */
+      ws[item].s = headStyle
+    }  else if (footerExtReg && footerExtReg.test(item)) {
+      /**
+       * 加载表尾样式：无边框，垂直居中
+       */
+      ws[item].s = footerStyle
+    } else if (bodyReg.test(item)) {
       /**
        * 表体样式： 垂直居中，边框
        */
-    if (/^[A-Z]+\d+$/.test(item)) {
-      if (parseInt(item.replace(/[^0-9]/ig, "")) % 2 == 0) {
+      if (parseInt(item.replace(/[^0-9]/ig, "")) % 2 === 0) {
         ws[item].s = bodyStyle
       } else {
         ws[item].s = bodyPairStyle
       }
-
     }
   })
   /* ws.B2 = {
@@ -200,210 +540,6 @@ function add_style(ws, title) {
 
 }
 
-function title_filter(data) {
-  let filterData = deepCopy(data);
-  for (var i = 0; i < filterData.length; i++) {
-    if ((filterData[i].hasOwnProperty('type') && filterData[i].type == typeRules[filterData[i].type]) ||
-      // !filterData[i].hasOwnProperty('title') ||
-      filterData[i].title == titleRules[filterData[i].title]) {
-      filterData.splice(i, 1)
-      i--
-    }
-  }
-  return filterData
-}
-
-function title_el_filter(data) {
-  let filterData = deepCopy(data);
-
-  for (var i = 0; i < filterData.length; i++) {
-    if ((filterData[i].hasOwnProperty('property') && filterData[i].property == typeRules[filterData[i].property]) ||
-      // !filterData[i].hasOwnProperty('title') ||
-      filterData[i].label == titleRules[filterData[i].label]) {
-      filterData.splice(i, 1)
-      i--
-    }
-  }
-  return filterData
-}
-
-function json_to_array(key, jsonData) {
-  return jsonData.map(v => key.map(j => {
-    return v[j]
-  }));
-}
-
-export const enums_to_value = (enums,jsonData)=>{
-  if(enums&&enums.length>0){
-    return jsonData.map(v => enums.map(e =>{
-      v[e.key] = e.enums[v[e.key]]
-      return v
-    })[0])
-  }else{
-    return jsonData
-  }
-}
-
-// fix data,return string
-function fixdata(data) {
-  let o = ''
-  let l = 0
-  const w = 10240
-  for (; l < data.byteLength / w; ++l) o += String.fromCharCode.apply(null, new Uint8Array(data.slice(l * w, l * w + w)))
-  o += String.fromCharCode.apply(null, new Uint8Array(data.slice(l * w)))
-  return o
-}
-
-// get head from excel file,return array
-function get_header_row(sheet) {
-  const headers = []
-  const range = XLSX.utils.decode_range(sheet['!ref'])
-  let C
-  const R = range.s.r
-  /* start in the first row */
-  for (C = range.s.c; C <= range.e.c; ++C) { /* walk every column in the range */
-    var cell = sheet[XLSX.utils.encode_cell({c: C, r: R})]
-    /* find the cell in the first row */
-    var hdr = 'UNKNOWN ' + C // <-- replace with your desired default
-    if (cell && cell.t) hdr = XLSX.utils.format_cell(cell)
-    headers.push(hdr)
-  }
-  return headers
-}
-
-export const title_transform = (paramData) => {
-  let result = []
-  let data1 = []
-  let data2 = []
-  let flag = false
-  let data = title_filter(paramData)
-  // let data =  paramData
-  for (var i = 0; i < data.length; i++) {
-    data1.push(data[i].title)
-    data2.push(data[i].title)
-    if ('children' in data[i]) {
-      flag = true
-      data2.pop()
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        data1.push(data[i].title)
-        data2.push(children[j].title)
-      }
-      data1.pop()
-    }
-  }
-  result.push(data1)
-  if (flag) {
-    result.push(data2)
-  }
-  return result
-}
-
-
-export const title_el_transform = (paramData) => {
-  let result = []
-  let data1 = []
-  let data2 = []
-  let flag = false
-  let data = title_el_filter(paramData)
-  // let data =  paramData
-  for (var i = 0; i < data.length; i++) {
-    data1.push(data[i].label)
-    data2.push(data[i].label)
-    if ('children' in data[i]) {
-      flag = true
-      data2.pop()
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        data1.push(data[i].label)
-        data2.push(children[j].label)
-      }
-      data1.pop()
-    }
-  }
-  result.push(data1)
-  if (flag) {
-    result.push(data2)
-  }
-  return result
-}
-
-export const key_transform = (paramData) => {
-  let result = []
-  let data = title_filter(paramData)
-  // let data = paramData
-  for (var i = 0; i < data.length; i++) {
-    if ("children" in data[i]) {
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        result.push(children[j].key)
-      }
-    } else {
-      result.push(data[i].key)
-    }
-  }
-  return result
-}
-
-export const key_el_transform = (paramData) => {
-  let result = []
-  let data = title_el_filter(paramData)
-  // let data = paramData
-  for (var i = 0; i < data.length; i++) {
-    if ("children" in data[i]) {
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        result.push(children[j].property)
-      }
-    } else {
-      result.push(data[i].property)
-    }
-  }
-  return result
-}
-
-export const enums_transform = (paramData) => {
-  let result = []
-  let data = title_filter(paramData)
-  // let data = paramData
-  for (var i = 0; i < data.length; i++) {
-    if ("children" in data[i]) {
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        if("enums" in children[j]){
-          result.push({key:children[j].key,enums:children[j].enums})
-        }
-      }
-    } else {
-      if("enums" in data[i]){
-        result.push({key:data[i].key,enums:data[i].enums})
-      }
-    }
-  }
-  return result
-}
-
-export const enums_el_transform = (paramData) => {
-  let result = []
-  let data = title_el_filter(paramData)
-  // let data = paramData
-  for (var i = 0; i < data.length; i++) {
-    if ("children" in data[i]) {
-      let children = data[i].children
-      for (var j = 0; j < children.length; j++) {
-        if("enums" in children[j]){
-          result.push({key:children[j].property,enums:children[j].enums})
-        }
-      }
-    } else {
-      if("enums" in data[i]){
-        result.push({key:data[i].property,enums:data[i].enums})
-      }
-    }
-  }
-  return result
-}
-
 export const export_table_to_excel = (id, filename) => {
   const table = document.getElementById(id);
   const wb = XLSX.utils.table_to_book(table);
@@ -417,6 +553,11 @@ export const export_table_to_excel = (id, filename) => {
   // XLSX.writeFile(wb, filename);
 }
 
+function json_to_array(key, jsonData) {
+  return jsonData.map(v => key.map(j => {
+    return v[j]
+  }));
+}
 export const export_json_to_excel = ({data, key, title, filename, spanColumns, autoWidth, format}) => {
   const wb = XLSX.utils.book_new();
   const ws = XLSX.utils.json_to_sheet(data, {header: key, skipHeader: false});
@@ -427,46 +568,100 @@ export const export_json_to_excel = ({data, key, title, filename, spanColumns, a
     const arr = json_to_array(key, data);
     auto_width(ws, arr);
   }
-  merge_content(ws, data, spanColumns)
-  XLSX.utils.book_append_sheet(wb, ws, filename);
+  merge_content(ws, data, spanColumns, title.length)
+  XLSX.utils.book_append_sheet(wb, ws, '');
   XLSX.writeFile(wb, filename + '.' + format);
 }
 
-export const export_array_to_excel = ({key, data, title, filename, spanColumns, autoWidth, format}) => {
+export const export_array_to_excel = ({key, data, title, header = [], footer = [], filename, spanColumns, autoWidth, format}) => {
   const wb = XLSX.utils.book_new();
-  const arr = json_to_array(key, data);
-  console.log(arr)
+  let arr = json_to_array(key, data);
   for (var i = title.length; i > 0; i--) {
     arr.unshift(title[i - 1])
   }
+  if(header){
+    arr = [...header,...arr]
+  }
+  if(footer){
+    arr = [...arr,...footer]
+  }
+
   const ws = XLSX.utils.aoa_to_sheet(arr);
   if (autoWidth) {
-    auto_width(ws, arr);
+    // 计算列宽时，查询条件行数据不参与计算
+    auto_width(ws, arr, header.length);
   }
   // 合并表头
   if (!ws['!merges']) ws['!merges'] = [];
   if (title.length > 1) {
-    merge_title(ws, title)
+    merge_cell({ws, data:title, startRow: header.length})
+  }
+  merge_cell({ws,data:header,startRow: 0 })
+  merge_content(ws, data, spanColumns, title.length + header.length)
+  add_style({ws, title, startRow: header.length, footerRow: footer.length})
+  XLSX.utils.book_append_sheet(wb, ws, '');
+  if(format === 'csv'){
+    XLSX.writeFile(wb, filename + '.' + format,{
+      // 要生成的文件类型
+      bookType: "csv",
+      // // 是否生成Shared String Table，官方解释是，如果开启生成速度会下降，但在低版本IOS设备上有更好的兼容性
+      bookSST: false});
+  }else{
+    // XLSX.writeFileXLSX(wb, filename + '.' + format);
+    const workbookBlob = workbook2blob(wb);
+    saveAs(workbookBlob, filename + '.' + format);
   }
-  merge_content(ws, data, spanColumns, title.length)
-  add_style(ws, title)
-  XLSX.utils.book_append_sheet(wb, ws, filename);
-  XLSX.writeFileXLSX(wb, filename + '.' + "xlsx");
 }
 
+/**
+ * 将一个字符串 s 转换为一个 ArrayBuffer 对象（如果可用）或一个普通数组（如果 ArrayBuffer 不可用）
+ * @param s
+ * @returns {any[]|ArrayBuffer}
+ */
 function s2ab(s) {
   if (typeof ArrayBuffer !== 'undefined') {
-    var buf = new ArrayBuffer(s.length);
-    var view = new Uint8Array(buf);
-    for (var i = 0; i != s.length; ++i) view[i] = s.charCodeAt(i) & 0xFF;
+    let buf = new ArrayBuffer(s.length);
+    let view = new Uint8Array(buf);
+    for (let i = 0; i !== s.length; ++i) view[i] = s.charCodeAt(i) & 0xFF;
     return buf;
   } else {
-    var buf = new Array(s.length);
-    for (var i = 0; i != s.length; ++i) buf[i] = s.charCodeAt(i) & 0xFF;
+    let buf = new Array(s.length);
+    for (let i = 0; i !== s.length; ++i) buf[i] = s.charCodeAt(i) & 0xFF;
     return buf;
   }
 }
 
+/**
+ * 将一个 ArrayBuffer 或 TypedArray 对象转换为一个字符串
+ * @param data
+ * @returns {string}
+ */
+function ab2s(data) {
+  let o = ''
+  let l = 0
+  const w = 10240
+  for (; l < data.byteLength / w; ++l) o += String.fromCharCode.apply(null, new Uint8Array(data.slice(l * w, l * w + w)))
+  o += String.fromCharCode.apply(null, new Uint8Array(data.slice(l * w)))
+  return o
+}
+
+// 将workbook装化成blob对象
+function workbook2blob(workbook) {
+  // 生成excel的配置项
+  const wopts = {
+    // 要生成的文件类型
+    bookType: "xlsx",
+    // // 是否生成Shared String Table，官方解释是，如果开启生成速度会下降，但在低版本IOS设备上有更好的兼容性
+    bookSST: false,
+    type: "binary"
+  };
+  let wbout = XLSXStyle.write(workbook, wopts);
+  let blob = new Blob([s2ab(wbout)], {
+    type: "application/octet-stream"
+  });
+  return blob;
+}
+
 export const saveAs = function(blob, fileName) {
   // IE10+
   if (window.navigator.msSaveOrOpenBlob) {
@@ -494,9 +689,26 @@ export const saveAs = function(blob, fileName) {
   }
 }
 
+// get head from excel file,return array
+function get_header_row(sheet) {
+  const headers = []
+  const range = XLSX.utils.decode_range(sheet['!ref'])
+  let C
+  const R = range.s.r
+  /* start in the first row */
+  for (C = range.s.c; C <= range.e.c; ++C) { /* walk every column in the range */
+    var cell = sheet[XLSX.utils.encode_cell({c: C, r: R})]
+    /* find the cell in the first row */
+    var hdr = 'UNKNOWN ' + C // <-- replace with your desired default
+    if (cell && cell.t) hdr = XLSX.utils.format_cell(cell)
+    headers.push(hdr)
+  }
+  return headers
+}
+
 export const read = (data, type) => {
   /* if type == 'base64' must fix data first */
-  // const fixedData = fixdata(data)
+  // const fixedData = ab2s(data)
   // const workbook = XLSX.read(btoa(fixedData), { type: 'base64' })
   const workbook = XLSX.read(data, {type: type});
   const firstSheetName = workbook.SheetNames[0];
@@ -512,10 +724,12 @@ export default {
   export_json_to_excel,
   key_transform,
   key_el_transform,
+  key_vxe_transform,
   title_transform,
   title_el_transform,
   enums_transform,
   enums_el_transform,
+  enums_vxe_transform,
   enums_to_value,
   read,
   s2ab,