Added draft command

Author: BennyHallett

lib/jekyll/commands/draft.rb

@@ -0,0 +1,53 @@
+module Jekyll
+  module Commands
+    class Draft < Command
+      def self.init_with_program(prog)
+        prog.command(:draft) do |c|
+          c.syntax 'draft NAME'
+          c.description 'Creates a new draft post with the given NAME'
+
+          c.option 'type', '-t TYPE', '--type TYPE', 'Specify the content type'
+          c.option 'layout', '-l LAYOUT', '--layout LAYOUT', 'Specify the post layout'
+          c.option 'force', '-f', '--force', 'Overwrite a draft if it already exists'
+
+          c.action do |args, options|
+            Jekyll::Commands::Draft.process(args, options)
+          end
+        end
+      end
+
+      def self.process(args, options = {})
+        raise ArgumentError.new('You must specify a name.') if args.empty?
+        
+        type = options["type"].nil? ? "markdown" : options["type"]
+        layout = options["layout"].nil? ? "post" : options["layout"]
+
+        title = args.shift
+        name = title.gsub(' ', '-').downcase
+
+        draft_path = draft_name(name, type)
+
+        raise ArgumentError.new("A draft already exists at ./#{draft_path}") if File.exist?(draft_path) and !options["force"]
+
+        File.open(draft_path, "w") do |f|
+          f.puts(front_matter(layout, title))
+        end
+
+        puts "New draft created at ./#{draft_path}.\n"
+      end 
+      # Internal: Gets the filename of the draft to be created
+      #
+      # Returns the filename of the draft, as a String
+      def self.draft_name(name, ext='markdown')
+        "_drafts/#{name}.#{ext}"
+      end
+
+      def self.front_matter(layout, title)
+        "---
+layout: #{layout}
+title: #{title}
+---"
+      end
+    end
+  end
+end

test/test_draft_command.rb

@@ -0,0 +1,77 @@
+require 'helper'
+require 'jekyll/commands/draft'
+
+class TestDraftCommand < Test::Unit::TestCase
+
+  context 'when no flags are given' do
+    setup do
+      @name = 'A test post'
+      @args = [@name]
+      @drafts_dir = '_drafts'
+      @path = File.join(@drafts_dir, 'a-test-post.markdown')
+      stub(Jekyll).configuration { Jekyll::Configuration::DEFAULTS }
+      @site = Site.new(Jekyll.configuration)
+      FileUtils.mkdir @drafts_dir
+    end
+
+    teardown do
+      FileUtils.rm_r @drafts_dir
+    end
+
+    should 'create a new draft post' do
+      assert !File.exist?(@path)
+      capture_stdout { Jekyll::Commands::Draft.process(@args) }
+      assert File.exist?(@path)
+    end
+
+    should 'write a success message' do
+      output = capture_stdout { Jekyll::Commands::Draft.process(@args) }
+      success_message = "New draft created at ./#{@path}.\n"
+      assert_equal success_message, output
+    end
+  end
+
+  context 'when the draft already exists' do
+    setup do
+      @name = 'An existing draft'
+      @args = [@name]
+      @drafts_dir = '_drafts'
+      @path = File.join(@drafts_dir, 'an-existing-draft.markdown')
+      stub(Jekyll).configuration { Jekyll::Configuration::DEFAULTS }
+      @site = Site.new(Jekyll.configuration)
+      FileUtils.mkdir @drafts_dir
+      FileUtils.touch @path
+    end
+
+    teardown do
+      FileUtils.rm_r @drafts_dir
+    end
+
+    should 'raise an ArgumentError' do
+      exception = assert_raise ArgumentError do
+        capture_stdout { Jekyll::Commands::Draft.process(@args) }
+      end
+      assert_equal "A draft already exists at ./#{@path}", exception.message
+    end
+
+    should 'overwrite the existing draft if --force is given' do
+      capture_stdout { Jekyll::Commands::Draft.process(@args, { "force" => true  }) }
+      assert File.readlines(@path).grep(/layout: post/).any?
+    end
+      
+  end
+
+  context 'when no args are given' do
+    setup do
+      @empty_args = []
+    end
+
+    should 'raise an ArgumentError' do
+      exception = assert_raise ArgumentError do
+        Jekyll::Commands::Draft.process(@empty_args)
+      end
+      assert_equal 'You must specify a name.', exception.message
+    end
+  end
+
+end