webshot 0.0.1 → 0.0.2

data/lib/webshot.rb

@@ -21,10 +21,12 @@ module Webshot
   mattr_accessor :user_agent
   @@user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.31 (KHTML, like Gecko) Chrome/26.0.1410.43 Safari/537.31"
 
+  # Customize settings
   def self.setup
     yield self
   end
 
+  # Capibara setup
   def self.capybara_setup!
     # By default Capybara will try to boot a rack application
     # automatically. You might want to switch off Capybara's

data/lib/webshot/screenshot.rb

@@ -6,12 +6,15 @@ module Webshot
       width  = opts.fetch(:width, Webshot.width)
       height = opts.fetch(:height, Webshot.height)
       user_agent = opts.fetch(:user_agent, Webshot.user_agent)
+
+      # Browser settings
       page.driver.resize(width, height)
       page.driver.headers = {
         "User-Agent" => user_agent,
       }
     end
 
+    # Captures a screenshot of +url+ saving it to +path+.
     def capture(url, path, opts = {})
       # Default settings
       width   = opts.fetch(:width, 120)
@@ -19,20 +22,27 @@ module Webshot
       gravity = opts.fetch(:gravity, "north")
       quality = opts.fetch(:quality, 85)
 
+      # Open page
       visit url
+
       if page.driver.status_code == 200
+        # Save screenshot
         page.driver.save_screenshot(path, :full => true)
+
+        # Resize screenshot
         thumb = MiniMagick::Image.open(path)
         thumb.combine_options do |c|
           c.thumbnail "#{width}x"
           c.background "white"
           c.extent "#{width}x#{height}"
-          c.gravity "north"
-          c.quality 85
+          c.gravity gravity
+          c.quality quality
         end
+
+        # Save thumbnail
         thumb.write path
       else
-        raise WebshotError.new("Could not fetch page: #{url.inspect}")
+        raise WebshotError.new("Could not fetch page: #{url.inspect}, error code: #{page.driver.status_code}")
       end
     end
   end

data/lib/webshot/version.rb

@@ -1,3 +1,3 @@
 module Webshot
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: webshot
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors: