90 lines
2.5 KiB
Ruby
90 lines
2.5 KiB
Ruby
module AASM
|
|
module Persistence
|
|
module Base
|
|
|
|
def self.included(base) #:nodoc:
|
|
base.extend ClassMethods
|
|
end
|
|
|
|
# Returns the value of the aasm.attribute_name - called from <tt>aasm.current_state</tt>
|
|
#
|
|
# If it's a new record, and the aasm state column is blank it returns the initial state
|
|
# (example provided here for ActiveRecord, but it's true for Mongoid as well):
|
|
#
|
|
# class Foo < ActiveRecord::Base
|
|
# include AASM
|
|
# aasm :column => :status do
|
|
# state :opened
|
|
# state :closed
|
|
# end
|
|
# end
|
|
#
|
|
# foo = Foo.new
|
|
# foo.current_state # => :opened
|
|
# foo.close
|
|
# foo.current_state # => :closed
|
|
#
|
|
# foo = Foo.find(1)
|
|
# foo.current_state # => :opened
|
|
# foo.aasm_state = nil
|
|
# foo.current_state # => nil
|
|
#
|
|
# NOTE: intended to be called from an event
|
|
#
|
|
# This allows for nil aasm states - be sure to add validation to your model
|
|
def aasm_read_state(name=:default)
|
|
state = send(self.class.aasm(name).attribute_name)
|
|
if !state || state.empty?
|
|
aasm_new_record? ? aasm(name).determine_state_name(self.class.aasm(name).initial_state) : nil
|
|
else
|
|
state.to_sym
|
|
end
|
|
end
|
|
|
|
def aasm_new_record?
|
|
new_record?
|
|
end
|
|
|
|
module ClassMethods
|
|
def aasm_column(attribute_name=nil)
|
|
warn "[DEPRECATION] aasm_column is deprecated. Use aasm.attribute_name instead"
|
|
aasm.attribute_name(attribute_name)
|
|
end
|
|
end # ClassMethods
|
|
|
|
end # Base
|
|
end # Persistence
|
|
|
|
class Base
|
|
# make sure to create a (named) scope for each state
|
|
def state_with_scope(*args)
|
|
names = state_without_scope(*args)
|
|
names.each do |name|
|
|
create_scopes(name)
|
|
end
|
|
end
|
|
alias_method :state_without_scope, :state
|
|
alias_method :state, :state_with_scope
|
|
|
|
private
|
|
|
|
def create_scope?(name)
|
|
@state_machine.config.create_scopes && !@klass.respond_to?(name) && @klass.respond_to?(:aasm_create_scope)
|
|
end
|
|
|
|
def create_scope(name)
|
|
@klass.aasm_create_scope(@name, name) if create_scope?(name)
|
|
end
|
|
|
|
def create_scopes(name)
|
|
if namespace?
|
|
# Create default scopes even when namespace? for backward compatibility
|
|
namepaced_name = "#{namespace}_#{name}"
|
|
create_scope(namepaced_name)
|
|
end
|
|
create_scope(name)
|
|
end
|
|
end # Base
|
|
|
|
end # AASM
|