class Selenium::WebDriver::Chromium::Profile

@private

Public Class Methods

new(model = nil) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 30
def initialize(model = nil)
  @model = verify_model(model)
  @extensions = []
  @encoded_extensions = []
  @directory = nil
end

Public Instance Methods

[](key) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 62
def [](key)
  parts = key.split('.')
  parts.inject(prefs) { |a, e| a.fetch(e) }
end
[]=(key, value) click to toggle source

Set a preference in the profile.

See src.chromium.org/viewvc/chrome/trunk/src/chrome/common/pref_names.cc

# File lib/selenium/webdriver/chromium/profile.rb, line 57
def []=(key, value)
  parts = key.split('.')
  parts[0..-2].inject(prefs) { |a, e| a[e] ||= {} }[parts.last] = value
end
add_encoded_extension(encoded) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 43
def add_encoded_extension(encoded)
  @encoded_extensions << encoded
end
add_extension(path) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 37
def add_extension(path)
  raise Error::WebDriverError, "could not find extension at #{path.inspect}" unless File.file?(path)

  @extensions << path
end
as_json(*) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 76
def as_json(*)
  extensions = @extensions.map do |crx_path|
    File.open(crx_path, 'rb') { |crx_file| Base64.strict_encode64 crx_file.read }
  end

  extensions.concat(@encoded_extensions)

  opts = {'directory' => directory || layout_on_disk}
  opts['extensions'] = extensions if extensions.any?
  opts
end
directory() click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 47
def directory
  @directory || layout_on_disk
end
layout_on_disk() click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 67
def layout_on_disk
  @directory = @model ? create_tmp_copy(@model) : Dir.mktmpdir('webdriver-chrome-profile')
  FileReaper << @directory

  write_prefs_to @directory

  @directory
end

Private Instance Methods

prefs() click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 97
def prefs
  @prefs ||= read_model_prefs
end
prefs_file_for(dir) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 107
def prefs_file_for(dir)
  File.join dir, 'Default', 'Preferences'
end
read_model_prefs() click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 101
def read_model_prefs
  return {} unless @model

  JSON.parse File.read(prefs_file_for(@model))
end
write_prefs_to(dir) click to toggle source
# File lib/selenium/webdriver/chromium/profile.rb, line 90
def write_prefs_to(dir)
  prefs_file = prefs_file_for(dir)

  FileUtils.mkdir_p File.dirname(prefs_file)
  File.open(prefs_file, 'w') { |file| file << JSON.generate(prefs) }
end