middleman--middleman/lib/middleman/core_extensions/routing.rb

module Middleman::CoreExtensions::Routing
  class << self
    def registered(app)
      app.extend ClassMethods
      
      app.set :proxied_paths, {}
      
      # Normalize the path and add index if we're looking at a directory
      app.before_processing do
        request.path_info = self.class.path_to_index(request.path)
        true
      end
    end
    alias :included :registered
  end
  
  module ClassMethods    
    def path_to_index(path)
      parts = path ? path.split('/') : []
      if parts.last.nil? || parts.last.split('.').length == 1
        path = File.join(path, settings.index_file) 
      end
      path.gsub(%r{^/}, '')
    end
  
    # Takes a block which allows many pages to have the same layout
    # with_layout :admin do
    #   page "/admin/"
    #   page "/admin/login.html"
    # end
    def with_layout(layout_name, &block)
      old_layout = settings.layout
    
      set :layout, layout_name
      class_eval(&block) if block_given?
    ensure
      set :layout, old_layout
    end
  
    def paths_for_url(url)
      url = url.gsub(%r{#{settings.index_file}$}, "")
      url = url.gsub(%r{(\/)$}, "") if url.length > 1
    
      paths = [url]
      paths << "#{url}/" if url.length > 1 && url.split("/").last.split('.').length <= 1
      paths << "/#{path_to_index(url)}"
      paths
    end
  
    # The page method allows the layout to be set on a specific path
    # page "/about.html", :layout => false
    # page "/", :layout => :homepage_layout
    def page(url, options={}, &block)
      has_block = block_given?
      options[:layout] = settings.layout if options[:layout].nil?
      
      if options.has_key?(:proxy)
        settings.proxied_paths[url] = options[:proxy]
      end
    
      paths_for_url(url).each do |p|
        get(p) do
          if settings.proxied_paths.has_key?(url)
            request.path_info = settings.proxied_paths[url]
          end
          
          instance_eval(&block) if has_block
          process_request(options)
        end
      end
    end
  end
end