aboutsummaryrefslogtreecommitdiffstats
path: root/activemodel/lib/active_model/state_machine/event.rb
blob: 30e9601dc2125447f9471a46ffd3b6154a4cd84e (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
module ActiveModel
  module StateMachine
    class Event
      attr_reader :name, :success

      def initialize(machine, name, options = {}, &block)
        @machine, @name, @transitions = machine, name, []
        if machine
          machine.klass.send(:define_method, "#{name}!") do |*args|
            machine.fire_event(name, self, true, *args)
          end

          machine.klass.send(:define_method, name.to_s) do |*args|
            machine.fire_event(name, self, false, *args)
          end
        end
        update(options, &block)
      end

      def fire(obj, to_state = nil, *args)
        transitions = @transitions.select { |t| t.from == obj.current_state(@machine ? @machine.name : nil) }
        raise InvalidTransition if transitions.size == 0

        next_state = nil
        transitions.each do |transition|
          next if to_state && !Array(transition.to).include?(to_state)
          if transition.perform(obj)
            next_state = to_state || Array(transition.to).first
            transition.execute(obj, *args)
            break
          end
        end
        next_state
      end

      def transitions_from_state?(state)
        @transitions.any? { |t| t.from? state }
      end

      def ==(event)
        if event.is_a? Symbol
          name == event
        else
          name == event.name
        end
      end

      def update(options = {}, &block)
        if options.key?(:success) then @success = options[:success] end
        if block                  then instance_eval(&block)        end
        self
      end

      private
        def transitions(trans_opts)
          Array(trans_opts[:from]).each do |s|
            @transitions << StateTransition.new(trans_opts.merge({:from => s.to_sym}))
          end
        end
    end
  end
end