Added post command

Author: BennyHallett

lib/jekyll/commands/post.rb

@@ -0,0 +1,56 @@
+module Jekyll
+  module Commands
+    class Post < Command
+      def self.init_with_program(prog)
+        prog.command(:post) do |c|
+          c.syntax 'post NAME'
+          c.description 'Creates a new post with the given NAME'
+
+          c.option 'type', '-t TYPE', '--type TYPE', 'Specify the content type'
+          c.option 'layout', '-t LAYOUT', '--layout LAYOUT', 'Specify the post layout'
+          c.option 'date', '-d DATE', '--date DATE', 'Specify the post date'
+          c.option 'force', '-f', '--force', 'Overwrite a post if it already exists'
+
+          c.action do |args, options|
+            Jekyll::Commands::Post.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"]
+
+        date = options["date"].nil? ? Time.now : DateTime.parse(options["date"])
+
+        title = args.shift
+        name = title.gsub(' ', '-').downcase
+
+        post_path = file_name(name, type, date)
+
+        raise ArgumentError.new("A post already exists at ./#{post_path}") if File.exist?(post_path) and !options["force"]
+
+        File.open(post_path, "w") do |f|
+          f.puts(front_matter(layout, title))
+        end
+
+        puts "New post created at ./#{post_path}.\n"
+      end 
+      # Internal: Gets the filename of the draft to be created
+      #
+      # Returns the filename of the draft, as a String
+      def self.file_name(name, ext, date)
+        "_posts/#{date.strftime('%Y-%m-%d')}-#{name}.#{ext}"
+      end
+
+      def self.front_matter(layout, title)
+        "---
+layout: #{layout}
+title: #{title}
+---"
+      end
+    end
+  end
+end

test/test_post_command.rb

@@ -0,0 +1,73 @@
+require 'helper'
+require 'jekyll/commands/post'
+
+class TestPostCommand < Test::Unit::TestCase
+
+  context 'when no flags are given' do
+    setup do
+      @name = 'A test post'
+      @args = [@name]
+      @posts_dir = '_posts'
+      @path = File.join(@posts_dir, "#{Time.now.strftime('%Y-%m-%d')}-a-test-post.markdown")
+      FileUtils.mkdir @posts_dir
+    end
+
+    teardown do
+      FileUtils.rm_r @posts_dir
+    end
+
+    should 'create a new post' do
+      assert !File.exist?(@path)
+      capture_stdout { Jekyll::Commands::Post.process(@args) }
+      assert File.exist?(@path)
+    end
+
+    should 'write a success message' do
+      output = capture_stdout { Jekyll::Commands::Post.process(@args) }
+      success_message = "New post created at ./#{@path}.\n"
+      assert_equal success_message, output
+    end
+  end
+
+  context 'when the post already exists' do
+    setup do
+      @name = 'An existing post'
+      @args = [@name]
+      @posts_dir = '_posts'
+      @path = File.join(@posts_dir, "#{Time.now.strftime('%Y-%m-%d')}-an-existing-post.markdown")
+      FileUtils.mkdir @posts_dir
+      FileUtils.touch @path
+    end
+
+    teardown do
+      FileUtils.rm_r @posts_dir
+    end
+
+    should 'raise an ArgumentError' do
+      exception = assert_raise ArgumentError do
+        capture_stdout { Jekyll::Commands::Post.process(@args) }
+      end
+      assert_equal "A post already exists at ./#{@path}", exception.message
+    end
+
+    should 'overwrite the existing post if --force is given' do
+      capture_stdout { Jekyll::Commands::Post.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::Post.process(@empty_args)
+      end
+      assert_equal 'You must specify a name.', exception.message
+    end
+  end
+
+end