url of an img tag is acquired from a link.

ytsuyuzaki · ytsuyuzaki · commit 7109985e9ad0 · 2013-06-02T14:24:41.000+09:00
diff --git a/plugins/filter/image_source.rb b/plugins/filter/image_source.rb
@@ -19,29 +19,52 @@ def initialize(config, pipeline=[])
     def run
       @return_feeds = []
       @pipeline.each {|feeds|
-        img_url = ""
+        dummyFeeds = Array.new
         unless feeds.nil?
           feeds.items.each {|feed|
-            arr = rewrite_link(feed.description)
+            arr = rewrite_link(feed)
             if arr.length > 0
-              feed.link = arr[0]
-            else
-              feed.link = nil
+              arr.each {|link|
+                Automatic::Log.puts("info", "Parsing: #{link}")
+                dummy = Hashie::Mash.new
+                dummy.title = 'FilterImageSource'
+                dummy.link = link
+                dummyFeeds << dummy
+              }
             end
           }
         end
-        @return_feeds << feeds
+        @return_feeds << Automatic::FeedParser.create(dummyFeeds)
       }
-      @return_feeds
+      @pipeline = @return_feeds
     end
 
     private
-    def rewrite_link(string)
+    def rewrite_link(feed)
       array = Array.new
-      string.scan(/<img src="(.*?)"/) { |matched|
+      feed.description.scan(/<img src="(.*?)"/) { |matched|
         array = array | matched
       }
+      if array.length === 0 && feed.link != nil
+        array = imgs(feed.link)
+      end
       array
     end
+
+    def imgs(link)
+      images = Array.new
+      html = open(link).read
+      unless html.nil?
+        doc = Nokogiri::HTML(html)
+        (doc/:img).each {|img|
+          image = img[:src]
+          unless /^http/ =~ image
+            image = link.sub(/\/([^\/]+)$/, image.sub(/^\./,''))
+          end
+          images << image
+        }
+      end
+      images
+    end
   end
 end
diff --git a/spec/plugins/filter/image_source_spec.rb b/spec/plugins/filter/image_source_spec.rb
@@ -54,7 +54,7 @@
 end
 
 describe Automatic::Plugin::FilterImageSource do
-  context "with no data" do
+  context "with link to tag image" do
     subject {
       Automatic::Plugin::FilterImageSource.new({},
         AutomaticSpec.generate_pipeline {
@@ -64,12 +64,26 @@
           }})}
 
     describe "#run" do
+      before do
+        subject.stub!(:rewrite_link).and_return(['http://huge.png'])
+      end
+
       its(:run) { should have(1).feeds }
       specify {
         subject.run
         subject.instance_variable_get(:@pipeline)[0].items[0].link.
-        should nil
+        should == 'http://huge.png'
       }
     end
+
+    describe "#imgs" do
+      before do
+        open = Hashie::Mash.new
+        open.read = '<img src="http://a.png"><br /><img src="http://b.png">'
+        subject.stub!(:open).and_return(open)
+      end
+
+      its(:run) { subject.run[0].items.length.should == 2 }
+    end
   end
 end
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -23,6 +23,7 @@
 end
 
 require 'lib/automatic'
+Automatic::Log.level('none')
 
 # Requires supporting files with custom matchers and macros, etc,
 # in ./support/ and its subdirectories.
