Created
December 11, 2008 19:04
-
-
Save ehrenmurdick/34824 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env ruby | |
# | |
# DarkRoom | |
# Takes fullsize screenshots of a web page. | |
# Copyright (c) 2007 Justin Palmer. | |
# | |
# Released under an MIT LICENSE | |
# | |
# Usage | |
# ==== | |
# ruby ./darkroom.rb http://activereload.net | |
# ruby ./darkroom.rb --output=google.png http://google.com | |
# | |
require 'optparse' | |
require 'osx/cocoa' | |
OSX.require_framework 'Webkit' | |
module ActiveReload | |
module DarkRoom | |
USER_AGENT = "DarkRoom/0.1" | |
class Photographer | |
def initialize | |
options = {} | |
opts = OptionParser.new do |opts| | |
opts.banner = "Usage: #$0 [options] URL" | |
opts.on('-w', '--width=[WIDTH]', Integer, 'Force width of the screenshot') do |v| | |
options[:width] = v | |
end | |
opts.on("-d", '--delay=[DELAY]', Integer, "Delay") do |v| | |
options[:delay] = v | |
end | |
opts.on('-h', '--height=[HEIGHT]', Integer, 'Force height of screenshot') do |v| | |
options[:height] = v | |
end | |
opts.on('-o', '--output=[FILENAME]', String, 'Specify filename for saving') do |v| | |
options[:output] = v | |
end | |
opts.on_tail('-h', '--help', 'Display this message and exit') do | |
puts opts | |
exit | |
end | |
end.parse! | |
options[:width] ||= 1024 | |
options[:height] ||= 0 | |
options[:website] = ARGV.first || 'http://google.com' | |
Camera.shoot(options) | |
end | |
end | |
class Camera | |
def self.shoot(options) | |
app = OSX::NSApplication.sharedApplication | |
delegate = Processor.alloc.init | |
delegate.options = options | |
app.setDelegate(delegate) | |
app.run | |
end | |
end | |
class Processor < OSX::NSObject | |
include OSX | |
attr_accessor :options, :web_view | |
def initialize | |
rect = [-16000.0, -16000.0, 100, 100] | |
win = NSWindow.alloc.initWithContentRect_styleMask_backing_defer(rect, NSBorderlessWindowMask, 2, 0) | |
@web_view = WebView.alloc.initWithFrame(rect) | |
@web_view.mainFrame.frameView.setAllowsScrolling(false) | |
@web_view.setApplicationNameForUserAgent(USER_AGENT) | |
NSNotificationCenter.defaultCenter.objc_send(:addObserver, self, | |
:selector, :webview_progress_finished, | |
:name, WebViewProgressFinishedNotification, | |
:object, @web_view) | |
win.setContentView(@web_view) | |
end | |
def webview_progress_finished(sender) | |
viewport = web_view.mainFrame.frameView.documentView | |
puts viewport.class | |
viewport.window.orderFront(nil) | |
viewport.window.display | |
viewport.window.setContentSize([@options[:width], (@options[:height] > 0 ? @options[:height] : viewport.bounds.height)]) | |
viewport.setFrame(viewport.bounds) | |
if options[:delay] | |
capture_and_save_with_delay(viewport, options[:delay]) | |
else | |
capture_and_save(viewport) | |
end | |
end | |
def applicationDidFinishLaunching(notification) | |
@options[:output] ||= "page.png" | |
@web_view.window.setContentSize([@options[:width], @options[:height]]) | |
@web_view.setFrameSize([@options[:width], @options[:height]]) | |
@web_view.mainFrame.loadRequest(NSURLRequest.requestWithURL(NSURL.URLWithString(@options[:website]))) | |
end | |
def capture_and_save(view) | |
view.lockFocus | |
bitmap = NSBitmapImageRep.alloc.initWithFocusedViewRect(view.bounds) | |
view.unlockFocus | |
bitmap.representationUsingType_properties(NSPNGFileType, nil).writeToFile_atomically(@options[:output], true) | |
NSApplication.sharedApplication.terminate(nil) | |
end | |
def capture_and_save_with_delay(view, delay) | |
Thread.new do | |
sleep delay | |
view.lockFocus | |
bitmap = NSBitmapImageRep.alloc.initWithFocusedViewRect(view.bounds) | |
view.unlockFocus | |
bitmap.representationUsingType_properties(NSPNGFileType, nil).writeToFile_atomically(@options[:output], true) | |
NSApplication.sharedApplication.terminate(nil) | |
end | |
end | |
end | |
end | |
end | |
ActiveReload::DarkRoom::Photographer.new | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment