middleman/bin/middleman

73 lines
No EOL
1.9 KiB
Ruby
Executable file

#!/usr/bin/env ruby
libdir = File.join(File.dirname(File.dirname(__FILE__)), "lib")
$LOAD_PATH.unshift(libdir) unless $LOAD_PATH.include?(libdir)
require 'pathname'
require 'rubygems'
module Middleman
module ProjectLocator
def self.locate_middleman_root!(args)
cwd = Dir.pwd
if !in_middleman_project? && !in_middleman_project_subdirectory?
$stderr.puts "== Error: Could not find a Middleman project config, perhaps you are in the wrong folder?"
return
end
if in_middleman_project?
did_locate_middleman_project(cwd, args)
return
end
Dir.chdir("..") do
# Recurse in a chdir block: if the search fails we want to be sure
# the application is generated in the original working directory.
locate_middleman_root!(args) unless cwd == Dir.pwd
end
rescue SystemCallError
# could not chdir, no problem just return
end
def self.in_middleman_project?
File.exists?('config.rb')
end
def self.in_middleman_project_subdirectory?(path = Pathname.new(Dir.pwd))
File.exists?(File.join(path, 'config.rb')) || !path.root? && in_middleman_project_subdirectory?(path.parent)
end
def self.did_locate_middleman_project(path, args)
# Set up gems listed in the Gemfile.
ENV['BUNDLE_GEMFILE'] ||= File.expand_path('Gemfile', path)
require 'bundler/setup' if File.exists?(ENV['BUNDLE_GEMFILE'])
start_cli!(args)
end
def self.start_cli!(args)
require 'middleman'
Middleman::CLI.start(args)
end
end
end
args = ARGV.dup
ARG_ALIASES = {
"s" => "server",
"b" => "build",
"i" => "init"
}
if ARG_ALIASES.has_key?(args[0])
args[0] = ARG_ALIASES[args[0]]
end
if args.length < 1 || %w(server build migrate).include?(args.first)
Middleman::ProjectLocator.locate_middleman_root!(args)
else
Middleman::ProjectLocator.start_cli!(args)
end