51 lines
1.6 KiB
Ruby
51 lines
1.6 KiB
Ruby
# typed: strict
|
|
# frozen_string_literal: true
|
|
|
|
class Hash
|
|
# Returns a new hash with all values converted by the block operation.
|
|
# This includes the values from the root hash and from all
|
|
# nested hashes and arrays.
|
|
#
|
|
# ### Example
|
|
#
|
|
# ```ruby
|
|
# hash = { person: { name: 'Rob', age: '28' } }
|
|
#
|
|
# hash.deep_transform_values{ |value| value.to_s.upcase }
|
|
# # => {person: {name: "ROB", age: "28"}}
|
|
# ```
|
|
def deep_transform_values(&block) = _deep_transform_values_in_object(self, &block)
|
|
|
|
# Destructively converts all values by using the block operation.
|
|
# This includes the values from the root hash and from all
|
|
# nested hashes and arrays.
|
|
def deep_transform_values!(&block) = _deep_transform_values_in_object!(self, &block)
|
|
|
|
private
|
|
|
|
# Support methods for deep transforming nested hashes and arrays.
|
|
sig { params(object: T.anything, block: T.proc.params(v: T.untyped).returns(T.untyped)).returns(T.untyped) }
|
|
def _deep_transform_values_in_object(object, &block)
|
|
case object
|
|
when Hash
|
|
object.transform_values { |value| _deep_transform_values_in_object(value, &block) }
|
|
when Array
|
|
object.map { |e| _deep_transform_values_in_object(e, &block) }
|
|
else
|
|
yield(object)
|
|
end
|
|
end
|
|
|
|
sig { params(object: T.anything, block: T.proc.params(v: T.untyped).returns(T.untyped)).returns(T.untyped) }
|
|
def _deep_transform_values_in_object!(object, &block)
|
|
case object
|
|
when Hash
|
|
object.transform_values! { |value| _deep_transform_values_in_object!(value, &block) }
|
|
when Array
|
|
object.map! { |e| _deep_transform_values_in_object!(e, &block) }
|
|
else
|
|
yield(object)
|
|
end
|
|
end
|
|
end
|