diff options
Diffstat (limited to 'lib/rss2html/feed.rb')
-rw-r--r-- | lib/rss2html/feed.rb | 56 |
1 files changed, 29 insertions, 27 deletions
diff --git a/lib/rss2html/feed.rb b/lib/rss2html/feed.rb index 0f3f01e..8df2055 100644 --- a/lib/rss2html/feed.rb +++ b/lib/rss2html/feed.rb @@ -18,43 +18,45 @@ require 'rss' require 'open-uri' require 'erb' -class Feed - attr_reader :title, :url +module Rss2Html + class Feed + attr_reader :title, :url - def initialize(title, args) - @title, @url = title, args['url'] + def initialize(title, args) + @title, @url = title, args['url'] - @feed_header = load_template('feed_header.html.erb') - @feed_footer = load_template('feed_footer.html.erb') - @item_template = load_template('item.html.erb') - end + @feed_header = load_template('feed_header.html.erb') + @feed_footer = load_template('feed_footer.html.erb') + @item_template = load_template('item.html.erb') + end - def fetch - open(url) do |rss| - feed = RSS::Parser.parse(rss) + def fetch + open(url) do |rss| + feed = RSS::Parser.parse(rss) - feed.entries.each do |item| - yield item + feed.entries.each do |item| + yield item + end end end - end - def render - feed = self - output = @feed_header.result(binding) + def render + feed = self + output = @feed_header.result(binding) - fetch do |entry| - item = FeedItem.new(entry) - output += @item_template.result(binding) - end + fetch do |entry| + item = FeedItem.new(entry) + output += @item_template.result(binding) + end - output += @feed_footer.result(binding) - end + output += @feed_footer.result(binding) + end - private + private - def load_template(t) - view_path = File.join(File.dirname(__FILE__), 'views') - ERB.new(IO.read(File.join(view_path, t))) + def load_template(t) + view_path = File.join(File.dirname(__FILE__), 'views') + ERB.new(IO.read(File.join(view_path, t))) + end end end |