Module: StatePattern::ActiveRecord

Defined in:
lib/state_pattern/active_record.rb

Class Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



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
 
# File 'lib/state_pattern/active_record.rb', line 3

def self.included(base)
  base.class_eval do
    include StatePattern

    after_initialize :set_state_from_db
    def set_state_from_db
      set_state(state_string_as_class || self.class.initial_state_class)
    end

    #enable after_initialize callback
    if ::ActiveRecord::VERSION::STRING >= "3.0.0"
      def self.after_initialize; end
    else
      def after_initialize; end
    end

    def set_state_with_active_record_attribute(state_class)
      set_state_without_active_record_attribute(state_class)
      write_attribute(self.class.state_attribute, @current_state_instance.class.name)
    end
    alias_method_chain :set_state, :active_record_attribute

    def state=(new_state_string)
      set_state(state_string_as_class(new_state_string))
    end

    def state_string_as_class(state_string = read_attribute(self.class.state_attribute))
      state_string.camelize.constantize unless state_string.nil?
    end

    def self.state_attribute
      @state_attribute ||= "state"
    end

    def self.set_state_attribute(state_attr)
      @state_attribute = state_attr.to_s
    end
  end
end