blob: c2d686f5146d6c9407d2bb8d55c4021556995cf6 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
require 'rack/utils'
module ActionDispatch
class Static
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
FILE_METHODS = %w(GET HEAD).freeze
def initialize(app, roots)
@app = app
roots = normalize_roots(roots)
@file_handlers = file_handlers(roots)
end
def call(env)
path = env['PATH_INFO'].chomp('/')
method = env['REQUEST_METHOD']
if FILE_METHODS.include?(method)
if file_handler = file_exist?(path)
return file_handler.call(env)
else
cached_path = directory_exist?(path) ? "#{path}/index" : path
cached_path += ::ActionController::Base.page_cache_extension
if file_handler = file_exist?(cached_path)
env['PATH_INFO'] = cached_path
return file_handler.call(env)
end
end
end
@app.call(env)
end
private
def file_exist?(path)
@file_handlers.detect { |f| f.file_exist?(path) }
end
def directory_exist?(path)
@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
end
end
end
|