powershell() persistent shell (win)

lib/index.js

@@ -509,3 +509,5 @@ exports.getAllData = getAllData;
 exports.get = get;
 exports.observe = observe;
 
+exports.powerShellStart = util.powerShellStart;
+exports.powerShellRelease = util.powerShellRelease;

lib/util.js

@@ -37,6 +37,17 @@ let _smartMonToolsInstalled = null;
 
 const WINDIR = process.env.WINDIR || 'C:\\Windows';
 
+// powerShell
+let _psChild;
+let _psResult = '';
+let _psCmds = [];
+let _psPersistent = false;
+const _psToUTF8 = '$OutputEncoding = [System.Console]::OutputEncoding = [System.Console]::InputEncoding = [System.Text.Encoding]::UTF8 ; ';
+const _psCmdStart = '--###START###--';
+const _psError = '--ERROR--';
+const _psCmdSeperator = '--###ENDCMD###--';
+const _psIdSeperator = '--##ID##--';
+
 const execOptsWin = {
   windowsHide: true,
   maxBuffer: 1024 * 20000,
@@ -352,15 +363,110 @@ function getVboxmanage() {
   return _windows ? `"${process.env.VBOX_INSTALL_PATH || process.env.VBOX_MSI_INSTALL_PATH}\\VBoxManage.exe"` : 'vboxmanage';
 }
 
+function powerShellProceedResults(data) {
+  let id = '';
+  let parts;
+  let res = '';
+  // startID
+  if (data.indexOf(_psCmdStart) >= 0) {
+    parts = data.split(_psCmdStart);
+    const parts2 = parts[1].split(_psIdSeperator);
+    id = parts2[0];
+    if (parts2.length > 1) {
+      data = parts2.slice(1).join(_psIdSeperator);
+    }
+  }
+  // result;
+  if (data.indexOf(_psCmdSeperator) >= 0) {
+    parts = data.split(_psCmdSeperator);
+    res = parts[0];
+  }
+  let remove = -1;
+  for (let i = 0; i < _psCmds.length; i++) {
+    if (_psCmds[i].id === id) {
+      remove = i;
+      // console.log(`----- TIME  : ${(new Date() - _psCmds[i].start) * 0.001} s`);
+
+      _psCmds[i].callback(res);
+    }
+  }
+  if (remove >= 0) {
+    _psCmds.splice(remove, 1);
+  }
+}
+
+function powerShellStart() {
+  _psChild = spawn('powershell.exe', ['-NoLogo', '-InputFormat', 'Text', '-NoExit', '-ExecutionPolicy', 'Unrestricted', '-Command', '-'], {
+    stdio: 'pipe',
+    windowsHide: true,
+    maxBuffer: 1024 * 20000,
+    encoding: 'UTF-8',
+    env: util._extend({}, process.env, { LANG: 'en_US.UTF-8' })
+  });
+  if (_psChild && _psChild.pid) {
+    _psPersistent = true;
+    _psChild.stdout.on('data', function (data) {
+      _psResult = _psResult + data.toString('utf8');
+      if (data.indexOf(_psCmdSeperator) >= 0) {
+        powerShellProceedResults(_psResult);
+        _psResult = '';
+      }
+    });
+    _psChild.stderr.on('data', function () {
+      powerShellProceedResults(_psResult + _psError);
+    });
+    _psChild.on('error', function () {
+      powerShellProceedResults(_psResult + _psError);
+    });
+    _psChild.on('close', function () {
+      _psChild.kill();
+    });
+  }
+}
+
+function powerShellRelease() {
+  try {
+    _psChild.stdin.write('exit' + os.EOL);
+    _psChild.stdin.end();
+    _psPersistent = false;
+  } catch (e) {
+    _psChild.kill();
+  }
+}
+
 function powerShell(cmd) {
 
+  if (_psPersistent) {
+    const id = Math.random().toString(36).substr(2, 10);
+    return new Promise((resolve) => {
+      process.nextTick(() => {
+        function callback(data) {
+          resolve(data);
+        }
+        _psCmds.push({
+          id,
+          cmd,
+          callback,
+          start: new Date()
+        });
+        try {
+          if (_psChild && _psChild.pid) {
+            _psChild.stdin.write(_psToUTF8 + 'echo ' + _psCmdStart + id + _psIdSeperator + '; ' + os.EOL + cmd + os.EOL + 'echo ' + _psCmdSeperator + os.EOL);
+          }
+        } catch (e) {
+          resolve('');
+        }
+      });
+    });
+
+  } else {
     let result = '';
-  const toUTF8 = '$OutputEncoding = [System.Console]::OutputEncoding = [System.Console]::InputEncoding = [System.Text.Encoding]::UTF8 ; ';
 
     return new Promise((resolve) => {
       process.nextTick(() => {
         try {
-        const child = spawn('powershell.exe', ['-NoLogo', '-NoProfile', '-InputFormat', 'Text', '-NoExit', '-ExecutionPolicy', 'Unrestricted', '-Command', '-'], { // added NoProfile
+          // const start = new Date();
+          const child = spawn('powershell.exe', ['-NoLogo', '-InputFormat', 'Text', '-NoExit', '-ExecutionPolicy', 'Unrestricted', '-Command', '-'], {
             stdio: 'pipe',
             windowsHide: true,
             maxBuffer: 1024 * 20000,
@@ -383,6 +489,8 @@ function powerShell(cmd) {
             });
             child.on('close', function () {
               child.kill();
+              // console.log(`----- TIME  : ${(new Date() - start) * 0.001} s`);
+
               resolve(result);
             });
             child.on('error', function () {
@@ -390,7 +498,7 @@ function powerShell(cmd) {
               resolve(result);
             });
             try {
-            child.stdin.write(toUTF8 + cmd + os.EOL);
+              child.stdin.write(_psToUTF8 + cmd + os.EOL);
               child.stdin.write('exit' + os.EOL);
               child.stdin.end();
             } catch (e) {
@@ -405,6 +513,7 @@ function powerShell(cmd) {
         }
       });
     });
+  }
 }
 
 function execSafe(cmd, args, options) {
@@ -1130,6 +1239,8 @@ exports.wmic = wmic;
 exports.darwinXcodeExists = darwinXcodeExists;
 exports.getVboxmanage = getVboxmanage;
 exports.powerShell = powerShell;
+exports.powerShellStart = powerShellStart;
+exports.powerShellRelease = powerShellRelease;
 exports.execSafe = execSafe;
 exports.nanoSeconds = nanoSeconds;
 exports.countUniqueLines = countUniqueLines;