aboutsummaryrefslogtreecommitdiffstats
path: root/activemodel/lib/active_model/validations/acceptance.rb
blob: 6fd54270f20dd2b49e74760bb1d6b0b081565257 (plain) (blame)
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
# frozen_string_literal: true

module ActiveModel
  module Validations
    class AcceptanceValidator < EachValidator # :nodoc:
      def initialize(options)
        super({ allow_nil: true, accept: ["1", true] }.merge!(options))
        setup!(options[:class])
      end

      def validate_each(record, attribute, value)
        unless acceptable_option?(value)
          record.errors.add(attribute, :accepted, options.except(:accept, :allow_nil))
        end
      end

      private

        def setup!(klass)
          klass.include(LazilyDefineAttributes.new(AttributeDefinition.new(attributes)))
        end

        def acceptable_option?(value)
          Array(options[:accept]).include?(value)
        end

        class LazilyDefineAttributes < Module
          def initialize(attribute_definition)
            define_method(:respond_to_missing?) do |method_name, include_private = false|
              super(method_name, include_private) || attribute_definition.matches?(method_name)
            end

            define_method(:method_missing) do |method_name, *args, &block|
              if attribute_definition.matches?(method_name)
                attribute_definition.define_on(self.class)
                send(method_name, *args, &block)
              else
                super(method_name, *args, &block)
              end
            end
          end
        end

        class AttributeDefinition
          def initialize(attributes)
            @attributes = attributes.map(&:to_s)
          end

          def matches?(method_name)
            attr_name = convert_to_reader_name(method_name)
            attributes.include?(attr_name)
          end

          def define_on(klass)
            attr_readers = attributes.reject { |name| klass.attribute_method?(name) }
            attr_writers = attributes.reject { |name| klass.attribute_method?("#{name}=") }
            klass.define_attribute_methods
            klass.attr_reader(*attr_readers)
            klass.attr_writer(*attr_writers)
          end

          private
            attr_reader :attributes

            def convert_to_reader_name(method_name)
              method_name.to_s.chomp("=")
            end
        end
    end

    module HelperMethods
      # Encapsulates the pattern of wanting to validate the acceptance of a
      # terms of service check box (or similar agreement).
      #
      #   class Person < ActiveRecord::Base
      #     validates_acceptance_of :terms_of_service
      #     validates_acceptance_of :eula, message: 'must be abided'
      #   end
      #
      # If the database column does not exist, the +terms_of_service+ attribute
      # is entirely virtual. This check is performed only if +terms_of_service+
      # is not +nil+ and by default on save.
      #
      # Configuration options:
      # * <tt>:message</tt> - A custom error message (default is: "must be
      #   accepted").
      # * <tt>:accept</tt> - Specifies a value that is considered accepted.
      #   Also accepts an array of possible values. The default value is
      #   an array ["1", true], which makes it easy to relate to an HTML
      #   checkbox. This should be set to, or include, +true+ if you are validating
      #   a database column, since the attribute is typecast from "1" to +true+
      #   before validation.
      #
      # There is also a list of default options supported by every validator:
      # +:if+, +:unless+, +:on+, +:allow_nil+, +:allow_blank+, and +:strict+.
      # See <tt>ActiveModel::Validations#validates</tt> for more information.
      def validates_acceptance_of(*attr_names)
        validates_with AcceptanceValidator, _merge_attributes(attr_names)
      end
    end
  end
end