B0RG/borg

module Borg
  class CLI < Capistrano::CLI
    module Assimilator
      def self.included(base) #:nodoc:
        base.send :alias_method, :execute_requested_actions_without_assimilator, :execute_requested_actions
        base.send :alias_method, :execute_requested_actions, :execute_requested_actions_with_assimilator
      end

      def execute_requested_actions_with_assimilator(config)
        config.assimilate!
        execute_requested_actions_without_assimilator config
      end
    end
  end
end