2009-09-26 12:37:42 -04:00
|
|
|
require 'rack/utils'
|
|
|
|
|
|
|
|
module ActionDispatch
|
|
|
|
class Static
|
2010-07-29 08:12:25 -04:00
|
|
|
class FileHandler
|
|
|
|
def initialize(at, root)
|
|
|
|
@at = at.chomp("/")
|
|
|
|
@file_server = ::Rack::File.new(root)
|
|
|
|
end
|
|
|
|
|
|
|
|
def file_exist?(path)
|
|
|
|
(path = full_readable_path(path)) && File.file?(path)
|
|
|
|
end
|
|
|
|
|
|
|
|
def directory_exist?(path)
|
|
|
|
(path = full_readable_path(path)) && File.directory?(path)
|
|
|
|
end
|
|
|
|
|
|
|
|
def call(env)
|
|
|
|
env["PATH_INFO"].gsub!(/^#{@at}/, "")
|
|
|
|
@file_server.call(env)
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
def includes_path?(path)
|
|
|
|
@at == "" || path =~ /^#{@at}/
|
|
|
|
end
|
|
|
|
|
|
|
|
def full_readable_path(path)
|
|
|
|
return unless includes_path?(path)
|
|
|
|
path = path.gsub(/^#{@at}/, "")
|
|
|
|
File.join(@file_server.root, ::Rack::Utils.unescape(path))
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-09-26 12:37:42 -04:00
|
|
|
FILE_METHODS = %w(GET HEAD).freeze
|
|
|
|
|
2010-07-29 08:12:25 -04:00
|
|
|
def initialize(app, roots)
|
2009-09-26 12:37:42 -04:00
|
|
|
@app = app
|
2010-07-29 08:12:25 -04:00
|
|
|
roots = normalize_roots(roots)
|
|
|
|
@file_handlers = file_handlers(roots)
|
2009-09-26 12:37:42 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def call(env)
|
|
|
|
path = env['PATH_INFO'].chomp('/')
|
|
|
|
method = env['REQUEST_METHOD']
|
|
|
|
|
|
|
|
if FILE_METHODS.include?(method)
|
2010-07-29 08:12:25 -04:00
|
|
|
if file_handler = file_exist?(path)
|
|
|
|
return file_handler.call(env)
|
2009-09-26 12:37:42 -04:00
|
|
|
else
|
|
|
|
cached_path = directory_exist?(path) ? "#{path}/index" : path
|
|
|
|
cached_path += ::ActionController::Base.page_cache_extension
|
|
|
|
|
2010-07-29 08:12:25 -04:00
|
|
|
if file_handler = file_exist?(cached_path)
|
2009-09-26 12:37:42 -04:00
|
|
|
env['PATH_INFO'] = cached_path
|
2010-07-29 08:12:25 -04:00
|
|
|
return file_handler.call(env)
|
2009-09-26 12:37:42 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
@app.call(env)
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
def file_exist?(path)
|
2010-07-29 08:12:25 -04:00
|
|
|
@file_handlers.detect { |f| f.file_exist?(path) }
|
2009-09-26 12:37:42 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def directory_exist?(path)
|
2010-07-29 08:12:25 -04:00
|
|
|
@file_handlers.detect { |f| f.directory_exist?(path) }
|
|
|
|
end
|
|
|
|
|
|
|
|
def normalize_roots(roots)
|
|
|
|
roots.is_a?(Hash) ? roots : { "/" => roots.chomp("/") }
|
|
|
|
end
|
|
|
|
|
|
|
|
def file_handlers(roots)
|
|
|
|
roots.map do |at, root|
|
|
|
|
FileHandler.new(at, root)
|
|
|
|
end
|
2009-09-26 12:37:42 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|