fix: type-safe overwriteCommand

src/browser/commands/moveCursorTo.ts

@@ -6,7 +6,7 @@ type MoveToOptions = {
 };
 
 // As of now, we can't export type of the command function directly. See: https://github.com/webdriverio/webdriverio/issues/12527
-export type MoveCursorToCommand = (options: MoveToOptions) => Promise<void>;
+export type MoveCursorToCommand = (this: WebdriverIO.Element, options: MoveToOptions) => Promise<void>;
 
 const makeMoveCursorToCommand = (session: WebdriverIO.Browser) =>
     async function (this: WebdriverIO.Element, { xOffset = 0, yOffset = 0 }: MoveToOptions = {}): Promise<void> {

src/browser/commands/openAndWait.ts

@@ -26,6 +26,7 @@ const is: Record<string, (match: Matches) => boolean> = {
 
 const makeOpenAndWaitCommand = (config: BrowserConfig, session: WebdriverIO.Browser) =>
     function openAndWait(
+        this: WebdriverIO.Browser,
         uri: string,
         {
             selector = [],

src/browser/commands/types.ts

@@ -81,5 +81,14 @@ export interface AssertViewOpts extends Partial<AssertViewOptsConfig> {
     ignoreDiffPixelCount?: `${number}%` | number;
 }
 
-export type AssertViewCommand = (state: string, selectors: string | string[], opts?: AssertViewOpts) => Promise<void>;
-export type AssertViewElementCommand = (state: string, opts?: AssertViewOpts) => Promise<void>;
+export type AssertViewCommand = (
+    this: WebdriverIO.Browser,
+    state: string,
+    selectors: string | string[],
+    opts?: AssertViewOpts,
+) => Promise<void>;
+export type AssertViewElementCommand = (
+    this: WebdriverIO.Element,
+    state: string,
+    opts?: AssertViewOpts,
+) => Promise<void>;

src/browser/types.ts

@@ -17,29 +17,37 @@ export interface Browser {
     applyState: (state: Record<string, unknown>) => void;
 }
 
+type FunctionProperties<T> = keyof T &
+    {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        [K in keyof T]: T[K] extends (...args: any[]) => any ? K : never;
+    }[keyof T];
+
 declare global {
     // eslint-disable-next-line @typescript-eslint/no-namespace
     namespace WebdriverIO {
-        type OverwriteCommandFn<IsElement extends boolean = false> = (
-            this: IsElement extends true ? WebdriverIO.Element : WebdriverIO.Browser,
-            origCommand: (...args: any[]) => Promise<any>, // eslint-disable-line @typescript-eslint/no-explicit-any
-            ...args: any[] // eslint-disable-line @typescript-eslint/no-explicit-any
-        ) => Promise<any>; // eslint-disable-line @typescript-eslint/no-explicit-any
-
         interface Browser {
-            getMeta(): Promise<BrowserMeta>;
-            getMeta(key: string): Promise<unknown>;
+            getMeta(this: WebdriverIO.Browser): Promise<BrowserMeta>;
+            getMeta(this: WebdriverIO.Browser, key: string): Promise<unknown>;
+
+            setMeta(this: WebdriverIO.Browser, key: string, value: unknown): Promise<void>;
 
-            setMeta(key: string, value: unknown): Promise<void>;
+            extendOptions(this: WebdriverIO.Browser, opts: { [name: string]: unknown }): Promise<void>;
 
-            extendOptions(opts: { [name: string]: unknown }): Promise<void>;
+            getConfig(this: WebdriverIO.Browser): Promise<BrowserConfig>;
 
-            getConfig(): Promise<BrowserConfig>;
+            overwriteCommand<CommandName extends FunctionProperties<WebdriverIO.Browser>>(
+                name: CommandName,
+                func: WebdriverIO.Browser[CommandName],
+                attachToElement?: false,
+                proto?: Record<string, any>, // eslint-disable-line @typescript-eslint/no-explicit-any
+                instances?: Record<string, WebdriverIO.Browser | WebdriverIO.MultiRemoteBrowser>,
+            ): void;
 
-            overwriteCommand<IsElement extends boolean = false>(
-                name: string,
-                func: OverwriteCommandFn<IsElement>,
-                attachToElement?: IsElement,
+            overwriteCommand<CommandName extends FunctionProperties<WebdriverIO.Element>>(
+                name: CommandName,
+                func: WebdriverIO.Element[CommandName],
+                attachToElement: true,
                 proto?: Record<string, any>, // eslint-disable-line @typescript-eslint/no-explicit-any
                 instances?: Record<string, WebdriverIO.Browser | WebdriverIO.MultiRemoteBrowser>,
             ): void;
@@ -96,7 +104,7 @@ declare global {
              * @param stepCb step callback
              * @returns {Promise<T>} value, returned by `stepCb`
              */
-            runStep<T = unknown>(stepName: string, stepCb: () => Promise<T> | T): Promise<T>;
+            runStep<T = unknown>(this: WebdriverIO.Browser, stepName: string, stepCb: () => Promise<T> | T): Promise<T>;
 
             // TODO: describe executionContext more precisely
             executionContext: (RunnerTest | RunnerHook) & {
@@ -111,9 +119,9 @@ declare global {
 
             openAndWait: OpenAndWaitCommand;
 
-            switchToRepl: (ctx?: Record<string, unknown>) => Promise<void>;
+            switchToRepl: (this: WebdriverIO.Browser, ctx?: Record<string, unknown>) => Promise<void>;
 
-            clearSession: () => Promise<void>;
+            clearSession: (this: WebdriverIO.Browser) => Promise<void>;
         }
 
         interface Element {