Merge pull request #32 from nilenso/whitelist

Adding ability to import posts only with whitelisted tags

README.md

@@ -89,7 +89,8 @@ The planet.yml file contains the basic config options for planet. It consists of
 ```yaml
 planet:
     posts_directory: source/_posts/                 # => Posts text files will be written into this directory
-    templates_directory: source/_layouts/           # => Planet specific layouts will be saved here, I suggest that it matches your Octopress/Jekyll layout directory.
+    templates_directory: source/_layouts/       # => Planet specific layouts will be saved here, I suggest that it matches your Octopress/Jekyll layout directory.
+    whitelisted_tags: []                                   # => Only posts that are tagged with any of these tags will be imported
 
 blogs:
   - author: "Cubox"                                 # => Name that we will use as the author of this post (soon you wont have to specify this :)

bin/planet

@@ -151,6 +151,7 @@ layout: post
 planet:
     posts_directory: source/_posts/
     templates_directory: source/_layouts/
+    whitelisted_tags: []
 
 blogs:
   # Bare minimum to get it working

lib/planet.rb

@@ -5,12 +5,13 @@
 
 class Planet
 
-  attr_accessor :config, :blogs
+  attr_accessor :config, :blogs, :whitelisted_tags
 
   def initialize(config_file_path)
     config_file = read_config_file(config_file_path)
     self.config = config_file[:planet]
     self.blogs  = config_file[:blogs]
+    self.whitelisted_tags  = self.config['whitelisted_tags']
   end
 
   def posts

lib/planet/blog.rb

@@ -40,6 +40,7 @@ def on_fetch_success(feed)
       end
 
       feed.entries.each do |entry|
+        next unless whitelisted?(entry)
         content = if entry.content
                     self.sanitize_images(entry.content.strip)
                   elsif entry.summary
@@ -73,5 +74,12 @@ def sanitize_images(html)
 
       html
     end
+
+    def whitelisted?(entry)
+      return true if self.planet.whitelisted_tags.empty?
+      result = !(entry.categories & self.planet.whitelisted_tags).empty?
+      puts "\t=> Ignored post titled: #{entry.title} with categories: [#{entry.categories.join(', ')}]" unless result
+      result
+    end
   end
 end