class RbScreenShot

# File lib/rbscreenshot.rb, line 16
def filename
  "#{Time.now.to_i.to_s}.png"
end

# File lib/rbscreenshot.rb, line 29
def initialize_capybara
  Capybara.javascript_driver = :poltergeist
  Capybara.register_driver :poltergeist do |app|
    Capybara::Poltergeist::Driver.new app, js_errors: false
  end
  Capybara.default_selector = :xpath
end

# File lib/rbscreenshot.rb, line 20
def session(url)
  initialize_capybara

  session = Capybara::Session.new(:poltergeist)
  session.driver.headers = {'User-Agent' => "Mozilla/5.0 (Macintosh; Intel Mac OS X)"}
  session.visit(url)
  session
end

# File lib/rbscreenshot.rb, line 6
def take(url)
  session(url).save_screenshot(filename, full: true)
  filename
end

# File lib/rbscreenshot.rb, line 11
def take_part(url)
  session(url).save_screenshot(filename)
  filename
end