Class: Ferrum::Network

Inherits:
Object
  • Object
show all
Defined in:
lib/ferrum/network.rb,
lib/ferrum/network/error.rb,
lib/ferrum/network/request.rb,
lib/ferrum/network/exchange.rb,
lib/ferrum/network/response.rb,
lib/ferrum/network/auth_request.rb,
lib/ferrum/network/intercepted_request.rb

Defined Under Namespace

Classes: AuthRequest, Error, Exchange, InterceptedRequest, Request, Response

Constant Summary collapse

CLEAR_TYPE = 
%i[traffic cache].freeze
AUTHORIZE_TYPE = 
%i[server proxy].freeze
RESOURCE_TYPES = 
%w[Document Stylesheet Image Media Font Script TextTrack
XHR Fetch EventSource WebSocket Manifest
SignedExchange Ping CSPViolationReport Other].freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(page) ⇒ Network

Returns a new instance of Network.



20
21
22
23
24
 
# File 'lib/ferrum/network.rb', line 20

def initialize(page)
  @page = page
  @traffic = []
  @exchange = nil
end

Instance Attribute Details

#traffic ⇒ Object (readonly)

Returns the value of attribute traffic.



18
19
20
 
# File 'lib/ferrum/network.rb', line 18

def traffic
  @traffic
end

Instance Method Details

#authorize(user:, password:, type: :server) ⇒ Object 



86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/ferrum/network.rb', line 86

def authorize(user:, password:, type: :server)
  unless AUTHORIZE_TYPE.include?(type)
    raise ArgumentError, ":type should be in #{AUTHORIZE_TYPE}"
  end

  @authorized_ids ||= {}
  @authorized_ids[type] ||= []

  intercept

  @page.on(:request) do |request|
    request.continue
  end

  @page.on(:auth) do |request, index, total|
    if request.auth_challenge?(type)
      response = authorized_response(@authorized_ids[type],
                                     request.request_id,
                                     user, password)

      @authorized_ids[type] << request.request_id
      request.continue(authChallengeResponse: response)
    elsif index + 1 < total
      next # There are other callbacks that can handle this
    else
      request.abort
    end
  end
end

#authorized_response(ids, request_id, username, password) ⇒ Object 



170
171
172
173
174
175
176
177
178
179
180
 
# File 'lib/ferrum/network.rb', line 170

def authorized_response(ids, request_id, username, password)
  if ids.include?(request_id)
    { response: "CancelAuth" }
  elsif username && password
    { response: "ProvideCredentials",
      username: username,
      password: password }
  else
    { response: "CancelAuth" }
  end
end

#build_exchange(id) ⇒ Object 



186
187
188
 
# File 'lib/ferrum/network.rb', line 186

def build_exchange(id)
  Network::Exchange.new(@page, id).tap { |e| @traffic << e }
end

#clear(type) ⇒ Object 



63
64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/ferrum/network.rb', line 63

def clear(type)
  unless CLEAR_TYPE.include?(type)
    raise ArgumentError, ":type should be in #{CLEAR_TYPE}"
  end

  if type == :traffic
    @traffic.clear
  else
    @page.command("Network.clearBrowserCache")
  end

  true
end

#finished_connections ⇒ Object 



43
44
45
 
# File 'lib/ferrum/network.rb', line 43

def finished_connections
  @traffic.count(&:finished?)
end

#idle?(connections = 0) ⇒ Boolean

Returns:

  • (Boolean) 


35
36
37
 
# File 'lib/ferrum/network.rb', line 35

def idle?(connections = 0)
  pending_connections <= connections
end

#intercept(pattern: "*", resource_type: nil) ⇒ Object 



77
78
79
80
81
82
83
84
 
# File 'lib/ferrum/network.rb', line 77

def intercept(pattern: "*", resource_type: nil)
  pattern = { urlPattern: pattern }
  if resource_type && RESOURCE_TYPES.include?(resource_type.to_s)
    pattern[:resourceType] = resource_type
  end

  @page.command("Fetch.enable", handleAuthRequests: true, patterns: [pattern])
end

#pending_connections ⇒ Object 



47
48
49
 
# File 'lib/ferrum/network.rb', line 47

def pending_connections
  total_connections - finished_connections
end

#request ⇒ Object 



51
52
53
 
# File 'lib/ferrum/network.rb', line 51

def request
  @exchange&.request
end

#response ⇒ Object 



55
56
57
 
# File 'lib/ferrum/network.rb', line 55

def response
  @exchange&.response
end

#select(request_id) ⇒ Object 



182
183
184
 
# File 'lib/ferrum/network.rb', line 182

def select(request_id)
  @traffic.select { |e| e.id == request_id }
end

#status ⇒ Object 



59
60
61
 
# File 'lib/ferrum/network.rb', line 59

def status
  response&.status
end

#subscribe ⇒ Object 



116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
 
# File 'lib/ferrum/network.rb', line 116

def subscribe
  @page.on("Network.requestWillBeSent") do |params|
    request = Network::Request.new(params)

    # We can build exchange in two places, here on the event or when request
    # is interrupted. So we have to be careful when to create new one. We
    # create new exchange only if there's no with such id or there's but
    # it's filled with request which means this one is new but has response
    # for a redirect. So we assign response from the params to previous
    # exchange and build new exchange to assign this request to it.
    exchange = select(request.id).last
    exchange = build_exchange(request.id) unless exchange&.blank?

    # On redirects Chrome doesn't change `requestId` and there's no
    # `Network.responseReceived` event for such request. If there's already
    # exchange object with this id then we got redirected and params has
    # `redirectResponse` key which contains the response.
    if params["redirectResponse"]
      previous_exchange = select(request.id)[-2]
      response = Network::Response.new(@page, params)
      previous_exchange.response = response
    end

    exchange.request = request

    if exchange.navigation_request?(@page.main_frame.id)
      @exchange = exchange
    end
  end

  @page.on("Network.responseReceived") do |params|
    if exchange = select(params["requestId"]).last
      response = Network::Response.new(@page, params)
      exchange.response = response
    end
  end

  @page.on("Network.loadingFinished") do |params|
    exchange = select(params["requestId"]).last
    if exchange && exchange.response
      exchange.response.body_size = params["encodedDataLength"]
    end
  end

  @page.on("Log.entryAdded") do |params|
    entry = params["entry"] || {}
    if entry["source"] == "network" && entry["level"] == "error"
      exchange = select(entry["networkRequestId"]).last
      error = Network::Error.new(entry)
      exchange.error = error
    end
  end
end

#total_connections ⇒ Object 



39
40
41
 
# File 'lib/ferrum/network.rb', line 39

def total_connections
  @traffic.size
end

#wait_for_idle(connections: 0, duration: 0.05, timeout: @page.browser.timeout) ⇒ Object 



26
27
28
29
30
31
32
33
 
# File 'lib/ferrum/network.rb', line 26

def wait_for_idle(connections: 0, duration: 0.05, timeout: @page.browser.timeout)
  start = Ferrum.monotonic_time

  until idle?(connections)
    raise TimeoutError if Ferrum.timeout?(start, timeout)
    sleep(duration)
  end
end