appium 2.0.0-beta.3 → 2.0.0-beta.30

package/lib/cli/extension.js

@@ -2,36 +2,53 @@
 
 import DriverCommand from './driver-command';
 import PluginCommand from './plugin-command';
-import DriverConfig from '../driver-config';
-import PluginConfig from '../plugin-config';
-import { DRIVER_TYPE } from '../extension-config';
+import { DRIVER_TYPE, PLUGIN_TYPE } from '../constants';
 import { errAndQuit, log, JSON_SPACES } from './utils';
 
+const commandClasses = Object.freeze(/** @type {const} */({
+  [DRIVER_TYPE]: DriverCommand,
+  [PLUGIN_TYPE]: PluginCommand
+}));
+
 /**
  * Run a subcommand of the 'appium driver' type. Each subcommand has its own set of arguments which
  * can be represented as a JS object.
  *
  * @param {Object} args - JS object where the key is the parameter name (as defined in
  * driver-parser.js)
+ * @template {import('../extension/manifest').ExtensionType} ExtType
+ * @param {import('../extension/extension-config').ExtensionConfig<ExtType>} configObject - Extension config object
  */
-async function runExtensionCommand (args, type) {
+async function runExtensionCommand (args, configObject) {
   // TODO driver config file should be locked while any of these commands are
   // running to prevent weird situations
   let jsonResult = null;
-  const {json, appiumHome} = args;
+  const {extensionType: type} = configObject;
+  const extCmd = args[`${type}Command`];
+  if (!extCmd) {
+    throw new TypeError(`Cannot call ${type} command without a subcommand like 'install'`);
+  }
+  let {json, suppressOutput} = args;
+  if (suppressOutput) {
+    json = true;
+  }
   const logFn = (msg) => log(json, msg);
-  const ConfigClass = type === DRIVER_TYPE ? DriverConfig : PluginConfig;
-  const CommandClass = type === DRIVER_TYPE ? DriverCommand : PluginCommand;
-  const config = new ConfigClass(appiumHome, logFn);
+  let config = configObject;
+  config.log = logFn;
+  const CommandClass = /** @type {ExtCommand<ExtType>} */(commandClasses[type]);
   const cmd = new CommandClass({config, json});
   try {
-    await config.read();
     jsonResult = await cmd.execute(args);
   } catch (err) {
+    // in the suppress output case, we are calling this function internally and should
+    // just throw instead of printing an error and ending the process
+    if (suppressOutput) {
+      throw err;
+    }
     errAndQuit(json, err);
   }
 
-  if (json) {
+  if (json && !suppressOutput) {
     console.log(JSON.stringify(jsonResult, null, JSON_SPACES));
   }
 
@@ -41,3 +58,8 @@ async function runExtensionCommand (args, type) {
 export {
   runExtensionCommand,
 };
+
+/**
+ * @template {import('../../types').ExtensionType} ExtType
+ * @typedef {ExtType extends import('../../types').DriverType ? import('@appium/types').Class<DriverCommand> : ExtType extends import('../../types').PluginType ? import('@appium/types').Class<PluginCommand> : never} ExtCommand
+ */

package/lib/cli/parser.js

@@ -1,98 +1,267 @@
-import path from 'path';
-import _ from 'lodash';
+
+import { fs } from '@appium/support';
 import { ArgumentParser } from 'argparse';
-import { sharedArgs, serverArgs, extensionArgs } from './args';
-import { DRIVER_TYPE, PLUGIN_TYPE } from '../extension-config';
-import { rootDir } from '../utils';
-
-function getParser (debug = false) {
-  const parser = new ArgumentParser({
-    version: require(path.resolve(rootDir, 'package.json')).version,
-    addHelp: true,
-    description: 'A webdriver-compatible server for use with native and hybrid iOS and Android applications.',
-    prog: process.argv[1] ? path.basename(process.argv[1]) : 'appium',
-    debug
-  });
-  const subParsers = parser.addSubparsers({dest: 'subcommand', debug});
-
-  // add the 'server' subcommand, and store the raw arguments on the parser
-  // object as a way for other parts of the code to work with the arguments
-  // conceptually rather than just through argparse
-  const serverArgs = addServerToParser(sharedArgs, subParsers, debug);
-  parser.rawArgs = serverArgs;
-
-  // add the 'driver' and 'plugin' subcommands
-  addExtensionsToParser(sharedArgs, subParsers, debug);
-
-  // modify the parseArgs function to insert the 'server' subcommand if the
-  // user hasn't specified a subcommand or the global help command
-  parser._parseArgs = parser.parseArgs;
-  parser.parseArgs = function (args, namespace) {
-    if (_.isUndefined(args)) {
-      args = [...process.argv.slice(2)];
-    }
-    if (!_.includes([DRIVER_TYPE, PLUGIN_TYPE, 'server', '-h'], args[0])) {
-      args.splice(0, 0, 'server');
+import _ from 'lodash';
+import path from 'path';
+import { DRIVER_TYPE, PLUGIN_TYPE, SERVER_SUBCOMMAND } from '../constants';
+import { finalizeSchema, getArgSpec, hasArgSpec } from '../schema';
+import { rootDir } from '../config';
+import {
+  getExtensionArgs,
+  getServerArgs
+} from './args';
+
+/**
+ * If the parsed args do not contain any of these values, then we
+ * will automatially inject the `server` subcommand.
+ */
+const NON_SERVER_ARGS = Object.freeze(
+  new Set([
+    DRIVER_TYPE,
+    PLUGIN_TYPE,
+    SERVER_SUBCOMMAND,
+    '-h',
+    '--help',
+    '-v',
+    '--version'
+  ])
+);
+
+const version = fs.readPackageJsonFrom(rootDir).version;
+
+/**
+ * A wrapper around `argparse`
+ *
+ * - Handles instantiation, configuration, and monkeypatching of an
+ *    `ArgumentParser` instance for Appium server and its extensions
+ * - Handles error conditions, messages, and exit behavior
+ */
+class ArgParser {
+  /**
+   * @param {boolean} [debug] - If true, throw instead of exit on error.
+   */
+  constructor (debug = false) {
+    const prog = process.argv[1] ? path.basename(process.argv[1]) : 'appium';
+    const parser = new ArgumentParser({
+      add_help: true,
+      description:
+        'A webdriver-compatible server that facilitates automation of web, mobile, and other types of apps across various platforms.',
+      prog,
+    });
+
+    ArgParser._patchExit(parser);
+
+    /**
+     * Program name (typically `appium`)
+     * @type {string}
+     */
+    this.prog = prog;
+
+    /**
+     * If `true`, throw an error on parse failure instead of printing help
+     * @type {boolean}
+     */
+    this.debug = debug;
+
+    /**
+     * Wrapped `ArgumentParser` instance
+     * @type {ArgumentParser}
+     */
+    this.parser = parser;
+
+    parser.add_argument('-v', '--version', {
+      action: 'version',
+      version,
+    });
+
+    const subParsers = parser.add_subparsers({dest: 'subcommand'});
+
+    // add the 'server' subcommand, and store the raw arguments on the parser
+    // object as a way for other parts of the code to work with the arguments
+    // conceptually rather than just through argparse
+    const serverArgs = ArgParser._addServerToParser(subParsers);
+
+    this.rawArgs = serverArgs;
+
+    // add the 'driver' and 'plugin' subcommands
+    ArgParser._addExtensionCommandsToParser(subParsers);
+
+    // backwards compatibility / drop-in wrapper
+    /**
+     * @type {ArgParser['parseArgs']}
+     */
+    this.parse_args = this.parseArgs;
+  }
+
+  /**
+   * Parse arguments from the command line.
+   *
+   * If no subcommand is passed in, this method will inject the `server` subcommand.
+   *
+   * `ArgParser.prototype.parse_args` is an alias of this method.
+   * @param {string[]} [args] - Array of arguments, ostensibly from `process.argv`. Gathers args from `process.argv` if not provided.
+   * @returns {import('../../types/cli').ParsedArgs} - The parsed arguments
+   */
+  parseArgs (args = process.argv.slice(2)) {
+    if (!NON_SERVER_ARGS.has(args[0])) {
+      args.unshift(SERVER_SUBCOMMAND);
     }
-    return this._parseArgs(args, namespace);
-  }.bind(parser);
-  return parser;
-}
 
-function addServerToParser (sharedArgs, subParsers, debug) {
-  const serverParser = subParsers.addParser('server', {
-    addHelp: true,
-    help: 'Run an Appium server',
-    debug
-  });
+    try {
+      const parsed = this.parser.parse_args(args);
+      return ArgParser._transformParsedArgs(parsed);
+    } catch (err) {
+      if (this.debug) {
+        throw err;
+      }
+      // this isn't tested via unit tests (we use `debug: true`) so may escape coverage.
 
-  for (const [flags, opts] of [...sharedArgs, ...serverArgs]) {
-    // addArgument mutates arguments so make copies
-    serverParser.addArgument([...flags], {...opts});
+      /* istanbul ignore next */
+      {
+        // eslint-disable-next-line no-console
+        console.error(); // need an extra space since argparse prints usage.
+        // eslint-disable-next-line no-console
+        console.error(err.message);
+        process.exit(1);
+      }
+    }
   }
 
-  return serverArgs;
-}
+  /**
+   * Given an object full of arguments as returned by `argparser.parse_args`,
+   * expand the ones for extensions into a nested object structure and rename
+   * keys to match the intended destination.
+   *
+   * E.g., `{'driver-foo-bar': baz}` becomes `{driver: {foo: {bar: 'baz'}}}`
+   * @param {object} args
+   * @returns {object}
+   */
+  static _transformParsedArgs (args) {
+    return _.reduce(
+      args,
+      (unpacked, value, key) => {
+        if (!_.isUndefined(value) && hasArgSpec(key)) {
+          const {dest} = /** @type {import('../schema/arg-spec').ArgSpec} */(getArgSpec(key));
+          _.set(unpacked, dest, value);
+        } else {
+          // this could be anything that _isn't_ a server arg
+          unpacked[key] = value;
+        }
+        return unpacked;
+      },
+      {},
+    );
+  }
 
-function getDefaultServerArgs () {
-  let defaults = {};
-  for (let [, arg] of serverArgs) {
-    defaults[arg.dest] = arg.defaultValue;
+  /**
+   * Patches the `exit()` method of the parser to throw an error, so we can handle it manually.
+   * @param {ArgumentParser} parser
+   */
+  static _patchExit (parser) {
+    parser.exit = (code, msg) => {
+      if (code) {
+        throw new Error(msg);
+      }
+      process.exit();
+    };
   }
-  return defaults;
-}
 
-function addExtensionsToParser (sharedArgs, subParsers, debug) {
-  for (const type of [DRIVER_TYPE, PLUGIN_TYPE]) {
-    const extParser = subParsers.addParser(type, {
-      addHelp: true,
-      help: `Access the ${type} management CLI commands`,
-      debug
-    });
-    const extSubParsers = extParser.addSubparsers({
-      dest: `${type}Command`,
-      debug
+  /**
+   *
+   * @param {import('argparse').SubParser} subParser
+   * @returns {import('./args').ArgumentDefinitions}
+   */
+  static _addServerToParser (subParser) {
+    const serverParser = subParser.add_parser('server', {
+      add_help: true,
+      help: 'Run an Appium server',
     });
-    const parserSpecs = [
-      {command: 'list', args: extensionArgs[type].list,
-       help: `List available and installed ${type}s`},
-      {command: 'install', args: extensionArgs[type].install,
-       help: `Install a ${type}`},
-      {command: 'uninstall', args: extensionArgs[type].uninstall,
-       help: `Uninstall a ${type}`},
-      {command: 'update', args: extensionArgs[type].update,
-       help: `Update installed ${type}s to the latest version`},
-    ];
-
-    for (const {command, args, help} of parserSpecs) {
-      const parser = extSubParsers.addParser(command, {help, debug});
-      for (const [flags, opts] of [...sharedArgs, ...args]) {
-        // addArgument mutates params so make sure to send in copies instead
-        parser.addArgument([...flags], {...opts});
+
+    ArgParser._patchExit(serverParser);
+
+    const serverArgs = getServerArgs();
+    for (const [flagsOrNames, opts] of serverArgs) {
+      // TS doesn't like the spread operator here.
+      // @ts-ignore
+      serverParser.add_argument(...flagsOrNames, {...opts});
+    }
+
+    return serverArgs;
+  }
+
+  /**
+   * Adds extension sub-sub-commands to `driver`/`plugin` subcommands
+   * @param {import('argparse').SubParser} subParsers
+   */
+  static _addExtensionCommandsToParser (subParsers) {
+    for (const type of [DRIVER_TYPE, PLUGIN_TYPE]) {
+      const extParser = subParsers.add_parser(type, {
+        add_help: true,
+        help: `Access the ${type} management CLI commands`,
+      });
+
+      ArgParser._patchExit(extParser);
+
+      const extSubParsers = extParser.add_subparsers({
+        dest: `${type}Command`,
+      });
+      const extensionArgs = getExtensionArgs();
+      const parserSpecs = [
+        {
+          command: 'list',
+          args: extensionArgs[type].list,
+          help: `List available and installed ${type}s`,
+        },
+        {
+          command: 'install',
+          args: extensionArgs[type].install,
+          help: `Install a ${type}`,
+        },
+        {
+          command: 'uninstall',
+          args: extensionArgs[type].uninstall,
+          help: `Uninstall a ${type}`,
+        },
+        {
+          command: 'update',
+          args: extensionArgs[type].update,
+          help: `Update installed ${type}s to the latest version`,
+        },
+        {
+          command: 'run',
+          args: extensionArgs[type].run,
+          help:
+            `Run a script (defined inside the ${type}'s package.json under the ` +
+            `“scripts” field inside the “appium” field) from an installed ${type}`,
+        },
+      ];
+
+      for (const {command, args, help} of parserSpecs) {
+        const parser = extSubParsers.add_parser(command, {help});
+
+        ArgParser._patchExit(parser);
+
+        for (const [flagsOrNames, opts] of args) {
+          // add_argument mutates params so make sure to send in copies instead
+          // @ts-ignore
+          parser.add_argument(...flagsOrNames, {...opts});
+        }
       }
     }
   }
 }
 
-export default getParser;
-export { getParser, getDefaultServerArgs };
+/**
+ * Creates a {@link ArgParser} instance; finalizes the config schema.
+ *
+ * @constructs ArgParser
+ * @param {boolean} [debug] - If `true`, throw instead of exit upon parsing error
+ * @returns {ArgParser}
+ */
+function getParser (debug) {
+  finalizeSchema();
+
+  return new ArgParser(debug);
+}
+
+export { getParser, ArgParser };

package/lib/cli/plugin-command.js

@@ -1,27 +1,34 @@
 import _ from 'lodash';
 import ExtensionCommand from './extension-command';
-import { PLUGIN_TYPE } from '../extension-config';
-import { KNOWN_PLUGINS } from '../plugins';
+import { KNOWN_PLUGINS } from '../constants';
 
 const REQ_PLUGIN_FIELDS = ['pluginName', 'mainClass'];
 
 export default class PluginCommand extends ExtensionCommand {
 
+  /**
+   *
+   * @param {PluginCommandOptions} opts
+   */
   constructor ({config, json}) {
-    super({config, json, type: PLUGIN_TYPE});
+    super({config, json});
     this.knownExtensions = KNOWN_PLUGINS;
   }
 
   async install ({plugin, installType, packageName}) {
-    return await super.install({ext: plugin, installType, packageName});
+    return await super._install({ext: plugin, installType, packageName});
   }
 
   async uninstall ({plugin}) {
-    return await super.uninstall({ext: plugin});
+    return await super._uninstall({ext: plugin});
   }
 
   async update ({plugin, unsafe}) {
-    return await super.update({ext: plugin, unsafe});
+    return await super._update({ext: plugin, unsafe});
+  }
+
+  async run ({plugin, scriptName}) {
+    return await super._run({ext: plugin, scriptName});
   }
 
   getPostInstallText ({extName, extData}) {
@@ -41,3 +48,9 @@ export default class PluginCommand extends ExtensionCommand {
   }
 
 }
+
+/**
+ * @typedef PluginCommandOptions
+ * @property {import('../extension/extension-config').ExtensionConfig<import('../extension/manifest').PluginType>} config
+ * @property {boolean} json
+ */

package/lib/cli/utils.js

@@ -7,7 +7,7 @@ const JSON_SPACES = 4;
 /***
  * Log an error to the console and exit the process.
  * @param {boolean} json - whether we should log json or text
- * @param {string} msg - error message
+ * @param {any} msg - error message, object, Error instance, etc.
  */
 function errAndQuit (json, msg) {
   if (json) {
@@ -52,9 +52,29 @@ async function spinWith (json, msg, fn) {
   }
 }
 
+class RingBuffer {
+  constructor (size = 50) {
+    this.size = size;
+    this.buffer = [];
+  }
+  getBuff () {
+    return this.buffer;
+  }
+  dequeue () {
+    this.buffer.shift();
+  }
+  enqueue (item) {
+    if (this.buffer.length >= this.size) {
+      this.dequeue();
+    }
+    this.buffer.push(item);
+  }
+}
+
 export {
   errAndQuit,
   log,
   spinWith,
-  JSON_SPACES
+  JSON_SPACES,
+  RingBuffer
 };