# CLI Module
module Middleman::Cli
  # A initializing Bundler
  class Bundle < Thor
    include Thor::Actions
    check_unknown_options!

    namespace :bundle

    desc 'bundle', 'Setup initial bundle', hide: true

    # The setup task
    def bundle
      run('bundle install') # , :capture => true)
    end
  end

  # A upgrading Bundler
  class Upgrade < Thor
    include Thor::Actions
    check_unknown_options!

    namespace :upgrade

    desc 'upgrade', 'Upgrade installed bundle'

    # The upgrade task
    def upgrade
      inside(ENV['MM_ROOT']) do
        run('bundle update') # , :capture => true)
      end
    end
  end

  # Map "u" to "upgrade"
  Base.map(
             'u' => 'upgrade'
  )
end