bidi2pdf 0.1.2 → 0.1.4

data/lib/bidi2pdf/bidi/client.rb

@@ -4,10 +4,9 @@ require "json"
 require "websocket-client-simple"
 
 require_relative "web_socket_dispatcher"
-require_relative "add_headers_interceptor"
-require_relative "auth_interceptor"
 require_relative "command_manager"
 require_relative "connection_manager"
+require_relative "commands"
 
 module Bidi2pdf
   module Bidi
@@ -18,29 +17,22 @@ module Bidi2pdf
 
       def initialize(ws_url)
         @ws_url = ws_url
-        @id = 0
-        @pending_responses = {}
-
-        @connected = false
-        @connection_mutex = Mutex.new
-        @next_id_mutex = Mutex.new
-        @connection_cv = ConditionVariable.new
-
         @started = false
+        @connection_manager = ConnectionManager.new(logger: Bidi2pdf.logger)
       end
 
       def start
         return @socket if started?
 
-        @socket = WebSocket::Client::Simple.connect(ws_url)
+        WebSocket::Client::Simple.connect(ws_url) do |socket|
+          @socket = socket
+          @command_manager = CommandManager.new(@socket, logger: Bidi2pdf.logger)
 
-        @connection_manager = ConnectionManager.new(logger: Bidi2pdf.logger)
-        @command_manager = CommandManager.new(@socket, logger: Bidi2pdf.logger)
-
-        dispatcher.on_open { @connection_manager.mark_connected }
-        dispatcher.on_message { |data| handle_response_to_cmd(data) }
+          dispatcher.on_open { @connection_manager.mark_connected }
+          dispatcher.on_message { |data| handle_response_to_cmd(data) }
+          dispatcher.start_listening
+        end
 
-        dispatcher.start_listening
         @started = true
 
         @socket
@@ -50,15 +42,23 @@ module Bidi2pdf
 
       def wait_until_open(timeout: Bidi2pdf.default_timeout)
         @connection_manager.wait_until_open(timeout: timeout)
+      rescue Bidi2pdf::WebsocketError => e
+        raise Bidi2pdf::WebsocketError, "Client#start must be called within #{timeout} sec." unless started?
+
+        raise e
       end
 
-      def send_cmd(method, params = {})
-        @command_manager.send_cmd(method, params)
+      def send_cmd(cmd)
+        raise Bidi2pdf::ClientError, "Client#start must be called before" unless started?
+
+        @command_manager.send_cmd(cmd)
       end
 
-      def send_cmd_and_wait(method, params = {}, timeout: Bidi2pdf.default_timeout, &block)
-        timed("Command #{method}") do
-          @command_manager.send_cmd_and_wait(method, params, timeout: timeout, &block)
+      def send_cmd_and_wait(cmd, timeout: Bidi2pdf.default_timeout, &block)
+        raise Bidi2pdf::ClientError, "Client#start must be called before" unless started?
+
+        timed("Command #{cmd.inspect}") do
+          @command_manager.send_cmd_and_wait(cmd, timeout: timeout, &block)
         end
       end
 
@@ -72,7 +72,8 @@ module Bidi2pdf
 
       def on_event(*names, &block)
         names.each { |name| dispatcher.on_event(name, &block) }
-        send_cmd("session.subscribe", { events: names }) if names.any?
+        cmd = Bidi2pdf::Bidi::Commands::SessionSubscribe.new(events: names)
+        send_cmd(cmd) if names.any?
       end
 
       def remove_message_listener(block) = dispatcher.remove_message_listener(block)
@@ -81,26 +82,13 @@ module Bidi2pdf
         names.each { |event_name| dispatcher.remove_event_listener(event_name, block) }
       end
 
-      def add_headers_interceptor(context:, url_patterns:, headers:)
-        add_interceptor(
-          context: context,
-          url_patterns: url_patterns,
-          phase: "beforeRequestSent",
-          event: "network.beforeRequestSent",
-          interceptor_class: AddHeadersInterceptor,
-          extra_args: { headers: headers }
-        )
-      end
+      def close
+        return unless @socket
 
-      def add_auth_interceptor(context:, url_patterns:, username:, password:)
-        add_interceptor(
-          context: context,
-          url_patterns: url_patterns,
-          phase: "authRequired",
-          event: "network.authRequired",
-          interceptor_class: AuthInterceptor,
-          extra_args: { username: username, password: password }
-        )
+        Bidi2pdf.logger.debug "Closing WebSocket connection"
+        @socket&.close
+        @socket = nil
+        @started = false
       end
 
       private
@@ -119,21 +107,6 @@ module Bidi2pdf
           Bidi2pdf.logger.warn "Unknown response: #{data.inspect}"
         end
       end
-
-      def add_interceptor(context:, url_patterns:, phase:, event:, interceptor_class:, extra_args: {})
-        send_cmd_and_wait("network.addIntercept", {
-          context: context,
-          phases: [phase],
-          urlPatterns: url_patterns
-        }) do |response|
-          id = response["result"]["intercept"]
-          Bidi2pdf.logger.debug "Interceptor added: #{id}"
-
-          interceptor_class.new(id, **extra_args, client: self).tap do |interceptor|
-            on_event(event, &interceptor.method(:handle_event))
-          end
-        end
-      end
     end
   end
 end

data/lib/bidi2pdf/bidi/command_manager.rb

@@ -3,18 +3,35 @@
 module Bidi2pdf
   module Bidi
     class CommandManager
+      class << self
+        def initialize_counter
+          @id = 0
+          @id_mutex = Mutex.new
+        end
+
+        def next_id = @id_mutex.synchronize { @id += 1 }
+      end
+
+      initialize_counter
+
       def initialize(socket, logger:)
         @socket = socket
         @logger = logger
 
-        @id = 0
-        @next_id_mutex = Mutex.new
         @pending_responses = {}
+        @initiated_cmds = {}
       end
 
-      def send_cmd(method, params = {})
+      def send_cmd(cmd, store_response: false)
         id = next_id
-        payload = { id: id, method: method, params: params }
+
+        if store_response
+          init_queue_for id
+        else
+          @initiated_cmds[id] = true
+        end
+
+        payload = cmd.as_payload(id)
 
         @logger.debug "Sending command: #{redact_sensitive_fields(payload).inspect}"
         @socket.send(payload.to_json)
@@ -22,40 +39,46 @@ module Bidi2pdf
         id
       end
 
-      def send_cmd_and_wait(method, params = {}, timeout: Bidi2pdf.default_timeout)
-        id = send_cmd(method, params)
-        queue = @pending_responses[id]
+      def send_cmd_and_wait(cmd, timeout: Bidi2pdf.default_timeout)
+        id = send_cmd(cmd, store_response: true)
+        response = pop_response id, timeout: timeout
 
-        response = queue.pop(timeout: timeout)
-        raise_timeout_error(id, method, params) if response.nil?
-        raise "Error response: #{response["error"]}" if response["error"]
+        raise_timeout_error(id, cmd) if response.nil?
+        raise CmdError, "Error response: #{response["error"]} #{cmd.inspect}" if response["error"]
 
         block_given? ? yield(response) : response
       ensure
         @pending_responses.delete(id)
       end
 
-      def queue_for(id)
-        @pending_responses[id]
+      def pop_response(id, timeout:)
+        raise CmdResponseNotStoredError, "No response stored for command ID #{id} or already popped or this command was not send" unless @pending_responses.key?(id)
+
+        @pending_responses[id].pop(timeout: timeout)
+      ensure
+        @pending_responses.delete(id)
       end
 
       def handle_response(data)
-        if (id = data["id"]) && @pending_responses.key?(id)
-          @pending_responses[id]&.push(data)
-        else
-          false
+        if (id = data["id"])
+          if @pending_responses.key?(id)
+            @pending_responses[id]&.push(data)
+            return true
+          elsif @initiated_cmds.key?(id)
+            @logger.error "Received error: #{data["error"]} for cmd: #{id}" if data["error"]
+
+            return @initiated_cmds.delete(id)
+          end
         end
+
+        false
       end
 
       private
 
-      def next_id
-        @next_id_mutex.synchronize do
-          @id += 1
-          @pending_responses[@id] = Thread::Queue.new
-          @id
-        end
-      end
+      def init_queue_for(id) = @pending_responses[id] = Thread::Queue.new
+
+      def next_id = self.class.next_id
 
       def redact_sensitive_fields(obj, sensitive_keys = %w[value token password authorization username])
         case obj
@@ -71,11 +94,10 @@ module Bidi2pdf
         end
       end
 
-      def raise_timeout_error(id, method, params)
-        # rubocop:disable Layout/LineLength
-        @logger.error "Timeout waiting for response to command #{id}, cmd: #{method}, params: #{redact_sensitive_fields(params).inspect}"
-        # rubocop:enable Layout/LineLength
-        raise "Timeout waiting for response to command ID #{id}"
+      def raise_timeout_error(id, cmd)
+        @logger.error "Timeout waiting for response to command #{id}, cmd: #{cmd.inspect}"
+
+        raise CmdTimeoutError, "Timeout waiting for response to command ID #{id}"
       end
     end
   end

data/lib/bidi2pdf/bidi/commands/add_intercept.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class AddIntercept
+        include Base
+
+        BEFORE_REQUEST = "beforeRequestSent"
+        RESPONSE_STARTED = "responseStarted"
+        AUTH_REQUIRED = "authRequired"
+
+        def initialize(context:, phases:, url_patterns:)
+          @context = context
+          @phases = phases
+          @url_patterns = url_patterns
+
+          validate_phases!
+        end
+
+        def method_name
+          "network.addIntercept"
+        end
+
+        def params
+          {
+            context: @context,
+            phases: @phases,
+            urlPatterns: @url_patterns
+          }.compact
+        end
+
+        def validate_phases!
+          valid_phases = [BEFORE_REQUEST, RESPONSE_STARTED, AUTH_REQUIRED]
+
+          raise ArgumentError, "Unsupported phase(s): #{@phases}" unless @phases.all? { |phase| valid_phases.include?(phase) }
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/base.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      module Base
+        def method_name = raise(NotImplementedError, "method_name must be implemented in subclass")
+
+        def params = {}
+
+        def as_payload(id)
+          {
+            id: id,
+            method: method_name,
+            params: params
+          }
+        end
+
+        # rubocop: disable Metrics/AbcSize
+        def ==(other)
+          return false unless other.respond_to?(:method_name) && other.respond_to?(:params)
+
+          return false unless method_name == other.method_name
+
+          return false unless params.keys.sort == other.params.keys.sort
+
+          params.all? { |key, value| other.params.key?(key) && value == other.params[key] }
+        end
+
+        # rubocop: enable Metrics/AbcSize
+
+        # Hash equality comparison
+        def eql?(other)
+          return false unless other.is_a?(Bidi2pdf::Bidi::Commands::Base)
+
+          self == other
+        end
+
+        def hash
+          [method_name, params].hash
+        end
+
+        def inspect
+          attributes = redact_sensitive_fields({ method_name: method_name, params: params })
+
+          "#<#{self.class}:#{object_id} #{attributes}>"
+        end
+
+        private
+
+        def redact_sensitive_fields(obj, sensitive_keys = %w[value token password authorization username])
+          case obj
+          when Hash
+            obj.transform_values.with_index do |v, idx|
+              k = obj.keys[idx]
+              sensitive_keys.include?(k.to_s.downcase) ? "[REDACTED]" : redact_sensitive_fields(v, sensitive_keys)
+            end
+          when Array
+            obj.map { |item| redact_sensitive_fields(item, sensitive_keys) }
+          else
+            obj
+          end
+        end
+
+        def raise_timeout_error(id, method, params)
+          @logger.error "Timeout waiting for response to command #{id}, cmd: #{method}, params: #{redact_sensitive_fields(params).inspect}"
+
+          raise CmdTimeoutError, "Timeout waiting for response to command ID #{id}"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/browser_close.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class BrowserClose
+        include Base
+
+        def method_name
+          "browser.close"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/browser_create_user_context.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class BrowserCreateUserContext
+        include Base
+
+        def method_name
+          "browser.createUserContext"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/browsing_context_close.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class BrowsingContextClose
+        include Base
+
+        def initialize(context:)
+          @context = context
+        end
+
+        def params
+          {
+            context: @context
+          }
+        end
+
+        def method_name
+          "browsingContext.close"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/browsing_context_navigate.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class BrowsingContextNavigate
+        include Base
+
+        def initialize(url:,
+                       context:,
+                       wait: "complete")
+          @url = url
+          @context = context
+          @wait = wait
+        end
+
+        def params
+          {
+            url: @url,
+            context: @context,
+            wait: @wait
+          }
+        end
+
+        def method_name
+          "browsingContext.navigate"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/browsing_context_print.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require_relative "print_parameters_validator"
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class BrowsingContextPrint
+        include Base
+
+        def initialize(context:, print_options:)
+          @context = context
+          @print_options = print_options || { background: true }
+
+          PrintParametersValidator.validate!(@print_options)
+        end
+
+        def params
+          @print_options.merge(context: @context)
+        end
+
+        def method_name
+          "browsingContext.print"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/cancel_auth.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class CancelAuth
+        include Base
+
+        def initialize(request:)
+          @request = request
+        end
+
+        def params
+          {
+            request: @request,
+            action: "cancel"
+          }
+        end
+
+        def method_name
+          "network.continueWithAuth"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/create_tab.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class CreateTab < CreateWindow
+        def type = "tab"
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/create_window.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class CreateWindow
+        include Base
+
+        def initialize(user_context_id: nil, reference_context: nil, background: false)
+          @user_context_id = user_context_id
+          @reference_context = reference_context
+          @background = background
+        end
+
+        def method_name
+          "browsingContext.create"
+        end
+
+        def params
+          {
+            type: type,
+            userContext: @user_context_id,
+            referenceContext: @reference_context,
+            background: @background
+          }.compact
+        end
+
+        def type = "window"
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/get_user_contexts.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class GetUserContexts
+        include Base
+
+        def method_name
+          "browser.getUserContexts"
+        end
+      end
+    end
+  end
+end

data/lib/bidi2pdf/bidi/commands/network_continue.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Bidi2pdf
+  module Bidi
+    module Commands
+      class NetworkContinue
+        include Base
+
+        attr_reader :request, :headers
+
+        def initialize(request:, headers:)
+          @headers = headers
+          @request = request
+        end
+
+        def method_name
+          "network.continueRequest"
+        end
+
+        def params
+          {
+            request: request,
+            headers: headers
+          }.compact
+        end
+      end
+    end
+  end
+end