[java][cdp] Simplify Augmentation for Dom Mutation

java/src/org/openqa/selenium/remote/AddHasLogEvents.java

@@ -0,0 +1,62 @@
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied. See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.openqa.selenium.remote;
+
+import static org.openqa.selenium.remote.Browser.CHROME;
+import static org.openqa.selenium.remote.Browser.EDGE;
+import static org.openqa.selenium.remote.Browser.OPERA;
+
+import org.openqa.selenium.Capabilities;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.devtools.HasDevTools;
+import org.openqa.selenium.logging.EventType;
+import org.openqa.selenium.logging.HasLogEvents;
+
+import java.util.function.Predicate;
+
+public class AddHasLogEvents
+ implements AugmenterProvider<HasLogEvents> {
+
+ private static final Predicate<String> IS_CHROMIUM_BROWSER = name ->
+ CHROME.is(name) ||
+ EDGE.is(name) ||
+ OPERA.is(name);
+
+ @Override
+ public Predicate<Capabilities> isApplicable() {
+ return caps -> IS_CHROMIUM_BROWSER.test(caps.getBrowserName());
+ }
+
+ @Override
+ public Class<HasLogEvents> getDescribedInterface() {
+ return HasLogEvents.class;
+ }
+
+ @Override
+ public HasLogEvents getImplementation(Capabilities capabilities, ExecuteMethod executeMethod) {
+ return new HasLogEvents() {
+ @Override
+ public <X> void onLogEvent(EventType<X> kind) {
+ if (((RemoteExecuteMethod) executeMethod).getWrappedDriver() instanceof HasDevTools) {
+ WebDriver driver = ((RemoteExecuteMethod) executeMethod).getWrappedDriver();
+ kind.initializeListener(driver);
+ }
+ }
+ };
+ }
+}

java/src/org/openqa/selenium/remote/Augmenter.java

@@ -33,6 +33,7 @@
 import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.WrapsDriver;
 import org.openqa.selenium.internal.Require;
+import org.openqa.selenium.logging.HasLogEvents;
 import org.openqa.selenium.remote.html5.AddWebStorage;
 
 import java.lang.reflect.Field;
@@ -219,6 +220,10 @@ private WebDriver addDependentAugmentations(WebDriver driver) {
  augmentationList.add(createAugmentation(new AddHasAuthentication()));
  }
 
+ if (!(driver instanceof HasLogEvents)) {
+ augmentationList.add(createAugmentation(new AddHasLogEvents()));
+ }
+
  if (!augmentationList.isEmpty()) {
  Capabilities caps = ImmutableCapabilities.copyOf(((HasCapabilities) driver).getCapabilities());