com.jdiai.jswraper.JSElement Maven / Gradle / Ivy

Go to download
package com.jdiai.jswraper;

import com.google.gson.JsonObject;
import com.jdiai.jsbuilder.IJSBuilder;
import com.jdiai.jsdriver.JSDriver;
import com.jdiai.jsproducer.JSProducer;
import com.jdiai.jsproducer.Json;
import com.jdiai.jswraper.scripts.NTC;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;

import java.util.List;
import java.util.function.Supplier;

import static com.epam.jdi.tools.LinqUtils.map;
import static com.epam.jdi.tools.LinqUtils.newList;
import static com.epam.jdi.tools.PrintUtils.print;
import static com.jdiai.jsbuilder.JSTemplates.XPATH_FUNC;

public class JSElement {
    protected JSDriver driver;

    public JSElement(Supplier driver, List locators) {
        this.driver = new JSDriver(driver, locators);
    }

    public JSElement(WebDriver driver, List locators) {
        this.driver = new JSDriver(driver, locators);
    }

    public JSElement(WebDriver driver, By... locators) {
        this.driver = new JSDriver(driver, locators);
    }

    public JSDriver jsDriver() {
        return driver;
    }

    public void updateDriver(JSDriver driver) {
        this.driver.context = driver.context;
        this.driver.strategy = driver.strategy;
        this.driver.setBuilder(driver.builder());
    }

    public void multiSearch() {
        driver.multiSearch();
    }

    public JSProducer jsGet(String script) {
        IJSBuilder builder = jsDriver().builder();
        builder.addJSCode(script);
        return new JSProducer(builder.executeQuery());
    }

    public String jsExecute(String script) {
        return jsGet(script).asString();
    }

    public void doAction(String action) {
        driver.doAction("element." + action);
    }

    // region Attributes
    public String getAttribute(String attribute) {
        return getValue(safeGet(attribute));
    }

    public String getValue(String valueFunc) {
        return driver.getOne(valueFunc).asString();
    }

    public List getAttributeList(String attribute) {
        return driver.getList(safeGet(attribute)).asString();
    }

    public Json getAttributes(String... attributes) {
        return getAttributes(newList(attributes));
    }

    public int getSize() {
        return driver.getSize();
    }

    public Json getAttributes(List attributes) {
        JsonObject json = driver.getOne(attributesToJson(attributes)).asJson();
        return new Json(attributes, s -> json.get(s).getAsString());
    }

    public List getJsonList(String json) {
        return driver.getList(json).asJson();
    }

    public Json getAsMap(String valueFunc) {
        return driver.getOne(valueFunc).asMap();
    }

    public String firstValue(String valueFunc) {
        return driver.getFirst(valueFunc).asString();
    }

    public List getValues(String valueFunc) {
        return driver.getList(valueFunc).asString();
    }

    public List getMultiAttributes(List attributes) {
        List objects = driver.getList(attributesToJson(attributes)).asJson();
        return map(objects, json -> new Json(attributes, v -> json.get(v).getAsString()));
    }

    public JsonObject getJson(String json) {
        return driver.getOne(json).asJson();
    }

    public List getMultiAttributes(String... attributes) {
        return getMultiAttributes(newList(attributes));
    }
    // endregion

    // region Styles
    public String getStyle(String style) {
        return driver.getOne("getComputedStyle(element)." + style).asString();
    }

    public Json getStyles(List styles) {
        String jsonObject = "{ " + print(map(styles, style -> "'" + style + "': getComputedStyle(element)." + style), ", ") + " }";
        JsonObject json = driver.getOne(jsonObject).asJson();
        return new Json(styles, json);
    }

    public String color() {
        return getColor("color");
    }

    public String bgColor() {
        return getColor("backgroundColor");
    }

    public String getColor(String name) {
        driver.builder().addJSCode(NTC.script).executeQuery();
        return driver.getOne("rgb = getComputedStyle(element)."+name+";\n" +
            "return jdi.ntc.name(rgb)[1]").asString();
    }

    public Json getStyles(String... styles) {
        return getStyles(newList(styles));
    }

    public Json getAllStyles() {
        JsonObject json =  driver.getOne("{ keys: [...getComputedStyle(element)], " +
            "values: [...getComputedStyle(element)].map(style=> getComputedStyle(element).getPropertyValue(style)) }").asJson();
        return new Json(json.get("keys"), json.get("values"));
    }

    public String pseudo(String name, String value) {
        return getValue("getComputedStyle(element, " + name + ").getPropertyValue(" + value + ")) }");
    }

    public List getStylesList(String style) {
        List styles = driver.getList("{ 'style': getComputedStyle(element)." + style + " }").asJson();
        return map(styles, s -> s.get("style").getAsString());
    }

    public List getMultiStyles(List styles) {
        String jsonObject = "{ " + print(map(styles, el -> "'" + el + "': getComputedStyle(element)." + el), ", ") + " }";
        List jsonList = driver.getList(jsonObject).asJson();
        return map(jsonList, j -> new Json(styles, s -> j.get(s).getAsString()));
    }

    public Json getObject(String objectMap) {
        return driver.getOne(validateXpath(objectMap)).asMap();
    }

    public List getMultiStyles(String... styles) {
        return getMultiStyles(newList(styles));
    }

    public Json getObject(List attributes) {
        return getObject(attributesToJson(attributes));
    }
    // Use json map like "{ 'tag': element.tagName, 'text': element.textContent... } with names equal to field names in class

    // endregion

    // region protected
    protected String safeGet(String attribute) {
        return "element?." + attribute + " ?? ''";
    }

    protected String attributesToJson(List attributes) {
        return  "{ " + print(map(attributes, attr -> "'" + attr + "': element." + attr), ", ") + " }";
    }

    protected String validateXpath(String objectMap) {
        if (objectMap.contains("': xpath(")) {
            driver.builder().registerFunction("xpath", XPATH_FUNC);
        }
        return objectMap;
    }
    // endregion
}