extract row

Author: marocchino

_plugins/generators.rb

@@ -1,5 +1,7 @@
-module Jekyll
-  class RSSGenerator < Generator
+module RSS
+  class Generator < Jekyll::Generator
+    ROW_MATCHER = /^-\s*\[(?<title>[^\]]+)\]\((?<url>[^)]+)\){:\s*.article(?<data>(\s*data-\w+="[^"]+")+)\s*}\s*$/
+    DATA_MATCHER = /data-(?<key>\w+)="(?<value>[^"]+)"/
     def generate(site)
       @index_page = site.pages.find { |page| page.name == "index.md" }
       @feed_page = site.pages.find { |page| page.name == "feed.xml" }
@@ -12,21 +14,26 @@ def generate(site)
     def articles
       unless @articles
         @articles = []
-        rows = @index_page.content.split(/#articles\s*}\n\n/).last.split(/## \[동영상\]/).first.split(/\n/)
         rows.each do |row|
-          @articles << {}
-          m = row.match(/^-\s*\[(?<title>[^\]]+)\]\((?<url>[^)]+)\){:\s*.article(?<data>(\s*data-\w+="[^"]+")+)\s*}\s*$/)
-          @articles.last["title"] = m[:title]
-          @articles.last["url"] = m[:url]
-          m[:data].scan(/data-(?<key>\w+)="(?<value>[^"]+)"/) do |key, value|
-            @articles.last[key] = value
+          article = {}
+          m = row.match(ROW_MATCHER)
+          m[:data].scan(DATA_MATCHER) do |key, value|
+            article[key] = value
           end
-          if articles.last["tags"].include?("translated")
-            @articles.last["title"] = "[번역] #{m[:title]}"
+          if article["tags"].include?("translated")
+            article["title"] = "[번역] #{m[:title]}"
+          else
+            article["title"] = m[:title]
           end
+          article["url"] = m[:url]
+          @articles << article
         end
       end
       @articles
     end
+
+    def rows
+      @index_page.content.split(/#articles\s*}\n\n/).last.split(/## \[동영상\]/).first.split(/\n/)
+    end
   end
 end