Separate lib and executable.

Add a separate executable, and move the rendering of the feeds into the feed class.

1 files changed, 24 insertions, 0 deletions

diff --git a/lib/rss2html/feed.rb b/lib/rss2html/feed.rb
index a2f3db4..0f3f01e 100644
--- a/lib/rss2html/feed.rb
+++ b/lib/rss2html/feed.rb
@@ -16,12 +16,17 @@
 
 require 'rss'
 require 'open-uri'
+require 'erb'
 
 class Feed
   attr_reader :title, :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
 
   def fetch
@@ -33,4 +38,23 @@ class Feed
       end
     end
   end
+
+  def render
+    feed = self
+    output = @feed_header.result(binding)
+
+    fetch do |entry|
+      item = FeedItem.new(entry)
+      output += @item_template.result(binding)
+    end
+
+    output += @feed_footer.result(binding)
+  end
+
+  private
+
+  def load_template(t)
+    view_path = File.join(File.dirname(__FILE__), 'views')
+    ERB.new(IO.read(File.join(view_path, t)))
+  end
 end