Contents

# encoding: UTF-8
require 'capybara'
require 'tmpdir'

module Simulacrum
  # The Renderer Class is responsible for driving Capybara and setting up
  # the desired page, screenshotting, croping, etc.
  class Renderer
    include Capybara::DSL

    attr_reader :url, :tmp_path

    def initialize(url)
      @url = url
      @tmp_dir = Dir.mktmpdir
    end

    def render
      page.visit(url)
      resize_window!
      save_screenshot!
    end

    def cleanup
      FileUtils.remove_entry(@tmp_dir)
    end

    def get_bounds_for_selector(selector)
      element = page.find(selector.to_s)
      location = element_location(element)
      size = element_size(element)
      [location.x, location.y, size.width, size.height]
    end

    private

    def element_location(element)
      element.native.location
    end

    def element_size(element)
      element.native.size
    end

    def resize_window!
      page.driver.browser.manage.window.resize_to(*window_size)
    rescue Selenium::WebDriver::Error::UnknownError
      return
    end

    def save_screenshot!
      page.driver.save_screenshot(tmp_path)
    end

    def window_size
      [1024, 768]
    end

    def tmp_path
      @tmp_path ||= File.join(@tmp_dir,
                              Simulacrum.configuration.candidate_filename)
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
simulacrum-0.3.2	lib/simulacrum/renderer.rb
simulacrum-0.3.1	lib/simulacrum/renderer.rb
simulacrum-0.3.0	lib/simulacrum/renderer.rb