feat: json转excel

Author: abigmiu

example/jsonToExcel/.DS_Store

@@ -3,6 +3,7 @@ const fs = require('fs')
 
 const { generateI18nJson } = require('./modules/copy-i18n/index')
 const { excelToJson } = require('./modules/jsonToExcel/excelToJson')
+const { jsonToExcel } = require('./modules/jsonToExcel/index')
 
 //获得命令运行时的路径
 const getCwd = () => process.cwd();
@@ -35,4 +36,10 @@ program
     excelToJson(filename, sheetName, languages)
   })
 
+program
+  .command('j2e fileName sheetName')
+  .description('读取当前文件夹下的json文件转成excel')
+  .action((filename, sheetName) => {
+    jsonToExcel(filename, sheetName)
+  })
 program.parse(process.argv);

index.js

@@ -2,37 +2,39 @@
 const fs = require('fs');
 const xlsx = require('node-xlsx').default
 
+const getCwd = () => process.cwd();
+
 /**
  *
  * @param {Array} jsonData
  * @returns
  */
 function flatObject(jsonData) {
-    let stack = []
-    const retObj = {}
-    function flat(obj, index) {
-        if (typeof obj === 'string') {
-            const flatKey = stack.join('.')
-            const value = retObj[flatKey]
-            if (!value) {
-                retObj[flatKey] = []
-            }
-            retObj[flatKey].push({index: index, value: obj})
-            return obj;
-        }
-        const keys = Object.keys(obj);
-        keys.forEach(key => {
-            stack.push(key);
-            flat(obj[key], index)
-            stack.pop();
-        })
+  let stack = []
+  const retObj = {}
+  function flat(obj, index) {
+    if (typeof obj === 'string') {
+      const flatKey = stack.join('.')
+      const value = retObj[flatKey]
+      if (!value) {
+        retObj[flatKey] = []
+      }
+      retObj[flatKey].push({ index: index, value: obj })
+      return obj;
     }
+    const keys = Object.keys(obj);
+    keys.forEach(key => {
+      stack.push(key);
+      flat(obj[key], index)
+      stack.pop();
+    })
+  }
 
-    jsonData.forEach((item, index) => {
-        flat(item, index + 1)
-    });
+  jsonData.forEach((item, index) => {
+    flat(item, index + 1)
+  });
 
-    return retObj;
+  return retObj;
 }
 
 /**
@@ -42,44 +44,57 @@ function flatObject(jsonData) {
  * @returns
  */
 function handleData(data, length) {
-    const retAry = []
-    for ([key, value] of Object.entries(data)) {
-        const tempAry = [key]
-        for (let i = 1; i <= length; i++) {
-            if (value.length && value[0].index === i) {
-                tempAry[i] = value.shift().value;
-            } else {
-                tempAry[i] = null
-            }
-        }
-        retAry.push(tempAry)
+  const retAry = []
+  for ([key, value] of Object.entries(data)) {
+    const tempAry = [key]
+    for (let i = 1; i <= length; i++) {
+      if (value.length && value[0].index === i) {
+        tempAry[i] = value.shift().value;
+      } else {
+        tempAry[i] = null
+      }
     }
-    return retAry;
+    retAry.push(tempAry)
+  }
+  return retAry;
 }
 
 function readFile() {
-    const ary = [];
-    ary.push(require('./zh.json'))
-    ary.push(require('./en.json'))
-    ary.push(require('./th.json'))
-    return ary;
+  const ary = [];
+  console.log(getCwd())
+  const files = fs.readdirSync('./')
+
+  files.forEach((file) => {
+    if (file.endsWith('.json')) {
+      ary.push(require(`${getCwd()}/${file}`))
+    }
+  })
+
+  return ary;
 }
 
 function write(fileName, sheetName) {
-    const files = readFile();
-    const data = flatObject(files, files.length);
-    const handledData = handleData(data, 3)
-    fs.writeFileSync('./test.json', JSON.stringify(handledData))
-    const buffer = xlsx.build(
-        [
-            {
-                name: sheetName,
-                data: handledData
-            }
-        ]
-    )
+  const files = readFile();
+  const data = flatObject(files, files.length);
+  const handledData = handleData(data, 3)
+  const buffer = xlsx.build(
+    [
+      {
+        name: sheetName,
+        data: handledData
+      }
+    ]
+  )
 
-    fs.writeFileSync(`./${fileName}.xlsx`, buffer)
+  fs.writeFileSync(`./${fileName}.xlsx`, buffer)
+  console.log('生成成功')
 }
 
-write('translate', 'sheetName')
+/**
+ *
+ * @param {*} fileName 文件名
+ * @param {*} sheetName 表名
+ */
+exports.jsonToExcel = (fileName, sheetName) => {
+  write(fileName, sheetName)
+}