@appium/base-driver 9.5.4 → 9.7.0

package/lib/express/middleware.js

@@ -2,8 +2,16 @@ import _ from 'lodash';
 import log from './logger';
 import {errors} from '../protocol';
 import {handleIdempotency} from './idempotency';
+import {pathToRegexp} from 'path-to-regexp';
 
-function allowCrossDomain(req, res, next) {
+/**
+ *
+ * @param {import('express').Request} req
+ * @param {import('express').Response} res
+ * @param {import('express').NextFunction} next
+ * @returns {any}
+ */
+export function allowCrossDomain(req, res, next) {
   try {
     res.header('Access-Control-Allow-Origin', '*');
     res.header('Access-Control-Allow-Methods', 'GET, POST, PUT, OPTIONS, DELETE');
@@ -22,7 +30,11 @@ function allowCrossDomain(req, res, next) {
   next();
 }
 
-function allowCrossDomainAsyncExecute(basePath) {
+/**
+ * @param {string} basePath
+ * @returns {import('express').RequestHandler}
+ */
+export function allowCrossDomainAsyncExecute(basePath) {
   return (req, res, next) => {
     // there are two paths for async responses, so cover both
     // https://regex101.com/r/txYiEz/1
@@ -36,12 +48,17 @@ function allowCrossDomainAsyncExecute(basePath) {
   };
 }
 
-function fixPythonContentType(basePath) {
+/**
+ *
+ * @param {string} basePath
+ * @returns {import('express').RequestHandler}
+ */
+export function fixPythonContentType(basePath) {
   return (req, res, next) => {
     // hack because python client library gives us wrong content-type
     if (
       new RegExp(`^${_.escapeRegExp(basePath)}`).test(req.path) &&
-      /^Python/.test(req.headers['user-agent'])
+      /^Python/.test(req.headers['user-agent'] ?? '')
     ) {
       if (req.headers['content-type'] === 'application/x-www-form-urlencoded') {
         req.headers['content-type'] = 'application/json; charset=utf-8';
@@ -51,14 +68,55 @@ function fixPythonContentType(basePath) {
   };
 }
 
-function defaultToJSONContentType(req, res, next) {
+/**
+ *
+ * @param {import('express').Request} req
+ * @param {import('express').Response} res
+ * @param {import('express').NextFunction} next
+ * @returns {any}
+ */
+export function defaultToJSONContentType(req, res, next) {
   if (!req.headers['content-type']) {
     req.headers['content-type'] = 'application/json; charset=utf-8';
   }
   next();
 }
 
-function catchAllHandler(err, req, res, next) {
+/**
+ *
+ * @param {import('@appium/types').StringRecord<import('@appium/types').WSServer>} webSocketsMapping
+ * @returns {import('express').RequestHandler}
+ */
+export function handleUpgrade(webSocketsMapping) {
+  return (req, res, next) => {
+    if (!req.headers?.upgrade || _.toLower(req.headers.upgrade) !== 'websocket') {
+      return next();
+    }
+    let currentPathname;
+    try {
+      currentPathname = new URL(req.url ?? '').pathname;
+    } catch {
+      currentPathname = req.url ?? '';
+    }
+    for (const [pathname, wsServer] of _.toPairs(webSocketsMapping)) {
+      if (pathToRegexp(pathname).test(currentPathname)) {
+        return wsServer.handleUpgrade(req, req.socket, Buffer.from(''), (ws) => {
+          wsServer.emit('connection', ws, req);
+        });
+      }
+    }
+    log.info(`Did not match the websocket upgrade request at ${currentPathname} to any known route`);
+    next();
+  };
+}
+
+/**
+ * @param {Error} err
+ * @param {import('express').Request} req
+ * @param {import('express').Response} res
+ * @param {import('express').NextFunction} next
+ */
+export function catchAllHandler(err, req, res, next) {
   if (res.headersSent) {
     return next(err);
   }
@@ -79,7 +137,11 @@ function catchAllHandler(err, req, res, next) {
   log.error(err);
 }
 
-function catch404Handler(req, res) {
+/**
+ * @param {import('express').Request} req
+ * @param {import('express').Response} res
+ */
+export function catch404Handler(req, res) {
   log.debug(`No route found for ${req.url}`);
   const error = errors.UnknownCommandError;
   res.status(error.w3cStatus()).json(
@@ -107,12 +169,4 @@ function patchWithSessionId(req, body) {
   return body;
 }
 
-export {
-  allowCrossDomain,
-  fixPythonContentType,
-  defaultToJSONContentType,
-  catchAllHandler,
-  allowCrossDomainAsyncExecute,
-  handleIdempotency,
-  catch404Handler,
-};
+export { handleIdempotency };

package/lib/express/server.js

@@ -14,6 +14,7 @@ import {
   catchAllHandler,
   allowCrossDomainAsyncExecute,
   handleIdempotency,
+  handleUpgrade,
   catch404Handler,
 } from './middleware';
 import {guineaPig, guineaPigScrollable, guineaPigAppBanner, welcome, STATIC_DIR} from './static';
@@ -109,6 +110,7 @@ async function server(opts) {
         allowCors,
         basePath,
         extraMethodMap,
+        webSocketsMapping: appiumServer.webSocketsMapping,
       });
       // allow extensions to update the app and http server objects
       for (const updater of serverUpdaters) {
@@ -139,6 +141,7 @@ function configureServer({
   allowCors = true,
   basePath = DEFAULT_BASE_PATH,
   extraMethodMap = {},
+  webSocketsMapping = {},
 }) {
   basePath = normalizeBasePath(basePath);
 
@@ -152,7 +155,7 @@ function configureServer({
   app.use(`${basePath}/produce_error`, produceError);
   app.use(`${basePath}/crash`, produceCrash);
 
-  // add middlewares
+  app.use(handleUpgrade(webSocketsMapping));
   if (allowCors) {
     app.use(allowCrossDomain);
   } else {
@@ -195,6 +198,7 @@ function configureHttp({httpServer, reject, keepAliveTimeout}) {
    * @type {AppiumServer}
    */
   const appiumServer = /** @type {any} */ (httpServer);
+  appiumServer.webSocketsMapping = {};
   appiumServer.addWebSocketHandler = addWebSocketHandler;
   appiumServer.removeWebSocketHandler = removeWebSocketHandler;
   appiumServer.removeAllWebSocketHandlers = removeAllWebSocketHandlers;
@@ -370,4 +374,5 @@ export {server, configureServer, normalizeBasePath};
  * @property {boolean} [allowCors]
  * @property {string} [basePath]
  * @property {MethodMap} [extraMethodMap]
+ * @property {import('@appium/types').StringRecord} [webSocketsMapping={}]
  */

package/lib/express/websocket.js

@@ -1,8 +1,6 @@
 /* eslint-disable require-await */
 import _ from 'lodash';
-import {URL} from 'url';
 import B from 'bluebird';
-import { pathToRegexp } from 'path-to-regexp';
 
 const DEFAULT_WS_PATHNAME_PREFIX = '/ws';
 
@@ -11,27 +9,6 @@ const DEFAULT_WS_PATHNAME_PREFIX = '/ws';
  * @type {AppiumServer['addWebSocketHandler']}
  */
 async function addWebSocketHandler(handlerPathname, handlerServer) {
-  if (_.isUndefined(this.webSocketsMapping)) {
-    this.webSocketsMapping = {};
-    // https://github.com/websockets/ws/pull/885
-    this.on('upgrade', (request, socket, head) => {
-      let currentPathname;
-      try {
-        currentPathname = new URL(request.url ?? '').pathname;
-      } catch {
-        currentPathname = request.url ?? '';
-      }
-      for (const [pathname, wsServer] of _.toPairs(this.webSocketsMapping)) {
-        if (pathToRegexp(pathname).test(currentPathname)) {
-          wsServer.handleUpgrade(request, socket, head, (ws) => {
-            wsServer.emit('connection', ws, request);
-          });
-          return;
-        }
-      }
-      socket.destroy();
-    });
-  }
   this.webSocketsMapping[handlerPathname] = handlerServer;
 }
 
@@ -40,10 +17,6 @@ async function addWebSocketHandler(handlerPathname, handlerServer) {
  * @type {AppiumServer['getWebSocketHandlers']}
  */
 async function getWebSocketHandlers(keysFilter = null) {
-  if (_.isEmpty(this.webSocketsMapping)) {
-    return {};
-  }
-
   return _.toPairs(this.webSocketsMapping).reduce((acc, [pathname, wsServer]) => {
     if (!_.isString(keysFilter) || pathname.includes(keysFilter)) {
       acc[pathname] = wsServer;

package/lib/jsonwp-proxy/protocol-converter.js

@@ -134,10 +134,12 @@ class ProtocolConverter {
     const bodyObj = util.safeJsonParse(body);
     if (_.isPlainObject(bodyObj)) {
       if (this.downstreamProtocol === W3C && _.has(bodyObj, 'name') && !_.has(bodyObj, 'handle')) {
-        this.log.debug(`Copied 'name' value '${bodyObj.name}' to 'handle' as per W3C spec`);
+        this.log.debug(
+          `Copied 'name' value '${/** @type {import('@appium/types').StringRecord} */ (bodyObj).name}' to 'handle' as per W3C spec`
+        );
         return await this.proxyFunc(url, method, {
-          ...bodyObj,
-          handle: bodyObj.name,
+          .../** @type {import('@appium/types').StringRecord} */ (bodyObj),
+          handle: /** @type {import('@appium/types').StringRecord} */ (bodyObj).name,
         });
       }
       if (
@@ -145,10 +147,12 @@ class ProtocolConverter {
         _.has(bodyObj, 'handle') &&
         !_.has(bodyObj, 'name')
       ) {
-        this.log.debug(`Copied 'handle' value '${bodyObj.handle}' to 'name' as per JSONWP spec`);
+        this.log.debug(
+          `Copied 'handle' value '${/** @type {import('@appium/types').StringRecord} */ (bodyObj).handle}' to 'name' as per JSONWP spec`
+        );
         return await this.proxyFunc(url, method, {
-          ...bodyObj,
-          name: bodyObj.handle,
+          .../** @type {import('@appium/types').StringRecord} */ (bodyObj),
+          name: /** @type {import('@appium/types').StringRecord} */ (bodyObj).handle,
         });
       }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@appium/base-driver",
-  "version": "9.5.4",
+  "version": "9.7.0",
   "description": "Base driver class for Appium drivers",
   "keywords": [
     "automation",
@@ -44,30 +44,30 @@
     "test:types": "tsd"
   },
   "dependencies": {
-    "@appium/support": "^4.2.4",
-    "@appium/types": "^0.16.2",
+    "@appium/support": "^4.2.6",
+    "@appium/types": "^0.18.0",
     "@colors/colors": "1.6.0",
     "@types/async-lock": "1.4.2",
     "@types/bluebird": "3.5.42",
     "@types/express": "4.17.21",
-    "@types/lodash": "4.17.0",
+    "@types/lodash": "4.17.4",
     "@types/method-override": "0.0.35",
     "@types/serve-favicon": "2.5.7",
     "async-lock": "1.4.1",
     "asyncbox": "3.0.0",
-    "axios": "1.6.8",
+    "axios": "1.7.2",
     "bluebird": "3.7.2",
     "body-parser": "1.20.2",
     "express": "4.19.2",
     "http-status-codes": "2.3.0",
     "lodash": "4.17.21",
-    "lru-cache": "10.2.0",
+    "lru-cache": "10.2.2",
     "method-override": "3.0.0",
     "morgan": "1.10.0",
     "path-to-regexp": "6.2.2",
     "serve-favicon": "2.5.0",
     "source-map-support": "0.5.21",
-    "type-fest": "4.10.1",
+    "type-fest": "4.18.3",
     "validate.js": "0.13.1"
   },
   "optionalDependencies": {
@@ -80,7 +80,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "407706cf5e28d3471decde8b56c780d40952e8d5",
+  "gitHead": "a4138c6ec9524594519f29938008500b7cdcc0b8",
   "tsd": {
     "directory": "test/types"
   }