import { ComponentFactory } from 'wix-ui-test-utils/driver-factory';
import {
    linearProgressBarDriverFactory as coreLinearProgressBarDriverFactory,
    LinearProgressBarDriver as CoreLinearProgressBarDriver
} from 'wix-ui-core/drivers/vanilla';
import { BaseDriver, DriverFactory } from 'wix-ui-test-utils/driver-factory';
import { tooltipDriverFactory } from '../Tooltip/Tooltip.driver'

export interface LinearProgressBarDriver extends CoreLinearProgressBarDriver {
    /* Returns true if the tooltip is shown */
    isTooltipShown: () => boolean;
    /* Returns true if the error icon is shown */
    isErrorIconShown: () => boolean;
    /* Returns true if the success icon is shown */
    isSuccessIconShown: () => boolean;
    /* Returns the tooltip driver */
    getTooltip: () => any;

}

export const linearProgressBarDriverFactory: DriverFactory<LinearProgressBarDriver> = ({ element, eventTrigger, wrapper }: ComponentFactory): LinearProgressBarDriver => {
    const createTooltipDriver = () => tooltipDriverFactory({ element: element.querySelector(`[data-hook='linear-progressbar-tooltip']`), wrapper, eventTrigger });
    const coreProgressBarDriver = coreLinearProgressBarDriverFactory({ element, wrapper, eventTrigger });
    const errorIcon = () => element.querySelector(`[data-hook='error-icon']`);
    const successIcon = () => element.querySelector(`[data-hook='success-icon']`);

    return {
        ...coreProgressBarDriver,
        isTooltipShown: () => createTooltipDriver().isContentElementExists(),
        getTooltip: () => createTooltipDriver(),
        isErrorIconShown: () => !!errorIcon(),
        isSuccessIconShown: () => !!successIcon()
    };
}