aboutsummaryrefslogtreecommitdiffstats
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/rss2html.rb31
-rw-r--r--lib/rss2html/feed.rb24
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