Class RemoteScript

java/src/org/openqa/selenium/remote/RemoteScript.java:47–158

import org.openqa.selenium.json.Json;

@Beta
class RemoteScript implements Script {

  private static final Json JSON = new Json();
  private final BiDi biDi;
  private final LogInspector logInspector;
  private final org.openqa.selenium.bidi.module.Script script;

  private final WebDriver driver;

  public RemoteScript(WebDriver driver) {
    this.driver = driver;
    this.biDi = ((HasBiDi) driver).getBiDi();
    this.logInspector = new LogInspector(driver);
    this.script = new org.openqa.selenium.bidi.module.Script(driver);
  }

  @Override
  public long addConsoleMessageHandler(Consumer<ConsoleLogEntry> consumer) {
    return this.logInspector.onConsoleEntry(consumer);
  }

  @Override
  public void removeConsoleMessageHandler(long id) {
    this.biDi.removeListener(id);
  }

  @Override
  public long addJavaScriptErrorHandler(Consumer<JavascriptLogEntry> consumer) {
    return this.logInspector.onJavaScriptException(consumer);
  }

  @Override
  public void removeJavaScriptErrorHandler(long id) {
    this.biDi.removeListener(id);
  }

  @Override
  public long addDomMutationHandler(Consumer<DomMutation> consumer) {
    String scriptValue =
        Read.resourceAsString(getClass(), "/org/openqa/selenium/remote/bidi-mutation-listener.js");

    this.script.addPreloadScript(scriptValue, List.of(new ChannelValue("channel_name")));

    return this.script.onMessage(
        message -> {
          String value = message.getData().getValue().get().toString();

          Map<String, Object> values = JSON.toType(value, MAP_TYPE);
          String id = (String) values.get("target");

          List<WebElement> elements;

          synchronized (this) {
            elements =
                this.driver.findElements(
                    By.cssSelector(String.format("*[data-__webdriver_id='%s']", id)));
          }