middleman-deploy/lib/middleman-deploy/commands.rb

require "middleman-core/cli"

require "middleman-deploy/extension"
require "middleman-deploy/methods"
require "middleman-deploy/strategies"
require "middleman-deploy/pkg-info"

module Middleman
  module Cli

    # This class provides a "deploy" command for the middleman CLI.
    class Deploy < Thor
      include Thor::Actions

      check_unknown_options!

      namespace :deploy

      # Tell Thor to exit with a nonzero exit code on failure
      def self.exit_on_failure?
        true
      end

      desc "deploy [options]", Middleman::Deploy::TAGLINE
      method_option "build_before",
        :type     => :boolean,
        :aliases  => "-b",
        :desc     => "Run `middleman build` before the deploy step"
      def deploy
        build_before(options)
        process
      end

      protected

      def build_before(options={})
        build_enabled = options.fetch('build_before', self.deploy_options.build_before)

        if build_enabled
          # http://forum.middlemanapp.com/t/problem-with-the-build-task-in-an-extension
          run('middleman build') || exit(1)
        end
      end

      def print_usage_and_die(message)
        usage_path    = File.join(File.dirname(__FILE__), '..', '..', 'USAGE')
        usage_message = File.read(usage_path)

        raise Error, "ERROR: #{message}\n#{usage_message}"
      end

      def process
        server_instance   = ::Middleman::Application.server.inst

        camelized_method  = self.deploy_options.method.to_s.split('_').map { |word| word.capitalize}.join
        method_class_name = "Middleman::Deploy::Methods::#{camelized_method}"
        method_instance   = method_class_name.constantize.new(server_instance, self.deploy_options)

        method_instance.process
      end

      def deploy_options
        options = nil

        begin
          options = ::Middleman::Application.server.inst.options
        rescue NoMethodError
          print_usage_and_die "You need to activate the deploy extension in config.rb."
        end

        unless options.method
          print_usage_and_die "The deploy extension requires you to set a method."
        end

        case options.method
        when :rsync, :sftp
          unless options.host && options.path
            print_usage_and_die "The #{options.method} method requires host and path to be set."
          end
        when :ftp
          unless options.host && options.user && options.password && options.path
            print_usage_and_die "The ftp deploy method requires host, path, user, and password to be set."
          end
        end

        options
      end
    end

    # Alias "d" to "deploy"
    Base.map({ "d" => "deploy" })
  end
end