Screenshotting w/ Selenium and Burp

watir-screen.rb

#!/usr/bin/env ruby

require 'watir-webdriver'

def screenshot(url)
	b = Watir::Browser.new
	b.goto url
	filename = url
	if filename =~ /:\/\//
		filename = filename.split('://')[1]
	end

	if filename =~ /\//
		filename = filename.split('/')[0]
	end

	filename = filename + ".png"
	b.screenshot.save filename
	b.close

end

a = ['http://www.google.com/', 'letmeoutofyour.net', '178.79.164.226', 'https://www.gmail.com']

a.each do |x| 
	screenshot(x)
end