|
11 | 11 | const fs = require('fs'); |
12 | 12 | const path = require('path'); |
13 | 13 | const child_process = require('child_process'); |
| 14 | +const EventEmitter = require('events').EventEmitter; |
14 | 15 | const os = require('os'); |
15 | 16 | const chalk = require('chalk'); |
16 | 17 | const shellQuote = require('shell-quote'); |
17 | 18 |
|
| 19 | +// Inspired by https://github.com/rannn505/node-powershell |
| 20 | +const EOI = 'EOI'; |
| 21 | +class PowerShell extends EventEmitter { |
| 22 | + constructor() { |
| 23 | + super(); |
| 24 | + |
| 25 | + this._proc = child_process.spawn( |
| 26 | + 'powershell.exe', |
| 27 | + ['-NoLogo', '-NoProfile', '-NoExit', '-Command', '-'], |
| 28 | + { |
| 29 | + stdio: 'pipe', |
| 30 | + } |
| 31 | + ); |
| 32 | + |
| 33 | + let output = []; |
| 34 | + this._proc.stdout.on('data', data => { |
| 35 | + if (data.indexOf(EOI) !== -1) { |
| 36 | + this.emit('resolve', output.join('')); |
| 37 | + output = []; |
| 38 | + } else { |
| 39 | + output.push(data); |
| 40 | + } |
| 41 | + }); |
| 42 | + |
| 43 | + this._proc.on('error', () => { |
| 44 | + this._proc = null; |
| 45 | + }); |
| 46 | + } |
| 47 | + |
| 48 | + invoke(cmd) { |
| 49 | + if (!this._proc) { |
| 50 | + return Promise.resolve(''); |
| 51 | + } |
| 52 | + |
| 53 | + return new Promise(resolve => { |
| 54 | + this.on('resolve', data => { |
| 55 | + resolve(data); |
| 56 | + this.removeAllListeners('resolve'); |
| 57 | + }); |
| 58 | + |
| 59 | + this._proc.stdin.write(cmd); |
| 60 | + this._proc.stdin.write(os.EOL); |
| 61 | + this._proc.stdin.write(`echo ${EOI}`); |
| 62 | + this._proc.stdin.write(os.EOL); |
| 63 | + }); |
| 64 | + } |
| 65 | +} |
| 66 | + |
18 | 67 | function isTerminalEditor(editor) { |
19 | 68 | switch (editor) { |
20 | 69 | case 'vim': |
@@ -104,57 +153,65 @@ function getArgumentsForLineNumber(editor, fileName, lineNumber, workspace) { |
104 | 153 | return [fileName]; |
105 | 154 | } |
106 | 155 |
|
107 | | -function guessEditor() { |
108 | | - // Explicit config always wins |
109 | | - if (process.env.REACT_EDITOR) { |
110 | | - return shellQuote.parse(process.env.REACT_EDITOR); |
| 156 | +let powerShellAgent = null; |
| 157 | +function launchPowerShellAgent() { |
| 158 | + if (!powerShellAgent) { |
| 159 | + powerShellAgent = new PowerShell(); |
111 | 160 | } |
| 161 | +} |
112 | 162 |
|
113 | | - // Using `ps x` on OSX or `Get-Process` on Windows we can find out which editor is currently running. |
114 | | - // Potentially we could use similar technique for Linux |
115 | | - try { |
116 | | - if (process.platform === 'darwin') { |
117 | | - const output = child_process.execSync('ps x').toString(); |
118 | | - const processNames = Object.keys(COMMON_EDITORS_OSX); |
119 | | - for (let i = 0; i < processNames.length; i++) { |
120 | | - const processName = processNames[i]; |
121 | | - if (output.indexOf(processName) !== -1) { |
122 | | - return [COMMON_EDITORS_OSX[processName]]; |
123 | | - } |
124 | | - } |
125 | | - } else if (process.platform === 'win32') { |
126 | | - const output = child_process |
127 | | - .execSync('powershell -Command "Get-Process | Select-Object Path"', { |
128 | | - stdio: ['pipe', 'pipe', 'ignore'], |
129 | | - }) |
130 | | - .toString(); |
131 | | - const runningProcesses = output.split('\r\n'); |
132 | | - for (let i = 0; i < runningProcesses.length; i++) { |
133 | | - // `Get-Process` sometimes returns empty lines |
134 | | - if (!runningProcesses[i]) { |
135 | | - continue; |
| 163 | +function guessEditor() { |
| 164 | + return new Promise(resolve => { |
| 165 | + // Explicit config always wins |
| 166 | + if (process.env.REACT_EDITOR) { |
| 167 | + return resolve(shellQuote.parse(process.env.REACT_EDITOR)); |
| 168 | + } |
| 169 | + |
| 170 | + // Using `ps x` on OSX or `Get-Process` on Windows we can find out which editor is currently running. |
| 171 | + // Potentially we could use similar technique for Linux |
| 172 | + try { |
| 173 | + if (process.platform === 'darwin') { |
| 174 | + const output = child_process.execSync('ps x').toString(); |
| 175 | + const processNames = Object.keys(COMMON_EDITORS_OSX); |
| 176 | + for (let i = 0; i < processNames.length; i++) { |
| 177 | + const processName = processNames[i]; |
| 178 | + if (output.indexOf(processName) !== -1) { |
| 179 | + return resolve([COMMON_EDITORS_OSX[processName]]); |
| 180 | + } |
136 | 181 | } |
| 182 | + } else if (process.platform === 'win32' && powerShellAgent) { |
| 183 | + return powerShellAgent |
| 184 | + .invoke('Get-Process | Select-Object Path') |
| 185 | + .then(output => { |
| 186 | + const runningProcesses = output.split('\r\n'); |
| 187 | + for (let i = 0; i < runningProcesses.length; i++) { |
| 188 | + // `Get-Process` sometimes returns empty lines |
| 189 | + if (!runningProcesses[i]) { |
| 190 | + continue; |
| 191 | + } |
137 | 192 |
|
138 | | - const fullProcessPath = runningProcesses[i].trim(); |
139 | | - const shortProcessName = path.basename(fullProcessPath); |
| 193 | + const fullProcessPath = runningProcesses[i].trim(); |
| 194 | + const shortProcessName = path.basename(fullProcessPath); |
140 | 195 |
|
141 | | - if (COMMON_EDITORS_WIN.indexOf(shortProcessName) !== -1) { |
142 | | - return [fullProcessPath]; |
143 | | - } |
| 196 | + if (COMMON_EDITORS_WIN.indexOf(shortProcessName) !== -1) { |
| 197 | + return resolve([fullProcessPath]); |
| 198 | + } |
| 199 | + } |
| 200 | + }); |
144 | 201 | } |
| 202 | + } catch (error) { |
| 203 | + // Ignore... |
145 | 204 | } |
146 | | - } catch (error) { |
147 | | - // Ignore... |
148 | | - } |
149 | 205 |
|
150 | | - // Last resort, use old skool env vars |
151 | | - if (process.env.VISUAL) { |
152 | | - return [process.env.VISUAL]; |
153 | | - } else if (process.env.EDITOR) { |
154 | | - return [process.env.EDITOR]; |
155 | | - } |
| 206 | + // Last resort, use old skool env vars |
| 207 | + if (process.env.VISUAL) { |
| 208 | + return resolve([process.env.VISUAL]); |
| 209 | + } else if (process.env.EDITOR) { |
| 210 | + return resolve([process.env.EDITOR]); |
| 211 | + } |
156 | 212 |
|
157 | | - return [null]; |
| 213 | + return resolve([null]); |
| 214 | + }); |
158 | 215 | } |
159 | 216 |
|
160 | 217 | function printInstructions(fileName, errorMessage) { |
@@ -195,64 +252,68 @@ function launchEditor(fileName, lineNumber) { |
195 | 252 | return; |
196 | 253 | } |
197 | 254 |
|
198 | | - let [editor, ...args] = guessEditor(); |
199 | | - if (!editor) { |
200 | | - printInstructions(fileName, null); |
201 | | - return; |
202 | | - } |
203 | | - |
204 | | - if ( |
205 | | - process.platform === 'linux' && |
206 | | - fileName.startsWith('/mnt/') && |
207 | | - /Microsoft/i.test(os.release()) |
208 | | - ) { |
209 | | - // Assume WSL / "Bash on Ubuntu on Windows" is being used, and |
210 | | - // that the file exists on the Windows file system. |
211 | | - // `os.release()` is "4.4.0-43-Microsoft" in the current release |
212 | | - // build of WSL, see: https://github.com/Microsoft/BashOnWindows/issues/423#issuecomment-221627364 |
213 | | - // When a Windows editor is specified, interop functionality can |
214 | | - // handle the path translation, but only if a relative path is used. |
215 | | - fileName = path.relative('', fileName); |
216 | | - } |
| 255 | + guessEditor().then(([editor, ...args]) => { |
| 256 | + if (!editor) { |
| 257 | + printInstructions(fileName, null); |
| 258 | + return; |
| 259 | + } |
217 | 260 |
|
218 | | - let workspace = null; |
219 | | - if (lineNumber) { |
220 | | - args = args.concat( |
221 | | - getArgumentsForLineNumber(editor, fileName, lineNumber, workspace) |
222 | | - ); |
223 | | - } else { |
224 | | - args.push(fileName); |
225 | | - } |
| 261 | + if ( |
| 262 | + process.platform === 'linux' && |
| 263 | + fileName.startsWith('/mnt/') && |
| 264 | + /Microsoft/i.test(os.release()) |
| 265 | + ) { |
| 266 | + // Assume WSL / "Bash on Ubuntu on Windows" is being used, and |
| 267 | + // that the file exists on the Windows file system. |
| 268 | + // `os.release()` is "4.4.0-43-Microsoft" in the current release |
| 269 | + // build of WSL, see: https://github.com/Microsoft/BashOnWindows/issues/423#issuecomment-221627364 |
| 270 | + // When a Windows editor is specified, interop functionality can |
| 271 | + // handle the path translation, but only if a relative path is used. |
| 272 | + fileName = path.relative('', fileName); |
| 273 | + } |
226 | 274 |
|
227 | | - if (_childProcess && isTerminalEditor(editor)) { |
228 | | - // There's an existing editor process already and it's attached |
229 | | - // to the terminal, so go kill it. Otherwise two separate editor |
230 | | - // instances attach to the stdin/stdout which gets confusing. |
231 | | - _childProcess.kill('SIGKILL'); |
232 | | - } |
| 275 | + let workspace = null; |
| 276 | + if (lineNumber) { |
| 277 | + args = args.concat( |
| 278 | + getArgumentsForLineNumber(editor, fileName, lineNumber, workspace) |
| 279 | + ); |
| 280 | + } else { |
| 281 | + args.push(fileName); |
| 282 | + } |
233 | 283 |
|
234 | | - if (process.platform === 'win32') { |
235 | | - // On Windows, launch the editor in a shell because spawn can only |
236 | | - // launch .exe files. |
237 | | - _childProcess = child_process.spawn( |
238 | | - 'cmd.exe', |
239 | | - ['/C', editor].concat(args), |
240 | | - { stdio: 'inherit' } |
241 | | - ); |
242 | | - } else { |
243 | | - _childProcess = child_process.spawn(editor, args, { stdio: 'inherit' }); |
244 | | - } |
245 | | - _childProcess.on('exit', function(errorCode) { |
246 | | - _childProcess = null; |
| 284 | + if (_childProcess && isTerminalEditor(editor)) { |
| 285 | + // There's an existing editor process already and it's attached |
| 286 | + // to the terminal, so go kill it. Otherwise two separate editor |
| 287 | + // instances attach to the stdin/stdout which gets confusing. |
| 288 | + _childProcess.kill('SIGKILL'); |
| 289 | + } |
247 | 290 |
|
248 | | - if (errorCode) { |
249 | | - printInstructions(fileName, '(code ' + errorCode + ')'); |
| 291 | + if (process.platform === 'win32') { |
| 292 | + // On Windows, launch the editor in a shell because spawn can only |
| 293 | + // launch .exe files. |
| 294 | + _childProcess = child_process.spawn( |
| 295 | + 'cmd.exe', |
| 296 | + ['/C', editor].concat(args), |
| 297 | + { stdio: 'inherit' } |
| 298 | + ); |
| 299 | + } else { |
| 300 | + _childProcess = child_process.spawn(editor, args, { stdio: 'inherit' }); |
250 | 301 | } |
251 | | - }); |
| 302 | + _childProcess.on('exit', function(errorCode) { |
| 303 | + _childProcess = null; |
252 | 304 |
|
253 | | - _childProcess.on('error', function(error) { |
254 | | - printInstructions(fileName, error.message); |
| 305 | + if (errorCode) { |
| 306 | + printInstructions(fileName, '(code ' + errorCode + ')'); |
| 307 | + } |
| 308 | + }); |
| 309 | + |
| 310 | + _childProcess.on('error', function(error) { |
| 311 | + printInstructions(fileName, error.message); |
| 312 | + }); |
255 | 313 | }); |
256 | 314 | } |
257 | 315 |
|
258 | | -module.exports = launchEditor; |
| 316 | +module.exports = { |
| 317 | + launchEditor, |
| 318 | + launchPowerShellAgent, |
| 319 | +}; |
0 commit comments