Class: Puppeteer::Browser

Inherits:
Object
  • Object
show all
Includes:
DebugPrint, EventCallbackable
Defined in:
lib/puppeteer/browser.rb

Class Method Summary collapse

Instance Method Summary collapse

Methods included from EventCallbackable

#add_event_listener, #emit_event, #on_event, #remove_event_listener

Methods included from DebugPrint

#debug_print, #debug_puts

Constructor Details

#initialize(connection:, context_ids:, ignore_https_errors:, default_viewport:, process:, close_callback:) ⇒ Browser 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/puppeteer/browser.rb', line 34

def initialize(connection:, context_ids:, ignore_https_errors:, default_viewport:, process:, close_callback:)
  @ignore_https_errors = ignore_https_errors
  @default_viewport = default_viewport
  @process = process
  # @screenshot_task_queue = TaskQueue.new
  @connection = connection
  @close_callback = close_callback

  @default_context = Puppeteer::BrowserContext.new(@connection, self, nil)
  @contexts = {}
  context_ids.each do |context_id|
    @contexts[context_id] = Puppeteer::BrowserContext.new(@connection, self. context_id)
  end
  @targets = {}
  @connection.on_event 'Events.CDPSession.Disconnected' do
    emit_event 'Events.Browser.Disconnected'
  end
  @connection.on_event 'Target.targetCreated', &method(:handle_target_created)
  @connection.on_event 'Target.targetDestroyed', &method(:handle_target_destroyed)
  @connection.on_event 'Target.targetInfoChanged', &method(:handle_target_info_changed)
end

Class Method Details

.create(connection:, context_ids:, ignore_https_errors:, default_viewport:, process:, close_callback:) ⇒ Object 



15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/puppeteer/browser.rb', line 15

def self.create(connection:, context_ids:, ignore_https_errors:, default_viewport:, process:, close_callback:)
  browser = Puppeteer::Browser.new(
    connection: connection,
    context_ids: context_ids,
    ignore_https_errors: ignore_https_errors,
    default_viewport: default_viewport,
    process: process,
    close_callback: close_callback,
  )
  connection.send_message('Target.setDiscoverTargets', discover: true)
  browser
end

Instance Method Details

#browser_contexts ⇒ Object 



68
69
70
 
# File 'lib/puppeteer/browser.rb', line 68

def browser_contexts
  [@default_context].concat(@contexts.values)
end

#close ⇒ Object 



224
225
226
227
 
# File 'lib/puppeteer/browser.rb', line 224

def close
  @close_callback.call
  disconnect
end

#connected? ⇒ Boolean 



233
234
235
 
# File 'lib/puppeteer/browser.rb', line 233

def connected?
  !@connection.closed?
end

#create_incognito_browser_context ⇒ Puppeteer::BrowserContext 



62
63
64
65
66
 
# File 'lib/puppeteer/browser.rb', line 62

def create_incognito_browser_context
  result = @connection.send_message('Target.createBrowserContext')
  browser_context_id = result['browserContextId']
  @contexts[browser_context_id] = Puppeteer::BrowserContext.new(@connection, self, browser_context_id)
end

#create_page_in_context(context_id) ⇒ !Promise<!Puppeteer.Page> 



152
153
154
155
156
157
158
159
160
161
162
 
# File 'lib/puppeteer/browser.rb', line 152

def create_page_in_context(context_id)
  create_target_params = { url: 'about:blank' }
  if context_id
    create_target_params[:browserContextId] = context_id
  end
  result = @connection.send_message('Target.createTarget', **create_target_params)
  target_id = result['targetId']
  target = @targets[target_id]
  await target.initialized_promise
  await target.page
end

#default_browser_context ⇒ Puppeteer::BrowserContext 



73
74
75
 
# File 'lib/puppeteer/browser.rb', line 73

def default_browser_context
  @default_context
end

#disconnect ⇒ Object 



229
230
231
 
# File 'lib/puppeteer/browser.rb', line 229

def disconnect
  @connection.dispose
end

#dispose_context(context_id) ⇒ Object 



78
79
80
81
 
# File 'lib/puppeteer/browser.rb', line 78

def dispose_context(context_id)
  @connection.send_message('Target.disposeBrowserContext', browserContextId: context_id)
  @contexts.remove(context_id)
end

#handle_target_created(event) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/puppeteer/browser.rb', line 84

def handle_target_created(event)
  target_info = Puppeteer::Target::TargetInfo.new(event['targetInfo'])
  browser_context_id = target_info.browser_context_id
  context =
    if browser_context_id && @contexts.has_key?(browser_context_id)
      @contexts[browser_context_id]
    else
      @default_context
    end

  target = Puppeteer::Target.new(
    target_info: target_info,
    browser_context: context,
    session_factory: -> { @connection.create_session(target_info) },
    ignore_https_errors: @ignore_https_errors,
    default_viewport: @default_viewport,
    screenshot_task_queue: @screenshot_task_queue,
  )
  #   assert(!this._targets.has(event.targetInfo.targetId), 'Target should not exist before targetCreated');
  @targets[target_info.target_id] = target

  target.on_initialize_succeeded do
    emit_event 'Events.Browser.TargetCreated', target
    context.emit_event 'Events.BrowserContext.TargetCreated', target
  end
end

#handle_target_destroyed(event) ⇒ Object 



113
114
115
116
117
118
119
120
121
122
123
 
# File 'lib/puppeteer/browser.rb', line 113

def handle_target_destroyed(event)
  target_id = event['targetId']
  target = @targets[target_id]
  target.handle_initialized(false)
  @targets.delete(target_id)
  target.handle_closed
  target.on_initialize_succeeded do
    emit_event 'Events.Browser.TargetDestroyed', target
    target.browser_context.emit_event 'Events.BrowserContext.TargetDestroyed', target
  end
end

#handle_target_info_changed(event) ⇒ Object 



126
127
128
129
130
131
132
133
134
135
136
137
138
139
 
# File 'lib/puppeteer/browser.rb', line 126

def handle_target_info_changed(event)
  target_info = Puppeteer::Target::TargetInfo.new(event['targetInfo'])
  target = @targets[target_info.target_id]
  if !target
    throw StandardError.new('target should exist before targetInfoChanged')
  end
  previous_url = target.url
  was_initialized = target.initialized?
  target.handle_target_info_changed(target_info)
  if was_initialized && previous_url != target.url
    emit_event 'Events.Browser.TargetChanged', target
    target.browser_context.emit_event 'Events.BrowserContext.TargetChanged', target
  end
end

#new_page ⇒ Object 



146
147
148
 
# File 'lib/puppeteer/browser.rb', line 146

def new_page
  @default_context.new_page
end

#pages ⇒ !Promise<!Array<!Puppeteer.Page>> 



210
211
212
 
# File 'lib/puppeteer/browser.rb', line 210

def pages
  browser_contexts.flat_map(&:pages)
end

#process ⇒ Puppeteer::BrowserRunner::BrowserProcess 



57
58
59
 
# File 'lib/puppeteer/browser.rb', line 57

def process
  @process
end

#target ⇒ !Target 



171
172
173
 
# File 'lib/puppeteer/browser.rb', line 171

def target
  targets.first { |target| target.type == 'browser' }
end

#targets ⇒ !Array<!Target> 



165
166
167
 
# File 'lib/puppeteer/browser.rb', line 165

def targets
  @targets.values.select { |target| target.initialized? }
end

#user_agent ⇒ String 



220
221
222
 
# File 'lib/puppeteer/browser.rb', line 220

def user_agent
  get_version.user_agent
end

#version ⇒ String 



215
216
217
 
# File 'lib/puppeteer/browser.rb', line 215

def version
  get_version.product
end

#wait_for_target(predicate:, timeout: nil) ⇒ !Promise<!Target> 



178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
 
# File 'lib/puppeteer/browser.rb', line 178

def wait_for_target(predicate:, timeout: nil)
  timeout_in_sec = (timeout || 30000).to_i / 1000.0
  existing_target = targets.first { |target| predicate.call(target) }
  return existing_target if existing_target

  event_listening_ids = []
  target_promise = resolvable_future
  event_listening_ids << add_event_listener('Events.Browser.TargetCreated') do |target|
    if predicate.call(target)
      target_promise.fulfill(target)
    end
  end
  event_listening_ids << add_event_listener('Events.Browser.TargetChanged') do |target|
    if predicate.call(target)
      target_promise.fulfill(target)
    end
  end

  begin
    if timeout_in_sec > 0
      Timeout.timeout(timeout_in_sec) do
        target_promise.value!
      end
    else
      target_promise.value!
    end
  ensure
    remove_event_listener(*event_listening_ids)
  end
end

#websocket_endpoint ⇒ String 



142
143
144
 
# File 'lib/puppeteer/browser.rb', line 142

def websocket_endpoint
  @connection.url
end