Contents

module Stateflow
  module Persistence
    module MongoMapper
      def self.install(base)
        ActiveSupport::Deprecation.silence do
          base.respond_to?(:before_validation_on_create) ? base.before_validation_on_create(:ensure_initial_state) : base.before_validation(:ensure_initial_state, :on => :create)
          base.send :include, InstanceMethods
        end
      end
      
      module InstanceMethods
        def load_from_persistence
          send machine.state_column.to_sym
        end

        def save_to_persistence(new_state, options = {})
          send("#{machine.state_column}=".to_sym, new_state)
          save if options[:save]
        end
        
        def ensure_initial_state
          send("#{machine.state_column.to_s}=", current_state.name.to_s) if send(machine.state_column.to_s).blank?
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
stateflow-0.4.2	lib/stateflow/persistence/mongo_mapper.rb