activesupport/lib/active_support/core_ext/class/attribute_accessors.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

# Extends the class object with class and instance accessors for class attributes,
# just like the native attr* accessors for instance attributes.
#
#  class Person
#    cattr_accessor :hair_colors
#  end
#
#  Person.hair_colors = [:brown, :black, :blonde, :red]
class Class
  def cattr_reader(*syms)
    syms.flatten.each do |sym|
      next if sym.is_a?(Hash)
      class_eval(<<-EOS, __FILE__, __LINE__ + 1)
        unless defined? @@#{sym}   # unless defined @@property
          @@#{sym} = nil           #   @@property = nil
        end                        # end

        def self.#{sym}            # def self.property
          @@#{sym}                 #   @@property
        end                        # end

        def #{sym}                 # def property
          @@#{sym}                 #   @@property
        end                        # end
      EOS
    end
  end

  def cattr_writer(*syms)
    options = syms.extract_options!
    syms.flatten.each do |sym|
      class_eval(<<-EOS, __FILE__, __LINE__ + 1)
        unless defined? @@#{sym}   # unless defined? @@property
          @@#{sym} = nil           #   @@property = nil
        end                        # end
                                   
        def self.#{sym}=(obj)      # def self.property=(obj)
          @@#{sym} = obj           #   @@property
        end                        # end
                                   
        #{"                        
        def #{sym}=(obj)           # def property=(obj)
          @@#{sym} = obj           #   @@property = obj
        end                        # end
        " unless options[:instance_writer] == false }
      EOS
    end
  end

  def cattr_accessor(*syms)
    cattr_reader(*syms)
    cattr_writer(*syms)
  end
end