Start the work on add_request_handler

Author: aguspe

rb/lib/selenium/webdriver/bidi/network.rb

@@ -60,11 +60,27 @@ def continue_with_auth(request_id, username, password)
           )
         end
 
+        def continue_with_request(**args)
+          @bidi.send_cmd(
+            'network.continueWithRequest',
+            request: args[:request_id],
+            'body' => args[:body],
+            'cookies' => args[:cookies],
+            'headers' => args[:headers],
+            'method' => args[:method],
+            'url' => args[:url]
+          )
+        end
+
         def on(event, &)
           event = EVENTS[event] if event.is_a?(Symbol)
           @bidi.add_callback(event, &)
         end
-      end # Network
-    end # BiDi
+      end
+
+      # Network
+    end
+
+    # BiDi
   end # WebDriver
 end # Selenium

rb/lib/selenium/webdriver/common/network.rb

@@ -20,11 +20,12 @@
 module Selenium
   module WebDriver
     class Network
-      attr_reader :auth_callbacks
+      attr_reader :auth_callbacks, :request_callbacks
 
       def initialize(bridge)
         @network = BiDi::Network.new(bridge.bidi)
         @auth_callbacks = {}
+        @request_callbacks = {}
       end
 
       def add_authentication_handler(username, password)
@@ -47,6 +48,18 @@ def remove_authentication_handler(id)
       def clear_authentication_handlers
         @auth_callbacks.each_key { |id| remove_authentication_handler(id) }
       end
+
+      def add_request_handler
+        intercept = @network.add_intercept(phases: [BiDi::Network::PHASES[:before_request]])
+        request_id = @network.on(:before_request) do |event|
+          request_id = event['requestId']
+          @network.continue_with_request(request_id: request_id)
+        end
+
+        @request_callbacks[request_id] = intercept
+
+        request_id
+      end
     end # Network
   end # WebDriver
 end # Selenium

rb/sig/lib/selenium/webdriver/bidi/network.rbs

@@ -12,6 +12,8 @@ module Selenium
 
         def add_intercept: (?phases: Array[String], ?contexts: BrowsingContext?, ?url_patterns: untyped?) -> Hash[String, String]
 
+        def continue_with_request: -> untyped
+
         def remove_intercept: (String intercept) -> untyped
 
         def continue_with_auth: (String request_id, String username, String password) -> untyped

rb/sig/lib/selenium/webdriver/common/network.rbs

@@ -3,12 +3,16 @@ module Selenium
     class Network
       @network: BiDi::Network
 
+      @request_callbacks: untyped
+
       attr_reader auth_callbacks: Hash[String, String]
 
       def initialize: (Remote::Bridge bridge) -> void
 
       def add_authentication_handler: (String username, String password) -> String
 
+      def add_request_handler: -> untyped
+
       def clear_authentication_handlers: -> Hash[nil, nil]
 
       def remove_authentication_handler: (String id) -> nil

rb/spec/integration/selenium/webdriver/bidi/network_spec.rb

@@ -56,6 +56,20 @@ class BiDi
             expect(driver.find_element(tag_name: 'h1').text).to eq('authorized')
           end
         end
+
+        it 'continues with request' do
+          reset_driver!(web_socket_url: true) do |driver|
+            network = described_class.new(driver.bidi)
+            network.add_intercept(phases: [described_class::PHASES[:before_request]])
+            network.on(:before_request) do |event|
+              request_id = event['requestId']
+              network.continue_with_request(request_id: request_id)
+            end
+
+            driver.navigate.to url_for('simple_page')
+            expect(driver.find_element(tag_name: 'h1').text).to eq('Simple Page')
+          end
+        end
       end
     end
   end

rb/spec/integration/selenium/webdriver/network_spec.rb

@@ -53,6 +53,14 @@ module WebDriver
           expect(network.auth_callbacks.count).to be 0
         end
       end
+
+      it 'adds a request handler' do
+        reset_driver!(web_socket_url: true) do |driver|
+          network = described_class.new(driver)
+          network.add_request_handler
+          expect(network.request_callbacks.count).to be 1
+        end
+      end
     end
   end
 end