sugoi_webpage_capture 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 30e446abb57a16193cedee7d908af361500a4800
-  data.tar.gz: ea17fb6b1e9e8a418476b32e7942a820b2d1caac
+  metadata.gz: 5421b6bd434aa669aa3af753e25e18dc5a2a641d
+  data.tar.gz: 5b6efc2710605b8f87ba9e86b4fc195decb6bfb3
 SHA512:
-  metadata.gz: 33ea9de1cb4adc273797a1a6b747bc76b9eef5455b734e1c561c494173f8f2119a1de1fc7262b6826479ee89a9182062892547709a2b28d527265b4f1d82ee09
-  data.tar.gz: 58e23dbd5619074e9a2016cc0b14f464c049993035e32eb3748d38a77224ffb99685130e1590ab08c811b24c84bfcfb4435b300b60d76ade0131762d9985c83a
+  metadata.gz: 62e34e47ce57da16ad7362b82edfdfb570a26fa3ce03742d18f36345c89bb1704f4d1336b4b07db7afc81473d4e0c58b311468341ddbf7b38a948de2aded5711
+  data.tar.gz: b31136c458191a15e6101e61caecc09481e1bf4268e5dcb385f3c2cc9c752272e3c88be64f423f18096219fc4992798983a9d254499fb48c0ce2d5a0db7cc7d5

data/README.md

@@ -23,13 +23,15 @@ Or install it yourself as:
 require "sugoi_webpage_capture"
 captured_url = "http://google.com"
 
-screenshot = SugoiWebpageCapture::Browser.new
-tempfile = screenshot.capture(captured_url) # return captured image.
-tempfile.unlink
-
-screenshot = SugoiWebpageCapture::Browser.new(:iphone5)
-tempfile = screenshot.capture(captured_url) # return captured image.
-tempfile.unlink
+browser = SugoiWebpageCapture::Browser.new
+screenshot = screenshot.capture(captured_url) # return tempfile wrapper instance.
+screenshot.path # /var/tmp.........
+browser.quit
+
+browser = SugoiWebpageCapture::Browser.new(:iphone5)
+screenshot = screenshot.capture(captured_url)
+screenshot.path
+browser.quit
 ```
 
 ## Contributing

data/lib/sugoi_webpage_capture.rb

@@ -2,6 +2,7 @@ require 'capybara/dsl'
 require "selenium-webdriver"
 require "sugoi_webpage_capture/version"
 require "sugoi_webpage_capture/browser"
+require "sugoi_webpage_capture/screenshot"
 
 module SugoiWebpageCapture
 end

data/lib/sugoi_webpage_capture/browser.rb

@@ -24,13 +24,18 @@ module SugoiWebpageCapture
       init
     end
 
-    def capture(captured_url)
+    def capture(captured_url, &block)
+      Capybara.reset_sessions!
       page.current_window.resize_to(*BROWSERS[@browser_name][:size])
-      tempfile = Tempfile.new(["ss", ".png"])
+      @screenshot = Screenshot.new(Tempfile.new(["ss", ".png"]))
       visit captured_url
       yield(self) if block_given?
-      page.driver.save_screenshot(tempfile, full: true) # TODO Chtome full size capture
-      tempfile
+      page.driver.save_screenshot(@screenshot.tempfile, full: true) # TODO Chtome full size capture
+      @screenshot
+    end
+
+    def quit
+      @screenshot.tempfile.unlink if @screenshot
     end
 
     private

data/lib/sugoi_webpage_capture/screenshot.rb

@@ -0,0 +1,17 @@
+module SugoiWebpageCapture
+  class Screenshot
+    attr_reader :tempfile
+
+    def initialize(tempfile)
+      @tempfile = tempfile
+    end
+
+    def unlink
+      @tempfile.unlink
+    end
+
+    def path
+      @tempfile.path
+    end
+  end
+end

data/lib/sugoi_webpage_capture/version.rb

@@ -1,3 +1,3 @@
 module SugoiWebpageCapture
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/sugoi_webpage_capture_spec.rb

@@ -15,14 +15,14 @@ describe SugoiWebpageCapture do
             x.fill_in 'yschsp', with: "あげぽよ"
             x.find(".b").click
           end
-          expect(tempfile).to be_a Tempfile
+          expect(tempfile).to be_a SugoiWebpageCapture::Screenshot
           expect(Capybara.current_driver).to eq :firefox
         end
       end
 
       it 'be success' do
         screenshot = SugoiWebpageCapture::Browser.new
-        expect(screenshot.capture(url)).to be_a Tempfile
+        expect(screenshot.capture(url)).to be_a SugoiWebpageCapture::Screenshot
         expect(Capybara.current_driver).to eq :firefox
       end
     end
@@ -31,7 +31,7 @@ describe SugoiWebpageCapture do
       it 'be success' do
         screenshot = SugoiWebpageCapture::Browser.new(:iphone5)
         expect(Capybara.current_driver).to eq :iphone5
-        expect(screenshot.capture(url)).to be_a Tempfile
+        expect(screenshot.capture(url)).to be_a SugoiWebpageCapture::Screenshot
       end
     end
 
@@ -39,7 +39,7 @@ describe SugoiWebpageCapture do
       it 'be success' do
         screenshot = SugoiWebpageCapture::Browser.new(:chrome)
         expect(Capybara.current_driver).to eq :chrome
-        expect(screenshot.capture(url)).to be_a Tempfile
+        expect(screenshot.capture(url)).to be_a SugoiWebpageCapture::Screenshot
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sugoi_webpage_capture
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - jiikko
@@ -110,6 +110,7 @@ files:
 - Rakefile
 - lib/sugoi_webpage_capture.rb
 - lib/sugoi_webpage_capture/browser.rb
+- lib/sugoi_webpage_capture/screenshot.rb
 - lib/sugoi_webpage_capture/version.rb
 - spec/spec_helper.rb
 - spec/sugoi_webpage_capture_spec.rb