middleman--middleman/middleman-core/lib/middleman-core/cli.rb

# Require thor since that's what the who CLI is built around
require 'thor'
require "thor/group"

# CLI Module
module Middleman::Cli
  
  module Cli
    autoload :Build,        "middleman-core/cli/build"
    autoload :Init,         "middleman-core/cli/init"
    autoload :Server,       "middleman-core/cli/server"
  end
  
  # The base task from which everything else etends
  class Base < Thor
    
    desc "version", "Show version"
    def version
      require 'middleman-core/version'
      say "Middleman #{Middleman::VERSION}"
    end
    
    # Override the Thor help method to find help for subtasks
    # @param [Symbol, String, nil] meth
    # @param [Boolean] subcommand
    # @return [void]
    def help(meth = nil, subcommand = false)
      if meth && !self.respond_to?(meth)
        klass, task = Thor::Util.find_class_and_task_by_namespace("#{meth}:#{meth}")
        klass.start(["-h", task].compact, :shell => self.shell)
      else
        list = []
        Thor::Util.thor_classes_in(Middleman::Cli).each do |klass|
          list += klass.printable_tasks(false)
        end
        list.sort!{ |a,b| a[0] <=> b[0] }
        
        shell.say "Tasks:"
        shell.print_table(list, :ident => 2, :truncate => true)
        shell.say
      end
    end
    
    # Intercept missing methods and search subtasks for them
    # @param [Symbol] meth
    def method_missing(meth, *args)
      meth = meth.to_s
      
      if self.class.map.has_key?(meth)
        meth = self.class.map[meth]
      end
      
      klass, task = Thor::Util.find_class_and_task_by_namespace("#{meth}:#{meth}")
      
      if klass.nil?
        tasks_dir = File.join(Dir.pwd, "tasks")

        if File.exists?(tasks_dir)
          Dir[File.join(tasks_dir, "**/*_task.rb")].each { |f| require f }
          klass, task = Thor::Util.find_class_and_task_by_namespace("#{meth}:#{meth}")
        end
      end
      
      if klass.nil?
        super
      else
        args.unshift(task) if task
        klass.start(args, :shell => self.shell)
      end
    end
  end
end

# Include the core CLI items
require "middleman-core/cli/init"
require "middleman-core/cli/server"
require "middleman-core/cli/build"