Update CDP Mode

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -766,6 +766,7 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.get_rd_port = CDPM.get_rd_port
     cdp.get_rd_url = CDPM.get_rd_url
     cdp.get_endpoint_url = CDPM.get_endpoint_url
+    cdp.get_websocket_url = CDPM.get_websocket_url
     cdp.get_port = CDPM.get_port
     cdp.find_element = CDPM.find_element
     cdp.find = CDPM.find_element

seleniumbase/core/sb_cdp.py

@@ -206,6 +206,11 @@ def get_port(self):
         """Same as get_rd_port(), which returns the remote-debugging port."""
         return self.get_rd_port()
 
+    def get_websocket_url(self):
+        """Returns the websocket URL of the active tab.
+        The websocket URL starts with `ws://`."""
+        return self.get_active_tab().websocket_url
+
     def add_handler(self, event, handler):
         self.page.add_handler(event, handler)
 

seleniumbase/undetected/__init__.py

@@ -179,6 +179,7 @@ def __init__(
         options.add_argument("--remote-debugging-host=%s" % debug_host)
         options.add_argument("--remote-debugging-port=%s" % debug_port)
         if user_data_dir:
+            user_data_dir = os.path.abspath(user_data_dir)
             options.add_argument("--user-data-dir=%s" % user_data_dir)
         language, keep_user_data_dir = None, bool(user_data_dir)
         # See if a custom user profile is specified in options

seleniumbase/undetected/cdp_driver/cdp_util.py

@@ -566,6 +566,8 @@ async def start(
                 platform_var = platform_var[1:-1]
     if IS_LINUX and not headless and not headed and not xvfb:
         xvfb = True  # The default setting on Linux
+    if port and not host:
+        host = "127.0.0.1"  # Assume localhost
     if not host or not port:
         # The browser hasn't been launched yet. (May need a virtual display)
         __activate_virtual_display_as_needed(
@@ -611,6 +613,8 @@ async def start(
             elif udd_string.startswith("'") and udd_string.endswith("'"):
                 udd_string = udd_string[1:-1]
             user_data_dir = udd_string
+    if user_data_dir:
+        user_data_dir = os.path.abspath(user_data_dir)
     if not browser_executable_path:
         browser = None
         if "browser" in kwargs: