import { type MemoryValue, When } from '@qavajs/core';
import { Locator } from './pageObject';
import { QavajsWdioWorld } from './QavajsWdioWorld';

function resolveFunction(script: string | Function) {
    return typeof script === 'function' ? script : new Function(script);
}
/**
 * Execute client function
 * @param {string} functionKey - memory key of function
 * @example I execute '$fn' function // fn is function reference
 * @example I execute 'window.scrollBy(0, 100)' function
 */
When('I execute {value} function/script', async function (this: QavajsWdioWorld, functionKey: MemoryValue) {
    const fn = resolveFunction(await functionKey.value()) as () => any;
    await this.wdio.browser.execute<any, any[]>(fn);
});

/**
 * Execute client function and save result into memory
 * @param {string} functionKey - memory key of function
 * @param {string} memoryKey - memory key to store result
 * @example I execute '$fn' function and save result as 'result' // fn is function reference
 * @example I execute 'window.scrollY' function and save result as 'scroll'
 */
When('I execute {value} function/script and save result as {value}', async function (this: QavajsWdioWorld, functionKey: MemoryValue, memoryKey: MemoryValue) {
    const fn = resolveFunction(await functionKey.value()) as () => any;
    memoryKey.set(await this.wdio.browser.execute<any, any[]>(fn));
});

/**
 * Execute client function on certain element
 * @param {string} functionKey - memory key of function
 * @param {string} alias - alias of target element
 * @example I execute '$fn' function on 'Component > Element' // fn is function reference
 * @example I execute 'arguments[0].scrollIntoView()' function on 'Component > Element'
 */
When('I execute {value} function/script on {wdioLocator}', async function (this: QavajsWdioWorld, functionKey: MemoryValue, locator: Locator) {
    const fn = resolveFunction(await functionKey.value()) as () => any;
    const element = locator();
    await this.wdio.browser.execute<any, any[]>(fn, await element.getElement());
});

/**
 * Execute client function on certain element
 * @param {string} functionKey - memory key of function
 * @param {string} alias - alias of target element
 * @example I execute '$fn' function on 'Component > Element' and save result as 'innerText' // fn is function reference
 * @example I execute 'arguments[0].innerText' function on 'Component > Element' and save result as 'innerText'
 */
When(
    'I execute {value} function/script on {wdioLocator} and save result as {value}',
    async function (functionKey: MemoryValue, locator: Locator, memoryKey: MemoryValue) {
        const fn = resolveFunction(await functionKey.value()) as () => any;
        const element = locator();
        memoryKey.set(await this.wdio.browser.execute(fn, await element.getElement()));
    }
);