Class: SimpleMaster::Master::Column::EnumColumn

Inherits:
SimpleMaster::Master::Column show all
Defined in:
lib/simple_master/master/column/enum_column.rb

Instance Attribute Summary collapse

Attributes inherited from SimpleMaster::Master::Column

#group_key, #options

Instance Method Summary collapse

Methods inherited from SimpleMaster::Master::Column

column_type, column_types, #db_column_name, inherited, register

Constructor Details

#initialize(name, options) ⇒ EnumColumn

Returns a new instance of EnumColumn.



13
14
15
16
17
18
19
20
 
# File 'lib/simple_master/master/column/enum_column.rb', line 13

def initialize(name, options)
  @enum = options[:enum]
  @const_name = "ENUM_FOR_#{name.upcase}"
  @prefix = "#{options[:prefix] == true ? name : options[:prefix]}_" if options[:prefix]
  @suffix = "_#{options[:suffix] == true ? name : options[:suffix]}" if options[:suffix]

  super
end

Instance Attribute Details

#const_name ⇒ Object (readonly)

Returns the value of attribute const_name.



9
10
11
 
# File 'lib/simple_master/master/column/enum_column.rb', line 9

def const_name
  @const_name
end

#enum ⇒ Object (readonly)

Returns the value of attribute enum.



8
9
10
 
# File 'lib/simple_master/master/column/enum_column.rb', line 8

def enum
  @enum
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



7
8
9
 
# File 'lib/simple_master/master/column/enum_column.rb', line 7

def name
  @name
end

#prefix ⇒ Object (readonly)

Returns the value of attribute prefix.



10
11
12
 
# File 'lib/simple_master/master/column/enum_column.rb', line 10

def prefix
  @prefix
end

#suffix ⇒ Object (readonly)

Returns the value of attribute suffix.



11
12
13
 
# File 'lib/simple_master/master/column/enum_column.rb', line 11

def suffix
  @suffix
end

Instance Method Details

#init(master_class, for_test = false) ⇒ Object 



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
 
# File 'lib/simple_master/master/column/enum_column.rb', line 22

def init(master_class, for_test = false)
  super

  master_class.simple_master_module.const_set(const_name, enum.freeze) unless master_class.const_defined?(const_name)

  master_class.class_eval <<-RUBY, __FILE__, __LINE__ + 1
    def self.#{name.to_s.pluralize}
      #{const_name}
    end
  RUBY

  master_class.simple_master_module.class_eval <<-RUBY, __FILE__, __LINE__ + 1
    def #{name}_to_enum(value)
      case value
      when Integer
        return #{const_name}.key(value)
      when String
        sym = value.to_sym
        return sym if #{const_name}.key?(sym)

        int_val = Integer(value, exception: false)
        if int_val
          result = #{const_name}.key(int_val)
          return result if result
        end

        fail "Unsupported type \#{value.inspect}."
      when Symbol, NilClass
        return value
      else
        fail "Unsupported type \#{value.inspect}."
      end
    end

    def #{name.to_s.pluralize}
      #{const_name}
    end

    def #{name}_before_type_cast
      #{const_name}[#{name}]
    end
  RUBY

  enum.each_key do |enum_name|
    # Skip generating helpers for names that start with a digit
    next if enum_name.match?(/\A\d/) && prefix.nil?
    master_class.simple_master_module.class_eval <<-RUBY, __FILE__, __LINE__ + 1
        def #{prefix}#{enum_name}#{suffix}?
          #{name} == :#{enum_name}
        end
    RUBY
  end
end