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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
|
require 'rack/mount/utils'
module Rack::Mount
class GeneratableRegexp < Regexp #:nodoc:
class DynamicSegment #:nodoc:
attr_reader :name, :requirement
def initialize(name, requirement)
@name, @requirement = name.to_sym, requirement
freeze
end
def ==(obj)
@name == obj.name && @requirement == obj.requirement
end
def =~(str)
@requirement =~ str
end
def to_hash
{ @name => @requirement }
end
def inspect
"/(?<#{@name}>#{@requirement.source})/"
end
end
module InstanceMethods
def self.extended(obj)
obj.segments
end
def defaults=(defaults)
@required_captures = nil
@required_params = nil
@required_defaults = nil
@defaults = defaults
end
def defaults
@defaults ||= {}
end
def generatable?
segments.any?
end
def generate(params = {}, recall = {}, options = {})
return nil unless generatable?
merged = recall.merge(params)
return nil unless required_params.all? { |p| merged.include?(p) }
return nil unless required_defaults.all? { |k, v| merged[k] == v }
generate_from_segments(segments, params, merged, options)
end
def segments
@segments ||= begin
defaults
segments = []
catch(:halt) do
expression = Utils.parse_regexp(self)
segments = parse_segments(expression)
end
segments
end
end
def captures
segments.flatten.find_all { |s| s.is_a?(DynamicSegment) }
end
def required_captures
@required_captures ||= segments.find_all { |s|
s.is_a?(DynamicSegment) && !@defaults.include?(s.name)
}.freeze
end
def required_params
@required_params ||= required_captures.map { |s| s.name }.freeze
end
def required_defaults
@required_defaults ||= begin
required_defaults = @defaults.dup
captures.inject({}) { |h, s| h.merge!(s.to_hash) }.keys.each { |name|
required_defaults.delete(name)
}
required_defaults
end
end
def freeze
segments
captures
required_captures
required_params
required_defaults
super
end
private
def parse_segments(segments)
s = []
segments.each_with_index do |part, index|
case part
when Regin::Anchor
# ignore
when Regin::Character
throw :halt unless part.literal?
if s.last.is_a?(String)
s.last << part.value.dup
else
s << part.value.dup
end
when Regin::Group
if part.name
s << DynamicSegment.new(part.name, part.expression.to_regexp(true))
else
s << parse_segments(part.expression)
end
when Regin::Expression
return parse_segments(part)
else
throw :halt
end
end
s
end
EMPTY_STRING = ''.freeze
def generate_from_segments(segments, params, merged, options, optional = false)
if optional
return EMPTY_STRING if segments.all? { |s| s.is_a?(String) }
return EMPTY_STRING unless segments.flatten.any? { |s|
params.has_key?(s.name) if s.is_a?(DynamicSegment)
}
return EMPTY_STRING if segments.any? { |segment|
if segment.is_a?(DynamicSegment)
value = merged[segment.name] || @defaults[segment.name]
value = parameterize(segment.name, value, options)
merged_value = parameterize(segment.name, merged[segment.name], options)
default_value = parameterize(segment.name, @defaults[segment.name], options)
if value.nil? || segment !~ value
true
elsif merged_value == default_value
# Nasty control flow
return :clear_remaining_segments
else
false
end
end
}
end
generated = segments.map do |segment|
case segment
when String
segment
when DynamicSegment
value = params[segment.name] || merged[segment.name] || @defaults[segment.name]
value = parameterize(segment.name, value, options)
if value && segment =~ value.to_s
value
else
return
end
when Array
value = generate_from_segments(segment, params, merged, options, true)
if value == :clear_remaining_segments
segment.each { |s| params.delete(s.name) if s.is_a?(DynamicSegment) }
EMPTY_STRING
elsif value.nil?
EMPTY_STRING
else
value
end
end
end
# Delete any used items from the params
segments.each { |s| params.delete(s.name) if s.is_a?(DynamicSegment) }
generated.join
end
def parameterize(name, value, options)
if block = options[:parameterize]
block.call(name, value)
else
value
end
end
end
include InstanceMethods
def initialize(regexp)
super
segments
end
end
end
|