package vid.automation.test.infra;

import org.openecomp.sdc.ci.tests.utilities.GeneralUIUtils;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.NoSuchElementException;
import vid.automation.test.Constants;

import java.util.concurrent.TimeUnit;
import java.util.function.Predicate;

public class Wait {
    public static boolean byText(String text) {
        return GeneralUIUtils.findAndWaitByText(text, Constants.generalTimeout);
    }

    public static <T> boolean waitFor(Predicate<T> predicate, T input, int numOfRetries, int interval, TimeUnit intervalUnit) {
        for (int i=0; i<numOfRetries; i++) {
            try {
                if (predicate.test(input)) {
                    return true;
                }
            }
            catch (Throwable t) {
                System.out.println("a retry failed duo to:" +t.getMessage());
            }
            try {
                intervalUnit.sleep(interval);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        return false;
    }

    public static <T> boolean waitFor(Predicate<T> predicate, T input, int numOfRetries, int interval) {
        return waitFor(predicate, input, numOfRetries, interval, TimeUnit.SECONDS);
    }

    public static boolean waitByClassAndText(String className, String text, int timeoutInSeconds) {
        return waitFor((x->Get.byClassAndText(className,text)!=null),null, timeoutInSeconds, 1);
    }

    public static boolean waitByTestId(String dataTestId,  int timeoutInSeconds) {
        return waitFor((x->Get.byTestId(dataTestId)!=null),null, timeoutInSeconds, 1);
    }

    public static void angularHttpRequestsLoaded() {
        JavascriptExecutor js = (JavascriptExecutor) GeneralUIUtils.getDriver();
        for (int i=0; i<Constants.generalRetries; i++) {
            Object result = js.executeScript("return window.angular.element('body').injector().get('$http').pendingRequests.length;");
            if(result.toString().equals("0")) {
                break;
            } else {
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public static void modalToDisappear() {
        for (int i=0; i<Constants.generalRetries; i++) {
            try {
                Object modalElement =  Get.byCssSelector(Constants.Modals.modalClass);
                if(modalElement == null) {
                    break;
                } else {
                    Thread.sleep(1000);
                }
            } catch (NoSuchElementException e) {
                break;
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    public static void modalToBeDisplayed() {
        for (int i=0; i<Constants.generalRetries; i++) {
            try {
                Object modalElement =  Get.byCssSelector(Constants.Modals.modalClass);
                if(modalElement == null) {
                    Thread.sleep(1000);
                } else {
                    break;
                }
            } catch (Exception e) {
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e1) {
                    e1.printStackTrace();
                }
                e.printStackTrace();
            }
        }
    }
}