rshot 0.0.1

data/ext/rshot/app_delegate.h

@@ -0,0 +1,11 @@
+#import <Foundation/Foundation.h>
+#import <AppKit/AppKit.h>
+
+@interface AppDelegate : NSObject
+{
+}
+
+- (void)applicationDidFinishLaunching:(NSNotification*)notice;
+- (void)threadStart:(NSApplication*)application;
+
+@end

data/ext/rshot/app_delegate.m

@@ -0,0 +1,15 @@
+#include "app_delegate.h"
+
+@implementation AppDelegate
+
+- (void)applicationDidFinishLaunching:(NSNotification*)notice
+{
+    NSLog(@"applicationDidFinishLaunching");
+}
+
+- (void)threadStart:(NSApplication*)application
+{
+    [application run];
+}
+
+@end

data/ext/rshot/extconf.rb

@@ -0,0 +1,5 @@
+require "mkmf"
+
+$DLDFLAGS << " -framework Foundation -framework AppKit -framework WebKit"
+$CFLAGS << " -std=c99"
+create_makefile "rshot/rshot"

data/ext/rshot/rshot.h

@@ -0,0 +1,26 @@
+#include "setjmp.h"
+#import <Foundation/Foundation.h>
+#import <AppKit/NSBitmapImageRep.h>
+#import <WebKit/WebKit.h>
+#import <WebKit/WebView.h>
+#import <WebKit/WebFrame.h>
+#import <WebKit/WebFrameLoadDelegate.h>
+
+extern jmp_buf RShot_finishedEnv;
+
+@interface RShot : NSObject
+{
+	WebView* webView;
+	NSString* path;
+}
+
+@property (retain) WebView* webView;
+@property (retain) NSString* path;
+
+- (void)renderUrl:(NSString*)url toFile:(NSString*)file;
+- (void)webView:(WebView*)sender didFinishLoadForFrame:(WebFrame*)frame;
+- (void)webView:(WebView*)sender didFailProvisionalLoadWithError:(NSError*)error forFrame:(WebFrame*)frame;
+- (void)webView:(WebView*)sender didFailLoadWithError:(NSError*)error forFrame:(WebFrame*)frame;
+- (void)dealloc;
+
+@end

data/ext/rshot/rshot.m

@@ -0,0 +1,56 @@
+#import "rshot.h"
+
+@implementation RShot
+
+@synthesize webView, path;
+
+- (void)renderUrl:(NSString*)url toFile:(NSString*)file
+{    
+    NSRect rect = NSMakeRect(0, 0, 800, 600);
+    NSWindow* window = [[NSWindow alloc]
+        initWithContentRect: rect
+        styleMask: NSBorderlessWindowMask
+        backing: NSBackingStoreBuffered
+        defer: NO];
+    
+    path = file;	
+    webView = [[WebView alloc] initWithFrame: rect];
+    [window setContentView: webView];
+    [webView setFrameLoadDelegate: self];
+    [[webView mainFrame] loadRequest: [NSURLRequest requestWithURL: [NSURL URLWithString: url]]];
+}
+
+- (void)webView:(WebView*)sender didFinishLoadForFrame:(WebFrame*)frame
+{
+    NSBitmapImageRep* bmp;
+    NSView* view = [[[webView mainFrame] frameView] documentView];
+    [[view window] setContentSize: [view bounds].size];
+    [view lockFocus];
+
+    bmp = [[NSBitmapImageRep alloc] initWithFocusedViewRect: [view bounds]];
+    [view unlockFocus];
+    NSLog(@"%@", path);
+    [[bmp representationUsingType: NSPNGFileType properties: nil] writeToFile: path atomically: YES];
+    
+    [NSApp stop: self];
+    longjmp(RShot_finishedEnv, 0);
+}
+
+- (void)webView:(WebView*)sender didFailProvisionalLoadWithError:(NSError*)error forFrame:(WebFrame*)frame
+{
+    NSLog(@"Error: %@", [error localizedDescription]);
+}
+
+- (void)webView:(WebView*)sender didFailLoadWithError:(NSError*)error forFrame:(WebFrame*)frame
+{
+    NSLog(@"Error: %@", [error localizedDescription]);
+}
+
+- (void)dealloc
+{
+    [webView release];
+    [path release];
+    [super dealloc];
+}
+
+@end

data/ext/rshot/ruby.m

@@ -0,0 +1,52 @@
+#import <AppKit/AppKit.h>
+#include "ruby.h"
+#import "rshot.h"
+#import "app_delegate.h"
+
+jmp_buf RShot_finishedEnv;
+
+static VALUE rb_cRShot = Qundef;
+static VALUE RShot_alloc(VALUE class);
+static void RShot_free(void* rshot);
+static VALUE RShot_render_url(VALUE self, VALUE url, VALUE png);
+static VALUE RShot_gogogo(VALUE self);
+static NSAutoreleasePool* pool;
+static NSThread* app_thread;
+static AppDelegate* app_delegate;
+
+void Init_rshot()
+{
+	pool = [[NSAutoreleasePool alloc] init];
+	[NSApplication sharedApplication];
+    app_delegate = [[AppDelegate alloc] init];
+    [NSApp setDelegate: app_delegate];
+	
+	rb_cRShot = rb_define_class("RShot", rb_cObject);
+	rb_define_alloc_func(rb_cRShot, RShot_alloc);
+	rb_define_method(rb_cRShot, "render_url", RShot_render_url, 2);
+}
+
+static void RShot_free(void* rshot)
+{
+	[(RShot*)rshot release];
+	[pool drain];
+}
+
+static VALUE RShot_alloc(VALUE class)
+{
+	RShot* rshot = [[RShot alloc] init];
+	return Data_Wrap_Struct(class, 0, RShot_free, rshot);
+}
+
+static VALUE RShot_render_url(VALUE self, VALUE url, VALUE png)
+{
+	RShot* rshot;
+	NSString* nsurl = [NSString stringWithUTF8String: rb_string_value_cstr(&url)];
+	NSString* nspng = [NSString stringWithUTF8String: rb_string_value_cstr(&png)];
+	Data_Get_Struct(self, RShot, rshot);
+	[rshot renderUrl: nsurl toFile: nspng];
+	if(!setjmp(RShot_finishedEnv)) {
+        [NSApp run];
+    }
+    return Qtrue;
+}

metadata

@@ -0,0 +1,51 @@
+--- !ruby/object:Gem::Specification
+name: rshot
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Charlie Somerville
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-11-26 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: charlie@charliesomerville.com
+executables: []
+extensions:
+- ext/rshot/extconf.rb
+extra_rdoc_files: []
+files:
+- ext/rshot/app_delegate.m
+- ext/rshot/rshot.m
+- ext/rshot/ruby.m
+- ext/rshot/app_delegate.h
+- ext/rshot/rshot.h
+- ext/rshot/extconf.rb
+homepage: https://github.com/charliesome/RShot
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Takes screenshots of web pages
+test_files: []