user_provided_default.rb 1.2 KB
Newer Older
1 2
# frozen_string_literal: true

3
require "active_model/attribute"
4

5
module ActiveModel
6
  class Attribute # :nodoc:
7
    class UserProvidedDefault < FromUser # :nodoc:
8
      def initialize(name, value, type, database_default)
9
        @user_provided_value = value
10
        super(name, value, type, database_default)
11 12
      end

13 14 15
      def value_before_type_cast
        if user_provided_value.is_a?(Proc)
          @memoized_value_before_type_cast ||= user_provided_value.call
16
        else
17
          @user_provided_value
18 19
        end
      end
20 21

      def with_type(type)
22
        self.class.new(name, user_provided_value, type, original_attribute)
23
      end
24

25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
      def marshal_dump
        result = [
          name,
          value_before_type_cast,
          type,
          original_attribute,
        ]
        result << value if defined?(@value)
        result
      end

      def marshal_load(values)
        name, user_provided_value, type, original_attribute, value = values
        @name = name
        @user_provided_value = user_provided_value
        @type = type
        @original_attribute = original_attribute
        if values.length == 5
          @value = value
        end
      end

47 48
      protected

49
        attr_reader :user_provided_value
50 51 52
    end
  end
end