imgkit 0.9.2

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,22 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+.bundle

data/.rspec

@@ -0,0 +1 @@
+--color

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.2@imgkit_devel

data/Gemfile

@@ -0,0 +1,9 @@
+source :rubygems
+
+group :development do
+  gem "rspec", "~> 2.0.0.beta.8"
+  gem "rspec-core", "~> 2.0.0.beta.8"
+  gem "mocha"
+  gem "jeweler"
+  gem "rack"
+end

data/Gemfile.lock

@@ -0,0 +1,35 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    diff-lcs (1.1.2)
+    gemcutter (0.6.1)
+    git (1.2.5)
+    jeweler (1.4.0)
+      gemcutter (>= 0.1.0)
+      git (>= 1.2.5)
+      rubyforge (>= 2.0.0)
+    json_pure (1.4.3)
+    mocha (0.9.8)
+      rake
+    rack (1.2.1)
+    rake (0.8.7)
+    rspec (2.0.0.beta.19)
+      rspec-core (= 2.0.0.beta.19)
+      rspec-expectations (= 2.0.0.beta.19)
+      rspec-mocks (= 2.0.0.beta.19)
+    rspec-core (2.0.0.beta.19)
+    rspec-expectations (2.0.0.beta.19)
+      diff-lcs (>= 1.1.2)
+    rspec-mocks (2.0.0.beta.19)
+    rubyforge (2.0.4)
+      json_pure (>= 1.1.7)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  jeweler
+  mocha
+  rack
+  rspec (~> 2.0.0.beta.8)
+  rspec-core (~> 2.0.0.beta.8)

data/LICENSE

@@ -0,0 +1,22 @@
+This is a derivative work.
+
+Original Copyright (c) 2009 jdpace
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/POST_INSTALL

@@ -0,0 +1,8 @@
+******************************************************************
+
+Now install wkhtmltoimage binaries:
+Global: sudo `which imgkit` --install-wkhtmltoimage
+or inside RVM folder: export TO=`which imgkit | sed 's:/imgkit:/wkhtmltoimage:'` && imgkit --install-wkhtmltoimage
+(run imgkit --help to see more options)
+
+******************************************************************

data/README.md

@@ -0,0 +1,82 @@
+# IMGKit
+
+Create JPGs using plain old HTML+CSS. Uses [wkhtmltoimage](http://github.com/antialize/wkhtmltopdf) on the backend which renders HTML using Webkit.
+
+## Install
+
+### IMGKit
+
+    gem install imgkit
+
+### wkhtmltoimage
+ * **Automatic**: `sudo imgkit --install-wkhtmltoimage`  
+ install latest version into /usr/local/bin  
+ (overwrite defaults with e.g. ARCHITECTURE=amd64 TO=/home/foo/bin)
+ * By hand: http://code.google.com/p/wkhtmltopdf/downloads/list
+
+## Usage
+    
+    # IMGKit.new takes the HTML and any options for wkhtmltoimage
+    # run `wkhtmltoimage --extended-help` for a full list of options
+    kit = IMGKit.new(html, :quality => 50)
+    kit.stylesheets << '/path/to/css/file'
+    
+    # Get an inline image
+    img = kit.to_img
+    
+    # Save the PDF to a file
+    file = kit.to_file('/path/to/save/file.jpg')
+    
+    # IMGKit.new can optionally accept a URL or a File.
+    # Stylesheets can not be added when source is provided as a URL of File.
+    kit = IMGKit.new('http://google.com')
+    kit = IMGKit.new(File.new('/path/to/html'))
+
+    # Add any kind of option through meta tags
+    IMGKit.new('<html><head><meta name="imgkit-quality" content="75")
+    
+## Configuration
+
+If you're on Windows or you installed wkhtmltopdf by hand to a location other than /usr/local/bin you will need to tell PDFKit where the binary is. You can configure PDFKit like so:
+
+    # config/initializers/imgkit.rb
+    IMGKit.configure do |config|
+      config.wkhtmltoimage = '/path/to/wkhtmltoimage'
+      config.default_options = {
+        :quality => 60
+      }
+    end
+
+
+## Mime Types
+register a .jpg mime type in: 
+
+    #config/initializers/mime_type.rb
+    Mime::Type.register       "image/jpeg", :jpg
+
+## Controllers
+You can then send JPGs with
+
+    format.jpg do
+      send_data(@kit.to_img, :type => "image/jpeg", :disposition => 'inline')
+    end
+
+This allows you to take advantage of rails page caching so you only generate the
+image when you need to.
+
+## Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Setup your development environment with: gem install bundler; bundle install
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+## Copyright
+
+Copyright (c) 2010 Chris Continanza 
+Based on work by Jared Pace  
+See LICENSE for details.

data/Rakefile

@@ -0,0 +1,45 @@
+require 'rubygems'
+require 'rake'
+require 'bundler'
+Bundler.require(:development)
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "imgkit"
+    gem.summary = %Q{HTML+CSS -> JPG}
+    gem.description = %Q{Uses wkhtmltoimage to create Images using HTML}
+    gem.email = "christopher.continanza@gmail.com"
+    gem.homepage = "http://github.com/csquared/IMGKit"
+    gem.authors = ["csquared"]
+    gem.add_development_dependency "rspec", "~> 2.0.0.beta.8"
+    gem.add_development_dependency "rspec-core", "~> 2.0.0.beta.8"
+    gem.add_development_dependency 'mocha'
+    gem.post_install_message = File.read('POST_INSTALL')
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+end
+
+RSpec::Core::RakeTask.new(:rcov) do |spec|
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "IMGKit #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.9.2

data/bin/imgkit

@@ -0,0 +1,72 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'optparse'
+require 'rbconfig'
+require 'open-uri'
+require 'imgkit/configuration'
+
+def detect_architecture
+  case Config::CONFIG['host_os']
+  when /x86_64-linux/i
+    'amd64'
+  when /linux/i
+    'i386'
+  when /darwin/i
+    'OS-X.i386'
+  else
+    raise "No binaries found for your system. Please install wkhtmltoimage by hand."
+  end
+end
+
+def cleanup(install_to)
+  `rm -rf wkhtmltoimage*`
+  `rm #{install_to}`
+end
+
+def download_wkhtmltoimage(arch)
+  page = open("http://code.google.com/p/wkhtmltopdf/downloads/list").read
+  download = page.match(/href=".*name=(.*wkhtmltoimage-.*#{arch}.*?)&/) || raise("File not found..")
+  download = download[1]
+  url = "http://wkhtmltopdf.googlecode.com/files/#{download}"
+  puts "Downloading #{download} from #{url}"
+  
+  `curl #{url} > #{download}`
+  download
+end
+
+def install(download, arch, install_to)
+  puts "Installing #{download} to #{install_to}"
+  if download =~ /.tar.bz2$/
+    `tar xjvf #{download}`
+    `mv wkhtmltoimage-#{arch} #{install_to}`
+  elsif download =~ /.tar.lzma$/
+    raise "couldn't extract archive: lzcat not found" unless system("which lzcat > /dev/null 2>/dev/null")
+    `lzcat #{download} | tar x`
+    `mv wkhtmltoimage-#{arch} #{install_to}`
+  else
+    `mv #{download} #{install_to}`
+  end
+  `sudo chmod +x #{install_to}`
+end
+
+OptionParser.new do |parser|
+  parser.banner = "IMGKit\n\nOptions are:"
+
+  parser.on("--install-wkhtmltoimage", "Install wkhtmltoimage binaries (TO=/usr/local/bin ARCHITECTURE=i386)") do
+    architecture = ENV['ARCHITECTURE'] || detect_architecture
+    install_to = ENV['TO'] || IMGKit.configuration.wkhtmltoimage
+    
+    Dir.chdir '/tmp'
+    
+    cleanup(install_to)
+    download = download_wkhtmltoimage(architecture)
+    install(download, architecture, install_to)
+  end
+
+  parser.on("--version", "Show Version.") do
+    root = File.dirname(File.dirname(__FILE__))
+    puts File.read(File.join(root, 'VERSION'))
+  end
+
+  parser.on("-h", "--help", "Show this.") { puts parser; exit }
+end.parse!

data/imgkit.gemspec

@@ -0,0 +1,84 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{imgkit}
+  s.version = "0.9.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["csquared"]
+  s.date = %q{2010-12-12}
+  s.default_executable = %q{imgkit}
+  s.description = %q{Uses wkhtmltoimage to create Images using HTML}
+  s.email = %q{christopher.continanza@gmail.com}
+  s.executables = ["imgkit"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.md"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     ".rspec",
+     ".rvmrc",
+     "Gemfile",
+     "Gemfile.lock",
+     "LICENSE",
+     "POST_INSTALL",
+     "README.md",
+     "Rakefile",
+     "VERSION",
+     "bin/imgkit",
+     "imgkit.gemspec",
+     "lib/imgkit.rb",
+     "lib/imgkit/configuration.rb",
+     "lib/imgkit/imgkit.rb",
+     "lib/imgkit/source.rb",
+     "spec/fixtures/example.css",
+     "spec/fixtures/example.html",
+     "spec/imgkit_spec.rb",
+     "spec/source_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/csquared/IMGKit}
+  s.post_install_message = %q{******************************************************************
+
+Now install wkhtmltoimage binaries:
+Global: sudo `which imgkit` --install-wkhtmltoimage
+or inside RVM folder: export TO=`which imgkit | sed 's:/imgkit:/wkhtmltoimage:'` && imgkit --install-wkhtmltoimage
+(run imgkit --help to see more options)
+
+******************************************************************
+}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{HTML+CSS -> JPG}
+  s.test_files = [
+    "spec/imgkit_spec.rb",
+     "spec/source_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, ["~> 2.0.0.beta.8"])
+      s.add_development_dependency(%q<rspec-core>, ["~> 2.0.0.beta.8"])
+      s.add_development_dependency(%q<mocha>, [">= 0"])
+    else
+      s.add_dependency(%q<rspec>, ["~> 2.0.0.beta.8"])
+      s.add_dependency(%q<rspec-core>, ["~> 2.0.0.beta.8"])
+      s.add_dependency(%q<mocha>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<rspec>, ["~> 2.0.0.beta.8"])
+    s.add_dependency(%q<rspec-core>, ["~> 2.0.0.beta.8"])
+    s.add_dependency(%q<mocha>, [">= 0"])
+  end
+end
+

data/lib/imgkit.rb

@@ -0,0 +1,5 @@
+require 'rubygems'
+require 'open3'
+require 'imgkit/source'
+require 'imgkit/imgkit'
+require 'imgkit/configuration'

data/lib/imgkit/configuration.rb

@@ -0,0 +1,33 @@
+class IMGKit
+  class Configuration
+    attr_accessor :meta_tag_prefix, :wkhtmltoimage, :default_options
+
+    def initialize
+      @meta_tag_prefix = 'imgkit-'
+      @wkhtmltoimage = '/usr/local/bin/wkhtmltoimage'
+      @default_options = {}
+    end
+  end
+
+  class << self
+    attr_accessor :configuration
+  end
+
+  # Configure IMGKit someplace sensible,
+  # like config/initializers/imgkit.rb
+  #
+  # @example
+  #   IMGKit.configure do |config|
+  #     config.wkhtmltoimage = '/usr/bin/wkhtmltoimage'
+  #   end
+  
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+  
+  
+  def self.configure
+    self.configuration 
+    yield(configuration)
+  end
+end

data/lib/imgkit/imgkit.rb

@@ -0,0 +1,147 @@
+class IMGKit
+
+  class NoExecutableError < StandardError
+    def initialize
+      msg  = "No wkhtmltoimage executable found at #{IMGKit.configuration.wkhtmltoimage}\n"
+      msg << ">> Install wkhtmltoimage by hand or try running `imgkit --install-wkhtmltoimage`"
+      super(msg)
+    end
+  end
+  
+  class ImproperSourceError < StandardError
+    def initialize(msg)
+      super("Improper Source: #{msg}")
+    end
+  end
+  
+  attr_accessor :source, :stylesheets
+  attr_reader :options
+  
+  def initialize(url_file_or_html, options = {})
+    @source = Source.new(url_file_or_html)
+    
+    @stylesheets = []
+
+    @options = IMGKit.configuration.default_options.merge(options)
+    @options.merge! find_options_in_meta(url_file_or_html) unless source.url?
+    @options = normalize_options(@options)
+    
+    raise NoExecutableError.new unless File.exists?(IMGKit.configuration.wkhtmltoimage)
+  end
+  
+  def command
+    args = [executable]
+    args += @options.to_a.flatten.compact
+    
+    if @source.html?
+      args << '-' # Get HTML from stdin
+    else
+      args << @source.to_s
+    end
+    
+    args << '-' # Read IMG from stdout
+    args
+  end
+
+  def executable
+    default = IMGKit.configuration.wkhtmltoimage
+    return default if default !~ /^\// # its not a path, so nothing we can do
+    if File.exist?(default)
+      default
+    else
+      default.split('/').last
+    end
+  end
+  
+  def to_img
+    append_stylesheets
+
+=begin  Need to use Open3 b.c wkhtmltoimage has to --quiet option
+        and we need to slience $stderr
+    image = Kernel.open('|-', "w+")
+    exec(*command) if image.nil?
+    image.puts(@source.to_s) if @source.html?
+    image.close_write
+    result = image.gets(nil)
+    image.close_read
+=end
+
+    result = nil
+    Open3.popen3(*command) do |stdin,stdout,stderr|
+      stdin << (@source.to_s) if @source.html?
+      stdin.close
+      result = stdout.gets(nil)
+      stdout.close
+      stderr.close
+    end
+    
+    raise "command failed: #{command.join(' ')}" unless result
+    return result
+  end
+  
+  def to_file(path)
+    File.open(path,'w') {|file| file << self.to_img}
+  end
+  
+  protected
+
+    def find_options_in_meta(body)
+      imgkit_meta_tags(body).inject({}) do |found, tag|
+        name = tag.attributes["name"].sub(/^#{IMGKit.configuration.meta_tag_prefix}/, '').to_sym
+        found.merge(name => tag.attributes["content"])
+      end
+    end
+
+    def imgkit_meta_tags(body)
+      require 'rexml/document'
+      xml_body = REXML::Document.new(body)
+      found = []
+      xml_body.elements.each("html/head/meta") do |tag|
+        found << tag if tag.attributes['name'].to_s =~ /^#{IMGKit.configuration.meta_tag_prefix}/
+      end
+      found
+    rescue # rexml random crash on invalid xml
+      []
+    end
+  
+    def style_tag_for(stylesheet)
+      "<style>#{File.read(stylesheet)}</style>"
+    end
+    
+    def append_stylesheets
+      raise ImproperSourceError.new('Stylesheets may only be added to an HTML source') if stylesheets.any? && !@source.html?
+      
+      stylesheets.each do |stylesheet|
+        if @source.to_s.match(/<\/head>/)
+          @source.to_s.gsub!(/(<\/head>)/, style_tag_for(stylesheet)+'\1')
+        else
+          @source.to_s.insert(0, style_tag_for(stylesheet))
+        end
+      end
+    end
+  
+    def normalize_options(options)
+      normalized_options = {}
+
+      options.each do |key, value|
+        next if !value
+        normalized_key = "--#{normalize_arg key}"
+        normalized_options[normalized_key] = normalize_value(value)
+      end
+      normalized_options
+    end
+    
+    def normalize_arg(arg)
+      arg.to_s.downcase.gsub(/[^a-z0-9]/,'-')
+    end
+    
+    def normalize_value(value)
+      case value
+      when TrueClass
+        nil
+      else
+        value.to_s
+      end
+    end
+  
+end

data/lib/imgkit/source.rb

@@ -0,0 +1,27 @@
+class IMGKit
+  
+  class Source
+        
+    def initialize(url_file_or_html)
+      @source = url_file_or_html
+    end
+    
+    def url?
+      @source.is_a?(String) && @source.match(/^http/)
+    end
+    
+    def file?
+      @source.kind_of?(File)
+    end
+    
+    def html?
+      !(url? || file?)
+    end
+    
+    def to_s
+      file? ? @source.path : @source
+    end
+    
+  end
+  
+end

data/spec/fixtures/example.css

@@ -0,0 +1 @@
+body { font-size: 20px; }

data/spec/fixtures/example.html

@@ -0,0 +1,5 @@
+<html>
+  <body>
+    <h1>Oh Hai!</h1>
+  </body>
+</html>

data/spec/imgkit_spec.rb

@@ -0,0 +1,182 @@
+require 'spec_helper' 
+
+describe IMGKit do
+  context "initialization" do
+    it "should accept HTML as the source" do
+      imgkit = IMGKit.new('<h1>Oh Hai</h1>')
+      imgkit.source.should be_html
+      imgkit.source.to_s.should == '<h1>Oh Hai</h1>'
+    end
+    
+    it "should accept a URL as the source" do
+      imgkit = IMGKit.new('http://google.com')
+      imgkit.source.should be_url
+      imgkit.source.to_s.should == 'http://google.com'
+    end
+    
+    it "should accept a File as the source" do
+      file_path = File.join(SPEC_ROOT,'fixtures','example.html')
+      imgkit = IMGKit.new(File.new(file_path))
+      imgkit.source.should be_file
+      imgkit.source.to_s.should == file_path
+    end
+    
+    it "should parse the options into a cmd line friedly format" do
+      imgkit = IMGKit.new('html', :quality => 75)
+      imgkit.options.should have_key('--quality')
+    end
+    
+    it "should provide no default options" do
+      imgkit = IMGKit.new('<h1>Oh Hai</h1>')
+      imgkit.options.should be_empty
+    end
+    
+=begin
+    it "should default to 'UTF-8' encoding" do
+      imgkit = IMGKit.new('Captación')
+    end
+=end
+    
+    it "should not have any stylesheedt by default" do
+      imgkit = IMGKit.new('<h1>Oh Hai</h1>')
+      imgkit.stylesheets.should be_empty
+    end
+  end
+  
+  context "command" do
+    it "should contstruct the correct command" do
+      imgkit = IMGKit.new('html')
+      imgkit.command[0].should include('wkhtmltoimage')
+      imgkit.command.should include('-')
+    end
+    
+    it "will not include default options it is told to omit" do
+      imgkit = IMGKit.new('html')
+      imgkit = IMGKit.new('html', :disable_smart_shrinking => false)
+      imgkit.command.should_not include('--disable-smart-shrinking')
+    end
+    
+    it "should encapsulate string arguments in quotes" do
+      imgkit = IMGKit.new('html', :header_center => "foo [page]")
+      imgkit.command[imgkit.command.index('--header-center') + 1].should == 'foo [page]'
+    end
+    
+    it "read the source from stdin if it is html" do
+      imgkit = IMGKit.new('html')
+      imgkit.command[-2..-1].should == ['-', '-']
+    end
+    
+    it "specify the URL to the source if it is a url" do
+      imgkit = IMGKit.new('http://google.com')
+      imgkit.command[-2..-1].should == ['http://google.com', '-']
+    end
+    
+    it "should specify the path to the source if it is a file" do
+      file_path = File.join(SPEC_ROOT,'fixtures','example.html')
+      imgkit = IMGKit.new(File.new(file_path))
+      imgkit.command[-2..-1].should == [file_path, '-']
+    end
+
+    it "should detect special imgkit meta tags" do
+      body = %{
+        <html>
+          <head>
+            <meta name="imgkit-page_size" content="Legal"/>
+            <meta name="imgkit-orientation" content="Landscape"/>
+          </head>
+        </html>
+      }
+      imgkit = IMGKit.new(body)
+      imgkit.command[imgkit.command.index('--page-size') + 1].should == 'Legal'
+      imgkit.command[imgkit.command.index('--orientation') + 1].should == 'Landscape'
+    end
+  end
+  
+  context "#to_img" do
+    def filetype_of(img) 
+      result = nil
+      tmpfile = Tempfile.new('imgkit') 
+      File.open(tmpfile.path, 'w') { |f| f << img }
+      result = `file #{tmpfile.path}`
+      tmpfile.unlink()
+      result
+    end
+
+    it "should generate a IMG of the HTML" do
+      imgkit = IMGKit.new('html')
+      img = imgkit.to_img
+      filetype_of(img).should include('JPEG')
+    end
+    
+    it "should generate an Image with a numerical parameter" do
+      imgkit = IMGKit.new('html', :quality => 50)
+      img = imgkit.to_img
+      filetype_of(img).should include('JPEG')
+    end
+    
+    it "should generate an Image with a symbol parameter" do
+      imgkit = IMGKit.new('html', :username => 'chris')
+      img = imgkit.to_img
+      filetype_of(img).should include('JPEG')
+    end
+    
+    it "should have the stylesheet added to the head if it has one" do
+      imgkit = IMGKit.new("<html><head></head><body>Hai!</body></html>")
+      css = File.join(SPEC_ROOT,'fixtures','example.css')
+      imgkit.stylesheets << css
+      imgkit.to_img
+      imgkit.source.to_s.should include("<style>#{File.read(css)}</style>")
+    end
+    
+    it "should prepend style tags if the HTML doesn't have a head tag" do
+      imgkit = IMGKit.new("<html><body>Hai!</body></html>")
+      css = File.join(SPEC_ROOT,'fixtures','example.css')
+      imgkit.stylesheets << css
+      imgkit.to_img
+      imgkit.source.to_s.should include("<style>#{File.read(css)}</style><html>")
+    end
+    
+    it "should throw an error if the source is not html and stylesheets have been added" do
+      imgkit = IMGKit.new('http://google.com')
+      css = File.join(SPEC_ROOT,'fixtures','example.css')
+      imgkit.stylesheets << css
+      lambda { imgkit.to_img }.should raise_error(IMGKit::ImproperSourceError)
+    end
+  end
+  
+  context "#to_file" do
+    before do
+      @file_path = File.join(SPEC_ROOT,'fixtures','test.jpg')
+      File.delete(@file_path) if File.exist?(@file_path)
+    end
+    
+    after do
+      File.delete(@file_path)
+    end
+    
+    it "should create a file with the result of :to_img  as content" do
+      imgkit = IMGKit.new('html', :quality => 50)
+      imgkit.expects(:to_img).returns('CONTENT')
+      file = imgkit.to_file(@file_path)
+      file.should be_instance_of(File)
+      File.read(file.path).should == 'CONTENT'
+    end
+  end
+  
+  context "security" do
+    before do
+      @test_path = File.join(SPEC_ROOT,'fixtures','security-oops')
+      File.delete(@test_path) if File.exist?(@test_path)
+    end
+    
+    after do
+      File.delete(@test_path) if File.exist?(@test_path)
+    end
+    
+    it "should not allow shell injection in options" do
+      imgkit = IMGKit.new('html', :password => "blah\"; touch #{@test_path} #")
+      imgkit.to_img
+      File.exist?(@test_path).should be_false
+    end
+  end
+end

data/spec/source_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+
+describe IMGKit::Source do
+  
+  describe "#url?" do
+    it "should return true if passed a url like string" do
+      source = IMGKit::Source.new('http://google.com')
+      source.should be_url
+    end
+    
+    it "should return false if passed a file" do
+      source = IMGKit::Source.new(File.new(__FILE__))
+      source.should_not be_url
+    end
+    
+    it "should return false if passed HTML" do
+      source = IMGKit::Source.new('<blink>Oh Hai!</blink>')
+      source.should_not be_url
+    end
+  end
+  
+  describe "#file?" do
+    it "should return true if passed a file" do
+      source = IMGKit::Source.new(File.new(__FILE__))
+      source.should be_file
+    end
+    
+    it "should return false if passed a url like string" do
+      source = IMGKit::Source.new('http://google.com')
+      source.should_not be_file
+    end
+    
+    it "should return false if passed HTML" do
+      source = IMGKit::Source.new('<blink>Oh Hai!</blink>')
+      source.should_not be_file
+    end
+  end
+  
+  describe "#html?" do
+    it "should return true if passed HTML" do
+      source = IMGKit::Source.new('<blink>Oh Hai!</blink>')
+      source.should be_html
+    end
+    
+    it "should return false if passed a file" do
+      source = IMGKit::Source.new(File.new(__FILE__))
+      source.should_not be_html
+    end
+    
+    it "should return false if passed a url like string" do
+      source = IMGKit::Source.new('http://google.com')
+      source.should_not be_html
+    end
+  end
+  
+  describe "#to_s" do
+    it "should return the HTML if passed HTML" do
+      source = IMGKit::Source.new('<blink>Oh Hai!</blink>')
+      source.to_s.should == '<blink>Oh Hai!</blink>'
+    end
+    
+    it "should return a path if passed a file" do
+      source = IMGKit::Source.new(File.new(__FILE__))
+      source.to_s.should == __FILE__
+    end
+    
+    it "should return the url if passed a url like string" do
+      source = IMGKit::Source.new('http://google.com')
+      source.to_s.should == 'http://google.com'
+    end
+  end
+  
+end

data/spec/spec_helper.rb

@@ -0,0 +1,19 @@
+SPEC_ROOT = File.dirname(__FILE__)
+$LOAD_PATH.unshift(SPEC_ROOT)
+$LOAD_PATH.unshift(File.join(SPEC_ROOT, '..', 'lib'))
+require 'imgkit'
+require 'rspec'
+require 'rspec/autorun'
+require 'mocha'
+require 'rack'
+require 'tempfile'
+
+RSpec.configure do |config|
+  
+  config.before do
+    IMGKit.any_instance.stubs(:wkhtmltoimage).returns(
+      File.join(SPEC_ROOT,'..','bin','wkhtmltoimage-proxy')
+    )
+  end
+  
+end

metadata

@@ -0,0 +1,143 @@
+--- !ruby/object:Gem::Specification 
+name: imgkit
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 9
+  - 2
+  version: 0.9.2
+platform: ruby
+authors: 
+- csquared
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-12 00:00:00 -06:00
+default_executable: imgkit
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 0
+        - 0
+        - beta
+        - 8
+        version: 2.0.0.beta.8
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec-core
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 0
+        - 0
+        - beta
+        - 8
+        version: 2.0.0.beta.8
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: Uses wkhtmltoimage to create Images using HTML
+email: christopher.continanza@gmail.com
+executables: 
+- imgkit
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.md
+files: 
+- .document
+- .gitignore
+- .rspec
+- .rvmrc
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- POST_INSTALL
+- README.md
+- Rakefile
+- VERSION
+- bin/imgkit
+- imgkit.gemspec
+- lib/imgkit.rb
+- lib/imgkit/configuration.rb
+- lib/imgkit/imgkit.rb
+- lib/imgkit/source.rb
+- spec/fixtures/example.css
+- spec/fixtures/example.html
+- spec/imgkit_spec.rb
+- spec/source_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/csquared/IMGKit
+licenses: []
+
+post_install_message: |
+  ******************************************************************
+  
+  Now install wkhtmltoimage binaries:
+  Global: sudo `which imgkit` --install-wkhtmltoimage
+  or inside RVM folder: export TO=`which imgkit | sed 's:/imgkit:/wkhtmltoimage:'` && imgkit --install-wkhtmltoimage
+  (run imgkit --help to see more options)
+  
+  ******************************************************************
+
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: HTML+CSS -> JPG
+test_files: 
+- spec/imgkit_spec.rb
+- spec/source_spec.rb
+- spec/spec_helper.rb