feat(client): add ability to set headers on requests

Signed-off-by: Muvaffak Onus <me@muvaf.com>

Author: muvaf

lib/types.ts

@@ -92,6 +92,7 @@ export interface WebDriverAgentArgs {
   resultBundleVersion?: string;
   reqBasePath?: string;
   launchTimeout?: number;
+  wdaRequestHeaders?: Record<string, string>;
 }
 
 export interface AppleDevice {
@@ -138,4 +139,5 @@ export interface XcodeBuildArgs {
   allowProvisioningDeviceRegistration?: boolean;
   resultBundlePath?: string;
   resultBundleVersion?: string;
+  wdaRequestHeaders?: Record<string, string>;
 }

lib/webdriveragent.ts

@@ -346,6 +346,7 @@ export class WebDriverAgent {
       timeout: this.wdaConnectionTimeout,
       keepAlive: true,
       scheme: this.url.protocol ? this.url.protocol.replace(':', '') : 'http',
+      headers: this.args.wdaRequestHeaders,
     };
     if (this.args.reqBasePath) {
       proxyOpts.reqBasePath = this.args.reqBasePath;
@@ -560,10 +561,12 @@ export class WebDriverAgent {
    */
   private async getStatus(timeoutMs: number = 0): Promise<StringRecord | null> {
     const noSessionProxy = new NoSessionProxy({
+      scheme: this.url.protocol ? this.url.protocol.replace(':', '') : 'http',
       server: this.url.hostname ?? undefined,
       port: parseInt(this.url.port ?? '', 10) || undefined,
       base: this.basePath,
       timeout: 3000,
+      headers: this.args.wdaRequestHeaders,
     });
 
     const sendGetStatus = async () =>

package.json

@@ -72,7 +72,7 @@
     "typescript": "^6.0.2"
   },
   "dependencies": {
-    "@appium/base-driver": "^10.0.0-rc.1",
+    "@appium/base-driver": "^10.3.0",
     "@appium/strongbox": "^1.0.0-rc.1",
     "@appium/support": "^7.0.0-rc.1",
     "appium-ios-device": "^3.0.0",