import { getBrowserObject } from '@wdio/utils'
import type { remote } from 'webdriver'

import { verifyArgsAndStripIfElement } from '../../utils/index.js'
import { LocalValue } from '../../utils/bidi/value.js'
import { parseScriptResult } from '../../utils/bidi/index.js'
import { getContextManager } from '../../session/context.js'
import { polyfillFn } from '../../scripts/polyfill.js'
import type { TransformElement } from '../../types.js'

/**
 * :::warning
 * The `executeAsync` command is deprecated and will be removed in a future version.
 * Please use the `execute` command instead as it provides better support for
 * error handling via `async`/`await`.
 * :::
 *
 * Inject a snippet of JavaScript into the page for execution in the context of the currently selected
 * frame. The executed script is assumed to be asynchronous and must signal that is done by invoking
 * the provided callback, which is always provided as the final argument to the function. The value
 * to this callback will be returned to the client.
 *
 * Asynchronous script commands may not span page loads. If an unload event is fired while waiting
 * for a script result, an error should be returned to the client.
 *
 * The script argument defines the script to execute in the form of a function body. The function will
 * be invoked with the provided args array and the values may be accessed via the arguments object
 * in the order specified. The final argument will always be a callback function that must be invoked
 * to signal that the script has finished.
 *
 * Arguments may be any JSON-primitive, array, or JSON object. JSON objects that define a WebElement
 * reference will be converted to the corresponding DOM element. Likewise, any WebElements in the script
 * result will be returned to the client as WebElement JSON objects.
 *
 * <example>
    :executeAsync.js
    it('should execute async JavaScript on the page', async () => {
        await browser.setTimeout({ script: 5000 })
        const result = await browser.executeAsync(function(a, b, c, d, done) {
            // browser context - you may not access client or console
            setTimeout(() => {
                done(a + b + c + d)
            }, 3000);
        }, 1, 2, 3, 4)
        // node.js context - client and console are available
        console.log(result) // outputs: 10
    });

    :executeAsync.ts
    // explicitly type the return value of the script to ensure type safety
    const result: number = await browser.executeAsync(function(a, b, c, d, done) {
        // browser context - you may not access client or console
        setTimeout(() => {
            done(a + b + c + d)
        }, 3000);
    }, 1, 2, 3, 4)
 * </example>
 *
 * @param {String|Function} script     The script to execute.
 * @param {*=}               arguments  script arguments
 *
 * @return {*}             The script result.
 *
 * @see  https://w3c.github.io/webdriver/webdriver-spec.html#dfn-execute-async-script
 * @type protocol
 * @deprecated Please use `execute` instead
 */
export async function executeAsync<ReturnValue, InnerArguments extends unknown[]>(
    this: WebdriverIO.Browser | WebdriverIO.MultiRemoteBrowser,
    script:
        string |
        (
            (
                ...args: [...innerArgs: { [K in keyof InnerArguments]: TransformElement<InnerArguments[K]> },
                callback: (result?: TransformElement<ReturnValue>) => void]
            ) => void
        ),
    ...args: InnerArguments
): Promise<ReturnValue> {
    /**
     * parameter check
     */
    if ((typeof script !== 'string' && typeof script !== 'function')) {
        throw new Error('number or type of arguments don\'t agree with execute protocol command')
    }

    if (this.isBidi && !this.isMultiremote) {
        const browser = getBrowserObject(this)
        const contextManager = getContextManager(browser)
        const context = await contextManager.getCurrentContext()
        const userScript = typeof script === 'string' ? new Function(script) : script
        const functionDeclaration = new Function(`
            const args = Array.from(arguments)
            return new Promise(async (resolve, reject) => {
                const cb = (result) => resolve(result)
                try {
                    await (${userScript.toString()}).apply(this, [...args, cb])
                } catch (err) {
                    return reject(err)
                }
            })
        `).toString()
        const params: remote.ScriptCallFunctionParameters = {
            functionDeclaration,
            awaitPromise: true,
            arguments: args.map((arg) => LocalValue.getArgument(arg)) as remote.ScriptLocalValue[],
            target: {
                context
            }
        }
        const result = await browser.scriptCallFunction(params)
        return parseScriptResult(params, result)
    }

    /**
     * instances started as multibrowserinstance can't getting called with
     * a function parameter, therefore we need to check if it starts with "function () {"
     */
    if (typeof script === 'function') {
        script = `
            ${polyfillFn}
            webdriverioPolyfill()
            return (${script}).apply(null, arguments)
        `
    }

    return this.executeAsyncScript(script, verifyArgsAndStripIfElement(args) as (string | number | boolean)[])
}