node-automator 1.3.15 → 1.4.1

package/commands/backup.js

@@ -0,0 +1,113 @@
+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,
+    processRename
+} = 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,
+                backupRename,
+                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);
+            }
+            if (backupRename) {
+                backupPath = processRename(backupPath, backupRename, backupData);
+            }
+            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 || bakCfg.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.15",
+  "version": "1.4.1",
   "description": "Execute automation with yaml configuration(compatible with json)",
   "main": "index.js",
   "repository": {

package/utils/file_tool.js

@@ -16,6 +16,8 @@ const iconv = require('iconv-lite');
 const chardet = require("chardet");
 const { get_full_path } = require("../commands/share_data");
 const { parse, stringify } = require("./parse_tool");
+const { hash } = require('./hash_tool');
+const { formatDate } = require('./transform_tool');
 /**
  * 
  * @param {Array | string} src 
@@ -213,7 +215,7 @@ async function read_cfg(src, type, options) {
     return parse(content, type, options);
 }
 
-function processRename(fullpath, params) {
+function processRename(fullpath, params, content) {
     if (!params) return fullpath;
     let dirname = path.dirname(fullpath);
     switch (typeof params) {
@@ -244,6 +246,8 @@ function processRename(fullpath, params) {
             if (params.dir_suffix) dirname = dirname + params.dir_suffix;
             if (params.base_prefix) basename = params.base_prefix + basename;
             if (params.base_suffix) basename = basename + params.base_suffix;
+            if (params.hash_suffix) basename = basename + "." + hash(content).substring(0, 8);
+            if (params.time_suffix) basename = basename + "." + formatDate(new Date(), "YYYYMMDDhhmmss");
             return path.join(dirname, basename + extname);
         }
     }
@@ -289,7 +293,7 @@ async function copy(src, dst, base, options) {
             } else {
                 dstFile = path.join(dst, path.relative(base, src));
             }
-            dstFile = processRename(dstFile, options.rename);
+            dstFile = processRename(dstFile, options.rename, fs.readFileSync(src));
             let dstDir = path.dirname(dstFile);
             if (!fs.existsSync(dstDir) || !fs.lstatSync(dstDir).isDirectory()) {
                 try {