appium 2.0.0-beta.4 → 2.0.0-beta.40

package/lib/main.js

@@ -1,42 +1,53 @@
 #!/usr/bin/env node
-// transpile:main
 
-import { init as logsinkInit } from './logsink';
-import logger from './logger'; // logger needs to remain first of imports
+import {init as logsinkInit} from './logsink'; // this import needs to come first since it sets up global npmlog
+import logger from './logger'; // logger needs to remain second
+// @ts-ignore
+import {routeConfiguringFunction as makeRouter, server as baseServer} from '@appium/base-driver';
+import {logger as logFactory, util, env} from '@appium/support';
+import {asyncify} from 'asyncbox';
 import _ from 'lodash';
-import { server as baseServer, routeConfiguringFunction as makeRouter } from 'appium-base-driver';
-import { asyncify } from 'asyncbox';
-import { default as getParser, getDefaultServerArgs } from './cli/parser';
-import { USE_ALL_PLUGINS } from './cli/args';
-import { logger as logFactory, util } from 'appium-support';
+import {AppiumDriver} from './appium';
+import {runExtensionCommand} from './cli/extension';
+import {getParser} from './cli/parser';
 import {
-  showConfig, checkNodeOk, validateServerArgs,
-  warnNodeDeprecations, validateTmpDir, getNonDefaultArgs,
-  getGitRev, APPIUM_VER
+  APPIUM_VER,
+  checkNodeOk,
+  getGitRev,
+  getNonDefaultServerArgs,
+  showConfig,
+  showBuildInfo,
+  validateTmpDir,
+  warnNodeDeprecations,
 } from './config';
-import DriverConfig from './driver-config';
-import PluginConfig from './plugin-config';
-import { DRIVER_TYPE, PLUGIN_TYPE } from './extension-config';
-import { runExtensionCommand } from './cli/extension';
-import { AppiumDriver } from './appium';
+import {readConfigFile} from './config-file';
+import {loadExtensions, getActivePlugins, getActiveDrivers} from './extension';
+import {DRIVER_TYPE, PLUGIN_TYPE, SERVER_SUBCOMMAND} from './constants';
 import registerNode from './grid-register';
-import { inspectObject } from './utils';
+import {getDefaultsForSchema, validate} from './schema/schema';
+import {inspect} from './utils';
 
+const {resolveAppiumHome} = env;
 
-async function preflightChecks ({parser, args, driverConfig, pluginConfig, throwInsteadOfExit = false}) {
+/**
+ *
+ * @param {ParsedArgs} args
+ * @param {boolean} [throwInsteadOfExit]
+ */
+async function preflightChecks(args, throwInsteadOfExit = false) {
   try {
     checkNodeOk();
     if (args.longStacktrace) {
       require('longjohn').async_trace_limit = -1;
     }
-    if (args.showConfig) {
-      await showConfig();
+    if (args.showBuildInfo) {
+      await showBuildInfo();
       process.exit(0);
     }
     warnNodeDeprecations();
-    validateServerArgs(parser, args);
-    await driverConfig.read();
-    await pluginConfig.read();
+
+    validate(args);
+
     if (args.tmpDir) {
       await validateTmpDir(args.tmpDir);
     }
@@ -50,18 +61,29 @@ async function preflightChecks ({parser, args, driverConfig, pluginConfig, throw
   }
 }
 
-function logNonDefaultArgsWarning (args) {
+/**
+ * @param {Args} args
+ */
+function logNonDefaultArgsWarning(args) {
   logger.info('Non-default server args:');
-  inspectObject(args);
+  inspect(args);
 }
 
-function logDefaultCapabilitiesWarning (caps) {
-  logger.info('Default capabilities, which will be added to each request ' +
-              'unless overridden by desired capabilities:');
-  inspectObject(caps);
+/**
+ * @param {Args['defaultCapabilities']} caps
+ */
+function logDefaultCapabilitiesWarning(caps) {
+  logger.info(
+    'Default capabilities, which will be added to each request ' +
+      'unless overridden by desired capabilities:'
+  );
+  inspect(caps);
 }
 
-async function logStartupInfo (parser, args) {
+/**
+ * @param {ParsedArgs} args
+ */
+async function logStartupInfo(args) {
   let welcome = `Welcome to Appium v${APPIUM_VER}`;
   let appiumRev = await getGitRev();
   if (appiumRev) {
@@ -69,7 +91,7 @@ async function logStartupInfo (parser, args) {
   }
   logger.info(welcome);
 
-  let showArgs = getNonDefaultArgs(parser, args);
+  let showArgs = getNonDefaultServerArgs(args);
   if (_.size(showArgs)) {
     logNonDefaultArgsWarning(showArgs);
   }
@@ -83,20 +105,79 @@ async function logStartupInfo (parser, args) {
   // }
 }
 
-function logServerPort (address, port) {
-  let logMessage = `Appium REST http interface listener started on ` +
-                   `${address}:${port}`;
+/**
+ * Logs the address and port the server is listening on
+ * @param {string} address - Address
+ * @param {number} port - Port
+ * @returns {void}
+ */
+function logServerPort(address, port) {
+  let logMessage = `Appium REST http interface listener started on ` + `${address}:${port}`;
   logger.info(logMessage);
 }
 
-async function main (args = null) {
-  let parser = getParser();
+/**
+ * Gets a list of `updateServer` functions from all extensions
+ * @param {DriverNameMap} driverClasses
+ * @param {PluginNameMap} pluginClasses
+ * @returns {import('@appium/types').UpdateServerCallback[]}
+ */
+function getServerUpdaters(driverClasses, pluginClasses) {
+  return _.compact(_.map([...driverClasses.keys(), ...pluginClasses.keys()], 'updateServer'));
+}
+
+/**
+ * Makes a big `MethodMap` from all the little `MethodMap`s in the extensions
+ * @param {DriverNameMap} driverClasses
+ * @param {PluginNameMap} pluginClasses
+ * @returns {import('@appium/types').MethodMap}
+ */
+function getExtraMethodMap(driverClasses, pluginClasses) {
+  return [...driverClasses.keys(), ...pluginClasses.keys()].reduce(
+    (map, klass) => ({
+      ...map,
+      ...(klass.newMethodMap ?? {}),
+    }),
+    {}
+  );
+}
+
+/**
+ * @template [T=WithServerSubcommand]
+ * @param {Args<T>} args
+ * @returns {args is Args<WithServerSubcommand>}
+ */
+function areServerCommandArgs(args) {
+  return args.subcommand === SERVER_SUBCOMMAND;
+}
+
+/**
+ * Initializes Appium, but does not start the server.
+ *
+ * Use this to get at the configuration schema.
+ *
+ * If `args` contains a non-empty `subcommand` which is not `server`, this function will return an empty object.
+ *
+ * @template [T=WithServerSubcommand]
+ * @param {Args<T>} [args] - Partial args (progammatic usage only)
+ * @returns {Promise<ServerInitResult | ExtCommandInitResult>}
+ * @example
+ * import {init, getSchema} from 'appium';
+ * const options = {}; // config object
+ * await init(options);
+ * const schema = getSchema(); // entire config schema including plugins and drivers
+ */
+async function init(args) {
+  const appiumHome = args?.appiumHome ?? (await resolveAppiumHome());
+
+  const {driverConfig, pluginConfig} = await loadExtensions(appiumHome);
+
+  const parser = getParser();
   let throwInsteadOfExit = false;
-  if (args) {
-    // a containing package passed in their own args, let's fill them out
-    // with defaults
-    args = Object.assign({}, getDefaultServerArgs(), args);
+  /** @type {Args<T>} */
+  let preConfigArgs;
 
+  if (args) {
     // if we have a containing package instead of running as a CLI process,
     // that package might not appreciate us calling 'process.exit' willy-
     // nilly, so give it the option to have us throw instead of exit
@@ -105,80 +186,161 @@ async function main (args = null) {
       // but remove it since it's not a real server arg per se
       delete args.throwInsteadOfExit;
     }
+    preConfigArgs = {...args, subcommand: args.subcommand ?? SERVER_SUBCOMMAND};
   } else {
     // otherwise parse from CLI
-    args = parser.parseArgs();
+    preConfigArgs = /** @type {Args<T>} */ (parser.parseArgs());
   }
-  await logsinkInit(args);
 
-  // if the user has requested the 'driver' CLI, don't run the normal server,
-  // but instead pass control to the driver CLI
-  if (args.subcommand === DRIVER_TYPE || args.subcommand === PLUGIN_TYPE) {
-    await runExtensionCommand(args, args.subcommand);
-    process.exit();
+  const configResult = await readConfigFile(preConfigArgs.configFile);
+
+  if (!_.isEmpty(configResult.errors)) {
+    throw new Error(
+      `Errors in config file ${configResult.filepath}:\n ${
+        configResult.reason ?? configResult.errors
+      }`
+    );
   }
 
-  if (args.logFilters) {
-    const {issues, rules} = await logFactory.loadSecureValuesPreprocessingRules(args.logFilters);
-    if (!_.isEmpty(issues)) {
-      throw new Error(`The log filtering rules config '${args.logFilters}' has issues: ` +
-        JSON.stringify(issues, null, 2));
+  // merge config and apply defaults.
+  // the order of precendece is:
+  // 1. command line args
+  // 2. config file
+  // 3. defaults from config file.
+  if (!areServerCommandArgs(preConfigArgs)) {
+    // if the user has requested the 'driver' CLI, don't run the normal server,
+    // but instead pass control to the driver CLI
+    if (preConfigArgs.subcommand === DRIVER_TYPE) {
+      await runExtensionCommand(preConfigArgs, driverConfig);
+      return {};
+    }
+    if (preConfigArgs.subcommand === PLUGIN_TYPE) {
+      await runExtensionCommand(preConfigArgs, pluginConfig);
+      return {};
     }
-    if (_.isEmpty(rules)) {
-      logger.warn(`Found no log filtering rules in '${args.logFilters}'. Is that expected?`);
-    } else {
-      logger.info(`Loaded ${util.pluralize('filtering rule', rules.length, true)} from '${args.logFilters}'`);
+    /* istanbul ignore next */
+    return {}; // should never happen
+  } else {
+    const defaults = getDefaultsForSchema(false);
+
+    /** @type {ParsedArgs} */
+    const serverArgs = _.defaultsDeep(preConfigArgs, configResult.config?.server, defaults);
+
+    if (preConfigArgs.showConfig) {
+      showConfig(getNonDefaultServerArgs(preConfigArgs), configResult, defaults, serverArgs);
+      return {};
+    }
+
+    await logsinkInit(serverArgs);
+
+    if (serverArgs.logFilters) {
+      const {issues, rules} = await logFactory.loadSecureValuesPreprocessingRules(
+        serverArgs.logFilters
+      );
+      if (!_.isEmpty(issues)) {
+        throw new Error(
+          `The log filtering rules config '${serverArgs.logFilters}' has issues: ` +
+            JSON.stringify(issues, null, 2)
+        );
+      }
+      if (_.isEmpty(rules)) {
+        logger.warn(
+          `Found no log filtering rules in '${serverArgs.logFilters}'. Is that expected?`
+        );
+      } else {
+        logger.info(
+          `Loaded ${util.pluralize('filtering rule', rules.length, true)} from '${
+            serverArgs.logFilters
+          }'`
+        );
+      }
     }
+
+    const appiumDriver = new AppiumDriver(serverArgs);
+    // set the config on the umbrella driver so it can match drivers to caps
+    appiumDriver.driverConfig = driverConfig;
+    await preflightChecks(serverArgs, throwInsteadOfExit);
+
+    return /** @type {ServerInitResult} */ ({
+      appiumDriver,
+      parsedArgs: serverArgs,
+      driverConfig,
+      pluginConfig,
+    });
   }
+}
+
+/**
+ * Initializes Appium's config.  Starts server if appropriate and resolves the
+ * server instance if so; otherwise resolves w/ `undefined`.
+ * @template [T=WithServerSubcommand]
+ * @param {Args<T>} [args] - Arguments from CLI or otherwise
+ * @returns {Promise<import('@appium/types').AppiumServer|undefined>}
+ */
+async function main(args) {
+  const {appiumDriver, parsedArgs, pluginConfig, driverConfig} = /** @type {ServerInitResult} */ (
+    await init(args)
+  );
 
-  let appiumDriver = new AppiumDriver(args);
-  const driverConfig = new DriverConfig(args.appiumHome);
-  const pluginConfig = new PluginConfig(args.appiumHome);
-  appiumDriver.driverConfig = driverConfig;
-  await preflightChecks({parser, args, driverConfig, pluginConfig, throwInsteadOfExit});
-  await logStartupInfo(parser, args);
+  if (!appiumDriver || !parsedArgs || !pluginConfig || !driverConfig) {
+    // if this branch is taken, we've run a different subcommand, so there's nothing
+    // left to do here.
+    return;
+  }
+
+  const pluginClasses = getActivePlugins(pluginConfig, parsedArgs.usePlugins);
+  // set the active plugins on the umbrella driver so it can use them for commands
+  appiumDriver.pluginClasses = pluginClasses;
+
+  await logStartupInfo(parsedArgs);
   let routeConfiguringFunction = makeRouter(appiumDriver);
 
-  // find any plugin name which has been installed, and which has been requested for activation by
-  // using the --plugins flag, and turn each one into an instantiated plugin object, so we can send
-  // them as objects to the server init. we also want to send/assign them to the umbrella driver so
-  // it can use them to wrap command execution
-  const plugins = Object.keys(pluginConfig.installedExtensions).filter((pluginName) =>
-    _.includes(args.plugins, pluginName) ||
-    (args.plugins.length === 1 && args.plugins[0] === USE_ALL_PLUGINS)
-  ).map((pluginName) => {
-    try {
-      const PluginClass = pluginConfig.require(pluginName);
-      return new PluginClass(pluginName);
-    } catch (err) {
-      logger.error(`Could not load plugin '${pluginName}', so it will not be available. Error ` +
-                   `in loading the plugin was: ${err}`);
-      return false;
-    }
-  }).filter(Boolean);
-  appiumDriver.plugins = plugins;
+  const driverClasses = getActiveDrivers(driverConfig, parsedArgs.useDrivers);
+  const serverUpdaters = getServerUpdaters(driverClasses, pluginClasses);
+  const extraMethodMap = getExtraMethodMap(driverClasses, pluginClasses);
 
-  let server = await baseServer({
+  const serverOpts = {
     routeConfiguringFunction,
-    port: args.port,
-    hostname: args.address,
-    allowCors: args.allowCors,
-    basePath: args.basePath,
-    plugins,
-  });
-  if (args.allowCors) {
-    logger.warn('You have enabled CORS requests from any host. Be careful not ' +
-                'to visit sites which could maliciously try to start Appium ' +
-                'sessions on your machine');
+    port: parsedArgs.port,
+    hostname: parsedArgs.address,
+    allowCors: parsedArgs.allowCors,
+    basePath: parsedArgs.basePath,
+    serverUpdaters,
+    extraMethodMap,
+  };
+  if (parsedArgs.keepAliveTimeout) {
+    serverOpts.keepAliveTimeout = parsedArgs.keepAliveTimeout * 1000;
   }
-  appiumDriver.server = server;
+  let server;
   try {
-    // TODO prelaunch if args.launch is set
-    // TODO: startAlertSocket(server, appiumServer);
+    server = await baseServer(serverOpts);
+  } catch (err) {
+    logger.error(
+      `Could not configure Appium server. It's possible that a driver or plugin tried ` +
+        `to update the server and failed. Original error: ${err.message}`
+    );
+    logger.debug(err.stack);
+    return process.exit(1);
+  }
 
+  if (parsedArgs.allowCors) {
+    logger.warn(
+      'You have enabled CORS requests from any host. Be careful not ' +
+        'to visit sites which could maliciously try to start Appium ' +
+        'sessions on your machine'
+    );
+  }
+  appiumDriver.server = server;
+  try {
     // configure as node on grid, if necessary
-    if (args.nodeconfig !== null) {
-      await registerNode(args.nodeconfig, args.address, args.port, args.basePath);
+    // falsy values should not cause this to run
+    if (parsedArgs.nodeconfig) {
+      await registerNode(
+        parsedArgs.nodeconfig,
+        parsedArgs.address,
+        parsedArgs.port,
+        parsedArgs.basePath
+      );
     }
   } catch (err) {
     await server.close();
@@ -186,7 +348,7 @@ async function main (args = null) {
   }
 
   for (const signal of ['SIGINT', 'SIGTERM']) {
-    process.once(signal, async function onSignal () {
+    process.once(signal, async function onSignal() {
       logger.info(`Received ${signal} - shutting down`);
       try {
         await appiumDriver.deleteAllSessions({
@@ -202,15 +364,56 @@ async function main (args = null) {
     });
   }
 
-  logServerPort(args.address, args.port);
+  logServerPort(parsedArgs.address, parsedArgs.port);
   driverConfig.print();
-  pluginConfig.print(plugins);
+  pluginConfig.print([...pluginClasses.values()]);
 
   return server;
 }
 
+// NOTE: this is here for backwards compat for any scripts referencing `main.js` directly
+// (more specifically, `build/lib/main.js`)
+// the executable is now `../index.js`, so that module will typically be `require.main`.
 if (require.main === module) {
   asyncify(main);
 }
 
-export { main };
+// everything below here is intended to be a public API.
+export {readConfigFile} from './config-file';
+export {finalizeSchema, getSchema, validate} from './schema/schema';
+export {main, init, resolveAppiumHome};
+
+/**
+ * @typedef {import('@appium/types').DriverType} DriverType
+ * @typedef {import('@appium/types').PluginType} PluginType
+ * @typedef {import('@appium/types').DriverClass} DriverClass
+ * @typedef {import('@appium/types').PluginClass} PluginClass
+ * @typedef {import('appium/types').WithServerSubcommand} WithServerSubcommand
+ * @typedef {import('./extension').DriverNameMap} DriverNameMap
+ * @typedef {import('./extension').PluginNameMap} PluginNameMap
+ */
+
+/**
+ * Literally an empty object
+ * @typedef { {} } ExtCommandInitResult
+ */
+
+/**
+ * @typedef ServerInitData
+ * @property {import('./appium').AppiumDriver} appiumDriver - The Appium driver
+ * @property {import('appium/types').ParsedArgs} parsedArgs - The parsed arguments
+ */
+
+/**
+ * @typedef {ServerInitData & import('./extension').ExtensionConfigs} ServerInitResult
+ */
+
+/**
+ * @template [T=WithServerSubcommand]
+ * @typedef {import('appium/types').Args<T>} Args
+ */
+
+/**
+ * @template [T=WithServerSubcommand]
+ * @typedef {import('appium/types').ParsedArgs<T>} ParsedArgs
+ */