module Middleman::Features::DirectoryIndexes

def registered(app)
  app.set :ignored_directory_indexes, []
  app.extend ClassMethods
  
  app.build_reroute do |destination, request_path|
    index_ext = File.extname(app.settings.index_file)
    new_index_path = "/#{app.settings.index_file}"
  
    indexed_path = request_path.gsub(/\/$/, "") + index_ext
    
    if app.settings.ignored_directory_indexes.include?(request_path)
      false
    elsif request_path =~ /#{new_index_path}$/
      false
    else
      [
        destination.gsub(/#{index_ext.gsub(".", "\\.")}$/, new_index_path),
        request_path
      ]
    end
  end
  
  app.before do
    indexed_path = request.path_info.gsub(/\/$/, "") + "/" + app.settings.index_file
    indexed_exists = resolve_template(indexed_path, :raise_exceptions => false)
    extensioned_path = request.path_info.gsub(/\/$/, "") + File.extname(app.settings.index_file)
    is_ingored = settings.ignored_directory_indexes.include?(extensioned_path)
    
    if !indexed_exists && !is_ingored
      parts = request.path_info.split("/")
      last_part = parts.last
      last_part_ext = File.extname(last_part)
    
      if last_part_ext.blank?
        # This is a folder, redirect to index
        request.path_info = extensioned_path
      end
    end
  end
end