Module: MakePrivate

Defined in:
lib/make_private.rb,
lib/make_private/version.rb

Constant Summary collapse

VERSION =
"0.0.1"

Class Method Summary collapse

Class Method Details

.included(klass) ⇒ Object



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
# File 'lib/make_private.rb', line 5

def self.included(klass)
  klass.class_eval do
    self.columns.each do |column|
      self.instance_eval do
        define_method(:"#{column.name}=") do |val|
          write_attribute column.name.to_sym, val
        end

        define_method(:"#{column.name}") do
          self[column.name.to_sym]
        end
        private :"#{column.name}=", column.name.to_sym
      end
    end

    class << self
      alias :__belongs_to_public :belongs_to
      alias :__has_one_public :has_one
      alias :__has_many_public :has_many
      alias :__has_and_belongs_to_many_public :has_and_belongs_to_many

      def belongs_to *args, &block
        __belongs_to_public *args, &block
        private args[0], :"#{args[0]}="
      end

      def has_one *args, &block
        __has_one_public *args, &block
        private args[0], :"#{args[0]}="
      end

      def has_many *args, &block
        __has_many_public *args, &block
        private args[0], :"#{args[0]}="
      end

      def has_and_belongs_to_many *args, &block
        __has_and_belongs_to_many_public *args, &block
        private args[0], :"#{args[0]}="
      end
    end
  end
end