silkscreen 0.1 → 0.1.1

data/bin/silkscreen

@@ -3,8 +3,7 @@ require 'rubygems'
 require 'thor'
 require 'headless'
 require 'selenium-webdriver'
-require 'chunky_png'
-require 'oily_png'
+require 'silkscreen'
 
 module Silkscreen
   class App < Thor
@@ -22,53 +21,18 @@ module Silkscreen
       headless.start
 
       driver = Selenium::WebDriver.for :chrome
-      window_width, window_height = options[:viewport].split('x').map { |x| x.to_i }
-      driver.manage.window.resize_to window_width, window_height
-      driver.navigate.to url
-      sleep options[:sleep]
-      img = driver.screenshot_as :png
+
+      img = Silkscreen.get_screenshot url, options, driver
       
       driver.quit
       headless.destroy
 
-      if options[:full]
-        if options[:max_width] || options[:max_height]
-          canvas = ChunkyPNG::Image.from_blob img
-          canvas = resize_canvas(canvas, options[:max_width], options[:max_height])
-          img = canvas.to_s
-        end
-      else
-        canvas = ChunkyPNG::Image.from_blob img
-        canvas.crop! 0, 0, [window_width, canvas.width].min, [window_height, canvas.height].min
-        canvas = resize_canvas(canvas, options[:max_width], options[:max_height])
-        img = canvas.to_s
-      end
-
-
-
       if options[:output]
         create_file options[:output], img
       else
         print img
       end
     end
-
-    private
-
-    def resize_canvas(canvas, max_width, max_height)
-      max_width ||= canvas.width
-      max_height ||= canvas.height
-      if canvas.width <= max_width && canvas.height <= max_height
-        canvas
-      else
-        width_ratio = max_width.to_f/canvas.width
-        height_ratio = max_height.to_f/canvas.height
-        lower_ratio = [width_ratio, height_ratio].min
-        new_width = (canvas.width * lower_ratio).round
-        new_height = (canvas.height * lower_ratio).round
-        canvas.resample_bilinear new_width, new_height
-      end
-    end
   end
 end
 

data/lib/silkscreen.rb

@@ -1,5 +1,42 @@
 require "silkscreen/version"
+require 'chunky_png'
+require 'oily_png'
 
 module Silkscreen
-  # Your code goes here...
+  def self.get_screenshot(url, options, driver)
+    window_width, window_height = options[:viewport].split('x').map { |x| x.to_i }
+    driver.manage.window.resize_to window_width, window_height
+    driver.navigate.to url
+    sleep options[:sleep]
+    img = driver.screenshot_as :png
+    if options[:full]
+      if options[:max_width] || options[:max_height]
+        canvas = ChunkyPNG::Image.from_blob img
+        canvas = resize_canvas(canvas, options[:max_width], options[:max_height])
+        img = canvas.to_s
+      end
+    else
+      canvas = ChunkyPNG::Image.from_blob img
+      canvas.crop! 0, 0, [window_width, canvas.width].min, [window_height, canvas.height].min
+      canvas = resize_canvas(canvas, options[:max_width], options[:max_height])
+      img = canvas.to_s
+    end
+  end
+
+  private
+
+  def self.resize_canvas(canvas, max_width, max_height)
+    max_width ||= canvas.width
+    max_height ||= canvas.height
+    if canvas.width <= max_width && canvas.height <= max_height
+      canvas
+    else
+      width_ratio = max_width.to_f/canvas.width
+      height_ratio = max_height.to_f/canvas.height
+      lower_ratio = [width_ratio, height_ratio].min
+      new_width = (canvas.width * lower_ratio).round
+      new_height = (canvas.height * lower_ratio).round
+      canvas.resample_bilinear new_width, new_height
+    end
+  end
 end

data/lib/silkscreen/version.rb

@@ -1,3 +1,3 @@
 module Silkscreen
-  VERSION = "0.1"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: silkscreen
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors: