add enable_webextensions method and exception for chrome webextensions

Author: navin772

py/selenium/webdriver/chrome/options.py

@@ -22,10 +22,47 @@
 
 
 class Options(ChromiumOptions):
+    def __init__(self) -> None:
+        super().__init__()
+        self._enable_webextensions = False
+
     @property
     def default_capabilities(self) -> dict:
         return DesiredCapabilities.CHROME.copy()
 
+    @property
+    def enable_webextensions(self) -> bool:
+        """Returns whether webextension support is enabled for Chrome.
+
+        :Returns: True if webextension support is enabled, False otherwise.
+        """
+        return self._enable_webextensions
+
+    @enable_webextensions.setter
+    def enable_webextensions(self, value: bool) -> None:
+        """Enables or disables webextension support for Chrome.
+
+        When enabled, this automatically adds the required Chrome flags:
+        - --enable-unsafe-extension-debugging
+        - --remote-debugging-pipe
+
+        :Args:
+         - value: True to enable webextension support, False to disable.
+        """
+        self._enable_webextensions = value
+        if value:
+            # Add required flags for Chrome webextension support
+            required_flags = ["--enable-unsafe-extension-debugging", "--remote-debugging-pipe"]
+            for flag in required_flags:
+                if flag not in self._arguments:
+                    self.add_argument(flag)
+        else:
+            # Remove webextension flags if disabling
+            flags_to_remove = ["--enable-unsafe-extension-debugging", "--remote-debugging-pipe"]
+            for flag in flags_to_remove:
+                if flag in self._arguments:
+                    self._arguments.remove(flag)
+
     def enable_mobile(
         self,
         android_package: Optional[str] = "com.android.chrome",

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

@@ -18,6 +18,7 @@
 from typing import Dict, Union
 
 from selenium.webdriver.common.bidi.common import command_builder
+from selenium.common.exceptions import WebDriverException
 
 
 class WebExtension:
@@ -54,8 +55,17 @@ def install(self, path=None, archive_path=None, base64_value=None) -> Dict:
             extension_data = {"type": "base64", "value": base64_value}
 
         params = {"extensionData": extension_data}
-        result = self.conn.execute(command_builder("webExtension.install", params))
-        return result
+
+        try:
+            result = self.conn.execute(command_builder("webExtension.install", params))
+            return result
+        except WebDriverException as e:
+            if "Method not available" in str(e):
+                raise WebDriverException(
+                    f"{str(e)}. If you are using Chrome, add '--enable-unsafe-extension-debugging' "
+                    "and '--remote-debugging-pipe' arguments or set options.enable_webextensions = True"
+                ) from e
+            raise
 
     def uninstall(self, extension_id_or_result: Union[str, Dict]) -> None:
         """Uninstalls a web extension from the remote end.