# encoding: UTF-8 $LOAD_PATH.unshift File.join(File.dirname(__FILE__)) module Maintain # We're not really interested in loading anything into memory if we don't need to, # so Maintainer, Value, and the Value subclasses are ignored until they're needed. autoload(:Maintainer, 'maintain/maintainer') autoload(:Backend, 'maintain/backend') autoload(:Value, 'maintain/value') autoload(:BitmaskValue, 'maintain/bitmask_value') autoload(:IntegerValue, 'maintain/integer_value') # The core class method of Maintain. Basic usage is: # # maintain :state do # state :new, :default => true # state :expired, :enter => :expire_children # state :reopened, :exit => lambda { children.each(&:reopen) } # aggregate :accessible, :as => [:new, :reopened] # end # # It also supports more complex configuration options, like bitmask columns # and integer values (for performance and portability) # # maintain :permissions, :bitmask => true do # state :edit, 1 # state :delete, 2 # state :manage, 3 # end # # This method is aliased as `maintains` with the intention of allowing developers # to code imperatively ("maintain, damn you!") or descriptively ("it maintains, man") def maintain(attribute, options = {}, &block) # Detect if this is ActiveRecord::Base or a subclass of it # TODO: Make this not suck if defined?(ActiveRecord::Base) active_record = self == ActiveRecord::Base superclass = self while !active_record && superclass.superclass active_record = superclass == ActiveRecord::Base superclass = superclass.superclass end options[:back_end] = 'active_record' if active_record end # Create an instance of the maintainer class. It handles all of the state # configuration, hooking, aggregation, named_scoping, etc. maintainer = Maintainer.new(self, attribute, options) if block_given? maintainer.instance_eval(&block) end # Define our getters and setters - these are the only methods Maintain will stomp # on if you've already defined them. This is because they're how Maintain works. class_eval <<-EOC, __FILE__ def #{attribute}=(value) # Find the maintainer on this attribute so we can use it to set values. if maintainer = self.class.maintainers[:#{attribute}] changed = #{attribute} != value # Run the exit hook if we're changing the value maintainer.hook(:exit, #{attribute}.name, self) if changed # Then set the value itself. Maintainer::State will return the value you set, # so if we're setting to nil we get rid of the attribute entirely - it's not # needed and we want the getter to return nil in that case. #{attribute}.set_value(value) # Allow the back end to write values in an ORM-specific way if maintainer.back_end maintainer.back_end.write(self, :#{attribute}, value) end # Last but not least, run the enter hooks for the new value - cause that's how # we do. maintainer.hook(:enter, #{attribute}.name, self) if changed end end def #{attribute} if maintainer = self.class.maintainers[:#{attribute}] if @#{attribute} @#{attribute} else @#{attribute} = maintainer.value(self) end else @#{attribute} end end EOC # Last! Not least! Save our maintainer directly on this class. We'll use it in our setters (as in above) # and we'll also modify it instead of replacing it outright, so subclasses or mixins can extend functionality # without replacing it. maintainers[attribute.to_sym] = maintainer end alias :maintains :maintain def maintainers #:nodoc: @maintainers ||= {} end if File.file?(version_path = File.join(File.dirname(__FILE__), '..', 'VERSION')) VERSION = File.read(version_path).strip else VERSION = '0.2.0' end end if defined?(ActiveRecord::Base) ActiveRecord::Base.extend Maintain end