middleman/bin/mm-server

#!/usr/bin/env ruby

require 'optparse'

# Require Middleman
require File.join(File.dirname(__FILE__), '..', 'lib', 'middleman')

env = ENV['MM_ENV'] || ENV['RACK_ENV'] || 'development'
options = { :Port => 4567, :AccessLog => [] }

# TODO: Switch to Thor
OptionParser.new { |opts|
  opts.banner = "Usage: mm-server [rack options]"
  opts.separator ""
  opts.separator "Rack options:"
  opts.on("-p", "--port PORT", "use PORT (default: 4567)") { |port|
    options[:Port] = port
  }
  opts.on("-E", "--env ENVIRONMENT", "use ENVIRONMENT for defaults (default: development)") { |e|
    env = e
  }
  opts.on("--debug", "Debug mode") {
    ::Middleman::Server.set :logging, true
  }

  opts.parse! ARGV
}

ENV['RACK_ENV'] = env

@current_path = Dir.pwd
@path_parts = @current_path.split("/")
@found_root = false

while (!@found_root && (@path_parts.length > 0))
  @current_path = File.join(*@path_parts)
  
  source_folder = File.join(@current_path, "source")
  
  if File.exists?(source_folder)
    @found_root = true
    next
  end
  
  @path_parts.pop
end

if !@found_root
  $stderr.puts "== Error: Could not find a Middleman project structure, perhaps you are in the wrong folder?"
  exit
end

# If the old init.rb exists, issue warning
old_config = File.join(@current_path, "init.rb")
if File.exists? old_config
  $stderr.puts "== Error: The init.rb file (deprecated) needs to be be renamed to config.rb"
  exit
end

# If the old directories exists, use it, but issue warning
old_views = File.join(@current_path, "views")
old_public = File.join(@current_path, "public")
if File.exists?(old_views) || File.exists?(old_public)
  $stderr.puts "== Error: The views and public folders are have been combined. Create a new 'source' folder, add the contents of views and public to it and then remove the empty views and public folders."
  exit
end

Middleman::Server.root = @current_path
app = Middleman::Server.new

# TODO: Remove this
require 'rubygems'

unless defined?(JRUBY_VERSION)
  require 'thin'
  handler = Rack::Handler::Thin
else
  require 'kirk'
  require 'rack/handler/kirk'
  handler = Rack::Handler::Kirk
end

handler.run app, options do |inst|
  puts "== The Middleman is standing watch on port #{options[:Port]}"
end