[🚀 Feature] [py]: Support FedCM commands for python

py/selenium/webdriver/chromium/webdriver.py

@@ -16,13 +16,14 @@
 # under the License.
 
 from selenium.webdriver.chromium.remote_connection import ChromiumRemoteConnection
+from selenium.webdriver.common.driver_extensions.has_fedcm_dialog import HasFedCmDialog
 from selenium.webdriver.common.driver_finder import DriverFinder
 from selenium.webdriver.common.options import ArgOptions
 from selenium.webdriver.common.service import Service
 from selenium.webdriver.remote.webdriver import WebDriver as RemoteWebDriver
 
 
-class ChromiumDriver(RemoteWebDriver):
+class ChromiumDriver(HasFedCmDialog, RemoteWebDriver):
     """Controls the WebDriver instance of ChromiumDriver and allows you to
     drive the browser."""
 

py/selenium/webdriver/common/driver_extensions/has_fedcm_dialog.py

@@ -0,0 +1,66 @@
+# 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 selenium.webdriver.common.fedcm.dialog import Dialog
+from selenium.webdriver.remote.webdriver import WebDriver
+
+
+class HasFedCmDialog(WebDriver):
+    """Mixin that provides FedCM-specific functionality."""
+
+    @property
+    def dialog(self):
+        """Returns the FedCM dialog object for interaction."""
+        return Dialog(self)
+
+    def enable_fedcm_delay(self):
+        """Re-enables the promise rejection delay for FedCM."""
+        self.fedcm.enable_delay()
+
+    def disable_fedcm_delay(self):
+        """Disables the promise rejection delay for FedCM.
+
+        FedCm by default delays promise resolution in failure cases for
+        privacy reasons. This method allows turning it off to let tests
+        run faster where this is not relevant.
+        """
+        self.fedcm.disable_delay()
+
+    def fedcm_cooldown(self):
+        """Resets the FedCm dialog cooldown.
+
+        If a user agent triggers a cooldown when the account chooser is
+        dismissed, this method resets that cooldown so that the dialog
+        can be triggered again immediately.
+        """
+        self.fedcm.reset_cooldown()
+
+    def fedcm_dialog(self, timeout=5, poll_frequency=0.5, ignored_exceptions=None):
+        """Waits for the FedCM dialog to appear."""
+        from selenium.common.exceptions import NoAlertPresentException
+        from selenium.webdriver.support.wait import WebDriverWait
+
+        if ignored_exceptions is None:
+            ignored_exceptions = (NoAlertPresentException,)
+
+        def _check_fedcm():
+            try:
+                return self.dialog if self.dialog.type else None
+            except NoAlertPresentException:
+                return None
+
+        wait = WebDriverWait(self, timeout, poll_frequency=poll_frequency, ignored_exceptions=ignored_exceptions)
+        return wait.until(lambda _: _check_fedcm())

py/selenium/webdriver/common/fedcm/account.py

@@ -0,0 +1,71 @@
+# 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 enum import Enum
+from typing import Optional
+
+
+class LoginState(Enum):
+    SIGN_IN = "SignIn"
+    SIGN_UP = "SignUp"
+
+
+class Account:
+    """Represents an account displayed in a FedCM account list.
+
+    See: https://w3c-fedid.github.io/FedCM/#dictdef-identityprovideraccount
+         https://w3c-fedid.github.io/FedCM/#webdriver-accountlist
+    """
+
+    def __init__(self, account_data):
+        self._account_data = account_data
+
+    @property
+    def account_id(self) -> Optional[str]:
+        return self._account_data.get("accountId")
+
+    @property
+    def email(self) -> Optional[str]:
+        return self._account_data.get("email")
+
+    @property
+    def name(self) -> Optional[str]:
+        return self._account_data.get("name")
+
+    @property
+    def given_name(self) -> Optional[str]:
+        return self._account_data.get("givenName")
+
+    @property
+    def picture_url(self) -> Optional[str]:
+        return self._account_data.get("pictureUrl")
+
+    @property
+    def idp_config_url(self) -> Optional[str]:
+        return self._account_data.get("idpConfigUrl")
+
+    @property
+    def terms_of_service_url(self) -> Optional[str]:
+        return self._account_data.get("termsOfServiceUrl")
+
+    @property
+    def privacy_policy_url(self) -> Optional[str]:
+        return self._account_data.get("privacyPolicyUrl")
+
+    @property
+    def login_state(self) -> Optional[str]:
+        return self._account_data.get("loginState")

py/selenium/webdriver/common/fedcm/dialog.py

@@ -0,0 +1,64 @@
+# 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 List
+from typing import Optional
+
+from .account import Account
+
+
+class Dialog:
+    """Represents a FedCM dialog that can be interacted with."""
+
+    DIALOG_TYPE_ACCOUNT_LIST = "AccountChooser"
+    DIALOG_TYPE_AUTO_REAUTH = "AutoReauthn"
+
+    def __init__(self, driver) -> None:
+        self._driver = driver
+
+    @property
+    def type(self) -> Optional[str]:
+        """Gets the type of the dialog currently being shown."""
+        return self._driver.fedcm.dialog_type
+
+    @property
+    def title(self) -> str:
+        """Gets the title of the dialog."""
+        return self._driver.fedcm.title
+
+    @property
+    def subtitle(self) -> Optional[str]:
+        """Gets the subtitle of the dialog."""
+        result = self._driver.fedcm.subtitle
+        return result.get("subtitle") if result else None
+
+    def get_accounts(self) -> List[Account]:
+        """Gets the list of accounts shown in the dialog."""
+        accounts = self._driver.fedcm.account_list
+        return [Account(account) for account in accounts]
+
+    def select_account(self, index: int) -> None:
+        """Selects an account from the dialog by index."""
+        self._driver.fedcm.select_account(index)
+
+    def click_continue(self) -> None:
+        """Clicks the continue button in the dialog."""
+        self._driver.fedcm.click_continue()
+
+    def cancel(self) -> None:
+        """Cancels/dismisses the dialog."""
+        self._driver.fedcm.cancel_dialog()

py/selenium/webdriver/remote/command.py

@@ -122,3 +122,13 @@ class Command:
     GET_DOWNLOADABLE_FILES: str = "getDownloadableFiles"
     DOWNLOAD_FILE: str = "downloadFile"
     DELETE_DOWNLOADABLE_FILES: str = "deleteDownloadableFiles"
+
+    # Federated Credential Management (FedCM)
+    GET_FEDCM_TITLE: str = "getFedcmTitle"
+    GET_FEDCM_DIALOG_TYPE: str = "getFedcmDialogType"
+    GET_FEDCM_ACCOUNT_LIST: str = "getFedcmAccountList"
+    SELECT_FEDCM_ACCOUNT: str = "selectFedcmAccount"
+    CLICK_FEDCM_DIALOG_BUTTON: str = "clickFedcmDialogButton"
+    CANCEL_FEDCM_DIALOG: str = "cancelFedcmDialog"
+    SET_FEDCM_DELAY: str = "setFedcmDelay"
+    RESET_FEDCM_COOLDOWN: str = "resetFedcmCooldown"

py/selenium/webdriver/remote/fedcm.py

@@ -0,0 +1,70 @@
+# 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 List
+from typing import Optional
+
+from .command import Command
+
+
+class FedCM:
+    def __init__(self, driver) -> None:
+        self._driver = driver
+
+    @property
+    def title(self) -> str:
+        """Gets the title of the dialog."""
+        return self._driver.execute(Command.GET_FEDCM_TITLE)["value"].get("title")
+
+    @property
+    def subtitle(self) -> Optional[str]:
+        """Gets the subtitle of the dialog."""
+        return self._driver.execute(Command.GET_FEDCM_TITLE)["value"].get("subtitle")
+
+    @property
+    def dialog_type(self) -> str:
+        """Gets the type of the dialog currently being shown."""
+        return self._driver.execute(Command.GET_FEDCM_DIALOG_TYPE).get("value")
+
+    @property
+    def account_list(self) -> List[dict]:
+        """Gets the list of accounts shown in the dialog."""
+        return self._driver.execute(Command.GET_FEDCM_ACCOUNT_LIST).get("value")
+
+    def select_account(self, index: int) -> None:
+        """Selects an account from the dialog by index."""
+        self._driver.execute(Command.SELECT_FEDCM_ACCOUNT, {"accountIndex": index})
+
+    def click_continue(self) -> None:
+        """Clicks the continue button in the dialog."""
+        self._driver.execute(Command.CLICK_FEDCM_DIALOG_BUTTON, {"dialogButton": "ConfirmIdpLoginContinue"})
+
+    def cancel_dialog(self) -> None:
+        """Cancels/dismisses the FedCM dialog."""
+        self._driver.execute(Command.CANCEL_FEDCM_DIALOG)
+
+    def enable_delay(self) -> None:
+        """Re-enables the promise rejection delay for FedCM."""
+        self._driver.execute(Command.SET_FEDCM_DELAY, {"enabled": True})
+
+    def disable_delay(self) -> None:
+        """Disables the promise rejection delay for FedCM."""
+        self._driver.execute(Command.SET_FEDCM_DELAY, {"enabled": False})
+
+    def reset_cooldown(self) -> None:
+        """Resets the FedCM dialog cooldown, allowing immediate retriggers."""
+        self._driver.execute(Command.RESET_FEDCM_COOLDOWN)

py/selenium/webdriver/remote/remote_connection.py

@@ -125,6 +125,15 @@
     Command.GET_DOWNLOADABLE_FILES: ("GET", "/session/$sessionId/se/files"),
     Command.DOWNLOAD_FILE: ("POST", "/session/$sessionId/se/files"),
     Command.DELETE_DOWNLOADABLE_FILES: ("DELETE", "/session/$sessionId/se/files"),
+    # Federated Credential Management (FedCM)
+    Command.GET_FEDCM_TITLE: ("GET", "/session/$sessionId/fedcm/gettitle"),
+    Command.GET_FEDCM_DIALOG_TYPE: ("GET", "/session/$sessionId/fedcm/getdialogtype"),
+    Command.GET_FEDCM_ACCOUNT_LIST: ("GET", "/session/$sessionId/fedcm/accountlist"),
+    Command.CLICK_FEDCM_DIALOG_BUTTON: ("POST", "/session/$sessionId/fedcm/clickdialogbutton"),
+    Command.CANCEL_FEDCM_DIALOG: ("POST", "/session/$sessionId/fedcm/canceldialog"),
+    Command.SELECT_FEDCM_ACCOUNT: ("POST", "/session/$sessionId/fedcm/selectaccount"),
+    Command.SET_FEDCM_DELAY: ("POST", "/session/$sessionId/fedcm/setdelayenabled"),
+    Command.RESET_FEDCM_COOLDOWN: ("POST", "/session/$sessionId/fedcm/resetcooldown"),
 }
 
 

py/selenium/webdriver/remote/webdriver.py

@@ -57,6 +57,7 @@
 from .client_config import ClientConfig
 from .command import Command
 from .errorhandler import ErrorHandler
+from .fedcm import FedCM
 from .file_detector import FileDetector
 from .file_detector import LocalFileDetector
 from .locator_converter import LocatorConverter
@@ -236,6 +237,7 @@ def __init__(
         self._authenticator_id = None
         self.start_client()
         self.start_session(capabilities)
+        self._fedcm = FedCM(self)
 
         self._websocket_connection = None
         self._script = None
@@ -1222,3 +1224,24 @@ def delete_downloadable_files(self) -> None:
             raise WebDriverException("You must enable downloads in order to work with downloadable files.")
 
         self.execute(Command.DELETE_DOWNLOADABLE_FILES)
+
+    @property
+    def fedcm(self) -> FedCM:
+        """
+        :Returns:
+            - FedCM: an object providing access to all Federated Credential Management (FedCM) dialog commands.
+
+        :Usage:
+            ::
+
+                title = driver.fedcm.title
+                subtitle = driver.fedcm.subtitle
+                dialog_type = driver.fedcm.dialog_type
+                accounts = driver.fedcm.account_list
+                driver.fedcm.select_account(0)
+                driver.fedcm.click_dialog_button()
+                driver.fedcm.cancel_dialog()
+                driver.fedcm.set_delay(False)
+                driver.fedcm.reset_cooldown()
+        """
+        return self._fedcm