[🚀 Feature] [py]: Implement BiDi add/remove_request handler

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

@@ -0,0 +1,59 @@
+# 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 dataclasses import dataclass
+from dataclasses import fields
+from dataclasses import is_dataclass
+from typing import get_type_hints
+
+
+@dataclass
+class BidiObject:
+    def to_json(self):
+        json = {}
+        for field in fields(self):
+            value = getattr(self, field.name)
+            if value is None:
+                continue
+            if is_dataclass(value):
+                value = value.to_json()
+            elif isinstance(value, list):
+                value = [v.to_json() if hasattr(v, "to_json") else v for v in value]
+            elif isinstance(value, dict):
+                value = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in value.items()}
+            key = field.name[:-1] if field.name.endswith("_") else field.name
+            json[key] = value
+        return json
+
+    @classmethod
+    def from_json(cls, json):
+        return cls(**json)
+
+
+@dataclass
+class BidiEvent(BidiObject):
+    @classmethod
+    def from_json(cls, json):
+        params = get_type_hints(cls)["params"].from_json(json)
+        return cls(params)
+
+
+@dataclass
+class BidiCommand(BidiObject):
+    def cmd(self):
+        result = yield self.to_json()
+        return result

py/selenium/webdriver/common/bidi/browsing_context.py

@@ -0,0 +1,48 @@
+# 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.
+
+import typing
+from dataclasses import dataclass
+
+from .bidi import BidiCommand
+from .bidi import BidiObject
+
+ReadinessState = typing.Literal["none", "interactive", "complete"]
+
+
+@dataclass
+class NavigateParameters(BidiObject):
+    context: str
+    url: str
+    wait: typing.Optional[ReadinessState] = None
+
+
+@dataclass
+class Navigate(BidiCommand):
+    params: NavigateParameters
+    method: typing.Literal["browsingContext.navigate"] = "browsingContext.navigate"
+
+
+Navigation = str
+
+class BrowsingContext:
+    def __init__(self, conn):
+        self.conn = conn
+
+    def navigate(self, params: NavigateParameters):
+        result = self.conn.execute(NavigateParameters(params).cmd())
+        return result

py/selenium/webdriver/common/bidi/network.py

@@ -0,0 +1,213 @@
+# 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.
+
+import typing
+from dataclasses import dataclass
+
+from selenium.webdriver.common.bidi.cdp import import_devtools
+
+from . import browsing_context
+from . import script
+from .bidi import BidiCommand
+from .bidi import BidiEvent
+from .bidi import BidiObject
+
+devtools = import_devtools("")
+event_class = devtools.util.event_class
+
+InterceptPhase = typing.Literal["beforeRequestSent", "responseStarted", "authRequired"]
+
+
+@dataclass
+class UrlPatternPattern(BidiObject):
+    type_: typing.Literal["pattern"] = "pattern"
+    protocol: typing.Optional[str] = None
+    hostname: typing.Optional[str] = None
+    port: typing.Optional[str] = None
+    pathname: typing.Optional[str] = None
+    search: typing.Optional[str] = None
+
+
+@dataclass
+class UrlPatternString(BidiObject):
+    pattern: str
+    type_: typing.Literal["string"] = "string"
+
+
+UrlPattern = typing.Union[UrlPatternPattern, UrlPatternString]
+
+
+@dataclass
+class AddInterceptParameters(BidiObject):
+    phases: typing.List[InterceptPhase]
+    contexts: typing.Optional[typing.List[browsing_context.BrowsingContext]] = None
+    urlPatterns: typing.Optional[typing.List[UrlPattern]] = None
+
+
+@dataclass
+class AddIntercept(BidiCommand):
+    params: AddInterceptParameters
+    method: typing.Literal["network.addIntercept"] = "network.addIntercept"
+
+
+Request = str
+
+
+@dataclass
+class StringValue(BidiObject):
+    value: str
+    type_: typing.Literal["string"] = "string"
+
+
+@dataclass
+class Base64Value(BidiObject):
+    value: str
+    type_: typing.Literal["base64"] = "base64"
+
+
+BytesValue = typing.Union[StringValue, Base64Value]
+
+
+@dataclass
+class CookieHeader(BidiObject):
+    name: str
+    value: BytesValue
+
+
+@dataclass
+class Header(BidiObject):
+    name: str
+    value: BytesValue
+
+
+@dataclass
+class ContinueRequestParameters(BidiObject):
+    request: Request
+    body: typing.Optional[BytesValue] = None
+    cookies: typing.Optional[typing.List[CookieHeader]] = None
+    headers: typing.Optional[typing.List[Header]] = None
+    method: typing.Optional[str] = None
+    url: typing.Optional[str] = None
+
+
+@dataclass
+class ContinueRequest(BidiCommand):
+    params: ContinueRequestParameters
+    method: typing.Literal["network.continueRequest"] = "network.continueRequest"
+
+
+Intercept = str
+
+
+@dataclass
+class RemoveInterceptParameters(BidiObject):
+    intercept: Intercept
+
+
+@dataclass
+class RemoveIntercept(BidiCommand):
+    params: RemoveInterceptParameters
+    method: typing.Literal["network.removeIntercept"] = "network.removeIntercept"
+
+
+SameSite = typing.Literal["strict", "lax", "none"]
+
+
+@dataclass
+class Cookie(BidiObject):
+    name: str
+    value: BytesValue
+    domain: str
+    path: str
+    size: int
+    httpOnly: bool
+    secure: bool
+    sameSite: SameSite
+    expiry: typing.Optional[int] = None
+
+
+@dataclass
+class FetchTimingInfo(BidiObject):
+    timeOrigin: float
+    requestTime: float
+    redirectStart: float
+    redirectEnd: float
+    fetchStart: float
+    dnsStart: float
+    dnsEnd: float
+    connectStart: float
+    connectEnd: float
+    tlsStart: float
+    requestStart: float
+    responseStart: float
+    responseEnd: float
+
+
+@dataclass
+class RequestData(BidiObject):
+    request: Request
+    url: str
+    method: str
+    headersSize: int
+    timings: FetchTimingInfo
+    headers: typing.Optional[typing.List[Header]] = None
+    cookies: typing.Optional[typing.List[Cookie]] = None
+    bodySize: typing.Optional[int] = None
+
+
+@dataclass
+class Initiator(BidiObject):
+    type_: typing.Literal["parser", "script", "preflight", "other"]
+    columnNumber: typing.Optional[int] = None
+    lineNumber: typing.Optional[int] = None
+    stackTrace: typing.Optional[script.StackTrace] = None
+    request: typing.Optional[Request] = None
+
+
+@dataclass
+class BeforeRequestSentParameters(BidiObject):
+    isBlocked: bool
+    redirectCount: int
+    request: RequestData
+    timestamp: int
+    initiator: Initiator
+    context: typing.Optional[browsing_context.BrowsingContext] = None
+    navigation: typing.Optional[browsing_context.Navigation] = None
+    intercepts: typing.Optional[typing.List[Intercept]] = None
+
+
+@dataclass
+@event_class("network.beforeRequestSent")
+class BeforeRequestSent(BidiEvent):
+    params: BeforeRequestSentParameters
+    method: typing.Literal["network.beforeRequestSent"] = "network.beforeRequestSent"
+
+
+class Network:
+    def __init__(self, conn):
+        self.conn = conn
+
+    def add_intercept(self, params: AddInterceptParameters):
+        result = self.conn.execute(AddIntercept(params).cmd())
+        return result
+
+    def continue_request(self, params: ContinueRequestParameters):
+        result = self.conn.execute(ContinueRequest(params).cmd())
+        return result
+
+    def remove_intercept(self, params: RemoveInterceptParameters):
+        self.conn.execute(RemoveIntercept(params).cmd())

py/selenium/webdriver/common/bidi/script.py

@@ -17,7 +17,9 @@
 
 from dataclasses import dataclass
 from typing import List
+from typing import Optional
 
+from .bidi import BidiObject
 from .session import session_subscribe
 from .session import session_unsubscribe
 
@@ -108,3 +110,16 @@ def from_json(cls, json):
             stacktrace=json["stackTrace"],
             type_=json["type"],
         )
+
+
+@dataclass
+class StackFrame(BidiObject):
+    columnNumber: int
+    functionName: str
+    lineNumber: int
+    url: str
+
+
+@dataclass
+class StackTrace(BidiObject):
+    callFrames: Optional[List[StackFrame]] = None