141 lines
4.2 KiB
Ruby
141 lines
4.2 KiB
Ruby
require 'middleman-core/application'
|
|
|
|
# CLI Module
|
|
module Middleman::Cli
|
|
# The CLI Build class
|
|
class Build < Thor::Group
|
|
include Thor::Actions
|
|
|
|
check_unknown_options!
|
|
|
|
class_option :environment,
|
|
aliases: '-e',
|
|
default: ENV['MM_ENV'] || ENV['RACK_ENV'] || :production
|
|
class_option :clean,
|
|
type: :boolean,
|
|
default: true,
|
|
desc: 'Remove orphaned files from build (--no-clean to disable)'
|
|
class_option :parallel,
|
|
type: :boolean,
|
|
default: true,
|
|
desc: 'Output files in parallel (--no-parallel to disable)'
|
|
class_option :glob,
|
|
type: :string,
|
|
aliases: '-g',
|
|
default: nil,
|
|
desc: 'Build a subset of the project'
|
|
class_option :verbose,
|
|
type: :boolean,
|
|
default: false,
|
|
desc: 'Print debug messages'
|
|
class_option :instrument,
|
|
type: :boolean,
|
|
default: false,
|
|
desc: 'Print instrument messages'
|
|
class_option :profile,
|
|
type: :boolean,
|
|
default: false,
|
|
desc: 'Generate profiling report for the build'
|
|
|
|
Middleman::Cli.import_config(self)
|
|
|
|
# Core build Thor command
|
|
# @return [void]
|
|
def build
|
|
unless ENV['MM_ROOT']
|
|
raise Thor::Error, 'Error: Could not find a Middleman project config, perhaps you are in the wrong folder?'
|
|
end
|
|
|
|
require 'middleman-core'
|
|
require 'middleman-core/logger'
|
|
require 'middleman-core/builder'
|
|
require 'fileutils'
|
|
|
|
verbose = options['verbose'] ? 0 : 1
|
|
instrument = options['instrument']
|
|
|
|
builder = nil
|
|
cli_options = options
|
|
|
|
::Middleman::Logger.singleton(verbose, instrument)
|
|
|
|
::Middleman::Util.instrument 'builder.setup' do
|
|
@app = ::Middleman::Application.new do
|
|
config[:mode] = :build
|
|
config[:show_exceptions] = false
|
|
config[:cli_options] = cli_options.each_with_object({}) do |(k, v), sum|
|
|
sum[k] = v
|
|
end
|
|
end
|
|
|
|
builder = Middleman::Builder.new(@app,
|
|
glob: options['glob'],
|
|
clean: options['clean'],
|
|
parallel: options['parallel'])
|
|
builder.thor = self
|
|
builder.on_build_event(&method(:on_event))
|
|
end
|
|
|
|
::Middleman::Util.instrument 'builder.run' do
|
|
if builder.run!
|
|
clean_directories! if options['clean']
|
|
shell.say 'Project built successfully.'
|
|
else
|
|
msg = 'There were errors during this build'
|
|
unless options['verbose']
|
|
msg << ', re-run with `middleman build --verbose` to see the full exception.'
|
|
end
|
|
shell.say msg, :red
|
|
|
|
exit(1)
|
|
end
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
# Handles incoming events from the builder.
|
|
# @param [Symbol] event_type The type of event.
|
|
# @param [String] contents The event contents.
|
|
# @param [String] extra The extra information.
|
|
# @return [void]
|
|
def on_event(event_type, target, extra=nil)
|
|
case event_type
|
|
when :error
|
|
say_status :error, target, :red
|
|
shell.say extra, :red if options['verbose']
|
|
when :deleted
|
|
say_status :remove, target, :green
|
|
when :created
|
|
say_status :create, target, :green
|
|
when :identical
|
|
say_status :identical, target, :blue
|
|
when :updated
|
|
say_status :updated, target, :yellow
|
|
else
|
|
say_status event_type, extra, :blue
|
|
end
|
|
end
|
|
|
|
# Find empty directories in the build folder and remove them.
|
|
# @return [Boolean]
|
|
def clean_directories!
|
|
all_build_files = File.join(@app.config[:build_dir], '**', '*')
|
|
|
|
empty_directories = Dir[all_build_files].select do |d|
|
|
File.directory?(d)
|
|
end
|
|
|
|
empty_directories.each do |d|
|
|
remove_file d, force: true if Pathname(d).children.empty?
|
|
end
|
|
end
|
|
|
|
# Add to CLI
|
|
Base.register(self, 'build', 'build [options]', 'Builds the static site for deployment')
|
|
|
|
# Map "b" to "build"
|
|
Base.map('b' => 'build')
|
|
end
|
|
end
|