osn-selenium 1.1.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

osn_selenium/trio_bidi/mapping/request_functions/_capture.py

@@ -0,0 +1,71 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi.mapping._js_snippets import CAPTURE
+from osn_selenium.trio_bidi._typehints import (
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+from osn_selenium.trio_bidi.mapping.request_functions._args_helpers import (
+	build_script_evaluate_request
+)
+
+
+__all__ = ["MAPPED_REQUEST_CAPTURE_COMMANDS"]
+
+
+def _map_screenshot_request(params: REQUEST_PARAMS_TYPEHINT, context_id: str) -> Dict[str, Any]:
+	"""
+	Maps the screenshot command to BiDi 'captureScreenshot'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (str): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.captureScreenshot",
+		"params": {"context": context_id}
+	}
+
+
+def _map_print_page_request(params: REQUEST_PARAMS_TYPEHINT, context_id: str) -> Dict[str, Any]:
+	"""
+	Maps the print command to BiDi 'browsingContext.print'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Print parameters.
+		context_id (str): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.print",
+		"params": {"context": context_id, **params}
+	}
+
+
+def _map_get_page_source_request(params: REQUEST_PARAMS_TYPEHINT, context_id: str) -> Dict[str, Any]:
+	"""
+	Maps the get page source command to a JS evaluation.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (str): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_script_evaluate_request(context_id=context_id, expression=CAPTURE.GET_PAGE_SOURCE)
+
+
+MAPPED_REQUEST_CAPTURE_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.SCREENSHOT: _map_screenshot_request,
+	Command.GET_PAGE_SOURCE: _map_get_page_source_request,
+	Command.PRINT_PAGE: _map_print_page_request,
+}

osn_selenium/trio_bidi/mapping/request_functions/_cdp.py

@@ -0,0 +1,40 @@
+from typing import Any, Dict
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_CDP_COMMANDS"]
+
+
+def _execute_cdp_command(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT,
+) -> Dict[str, Any]:
+	"""
+	Maps a generic CDP send command to the BiDi Chrome-specific protocol.
+
+	Args:
+		params (Optional[Dict[str, Any]]): Parameters containing 'cmd' and 'params'.
+		context_id (Optional[str]): The target browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The mapped BiDi request.
+	"""
+	
+	cdp_method = params.get("cmd", "")
+	cdp_params = params.get("params", {})
+	
+	return {
+		"method": "goog:cdp.sendCommand",
+		"params": {
+			"method": cdp_method,
+			"params": cdp_params,
+			"browsingContext": context_id,
+		}
+	}
+
+
+MAPPED_REQUEST_CDP_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {"executeCdpCommand": _execute_cdp_command}

osn_selenium/trio_bidi/mapping/request_functions/_element.py

@@ -0,0 +1,55 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi.mapping.request_functions._args_helpers import (
+	build_locate_nodes_request
+)
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_ELEMENT_COMMANDS"]
+
+
+def _map_find_elements_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the 'find elements' command to BiDi 'locateNodes'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Strategy and value.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_locate_nodes_request(context_id=context_id, params=params)
+
+
+def _map_find_element_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the 'find element' command to BiDi 'locateNodes' with max count of 1.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Strategy and value.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_locate_nodes_request(context_id=context_id, params=params, max_nodes=1)
+
+
+MAPPED_REQUEST_ELEMENT_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.FIND_ELEMENT: _map_find_element_request,
+	Command.FIND_ELEMENTS: _map_find_elements_request
+}

osn_selenium/trio_bidi/mapping/request_functions/_file.py

@@ -0,0 +1,32 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi._internal_mappings import UPLOAD_INTERNAL_FILE
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_FILE_COMMANDS"]
+
+
+def _map_upload_file_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the file upload command to an internal bridge command.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The internal BiDi request.
+	"""
+	
+	return {"method": UPLOAD_INTERNAL_FILE, "params": {}}
+
+
+MAPPED_REQUEST_FILE_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {Command.UPLOAD_FILE: _map_upload_file_request}

osn_selenium/trio_bidi/mapping/request_functions/_navigation.py

@@ -0,0 +1,147 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi.mapping._js_snippets import NAVIGATION
+from osn_selenium.trio_bidi.mapping.request_functions._args_helpers import (
+	build_script_evaluate_request
+)
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_NAVIGATION_COMMANDS"]
+
+
+def _map_refresh_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the refresh command to BiDi 'browsingContext.reload'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.reload",
+		"params": {"context": context_id, "ignoreCache": True, "wait": "interactive"}
+	}
+
+
+def _map_get_title_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the get title command to a JS evaluation.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_script_evaluate_request(context_id=context_id, expression=NAVIGATION.GET_TITLE)
+
+
+def _map_get_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the navigate command to BiDi 'browsingContext.navigate'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Parameters containing 'url'.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.navigate",
+		"params": {"context": context_id, "url": params["url"], "wait": "interactive"}
+	}
+
+
+def _map_get_current_url_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the get current URL command to BiDi 'browsingContext.getTree'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.getTree",
+		"params": {"root": context_id, "maxDepth": 0}
+	}
+
+
+def _map_forward_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the forward history command to BiDi history traversal.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.traverseHistory",
+		"params": {"context": context_id, "delta": 1}
+	}
+
+
+def _map_back_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the back history command to BiDi history traversal.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {
+		"method": "browsingContext.traverseHistory",
+		"params": {"context": context_id, "delta": -1}
+	}
+
+
+MAPPED_REQUEST_NAVIGATION_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.GET: _map_get_request,
+	Command.GET_CURRENT_URL: _map_get_current_url_request,
+	Command.REFRESH: _map_refresh_request,
+	Command.GO_BACK: _map_back_request,
+	Command.GO_FORWARD: _map_forward_request,
+	Command.GET_TITLE: _map_get_title_request,
+}

osn_selenium/trio_bidi/mapping/request_functions/_script.py

@@ -0,0 +1,85 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi.mapping.request_functions._args_helpers import (
+	build_script_call_request,
+	convert_w3c_to_bidi_js
+)
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_SCRIPT_COMMANDS"]
+
+
+def _map_w3c_execute_script_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps synchronous script execution to BiDi 'script.callFunction'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Script and arguments.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	script = params.get("script", "")
+	args = params.get("args", [])
+	bidi_args = [convert_w3c_to_bidi_js(w3c_value=arg) for arg in args]
+	
+	return build_script_call_request(
+			context_id=context_id,
+			function=f"function() {{ {script} }}",
+			args=bidi_args,
+			ownership="none",
+	)
+
+
+def _map_w3c_execute_script_async_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps asynchronous script execution to a Promise-based BiDi 'script.callFunction'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Script and arguments.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	script = params.get("script", "")
+	args = params.get("args", [])
+	bidi_args = [convert_w3c_to_bidi_js(w3c_value=arg) for arg in args]
+	
+	wrapper_script = f"""
+function() {{
+	return new Promise((resolve) => {{
+		const args = Array.from(arguments);
+		const callback = (result) => resolve(result);
+		args.push(callback);
+		(function() {{ {script} }}).apply(null, args);
+	}});
+}}
+"""
+	
+	return build_script_call_request(
+			context_id=context_id,
+			function=wrapper_script,
+			args=bidi_args,
+			ownership="none",
+	)
+
+
+MAPPED_REQUEST_SCRIPT_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.W3C_EXECUTE_SCRIPT: _map_w3c_execute_script_request,
+	Command.W3C_EXECUTE_SCRIPT_ASYNC: _map_w3c_execute_script_async_request,
+}

osn_selenium/trio_bidi/mapping/request_functions/_shadow_root.py

@@ -0,0 +1,65 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi.mapping.request_functions._args_helpers import (
+	build_locate_nodes_request,
+	get_shadow_root_argument
+)
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_SHADOW_ROOT_COMMANDS"]
+
+
+def _map_find_elements_from_shadow_root_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the find elements from shadow root command to BiDi 'locateNodes'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Strategy, value and shadow ID.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_locate_nodes_request(
+			context_id=context_id,
+			params=params,
+			start_nodes=[get_shadow_root_argument(params=params)],
+	)
+
+
+def _map_find_element_from_shadow_root_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps the find element from shadow root command to BiDi 'locateNodes' (max 1).
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Strategy, value and shadow ID.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return build_locate_nodes_request(
+			context_id=context_id,
+			params=params,
+			max_nodes=1,
+			start_nodes=[get_shadow_root_argument(params=params)],
+	)
+
+
+MAPPED_REQUEST_SHADOW_ROOT_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.FIND_ELEMENT_FROM_SHADOW_ROOT: _map_find_element_from_shadow_root_request,
+	Command.FIND_ELEMENTS_FROM_SHADOW_ROOT: _map_find_elements_from_shadow_root_request,
+}

osn_selenium/trio_bidi/mapping/request_functions/_switch_to.py

@@ -0,0 +1,59 @@
+from typing import Any, Dict
+from selenium.webdriver.remote.command import Command
+from osn_selenium.trio_bidi._typehints import (
+	CURRENT_BROWSING_CONTEXT_TYPEHINT,
+	MAP_REQUEST_FUNCTION_TYPEHINT,
+	REQUEST_PARAMS_TYPEHINT
+)
+
+
+__all__ = ["MAPPED_REQUEST_SWITCH_TO_COMMANDS"]
+
+
+def _map_switch_to_window_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps switch to window command to BiDi 'browsingContext.getTree'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Parameters containing 'handle'.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The current browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	target_handle = params.get("handle")
+	
+	return {
+		"method": "browsingContext.getTree",
+		"params": {"root": target_handle, "maxDepth": 0}
+	}
+
+
+def _map_new_window_request(
+		params: REQUEST_PARAMS_TYPEHINT,
+		context_id: CURRENT_BROWSING_CONTEXT_TYPEHINT
+) -> Dict[str, Any]:
+	"""
+	Maps new window command to BiDi 'browsingContext.create'.
+
+	Args:
+		params (REQUEST_PARAMS_TYPEHINT): Request parameters.
+		context_id (CURRENT_BROWSING_CONTEXT_TYPEHINT): The browsing context ID.
+
+	Returns:
+		Dict[str, Any]: The BiDi request.
+	"""
+	
+	return {"method": "browsingContext.create", "params": {"type": "tab"}}
+
+
+MAPPED_REQUEST_SWITCH_TO_COMMANDS: Dict[str, MAP_REQUEST_FUNCTION_TYPEHINT] = {
+	Command.SWITCH_TO_WINDOW: _map_switch_to_window_request,
+	Command.NEW_WINDOW: _map_new_window_request,
+}
+
+# TODO create mappings for Command.SWITCH_TO_FRAME, Command.SWITCH_TO_PARENT_FRAME and Command.W3C_GET_ACTIVE_ELEMENT