Implement WebDriver tunnel.

Author: edhager

src/SeleniumTunnel.ts

@@ -184,7 +184,7 @@ export default class SeleniumTunnel extends Tunnel
           });
         });
 
-        resolve(Task.all(tasks).then(() => {}));
+        Task.all(tasks).then(() => resolve());
       },
       () => {
         tasks &&

src/Tunnel.ts

@@ -8,12 +8,16 @@ import {
   request,
   Response
 } from '@theintern/common';
-import { spawn, ChildProcess } from 'child_process';
+import { ChildProcess } from 'child_process';
 import { join } from 'path';
 import { format as formatUrl, Url } from 'url';
-import { fileExists, kill, on } from './lib/util';
+import { fileExists, on } from './lib/util';
 import { JobState } from './interfaces';
 import * as decompress from 'decompress';
+import {
+  makeChildWithCommand,
+  stopChildProcess
+} from './lib/tunnelChildProcesses';
 
 /**
  * A Tunnel is a mechanism for connecting to a WebDriver service provider that
@@ -121,6 +125,12 @@ export default class Tunnel extends Evented<TunnelEvents, string>
   /** Whether or not to tell the tunnel to provide verbose logging output. */
   verbose!: boolean;
 
+  /**
+   * The directory in which the tunnel implementations will download
+   * any necessary applications.
+   */
+  basePath: string = __dirname;
+
   protected _startTask: CancellablePromise<any> | undefined;
   protected _stopTask: Promise<number | void> | undefined;
   protected _handle: Handle | undefined;
@@ -259,8 +269,8 @@ export default class Tunnel extends Evented<TunnelEvents, string>
   /**
    * Sends information about a job to the tunnel provider.
    *
-   * @param jobId The job to send data about. This is usually a session ID.
-   * @param data Data to send to the tunnel provider about the job.
+   * @param _jobId The job to send data about. This is usually a session ID.
+   * @param _data Data to send to the tunnel provider about the job.
    * @returns A promise that resolves once the job state request is complete.
    */
   sendJobState(_jobId: string, _data: JobState): CancellablePromise<void> {
@@ -446,89 +456,12 @@ export default class Tunnel extends Evented<TunnelEvents, string>
     executor: ChildExecutor,
     ...values: string[]
   ): CancellablePromise {
-    const command = this.executable;
-    const args = this._makeArgs(...values);
-    const options = this._makeOptions(...values);
-    const child = spawn(command, args, options);
-
-    child.stdout.setEncoding('utf8');
-    child.stderr.setEncoding('utf8');
-
-    let handle: Handle;
-    let canceled = false;
-
-    // Ensure child process is killed when parent exits
-    process.on('exit', () => kill(child.pid));
-    process.on('SIGINT', () => kill(child.pid));
-
-    const task = new Task(
-      (resolve, reject) => {
-        let errorMessage = '';
-        let exitCode: number | undefined;
-        let stderrClosed = false;
-        let exitted = false;
-
-        function handleChildExit() {
-          reject(
-            new Error(
-              `Tunnel failed to start: ${errorMessage ||
-                `Exit code: ${exitCode}`}`
-            )
-          );
-        }
-
-        handle = createCompositeHandle(
-          on(child, 'error', reject),
-
-          on(child.stderr, 'data', (data: string) => {
-            errorMessage += data;
-          }),
-
-          on(child, 'exit', () => {
-            exitted = true;
-            if (stderrClosed) {
-              handleChildExit();
-            }
-          }),
-
-          // stderr might still have data in buffer at the time the
-          // exit event is sent, so we have to store data from stderr
-          // and the exit code and reject only once stderr closes
-          on(child.stderr, 'close', () => {
-            stderrClosed = true;
-            if (exitted) {
-              handleChildExit();
-            }
-          })
-        );
-
-        const result = executor(child, resolve, reject);
-        if (result) {
-          handle = createCompositeHandle(handle, result);
-        }
-      },
-      () => {
-        canceled = true;
-
-        // Make a best effort to kill the process, but don't throw
-        // exceptions
-        try {
-          kill(child.pid);
-        } catch (error) {}
-      }
+    return makeChildWithCommand(
+      this.executable,
+      executor,
+      this._makeArgs(...values),
+      this._makeOptions(...values)
     );
-
-    return task.finally(() => {
-      handle.destroy();
-      if (canceled) {
-        // We only want this to run when cancelation has occurred
-        return new Promise(resolve => {
-          child.once('exit', () => {
-            resolve();
-          });
-        });
-      }
-    });
   }
 
   /**
@@ -620,21 +553,7 @@ export default class Tunnel extends Evented<TunnelEvents, string>
    */
   protected _stop() {
     return new Promise<number | void>((resolve, reject) => {
-      const childProcess = this._process;
-      if (!childProcess) {
-        resolve();
-        return;
-      }
-
-      childProcess.once('exit', code => {
-        resolve(code == null ? undefined : code);
-      });
-
-      try {
-        kill(childProcess.pid);
-      } catch (error) {
-        reject(error);
-      }
+      stopChildProcess(this._process, resolve, reject);
     });
   }
 }
@@ -768,6 +687,9 @@ export interface TunnelProperties extends DownloadProperties {
 
   /** [[Tunnel.Tunnel.verbose|More info]] */
   verbose: boolean;
+
+  /** [[Tunnel.Tunnel.basePath|More info]] */
+  basePath: string | undefined;
 }
 
 export type TunnelOptions = Partial<TunnelProperties>;