2 files changed, 26 insertions, 29 deletions
diff --git a/lib/rss2html.rb b/lib/rss2html.rb
index 4890fd4..6d95382 100644
--- a/lib/rss2html.rb
+++ b/lib/rss2html.rb
@@ -14,32 +14,5 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-require_relative 'lib/feed_item'
-require_relative 'lib/feed'
-
-require 'yaml'
-require 'erb'
-
-config_file = File.expand_path('~/.config/feeds.yml')
-
-unless File.exists?(config_file)
-  puts "Config file #{config_file} not found. Please create it!"
-  exit
-end
-
-feeds = YAML.load(IO.read(config_file))
-items = []
-
-feed_header = ERB.new(IO.read('lib/views/feed_header.html.erb'))
-feed_footer = ERB.new(IO.read('lib/views/feed_footer.html.erb'))
-item_template = ERB.new(IO.read('lib/views/item.html.erb'))
-
-feeds.each do |t, f|
-  feed = Feed.new(t, f)
-  puts feed_header.result(binding)
-
-  feed.fetch do |entry|
-    item = FeedItem.new(entry)
-    puts item_template.result(binding)
-  end
-end
+require 'rss2html/feed'
+require 'rss2html/feed_item'
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