forked from ruby-grape/grape
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathparameters.rb
204 lines (185 loc) · 8 KB
/
parameters.rb
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
require 'active_support/concern'
module Grape
module DSL
# Defines DSL methods, meant to be applied to a ParamsScope, which define
# and describe the parameters accepted by an endpoint, or all endpoints
# within a namespace.
module Parameters
extend ActiveSupport::Concern
# Include reusable params rules among current.
# You can define reusable params with helpers method.
#
# @example
#
# class API < Grape::API
# helpers do
# params :pagination do
# optional :page, type: Integer
# optional :per_page, type: Integer
# end
# end
#
# desc "Get collection"
# params do
# use :pagination
# end
# get do
# Collection.page(params[:page]).per(params[:per_page])
# end
# end
def use(*names)
named_params = @api.namespace_stackable_with_hash(:named_params) || {}
options = names.extract_options!
names.each do |name|
params_block = named_params.fetch(name) do
raise "Params :#{name} not found!"
end
instance_exec(options, ¶ms_block)
end
end
alias use_scope use
alias includes use
# Require one or more parameters for the current endpoint.
#
# @param attrs list of parameter names, or, if :using is
# passed as an option, which keys to include (:all or :none) from
# the :using hash. The last key can be a hash, which specifies
# options for the parameters
# @option attrs :type [Class] the type to coerce this parameter to before
# passing it to the endpoint. See {Grape::Validations::Types} for a list of
# types that are supported automatically. Custom classes may be used
# where they define a class-level `::parse` method, or in conjunction
# with the `:coerce_with` parameter. `JSON` may be supplied to denote
# `JSON`-formatted objects or arrays of objects. `Array[JSON]` accepts
# the same values as `JSON` but will wrap single objects in an `Array`.
# @option attrs :types [Array<Class>] may be supplied in place of +:type+
# to declare an attribute that has multiple allowed types. See
# {Validations::Types::MultipleTypeCoercer} for more details on coercion
# and validation rules for variant-type parameters.
# @option attrs :desc [String] description to document this parameter
# @option attrs :default [Object] default value, if parameter is optional
# @option attrs :values [Array] permissable values for this field. If any
# other value is given, it will be handled as a validation error
# @option attrs :using [Hash[Symbol => Hash]] a hash defining keys and
# options, like that returned by {Grape::Entity#documentation}. The value
# of each key is an options hash accepting the same parameters
# @option attrs :except [Array[Symbol]] a list of keys to exclude from
# the :using Hash. The meaning of this depends on if :all or :none was
# passed; :all + :except will make the :except fields optional, whereas
# :none + :except will make the :except fields required
# @option attrs :coerce_with [#parse, #call] method to be used when coercing
# the parameter to the type named by `attrs[:type]`. Any class or object
# that defines `::parse` or `::call` may be used.
#
# @example
#
# params do
# # Basic usage: require a parameter of a certain type
# requires :user_id, type: Integer
#
# # You don't need to specify type; String is default
# requires :foo
#
# # Multiple params can be specified at once if they share
# # the same options.
# requires :x, :y, :z, type: Date
#
# # Nested parameters can be handled as hashes. You must
# # pass in a block, within which you can use any of the
# # parameters DSL methods.
# requires :user, type: Hash do
# requires :name, type: String
# end
# end
def requires(*attrs, &block)
orig_attrs = attrs.clone
opts = attrs.extract_options!.clone
opts[:presence] = { value: true, message: opts[:message] }
if opts[:using]
require_required_and_optional_fields(attrs.first, opts)
else
validate_attributes(attrs, opts, &block)
block_given? ? new_scope(orig_attrs, &block) : push_declared_params(attrs)
end
end
# Allow, but don't require, one or more parameters for the current
# endpoint.
# @param (see #requires)
# @option (see #requires)
def optional(*attrs, &block)
orig_attrs = attrs.clone
opts = attrs.extract_options!.clone
type = opts[:type]
# check type for optional parameter group
if attrs && block_given?
raise Grape::Exceptions::MissingGroupTypeError.new if type.nil?
raise Grape::Exceptions::UnsupportedGroupTypeError.new unless Grape::Validations::Types.group?(type)
end
if opts[:using]
require_optional_fields(attrs.first, opts)
else
validate_attributes(attrs, opts, &block)
block_given? ? new_scope(orig_attrs, true, &block) : push_declared_params(attrs)
end
end
# Disallow the given parameters to be present in the same request.
# @param attrs [*Symbol] parameters to validate
def mutually_exclusive(*attrs)
validates(attrs, mutual_exclusion: { value: true, message: extract_message_option(attrs) })
end
# Require exactly one of the given parameters to be present.
# @param (see #mutually_exclusive)
def exactly_one_of(*attrs)
validates(attrs, exactly_one_of: { value: true, message: extract_message_option(attrs) })
end
# Require at least one of the given parameters to be present.
# @param (see #mutually_exclusive)
def at_least_one_of(*attrs)
validates(attrs, at_least_one_of: { value: true, message: extract_message_option(attrs) })
end
# Require that either all given params are present, or none are.
# @param (see #mutually_exclusive)
def all_or_none_of(*attrs)
validates(attrs, all_or_none_of: { value: true, message: extract_message_option(attrs) })
end
# Define a block of validations which should be applied if and only if
# the given parameter is present. The parameters are not nested.
# @param attr [Symbol] the parameter which, if present, triggers the
# validations
# @raise Grape::Exceptions::UnknownParameter if `attr` has not been
# defined in this scope yet
# @yield a parameter definition DSL
def given(*attrs, &block)
attrs.each do |attr|
raise Grape::Exceptions::UnknownParameter.new(attr) unless declared_param?(attr)
end
new_lateral_scope(dependent_on: attrs, &block)
end
# Test for whether a certain parameter has been defined in this params
# block yet.
# @return [Boolean] whether the parameter has been defined
def declared_param?(param)
# @declared_params also includes hashes of options and such, but those
# won't be flattened out.
@declared_params.flatten.include?(param)
end
alias group requires
# @param params [Hash] initial hash of parameters
# @return hash of parameters relevant for the current scope
# @api private
def params(params)
params = @parent.params(params) if @parent
if @element
params = if params.is_a?(Array)
params.flat_map { |el| el[@element] || {} }
elsif params.is_a?(Hash)
params[@element] || {}
else
{}
end
end
params
end
end
end
end