middleman--middleman/lib/middleman.rb

require 'rubygems'
require 'haml'
require 'compass' #must be loaded before sinatra
require 'sinatra/base'

# Include markaby support
require File.join(File.dirname(__FILE__), '..', 'vendor', 'sinatra-markaby', 'lib', 'sinatra', 'markaby')

# Include maruku support
require File.join(File.dirname(__FILE__), '..', 'vendor', 'sinatra-maruku', 'lib', 'sinatra', 'maruku')

class Middleman < Sinatra::Base
  set :app_file, __FILE__
  helpers Sinatra::Markaby
  helpers Sinatra::Maruku
  
  def self.run!(options={}, &block)
    set options
    handler      = detect_rack_handler
    handler_name = handler.name.gsub(/.*::/, '')
    puts "== The Middleman is standing watch on port #{port}"
    handler.run self, :Host => host, :Port => port do |server|
      trap(:INT) do
        ## Use thins' hard #stop! if available, otherwise just #stop
        server.respond_to?(:stop!) ? server.stop! : server.stop
        puts "\n== The Middleman has ended his patrol"
      end
      
      if block_given?
        block.call
        ## Use thins' hard #stop! if available, otherwise just #stop
        server.respond_to?(:stop!) ? server.stop! : server.stop
      end
    end
  rescue Errno::EADDRINUSE => e
    puts "== The Middleman is already standing watch on port #{port}!"
  end
  
  configure do
    Compass.configuration do |config|
      config.project_path     = Dir.pwd
      config.sass_dir         = File.join(File.expand_path(self.views), "stylesheets")
      config.output_style     = :nested
      config.images_dir       = File.join(File.expand_path(self.public), "images")
      config.http_images_path = "/images/"
    end
  end
  
  get /(.*)/ do |path|
    path << "index.html" if path.match(%r{/$})
    path.gsub!(%r{^/}, '')
    
    template = path.gsub(File.extname(path), '').to_sym
    if path.match /.html$/
      if File.exists? File.join(options.views, "#{template}.haml")
        haml(template)
      elsif File.exists? File.join(options.views, "#{template}.maruku")
        maruku(template)
      elsif File.exists? File.join(options.views, "#{template}.mab")
        markaby(template)
      else
        pass
      end
    elsif path.match /.css$/
      content_type 'text/css', :charset => 'utf-8'
      sass(template, Compass.sass_engine_options)
    else
      pass
    end
  end
end