feat: Added reload from the terminal by pressing a key

package.json

@@ -61,7 +61,7 @@
     "minimatch": "3.0.3",
     "mkdirp": "0.5.1",
     "mz": "2.6.0",
-    "node-firefox-connect": "1.2.0",    
+    "node-firefox-connect": "1.2.0",
     "open": "0.0.5",
     "node-notifier": "5.0.2",
     "parse-json": "2.2.0",

src/cmd/run.js

@@ -1,4 +1,7 @@
 /* @flow */
+import readline from 'readline';
+import tty from 'tty';
+
 import type FirefoxProfile from 'firefox-profile';
 import type Watchpack from 'watchpack';
 
@@ -41,9 +44,9 @@ export type WatcherCreatorParams = {|
   sourceDir: string,
   artifactsDir: string,
   onSourceChange?: OnSourceChangeFn,
-  desktopNotifications?: typeof defaultDesktopNotifications,
   ignoreFiles?: Array<string>,
   createFileFilter?: FileFilterCreatorFn,
+  addonReload?: typeof defaultAddonReload,
 |};
 
 export type WatcherCreatorFn = (params: WatcherCreatorParams) => Watchpack;
@@ -52,8 +55,8 @@ export function defaultWatcherCreator(
   {
     addonId, client, sourceDir, artifactsDir, ignoreFiles,
     onSourceChange = defaultSourceWatcher,
-    desktopNotifications = defaultDesktopNotifications,
     createFileFilter = defaultFileFilterCreator,
+    addonReload = defaultAddonReload,
   }: WatcherCreatorParams
  ): Watchpack {
   const fileFilter = createFileFilter(
@@ -62,24 +65,36 @@ export function defaultWatcherCreator(
   return onSourceChange({
     sourceDir,
     artifactsDir,
-    onChange: () => {
-      log.debug(`Reloading add-on ID ${addonId}`);
-
-      return client.reloadAddon(addonId)
-        .catch((error) => {
-          log.error('\n');
-          log.error(error.stack);
-          desktopNotifications({
-            title: 'web-ext run: error occurred',
-            message: error.message,
-          });
-          throw error;
-        });
-    },
+    onChange: () => addonReload({addonId, client}),
     shouldWatchFile: (file) => fileFilter.wantFile(file),
   });
 }
 
+export type ReloadParams = {|
+  addonId: string,
+  client: RemoteFirefox,
+  desktopNotifications?: typeof defaultDesktopNotifications,
+|};
+
+export function defaultAddonReload(
+  {
+    addonId, client,
+    desktopNotifications = defaultDesktopNotifications,
+  }: ReloadParams
+): Promise<void> {
+  log.debug(`Reloading add-on ID ${addonId}`);
+  return client.reloadAddon(addonId)
+    .catch((error) => {
+      log.error('\n');
+      log.error(error.stack);
+      desktopNotifications({
+        title: 'web-ext run: error occurred',
+        message: error.message,
+      });
+      throw error;
+    });
+}
+
 
 // defaultReloadStrategy types and implementation.
 
@@ -180,24 +195,30 @@ export type CmdRunParams = {|
   customPrefs?: FirefoxPreferences,
   startUrl?: string | Array<string>,
   ignoreFiles?: Array<string>,
+  stdin: stream$Readable,
 |};
 
 export type CmdRunOptions = {|
   firefoxApp: typeof defaultFirefoxApp,
   firefoxClient: typeof defaultFirefoxClient,
   reloadStrategy: typeof defaultReloadStrategy,
+  addonReload: typeof defaultAddonReload,
+  AddonRunner: typeof ExtensionRunner
 |};
 
 export default async function run(
   {
     sourceDir, artifactsDir, firefox, firefoxProfile,
     keepProfileChanges = false, preInstall = false, noReload = false,
     browserConsole = false, customPrefs, startUrl, ignoreFiles,
+    stdin = process.stdin,
   }: CmdRunParams,
   {
     firefoxApp = defaultFirefoxApp,
     firefoxClient = defaultFirefoxClient,
     reloadStrategy = defaultReloadStrategy,
+    addonReload = defaultAddonReload,
+    AddonRunner = ExtensionRunner,
   }: CmdRunOptions = {}): Promise<Object> {
 
   log.info(`Running web extension from ${sourceDir}`);
@@ -216,7 +237,7 @@ export default async function run(
 
   const manifestData = await getValidatedManifest(sourceDir);
 
-  const runner = new ExtensionRunner({
+  const runner = new AddonRunner({
     sourceDir,
     firefoxApp,
     firefox,
@@ -272,6 +293,36 @@ export default async function run(
         );
       }
 
+      if (stdin.isTTY && stdin instanceof tty.ReadStream) {
+        readline.emitKeypressEvents(stdin);
+        stdin.setRawMode(true);
+
+        // NOTE: this `Promise.resolve().then(...)` is basically used to spawn a "co-routine" that is executed
+        // before the callback attached to the Promise returned by this function (and it allows the `run` function
+        // to not be stuck in the while loop).
+        Promise.resolve().then(async function() {
+          log.info('Press R to reload (and Ctrl-C to quit)');
+
+          let userExit = false;
+
+          while (!userExit) {
+            const keyPressed = await new Promise((resolve) => {
+              stdin.once('keypress', (str, key) => resolve(key));
+            });
+
+            if (keyPressed.ctrl && keyPressed.name === 'c') {
+              userExit = true;
+            } else if (keyPressed.name === 'r' && addonId) {
+              await addonReload({addonId, client});
+            }
+          }
+
+          log.info('\nExiting web-ext on user request');
+          runningFirefox.kill();
+          stdin.pause();
+        });
+      }
+
       log.info('The extension will reload if any source file changes');
       reloadStrategy({
         firefoxProcess: runningFirefox,

src/firefox/index.js

@@ -96,6 +96,7 @@ export type FirefoxRunnerParams = {|
 export interface FirefoxProcess extends events$EventEmitter {
   stderr: events$EventEmitter;
   stdout: events$EventEmitter;
+  kill: Function;
 }
 
 export type FirefoxRunnerResults = {|

tests/unit/helpers.js

@@ -149,7 +149,9 @@ export function makeSureItFails(): Function {
  *
  */
  // $FLOW_IGNORE: fake can return any kind of object and fake a defined set of methods for testing.
-export function fake<T>(original: Object, methods: Object = {}): T {
+export function fake<T>(
+  original: Object, methods: Object = {}, skipProperties: Array<string> = []
+): T {
   var stub = {};
 
   // Provide stubs for all original members:
@@ -162,7 +164,8 @@ export function fake<T>(original: Object, methods: Object = {}): T {
 
   var proto = Object.getPrototypeOf(original);
   for (const key of props) {
-    if (!original.hasOwnProperty(key) && !proto.hasOwnProperty(key)) {
+    if (skipProperties.indexOf(key) >= 0 ||
+       (!original.hasOwnProperty(key) && !proto.hasOwnProperty(key))) {
       continue;
     }
     const definition = original[key] || proto[key];
@@ -190,6 +193,10 @@ export function fake<T>(original: Object, methods: Object = {}): T {
   return stub;
 }
 
+export function createFakeProcess() {
+  return fake(process, {}, ['EventEmitter', 'stdin']);
+}
+
 
 /*
  * Returns a fake Firefox client as would be returned by