module Selenium::WebDriver::TakesScreenshot

def screenshot_as(format, full_page: false)
  case format
  when :base64
    full_page ? full_screenshot : screenshot
  when :png
    screenshot_as(:base64, full_page: full_page).unpack1('m')
  else
    raise Error::UnsupportedOperationError, "unsupported format: #{format.inspect}"
  end
rescue NameError
  raise Error::UnsupportedOperationError, "Full Page Screenshots are not supported for #{inspect}"
end