add bidi webextension module

Author: navin772

py/selenium/webdriver/common/bidi/webextension.py

@@ -0,0 +1,74 @@
+# 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.
+
+from typing import Dict, Union
+
+from selenium.webdriver.common.bidi.common import command_builder
+
+
+class WebExtension:
+    """
+    BiDi implementation of the webExtension module.
+    """
+
+    def __init__(self, conn):
+        self.conn = conn
+
+    def install(self, path=None, archive_path=None, base64_value=None) -> Dict:
+        """Installs a web extension in the remote end.
+
+        You must provide exactly one of the parameters.
+
+        Parameters:
+        -----------
+            path: Path to an extension directory
+            archive_path: Path to an extension archive file
+            base64_value: Base64 encoded string of the extension archive
+
+        Returns:
+        -------
+            Dict: A dictionary containing the extension ID.
+        """
+        if sum(x is not None for x in (path, archive_path, base64_value)) != 1:
+            raise ValueError("Exactly one of path, archive_path, or base64_value must be provided")
+
+        if path is not None:
+            extension_data = {"type": "path", "path": path}
+        elif archive_path is not None:
+            extension_data = {"type": "archivePath", "path": archive_path}
+        elif base64_value is not None:
+            extension_data = {"type": "base64", "value": base64_value}
+
+        params = {"extensionData": extension_data}
+        result = self.conn.execute(command_builder("webExtension.install", params))
+        return result
+
+    def uninstall(self, extension_id_or_result: Union[str, Dict]) -> None:
+        """Uninstalls a web extension from the remote end.
+
+        Parameters:
+        -----------
+            extension_id_or_result: Either the extension ID as a string or the result dictionary
+                                   from a previous install() call containing the extension ID.
+        """
+        if isinstance(extension_id_or_result, dict):
+            extension_id = extension_id_or_result.get("extension")
+        else:
+            extension_id = extension_id_or_result
+
+        params = {"extension": extension_id}
+        self.conn.execute(command_builder("webExtension.uninstall", params))

py/selenium/webdriver/remote/webdriver.py

@@ -50,6 +50,7 @@
 from selenium.webdriver.common.bidi.script import Script
 from selenium.webdriver.common.bidi.session import Session
 from selenium.webdriver.common.bidi.storage import Storage
+from selenium.webdriver.common.bidi.webextension import WebExtension
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.options import ArgOptions
 from selenium.webdriver.common.options import BaseOptions
@@ -269,6 +270,7 @@ def __init__(
         self._bidi_session = None
         self._browsing_context = None
         self._storage = None
+        self._webextension = None
 
     def __repr__(self):
         return f'<{type(self).__module__}.{type(self).__name__} (session="{self.session_id}")>'
@@ -1343,6 +1345,28 @@ def storage(self):
 
         return self._storage
 
+    @property
+    def webextension(self):
+        """Returns a webextension module object for BiDi webextension commands.
+
+        Returns:
+        --------
+        WebExtension: an object containing access to BiDi webextension commands.
+
+        Examples:
+        ---------
+        >>> extension_path = "/path/to/extension"
+        >>> extension_result = driver.webextension.install(path=extension_path)))
+        >>> driver.webextension.uninstall(extension_result)
+        """
+        if not self._websocket_connection:
+            self._start_bidi()
+
+        if self._webextension is None:
+            self._webextension = WebExtension(self._websocket_connection)
+
+        return self._webextension
+
     def _get_cdp_details(self):
         import json