class Middleman::Directory

def handle_directory(lookup, &block)
  lookup = File.join(lookup, '*')
  
  results = Dir[lookup].sort do |a, b|
    simple_a = a.gsub(SHARED_SERVER.root + "/", '').gsub(SHARED_SERVER.views + "/", '') 
    simple_b = b.gsub(SHARED_SERVER.root + "/", '').gsub(SHARED_SERVER.views + "/", '')
    
    a_dir = simple_a.split("/").first
    b_dir = simple_b.split("/").first
    
    if a_dir == SHARED_SERVER.images_dir
      -1
    elsif b_dir == SHARED_SERVER.images_dir
      1
    else
      0
    end
  end
  
  results = results.select(&block) if block_given?
  
  results.each do |file_source|
    if File.directory?(file_source)
      handle_directory(file_source)
      next
    end
    
    # Skip partials prefixed with an underscore
    next unless file_source.gsub(SHARED_SERVER.root, '').split('/').select { |p| p[0,1] == '_' }.empty?
    
    file_extension = File.extname(file_source)
    file_destination = File.join(given_destination, file_source.gsub(source, '.'))
    file_destination.gsub!('/./', '/')
    
    handled_by_tilt = ::Tilt.mappings.has_key?(file_extension.gsub(/^\./, ""))
    if handled_by_tilt
      file_destination.gsub!(file_extension, "")
    end
    
    destination = base.tilt_template(file_source, file_destination, config, &@block)
  end
end