136 lines
3.8 KiB
Ruby
136 lines
3.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "active_model/attribute_set"
|
|
require "active_model/attribute/user_provided_default"
|
|
|
|
module ActiveModel
|
|
module Attributes # :nodoc:
|
|
extend ActiveSupport::Concern
|
|
include ActiveModel::AttributeMethods
|
|
|
|
included do
|
|
attribute_method_suffix "=", parameters: "value"
|
|
class_attribute :attribute_types, :_default_attributes, instance_accessor: false
|
|
self.attribute_types = Hash.new(Type.default_value)
|
|
self._default_attributes = AttributeSet.new({})
|
|
end
|
|
|
|
module ClassMethods
|
|
def attribute(name, cast_type = nil, default: NO_DEFAULT_PROVIDED, **options)
|
|
name = name.to_s
|
|
|
|
cast_type = Type.lookup(cast_type, **options) if Symbol === cast_type
|
|
cast_type ||= attribute_types[name]
|
|
|
|
self.attribute_types = attribute_types.merge(name => cast_type)
|
|
define_default_attribute(name, default, cast_type)
|
|
define_attribute_method(name)
|
|
end
|
|
|
|
# Returns an array of attribute names as strings
|
|
#
|
|
# class Person
|
|
# include ActiveModel::Attributes
|
|
#
|
|
# attribute :name, :string
|
|
# attribute :age, :integer
|
|
# end
|
|
#
|
|
# Person.attribute_names
|
|
# # => ["name", "age"]
|
|
def attribute_names
|
|
attribute_types.keys
|
|
end
|
|
|
|
private
|
|
def define_method_attribute=(name, owner:)
|
|
ActiveModel::AttributeMethods::AttrNames.define_attribute_accessor_method(
|
|
owner, name, writer: true,
|
|
) do |temp_method_name, attr_name_expr|
|
|
owner.define_cached_method("#{name}=", as: temp_method_name, namespace: :active_model) do |batch|
|
|
batch <<
|
|
"def #{temp_method_name}(value)" <<
|
|
" _write_attribute(#{attr_name_expr}, value)" <<
|
|
"end"
|
|
end
|
|
end
|
|
end
|
|
|
|
NO_DEFAULT_PROVIDED = Object.new # :nodoc:
|
|
private_constant :NO_DEFAULT_PROVIDED
|
|
|
|
def define_default_attribute(name, value, type)
|
|
self._default_attributes = _default_attributes.deep_dup
|
|
if value == NO_DEFAULT_PROVIDED
|
|
default_attribute = _default_attributes[name].with_type(type)
|
|
else
|
|
default_attribute = Attribute::UserProvidedDefault.new(
|
|
name,
|
|
value,
|
|
type,
|
|
_default_attributes.fetch(name.to_s) { nil },
|
|
)
|
|
end
|
|
_default_attributes[name] = default_attribute
|
|
end
|
|
end
|
|
|
|
def initialize(*)
|
|
@attributes = self.class._default_attributes.deep_dup
|
|
super
|
|
end
|
|
|
|
def initialize_dup(other) # :nodoc:
|
|
@attributes = @attributes.deep_dup
|
|
super
|
|
end
|
|
|
|
# Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
|
|
#
|
|
# class Person
|
|
# include ActiveModel::Attributes
|
|
#
|
|
# attribute :name, :string
|
|
# attribute :age, :integer
|
|
# end
|
|
#
|
|
# person = Person.new(name: 'Francesco', age: 22)
|
|
# person.attributes
|
|
# # => {"name"=>"Francesco", "age"=>22}
|
|
def attributes
|
|
@attributes.to_hash
|
|
end
|
|
|
|
# Returns an array of attribute names as strings
|
|
#
|
|
# class Person
|
|
# include ActiveModel::Attributes
|
|
#
|
|
# attribute :name, :string
|
|
# attribute :age, :integer
|
|
# end
|
|
#
|
|
# person = Person.new
|
|
# person.attribute_names
|
|
# # => ["name", "age"]
|
|
def attribute_names
|
|
@attributes.keys
|
|
end
|
|
|
|
def freeze
|
|
@attributes = @attributes.clone.freeze unless frozen?
|
|
super
|
|
end
|
|
|
|
private
|
|
def _write_attribute(attr_name, value)
|
|
@attributes.write_from_user(attr_name, value)
|
|
end
|
|
alias :attribute= :_write_attribute
|
|
|
|
def attribute(attr_name)
|
|
@attributes.fetch_value(attr_name)
|
|
end
|
|
end
|
|
end
|