node-automator 1.0.0

package/utils/shell_tool.js

@@ -0,0 +1,162 @@
+var child_process = require('child_process');
+const {
+    get_fst_file
+} = require('../utils/file_tool');
+const {
+    dirname
+} = require('path');
+const {
+    option
+} = require('yargs');
+
+const iconv = require('iconv-lite');
+const {
+    info,
+    warn,
+    vital,
+    error,
+    whisper,
+    success,
+    log,
+    isStdOut,
+    setLastError,
+    resetError
+} = require('../utils/log_tool');
+
+/**
+ * 
+ * @param {{ cmd: string | string[], cwd?: string, encoding?:string = "cp936", capture_stdout?: boolean = false, ignore_code?: boolean = false, shell?: boolean = false, inputs?: any[]}} data 
+ */
+async function exec_shell(data) {
+    let instant_stdout = data.instant_stdout;
+    let capture_stdout = isStdOut() || data.capture_stdout;
+    let acceptableErrors = data.acceptableErrors || [];
+    const result = await new Promise((resolve, reject) => {
+        /** @type {string[]} */
+        let cmd = data.cmd;
+        if (typeof cmd === "string") {
+            cmd = generateTokens(cmd);
+        } else {
+            cmd = cmd.map(v => "" + v);
+        }
+        let cwd = data.cwd;
+        if (!cwd) {
+            let filepath = get_fst_file(cmd);
+            if (filepath) {
+                cwd = dirname(filepath);
+            }
+        }
+        if (cwd) {
+            let exist = get_fst_file(cwd);
+            if (!exist) {
+                throw (`未找到指定目录 ${cwd}`);
+            } else {
+                cwd = exist;
+            }
+        }
+        let options = {
+            cwd: cwd,
+            // shell 为 false(默认值), 可能会影响后续的控制台打印
+            // shell 为 true, 可能会影响 vbs 脚本调用
+            shell: data.shell == undefined ? false : data.shell,
+        };
+        let results = [];
+        let errors = [];
+        let encoding = data.encoding || "cp936";
+        whisper(`cmd:[${cmd}] cwd:${data.cwd} inputs:${data.inputs}`);
+        options.stdio = ["pipe", capture_stdout ? "pipe" : process.stdout, "pipe"];
+        const child = child_process.spawn(cmd[0], cmd.slice(1), options);
+        if (child.stdout) {
+            child.stdout.on("data", (chunk) => {
+                results.push(chunk);
+                if (instant_stdout) {
+                    warn(iconv.decode(chunk, encoding));
+                }
+            });
+        }
+        if (child.stderr) {
+            child.stderr.on("data", (chunk) => {
+                errors.push(chunk);
+            });
+        }
+        try {
+            if (child.stdin) {
+                if (data.inputs) {
+                    child.stdin.write(data.inputs.join("\n"));
+                }
+                child.stdin.end();
+            }
+        } catch (err) {
+            setLastError(`${err}`);
+            resolve("");
+        }
+        child.on("error", (err) => {
+            setLastError(err);
+            resolve("");
+        });
+        // child.on('close', code => {
+        child.on('exit', code => {
+            if (!data.ignore_code && code != 0) {
+                let errMsg = iconv.decode(Buffer.concat(errors), encoding);
+                let isAcceptableErrors = errMsg && acceptableErrors.some(err => {
+                    return errMsg.indexOf(err) >= 0;
+                });
+                if (!isAcceptableErrors) {
+                    setLastError(`错误码 ${code} ${errMsg}`, code);
+                }
+            }
+            if (results.length) {
+                let result = iconv.decode(Buffer.concat(results), encoding);
+                if (data.trim) {
+                    result = result.trim();
+                }
+                resolve(result);
+            } else {
+                resolve("");
+            }
+        });
+    });
+    if (!capture_stdout) { // 不捕获输出
+        // 换行使打印数据显示更干净
+        // log("");
+    }
+    return result;
+}
+
+/**
+ * 解析字符串为token数组
+ * @param {string} cmd 
+ * @returns 
+ */
+function generateTokens(cmd) {
+    let tokens = [];
+    let buffer = "";
+    for (let i = 0, len = cmd.length; i < len; ++i) {
+        let char = cmd[i];
+        if (buffer[0] === "\"") {
+            if (char === "\"") {
+                tokens.push(buffer.substring(1));
+                buffer = "";
+                continue;
+            }
+        } else {
+            if (char === " ") {
+                if (buffer.length) {
+                    tokens.push(buffer);
+                    buffer = "";
+                }
+                continue;
+            }
+        }
+        buffer += char;
+    }
+    if (buffer.length) {
+        tokens.push(buffer);
+        buffer = "";
+    }
+    return tokens;
+}
+
+module.exports = {
+    exec_shell,
+};

package/utils/svn_diff/diff_file.js

@@ -0,0 +1,44 @@
+(function() {
+  var DiffFile;
+
+  module.exports = DiffFile = (function() {
+    function DiffFile(firstLine) {
+      this.name = firstLine.split(/\s/)[1];
+      this.lines = [];
+    }
+
+    DiffFile.prototype.setIndex = function(indexLine) {
+      var match;
+      match = indexLine.split(/\s/)[2].replace(')', '');
+      return this.from = match;
+    };
+
+    DiffFile.prototype.finishHeaderParse = function(line) {
+      var match;
+      match = line.split(/\s/)[2].replace(')', '');
+      this.to = match;
+      return this.headerParsed = true;
+    };
+
+    DiffFile.prototype.appendLine = function(line) {
+      var match;
+      this.lines.push(line);
+      if (line.type === 'chunk') {
+        match = line.content.match(/^@@\s+\-([^\s]+)\s+\+([^\s]+)/);
+        this.deletionLineNum = match[1].split(',')[0];
+        return this.additionLineNum = match[2].split(',')[0];
+      } else if (line.type === 'normal') {
+        line.deletionLineNum = this.deletionLineNum++;
+        return line.additionLineNum = this.additionLineNum++;
+      } else if (line.type === 'deletion') {
+        return line.deletionLineNum = this.deletionLineNum++;
+      } else if (line.type === 'addition') {
+        return line.additionLineNum = this.additionLineNum++;
+      }
+    };
+
+    return DiffFile;
+
+  })();
+
+}).call(this);

package/utils/svn_diff/diff_parser.js

@@ -0,0 +1,55 @@
+(function() {
+  var DiffFile;
+
+  DiffFile = require('./diff_file');
+
+  module.exports.parse = function(diffStr) {
+    var currentFile, files, lines;
+    files = [];
+    currentFile = null;
+    lines = diffStr.split("\n");
+    lines.pop();
+    lines.forEach(function(line) {
+      if (line.match(/^Index/)) {
+        currentFile = new DiffFile(line);
+        return files.push(currentFile);
+      } else {
+        if (!currentFile.headerParsed) {
+          if (line.match(/^\-\-\-/)) {
+            currentFile.setIndex(line);
+          }
+          if(line.match(/^={67}$/)) {
+            return;
+          }
+          if (line.match(/^\+\+\+/)) {
+            return currentFile.finishHeaderParse(line);
+          }
+        } else {
+          if (line.match(/^@@/)) {
+            return currentFile.appendLine({
+              content: line,
+              type: 'chunk'
+            });
+          } else if (line.match(/^-/)) {
+            return currentFile.appendLine({
+              content: line,
+              type: 'deletion'
+            });
+          } else if (line.match(/^\+/)) {
+            return currentFile.appendLine({
+              content: line,
+              type: 'addition'
+            });
+          } else {
+            return currentFile.appendLine({
+              content: line,
+              type: 'normal'
+            });
+          }
+        }
+      }
+    });
+    return files;
+  };
+
+}).call(this);