jekyll-import.rb in jekyll-import-0.1.0.beta4

- old
+ new
@@ -1,7 +1,56 @@
 $:.unshift File.dirname(__FILE__) # For use/testing when no gem is installed
 require 'rubygems'
 require 'jekyll/commands/import'
+require 'jekyll/stevenson'
+require 'colorator'
 
+require 'jekyll-import/importer'
+require 'jekyll-import/importers'
+
 module JekyllImport
-  VERSION = '0.1.0.beta3'
+  VERSION = '0.1.0.beta4'
+
+  def self.logger
+    @logger ||= Jekyll::Stevenson.new
+  end
+
+  # Public: Add the subcommands for each importer
+  #
+  # cmd - the instance of Mercenary::Command from the
+  #
+  # Returns a list of valid subcommands
+  def self.add_importer_commands(cmd)
+    commands = []
+    JekyllImport::Importer.subclasses.each do |importer|
+      name = importer.to_s.split("::").last.downcase
+      commands << name
+      cmd.command(name.to_sym) do |c|
+        c.syntax "jekyll import #{name} [options]"
+        importer.specify_options(c)
+        c.action do |args, options|
+          importer.require_deps
+          importer.validate(options) if importer.respond_to?(:validate)
+          importer.process(options)
+        end
+      end
+    end
+    commands
+  end
+
+  def self.require_with_fallback(gems)
+    Array[gems].flatten.each do |gem|
+      begin
+        require gem
+      rescue LoadError
+        logger.error "Whoops! Looks like you need to install '#{gem}' before you can use this importer."
+        logger.error ""
+        logger.error "If you're using bundler:"
+        logger.error "  1. Add 'gem \"#{gem}\"' to your Gemfile"
+        logger.error "  2. Run 'bundle install'"
+        logger.error ""
+        logger.error "If you're not using bundler:"
+        logger.abort_with "  1. Run 'gem install #{gem}'."
+      end
+    end
+  end
 end