lib/gnawrnip/screenshot.rb in gnawrnip-0.1.3 vs lib/gnawrnip/screenshot.rb in gnawrnip-0.2.0

@@ -27,10 +27,11 @@
         start_time = Time.now
 
         begin
           tempfile = Tempfile.new(['gnawrnip', '.png'])
           session.save_screenshot(tempfile.path)
+          resize(tempfile.path) if need_resize?
           tempfile
         rescue Capybara::NotSupportedByDriverError => e
           raise e
         rescue => e
           raise e if (Time.now - start_time) >= wait_second
@@ -39,11 +40,48 @@
         end
       end
 
       private
 
-      def session
-        Capybara.current_session
-      end
+        def session
+          Capybara.current_session
+        end
+
+        def need_resize?
+          !Gnawrnip.max_frame_size.nil?
+        end
+
+        def resize(path)
+          require 'oily_png'
+
+          image = OilyPNG::Canvas.from_file(path)
+          new_width, new_height = calculate_new_size(image.width, image.height)
+
+          image.resample_bilinear!(new_width, new_height)
+          image.save(path)
+        end
+
+        #
+        # Return new frame size (width and height).
+        # This size is keeping original aspect ratio.
+        #
+        # @return  [Array]  New width and height size. [width, height]
+        #
+        def calculate_new_size(width, height)
+          ratio  = width / height
+          target = Gnawrnip.max_frame_size
+
+          return [width, height] if target > [width, height].max
+
+          if ratio < 0
+            new_width  = target * ratio
+            new_height = target
+          else
+            new_width  = target
+            new_height = target / ratio
+          end
+
+          return [new_width, new_height]
+        end
     end
   end
 end