ero_getter 1.2.0 → 1.3.0

data/VERSION

@@ -1 +1 @@
-1.2.0
+1.3.0

data/ero_getter.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "ero_getter"
-  s.version = "1.2.0"
+  s.version = "1.3.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["masarakki"]
-  s.date = "2012-06-01"
+  s.date = "2012-06-02"
   s.description = "ero getter"
   s.email = "masaki@hisme.net"
   s.extra_rdoc_files = [
@@ -28,21 +28,27 @@ Gem::Specification.new do |s|
     "Rakefile",
     "VERSION",
     "ero_getter.gemspec",
+    "lib/downloader/gazou_sokuhou.rb",
     "lib/downloader/nijigazou_sokuhou.rb",
     "lib/downloader/pm_style.rb",
+    "lib/downloader/wakuteka_sokuhou.rb",
     "lib/ero_getter.rb",
     "lib/ero_getter/base.rb",
     "lib/ero_getter/utils.rb",
+    "spec/downloader/gazou_sokuhou_spec.rb",
     "spec/downloader/nijigazou_sokuhou_spec.rb",
     "spec/downloader/pm_style_spec.rb",
+    "spec/downloader/wakuteka_sokuhou_spec.rb",
     "spec/ero_getter/base_spec.rb",
     "spec/ero_getter_spec.rb",
+    "spec/samples/gazou_sokuhou/sample.html",
     "spec/samples/nijigazou_sokuhou/first.html",
     "spec/samples/nijigazou_sokuhou/global_last.html",
     "spec/samples/nijigazou_sokuhou/last.html",
     "spec/samples/nijigazou_sokuhou/middle.html",
     "spec/samples/pm_style/test.html",
     "spec/samples/sample.html",
+    "spec/samples/wakuteka_sokuhou/sample.html",
     "spec/spec_helper.rb"
   ]
   s.homepage = "http://github.com/masarakki/ero_getter"

data/lib/downloader/gazou_sokuhou.rb

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+class GazouSokuhou < EroGetter::Base
+
+  name 'がぞー速報'
+  url %r{http://stalker.livedoor.biz/archives/\d+.html}
+
+  target "img.pict" do |path|
+    if path.parent.name == "a" && path.parent[:href] =~ /jpe?g|gif|png$/
+      path.parent[:href]
+    else
+      path[:src]
+    end
+  end
+
+  sub_directory do
+    url.match(/(\d+).html/)[1]
+  end
+
+  filename { |attr| "%04d%s" % [attr[:index], attr[:ext]] }
+
+end

data/lib/downloader/wakuteka_sokuhou.rb

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+class WakutekaSokuhou < EroGetter::Base
+
+  name 'わくてか速報'
+  url %r{http://blog.livedoor.jp/wakusoku/archives/(\d+).html}
+
+  target ".article-body-inner a > img" do |path|
+    path.parent[:href] if path.parent[:href] =~ /jpe?g|gif|png$/
+  end
+
+  sub_directory do
+    url.match(/(\d+)\.html/)[1]
+  end
+
+  filename {|attr| "%04d%s" % [attr[:index], attr[:ext]] }
+
+end

data/lib/ero_getter/base.rb

@@ -44,11 +44,11 @@ class EroGetter::Base
   end
 
   def run
-    targets.each do |target_url|
+    targets.each_with_index do |target_url, index|
       if target_url =~ /.*\.zip$/
-        save_zip(target_url)
+        save_zip(target_url, index)
       else
-        save_image(target_url)
+        save_image(target_url, index)
       end
     end
     self.class.new(self.prev, :prev).run if run_prev?
@@ -61,16 +61,16 @@ class EroGetter::Base
     response
   end
 
-  def save_image(target_url)
-    filename = File.basename(target_url)
+  def save_image(target_url, index)
+    _filename = filename(File.basename(target_url), index)
     response = get_target(target_url)
-    File.open(File.join(directory, filename), "wb") {|f| f.write response.body }
+    File.open(File.join(directory, _filename), "wb") {|f| f.write response.body }
   end
 
-  def save_zip(target_url)
+  def save_zip(target_url, index)
     response = get_target(target_url)
-    unzip(response.body).each do |filename, data|
-      File.open(File.join(directory, filename), "wb") {|f| f.write data }
+    unzip(response.body).each do |_filename, data|
+      File.open(File.join(directory, _filename), "wb") {|f| f.write data }
     end
   end
 
@@ -135,6 +135,12 @@ class EroGetter::Base
         end
       end
     end
+
+    def filename(&block)
+      define_method(:_filename) do |attr|
+        yield(attr)
+      end
+    end
   end
 
   private
@@ -145,4 +151,12 @@ class EroGetter::Base
   def run_prev?
     direction != :next && respond_to?(:prev) && self.prev != nil
   end
+
+  def filename(basename, index)
+    if respond_to?(:_filename)
+      _filename(index: index, basename: basename, ext: File.extname(basename))
+    else
+      basename
+    end
+  end
 end

data/spec/downloader/gazou_sokuhou_spec.rb

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe GazouSokuhou do
+  let(:url) { 'http://stalker.livedoor.biz/archives/51977311.html' }
+  subject { @dl }
+
+  before do
+    EroGetter.stub(:mkdir)
+    @dl = GazouSokuhou.new(url)
+    fake(:get, url, 'gazou_sokuhou/sample.html')
+  end
+  its(:name) { should == 'がぞー速報' }
+  its(:title) { should == '6月になったし2次水着画像どんどん貼ってくよー  -  がぞ～速報' }
+  its(:base_dir) { should == 'gazou_sokuhou' }
+  its(:sub_directory) { should == '51977311' }
+  its("targets.count") { should == 31 }
+end

data/spec/downloader/wakuteka_sokuhou_spec.rb

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe WakutekaSokuhou do
+  let(:url) { 'http://blog.livedoor.jp/wakusoku/archives/1461175.html' }
+  subject { @dl }
+  before do
+    EroGetter.stub(:mkdir)
+    @dl = WakutekaSokuhou.new(url)
+    fake(:get, url, 'wakuteka_sokuhou/sample.html')
+  end
+
+  its(:title) { should == 'わくてか速報 : 【エロ画像】抜かずにはいられない虹エロ画像パート１１９４【詳細付き】 - livedoor Blog（ブログ）' }
+  its(:sub_directory) { should == '1461175' }
+  its("targets.count") { should == 37 }
+end

data/spec/ero_getter/base_spec.rb

@@ -48,6 +48,7 @@ describe EroGetter::Base do
             'https://github.com/masarakki/ero_getter_chrome_extension'] }
         its(:sub_directory) { should == 'ero_getter_server/ero_getter_chrome_extension' }
         its(:directory) { should == '/tmp/test_class/ero_getter_server/ero_getter_chrome_extension' }
+        it { subject.send(:filename, "hogehoge.jpg", 1).should == 'hogehoge.jpg' }
 
         describe :after_run do
           context :not_set_after do
@@ -152,4 +153,17 @@ describe EroGetter::Base do
       its(:next) { should == 'unko' }
     end
   end
+  context :with_filename do
+    before do
+      _regex = regex
+      klazz = Class.new(EroGetter::Base) do
+        url _regex
+        filename do |attr|
+          "%04d%s" % [attr[:index], attr[:ext]]
+        end
+      end
+      @dl = klazz.new(url)
+    end
+    it { @dl.send(:filename, 'hogehoge.jpg', 0).should == '0000.jpg' }
+  end
 end