node-automator 1.3.14 → 1.4.0

package/commands/backup.js

@@ -0,0 +1,108 @@
+const {
+    BaseCommand
+} = require("./base");
+
+const glob = require("glob");
+const fs = require("fs");
+const path = require("path");
+const {
+    success,
+    error
+} = require('../utils/log_tool');
+const {
+    parse,
+    stringify
+} = require('../utils/file_tool');
+
+class BackupCommand extends BaseCommand {
+    async execute() {
+        const {
+            sourceFolder,
+            backupFolder,
+            backupCfgs,
+            quiet,
+        } = this.selfData;
+
+        async function backup(bakCfg, filename) {
+            const {
+                sourceEncoding,
+                sourceKeys,
+                sourceType,
+                backupFormat,
+                backupFilename,
+                backupEncoding
+            } = bakCfg;
+            const filepath = path.join(sourceFolder, filename);
+            let backupPath = path.join(backupFolder, backupFilename || filename);
+            let backupData;
+            if (sourceType) {
+                backupData = await parse(fs.readFileSync(filepath, sourceEncoding), sourceType);
+            } else {
+                backupData = fs.readFileSync(filepath, {
+                    encoding: backupEncoding,
+                });
+            }
+            if (sourceKeys) {
+                for (const key of sourceKeys) {
+                    backupData = backupData[key];
+                }
+            }
+            if (backupFormat) {
+                backupData = await stringify(backupData, backupFormat);
+            }
+            const backupDir = path.dirname(backupPath);
+            // 创建目录
+            if (!fs.existsSync(backupDir)) {
+                fs.mkdirSync(backupDir, {
+                    recursive: true
+                });
+            }
+            fs.writeFileSync(backupPath, backupData, {
+                encoding: backupEncoding
+            });
+        }
+
+
+        backupCfgs.forEach((bakCfg, index) => {
+            if (typeof bakCfg === "string") {
+                backupCfgs[index] = {
+                    sourceFilename: bakCfg,
+                }
+            }
+        })
+        // 首次备份所有
+        backupCfgs.forEach(cfg => {
+            const files = glob.sync(path.join(sourceFolder, cfg.sourceFilename)).map(v => path.relative(sourceFolder, v));
+            files.forEach(file => {
+                backup(cfg, file);
+            });
+        });
+
+        // 监听文件夹变化
+        fs.watch(sourceFolder, {
+                recursive: true
+            }, (eventType, filename) => {
+                if (eventType === "change") {
+                    const bakCfg = backupCfgs.find(cfg => {
+                        const files = glob.sync(path.join(sourceFolder, cfg.sourceFilename)).map(v => path.relative(sourceFolder, v));
+                        return files.some(file => file === filename);
+                    });
+                    if (bakCfg) {
+                        backup(bakCfg, filename).then(() => {
+                                !quiet && success(`${filename} 备份成功`);
+                            })
+                            .catch(err => {
+                                error(`${filename} 备份异常 ${err}`);
+                            })
+                    }
+                }
+            })
+            .on('error', (err) => {
+                error(`备份异常: ${err}`);
+            });
+    }
+}
+
+module.exports = {
+    BackupCommand,
+};

package/commands/mgr.js

@@ -116,6 +116,7 @@ const { CursorUpCommand } = require('./cursor_up');
 const { RegisterSearchInstallLocationCommand } = require('./register_search_install_location');
 const { DeobfuscateCommand } = require('./deobfuscate');
 const { AliOssCommand } = require('./ali_oss');
+const { BackupCommand } = require('./backup');
 
 const globalData = {
     "executed_cfg": [], // 执行过的配置文件
@@ -475,6 +476,7 @@ function init() {
     register("register_search_install_location", RegisterSearchInstallLocationCommand, false);
     register("deobfuscate", DeobfuscateCommand, false);
     register("ali_oss", AliOssCommand, false);
+    register("backup", BackupCommand, false);
 }
 
 module.exports = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-automator",
-  "version": "1.3.14",
+  "version": "1.4.0",
   "description": "Execute automation with yaml configuration(compatible with json)",
   "main": "index.js",
   "repository": {

package/utils/func_tool.js

@@ -83,15 +83,11 @@ function queueAsync(tasks, limit, onProgress) {
             onProgress && onProgress(numDone, tasks.length, currIndex);
             task().then(function (val) {
                 numDone++;
-                result.push({
-                    val,
-                });
+                result[currIndex] = { val };
                 checkDone(currIndex, { val });
             }).catch(function (err) {
                 numDone++;
-                result.push({
-                    err,
-                });
+                result[currIndex] = { err };
                 checkDone(currIndex, { err });
             });
         }