Add unpublish command

Author: courajs

lib/jekyll-compose.rb

@@ -11,6 +11,6 @@ module Compose
   end
 end
 
-%w{draft post publish page}.each do |file|
+%w{draft post publish unpublish page}.each do |file|
   require File.expand_path("jekyll/commands/#{file}.rb", File.dirname(__FILE__))
 end

lib/jekyll/commands/unpublish.rb

@@ -0,0 +1,88 @@
+module Jekyll
+  module Commands
+    class Unpublish < Command
+      def self.init_with_program(prog)
+        prog.command(:unpublish) do |c|
+          c.syntax 'unpublish POST_PATH'
+          c.description 'Moves a post back into the _drafts directory'
+
+          c.action do |args, options|
+            process(args, options)
+          end
+        end
+      end
+
+      def self.process(args = [], options = {})
+        params = UnpublishArgParser.new args, options
+        params.validate!
+
+        movement = PostMovementInfo.new params
+
+        mover = PostMover.new movement
+        mover.move
+      end
+
+    end
+
+    class UnpublishArgParser
+      attr_reader :args, :options
+      def initialize(args, options)
+        @args = args
+        @options = options
+      end
+
+      def validate!
+        raise ArgumentError.new('You must specify a post path.') if args.empty?
+      end
+
+      def post_path
+        args.join ' '
+      end
+
+      def post_name
+        File.basename(post_path).sub /\d{4}-\d{2}-\d{2}-/, ''
+      end
+    end
+
+    class PostMovementInfo
+      attr_reader :params
+      def initialize(params)
+        @params = params
+      end
+
+      def from
+        params.post_path
+      end
+
+      def to
+        "_drafts/#{params.post_name}"
+      end
+    end
+
+    class PostMover
+      attr_reader :movement
+      def initialize(movement)
+        @movement = movement
+      end
+
+      def move
+        validate_source
+        ensure_directory_exists
+        move_file
+      end
+
+      def validate_source
+        raise ArgumentError.new("There was no post found at '#{movement.from}'.") unless File.exist? movement.from
+      end
+
+      def ensure_directory_exists
+        Dir.mkdir("_drafts") unless Dir.exist?("_drafts")
+      end
+
+      def move_file
+        FileUtils.mv(movement.from, movement.to)
+        puts "Post #{movement.from} was moved to #{movement.to}"
+      end
+    end
+  end
+end

spec/unpublish_spec.rb

@@ -0,0 +1,60 @@
+RSpec.describe(Jekyll::Commands::Unpublish) do
+  let(:drafts_dir) { Pathname.new(source_dir('_drafts')) }
+  let(:posts_dir)  { Pathname.new(source_dir('_posts')) }
+  let(:post_name) { "a-test-post.md" }
+  let(:post_filename) { "2012-03-04-#{post_name}" }
+  let(:post_path) { posts_dir.join post_filename }
+  let(:draft_path) { drafts_dir.join post_name }
+
+  let(:args) { ["_posts/#{post_filename}"] }
+
+  before(:all) do
+    FileUtils.mkdir_p source_dir unless File.directory? source_dir
+    Dir.chdir source_dir
+  end
+
+  before(:each) do
+    FileUtils.mkdir_p drafts_dir unless File.directory? drafts_dir
+    FileUtils.mkdir_p posts_dir unless File.directory? posts_dir
+    FileUtils.touch post_path
+  end
+
+  after(:each) do
+    FileUtils.rm_r drafts_dir if File.directory? drafts_dir
+    FileUtils.rm_r posts_dir if File.directory? posts_dir
+  end
+
+  it 'moves a post back to _drafts' do
+    expect(post_path).to exist
+    expect(draft_path).not_to exist
+    capture_stdout { described_class.process(args) }
+    expect(post_path).not_to exist
+    expect(draft_path).to exist
+  end
+
+  it 'writes a helpful message on success' do
+    expect(post_path).to exist
+    output = capture_stdout { described_class.process(args) }
+    expect(output).to eql("Post _posts/#{post_filename} was moved to _drafts/#{post_name}\n")
+  end
+
+  it 'creates the drafts folder if necessary' do
+    FileUtils.rm_r drafts_dir if File.directory? drafts_dir
+    capture_stdout { described_class.process(args) }
+    expect(drafts_dir).to exist
+  end
+
+  it 'errors if there is no argument' do
+    expect(-> {
+      capture_stdout { described_class.process }
+    }).to raise_error('You must specify a post path.')
+  end
+
+  it 'errors if no file exists at given path' do
+    weird_path = '_posts/i-forgot-the-date.md'
+    expect(-> {
+      capture_stdout { described_class.process [weird_path] }
+    }).to raise_error("There was no post found at '#{weird_path}'.")
+  end
+
+end