From abecc433d057620d421ced65b8e062930ed6ee5e Mon Sep 17 00:00:00 2001
From: Evandro Araujo <evandroprogram@gmail.com>
Date: Wed, 9 Sep 2020 12:01:22 -0300
Subject: [PATCH] =?UTF-8?q?Implements=20process=20timeout=20and=20fixes=20?=
 =?UTF-8?q?#9=20=E2=9C=94?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Implements process timeout and alternative method for tracking missed process. Finally fixes #9 lol 🎉
---
 src/renderer/ProcessHandlerWin.js             | 40 ++++++++++++++++++-
 src/renderer/ProcessManager.js                | 10 +++++
 src/renderer/components/MainWindow/index.vue  | 25 +++++++++++-
 .../components/SettingsWindow/index.vue       |  5 +++
 src/renderer/store/modules/Settings.js        |  6 ++-
 5 files changed, 81 insertions(+), 5 deletions(-)

diff --git a/src/renderer/ProcessHandlerWin.js b/src/renderer/ProcessHandlerWin.js
index 885400d..c031a53 100644
--- a/src/renderer/ProcessHandlerWin.js
+++ b/src/renderer/ProcessHandlerWin.js
@@ -46,8 +46,6 @@ class ProcessHandlerWin {
         ]
       }
 
-      console.log(cmdArgs)
-
       if (conn.authType === 'password' || conn.authType === 'password-ask') {
         cmdArgs.push('-oPreferredAuthentications=password')
         cmdArgs.push('-opassword_stdin')
@@ -119,6 +117,44 @@ class ProcessHandlerWin {
       })
     })
   }
+
+  getLastSpawnedProcess () {
+    return new Promise((resolve, reject) => {
+      exec(`wmic process where '(name="sshfs.exe")' get processid, creationdate /value`, (err, stdout) => {
+        if (!err) {
+          let data = stdout.toString().trim().split('\n')
+          let pid = null
+          let creationDate = null
+
+          data.forEach(item => {
+            let itemParts = item.split('=')
+
+            let key = itemParts[0].trim().toLowerCase()
+            let value = itemParts[1].trim().toLowerCase()
+
+            if (key === 'processid') {
+              pid = parseInt(value)
+            }
+
+            if (key === 'creationdate') {
+              let year = parseInt(value.substr(0, 4))
+              let month = parseInt(value.substr(4, 2)) - 1
+              let day = parseInt(value.substr(6, 2))
+              let hours = parseInt(value.substr(8, 2))
+              let minutes = parseInt(value.substr(10, 2))
+              let seconds = parseInt(value.substr(13, 2))
+
+              creationDate = new Date(year, month, day, hours, minutes, seconds)
+            }
+          })
+
+          resolve({ pid, creationDate })
+        } else {
+          reject(new Error('Process not found'))
+        }
+      })
+    })
+  }
 }
 
 export default ProcessHandlerWin
diff --git a/src/renderer/ProcessManager.js b/src/renderer/ProcessManager.js
index da8cafc..9dfede7 100644
--- a/src/renderer/ProcessManager.js
+++ b/src/renderer/ProcessManager.js
@@ -17,6 +17,10 @@ class ProcessManager extends EventEmitter {
   }
 
   create (conn) {
+    this.timeoutTimer = setTimeout(() => {
+      this.emit('timeout', conn)
+    }, (this.processHandler.settings.processTrackTimeout * 1000))
+
     return new Promise((resolve, reject) => {
       this.processHandler.create(conn).then((pid, process) => {
         this.emit('created', {
@@ -32,6 +36,8 @@ class ProcessManager extends EventEmitter {
         resolve(pid)
       }).catch(error => {
         reject(error)
+      }).finally(() => {
+        clearTimeout(this.timeoutTimer)
       })
     })
   }
@@ -79,6 +85,10 @@ class ProcessManager extends EventEmitter {
   exists (pid) {
     return this.processHandler.exists(pid)
   }
+
+  getLastSpawnedProcess () {
+    return this.processHandler.getLastSpawnedProcess()
+  }
 }
 
 const settings = store.state.Settings.settings
diff --git a/src/renderer/components/MainWindow/index.vue b/src/renderer/components/MainWindow/index.vue
index 2a9a977..e2436ee 100644
--- a/src/renderer/components/MainWindow/index.vue
+++ b/src/renderer/components/MainWindow/index.vue
@@ -47,6 +47,7 @@
 </template>
 
 <script>
+import fs from 'fs'
 import { remote } from 'electron'
 
 import { v4 as uuid } from 'uuid'
@@ -198,7 +199,7 @@ export default {
 
     settings () {
       const window = windowManager.createNew('settings-window', '', '/index.html#settings', null, {
-        height: 265,
+        height: 340,
         width: 500,
         useContentSize: true,
         frame: false,
@@ -309,6 +310,28 @@ export default {
         this.notify(`'${conn.name}' was disconnected due to a connection error.\nCheck your internet connection`, 'error-icon')
       }
     })
+
+    ProcessManager.on('timeout', conn => {
+      if (fs.existsSync(conn.mountPoint)) {
+        ProcessManager.getLastSpawnedProcess().then(process => {
+          let foundConnection = this.connections.find(i => i.pid === process.pid)
+
+          if (!foundConnection) {
+            conn.pid = process.pid
+            conn.status = 'connected'
+
+            ProcessManager.watch(process.pid)
+
+            this.notify(`'${conn.name}' tracked using alternative method`)
+          }
+        })
+      } else {
+        conn.pid = null
+        conn.status = 'disconnected'
+
+        this.notify(`Process Timeout: Couldn't connect to '${conn.name}'`, 'error-icon')
+      }
+    })
   }
 }
 </script>
diff --git a/src/renderer/components/SettingsWindow/index.vue b/src/renderer/components/SettingsWindow/index.vue
index 4c41755..46110e2 100644
--- a/src/renderer/components/SettingsWindow/index.vue
+++ b/src/renderer/components/SettingsWindow/index.vue
@@ -6,6 +6,11 @@
         <input type="text" autofocus placeholder="eg. C:\Program Files\SSHFS-Win\bin\sshfs-win.exe" v-model="data.sshfsBinary">
       </div>
 
+      <div class="form-item">
+        <label>Process Timeout</label>
+        <input type="text" autofocus placeholder="Time in seconds" v-model.number="data.processTrackTimeout" style="width: 100px; text-align: right;">
+      </div>
+
       <div class="form-item" style="margin: 10px 0">
         <SwitchLabel label="Startup with Windows" v-model="data.startupWithOS"/>
         <SwitchLabel label="Display system tray message on close" v-model="data.displayTrayMessageOnClose"/>
diff --git a/src/renderer/store/modules/Settings.js b/src/renderer/store/modules/Settings.js
index 232d458..359fb1a 100644
--- a/src/renderer/store/modules/Settings.js
+++ b/src/renderer/store/modules/Settings.js
@@ -2,7 +2,8 @@ const state = {
   settings: {
     sshfsBinary: 'C:\\Program Files\\SSHFS-Win\\bin\\sshfs-win.exe',
     startupWithOS: true,
-    displayTrayMessageOnClose: true
+    displayTrayMessageOnClose: true,
+    processTrackTimeout: 15
   }
 }
 
@@ -15,7 +16,8 @@ const mutations = {
     state.settings = {
       sshfsBinary: 'C:\\Program Files\\SSHFS-Win\\bin\\sshfs-win.exe',
       startupWithOS: true,
-      displayTrayMessageOnClose: true
+      displayTrayMessageOnClose: true,
+      processTrackTimeout: 15
     }
   }
 }