1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/actionpack/lib/action_dispatch/middleware/static.rb

68 lines
1.6 KiB
Ruby
Raw Normal View History

2009-09-26 12:37:42 -04:00
require 'rack/utils'
require 'active_support/core_ext/uri'
2009-09-26 12:37:42 -04:00
module ActionDispatch
2010-09-03 06:48:21 -04:00
class FileHandler
def initialize(root, cache_control)
@root = root.chomp('/')
@compiled_root = /^#{Regexp.escape(root)}/
headers = cache_control && { 'Cache-Control' => cache_control }
@file_server = ::Rack::File.new(@root, headers)
2010-09-03 06:48:21 -04:00
end
2010-09-03 06:48:21 -04:00
def match?(path)
path = path.dup
full_path = path.empty? ? @root : File.join(@root, escape_glob_chars(unescape_path(path)))
paths = "#{full_path}#{ext}"
matches = Dir[paths]
match = matches.detect { |m| File.file?(m) }
if match
match.sub!(@compiled_root, '')
::Rack::Utils.escape(match)
2010-09-03 06:48:21 -04:00
end
end
2010-09-03 06:48:21 -04:00
def call(env)
@file_server.call(env)
end
def ext
@ext ||= begin
ext = ::ActionController::Base.default_static_extension
"{,#{ext},/index#{ext}}"
end
end
def unescape_path(path)
URI.parser.unescape(path)
end
def escape_glob_chars(path)
path.force_encoding('binary') if path.respond_to? :force_encoding
2012-02-17 09:44:28 -05:00
path.gsub(/[*?{}\[\]]/, "\\\\\\&")
end
2010-09-03 06:48:21 -04:00
end
2010-09-03 06:48:21 -04:00
class Static
def initialize(app, path, cache_control=nil)
2009-09-26 12:37:42 -04:00
@app = app
@file_handler = FileHandler.new(path, cache_control)
2009-09-26 12:37:42 -04:00
end
def call(env)
case env['REQUEST_METHOD']
when 'GET', 'HEAD'
path = env['PATH_INFO'].chomp('/')
if match = @file_handler.match?(path)
env["PATH_INFO"] = match
return @file_handler.call(env)
2009-09-26 12:37:42 -04:00
end
end
@app.call(env)
2009-09-26 12:37:42 -04:00
end
end
end