stateflow 0.0.1 → 0.0.2

data/Manifest

@@ -2,10 +2,14 @@ LICENCE
 Manifest
 README.rdoc
 Rakefile
+examples/robot.rb
 examples/test.rb
 init.rb
 lib/stateflow.rb
 lib/stateflow/event.rb
 lib/stateflow/machine.rb
+lib/stateflow/persistence.rb
+lib/stateflow/persistence/active_record.rb
+lib/stateflow/persistence/mongo_mapper.rb
 lib/stateflow/state.rb
 lib/stateflow/transition.rb

data/Rakefile

@@ -2,7 +2,7 @@ require 'rubygems'
 require 'rake'
 require 'echoe'
 
-Echoe.new('stateflow', '0.0.1') do |p|
+Echoe.new('stateflow', '0.0.2') do |p|
   p.description    = "State machine that allows dynamic transitions for business workflows"
   p.url            = "http://github.com/ryanza/stateflow"
   p.author         = "Ryan Oberholzer"

data/examples/robot.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'stateflow'
+
+class Robot
+  include Stateflow
+  
+  stateflow do
+    initial :green
+    
+    state :green, :yellow, :red
+    
+    event :change_color do
+      transitions :from => :green, :to => :yellow
+      transitions :from => :yellow, :to => :red
+      transitions :from => :red, :to => :green
+    end
+  end
+end  

data/examples/test.rb

@@ -35,7 +35,7 @@ class Test
   end
   
   def likes_ice_cream?
-    rand(10) > 5 ? :mixed : :hate
+    rand(10) > 5 ? :mixeds : :hate
   end
   
   def exit_love

data/lib/stateflow.rb

@@ -1,11 +1,12 @@
 module Stateflow
   def self.included(base)
     base.send :include, InstanceMethods
+    Stateflow::Persistence.set(base)
     base.extend ClassMethods
   end
   
   def self.persistence
-    @@persistence ||= "active_record"
+    @@persistence ||= :active_record
   end
   
   def self.persistence=(persistence)
@@ -34,11 +35,12 @@ module Stateflow
   end
   
   module InstanceMethods
-    def current_state
-      @current_state ||= machine.initial_state
+    def current_state  
+      @current_state ||= load_from_persistence.nil? ? machine.initial_state : machine.states[load_from_persistence.to_sym]
     end
     
     def current_state=(new_state)
+      save_to_persistence(new_state.name.to_s)
       @current_state = new_state
     end
     
@@ -58,4 +60,5 @@ module Stateflow
   autoload :State, 'stateflow/state'
   autoload :Event, 'stateflow/event'
   autoload :Transition, 'stateflow/transition'
-end
+  autoload :Persistence, 'stateflow/persistence'
+end

data/lib/stateflow/machine.rb

@@ -6,6 +6,10 @@ module Stateflow
       @states, @events = Hash.new, Hash.new
       instance_eval(&machine)
     end
+    
+    def state_column(name = nil)
+      @state_column ||= name.nil? ? :state : name
+    end
 
     private
     def initial(name)

data/lib/stateflow/persistence.rb

@@ -0,0 +1,15 @@
+module Stateflow
+  module Persistence
+    def self.set(base)
+      case Stateflow.persistence
+        when :mongo_mapper
+          Stateflow::Persistence::MongoMapper.install(base)
+        when :active_record
+          Stateflow::Persistence::ActiveRecord.install(base)
+      end
+    end
+    
+    autoload :MongoMapper, 'stateflow/persistence/mongo_mapper'
+    autoload :ActiveRecord, 'stateflow/persistence/active_record'
+  end
+end

data/lib/stateflow/persistence/active_record.rb

@@ -0,0 +1,24 @@
+module Stateflow
+  module Persistence
+    module ActiveRecord
+      def self.install(base)
+        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
+      
+      module InstanceMethods
+        def load_from_persistence
+          self.send machine.state_column.to_sym
+        end
+
+        def save_to_persistence(new_state)
+          self.update_attribute(machine.state_column.to_sym, new_state)
+        end
+        
+        def ensure_initial_state
+          send("#{self.machine.state_column.to_s}=", self.current_state.name.to_s) if send(self.machine.state_column.to_s).blank?
+        end
+      end
+    end
+  end
+end

data/lib/stateflow/persistence/mongo_mapper.rb

@@ -0,0 +1,25 @@
+module Stateflow
+  module Persistence
+    module MongoMapper
+      def self.install(base)
+        base.send :key, :state, String
+        base.before_validation_on_create :ensure_initial_state
+        base.send :include, InstanceMethods
+      end
+      
+      module InstanceMethods
+        def load_from_persistence
+          self.send machine.state_column.to_sym
+        end
+
+        def save_to_persistence(new_state)
+          self.update_attributes(machine.state_column.to_sym => new_state)
+        end
+        
+        def ensure_initial_state
+          send("#{self.machine.state_column.to_s}=", self.current_state.name.to_s) if send(self.machine.state_column.to_s).blank?
+        end
+      end
+    end
+  end
+end

data/stateflow.gemspec

@@ -2,15 +2,15 @@
 
 Gem::Specification.new do |s|
   s.name = %q{stateflow}
-  s.version = "0.0.1"
+  s.version = "0.0.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
   s.authors = ["Ryan Oberholzer"]
-  s.date = %q{2010-03-22}
+  s.date = %q{2010-03-24}
   s.description = %q{State machine that allows dynamic transitions for business workflows}
   s.email = %q{ryan@platform45.com}
-  s.extra_rdoc_files = ["README.rdoc", "lib/stateflow.rb", "lib/stateflow/event.rb", "lib/stateflow/machine.rb", "lib/stateflow/state.rb", "lib/stateflow/transition.rb"]
-  s.files = ["LICENCE", "Manifest", "README.rdoc", "Rakefile", "examples/test.rb", "init.rb", "lib/stateflow.rb", "lib/stateflow/event.rb", "lib/stateflow/machine.rb", "lib/stateflow/state.rb", "lib/stateflow/transition.rb", "stateflow.gemspec"]
+  s.extra_rdoc_files = ["README.rdoc", "lib/stateflow.rb", "lib/stateflow/event.rb", "lib/stateflow/machine.rb", "lib/stateflow/persistence.rb", "lib/stateflow/persistence/active_record.rb", "lib/stateflow/persistence/mongo_mapper.rb", "lib/stateflow/state.rb", "lib/stateflow/transition.rb"]
+  s.files = ["LICENCE", "Manifest", "README.rdoc", "Rakefile", "examples/robot.rb", "examples/test.rb", "init.rb", "lib/stateflow.rb", "lib/stateflow/event.rb", "lib/stateflow/machine.rb", "lib/stateflow/persistence.rb", "lib/stateflow/persistence/active_record.rb", "lib/stateflow/persistence/mongo_mapper.rb", "lib/stateflow/state.rb", "lib/stateflow/transition.rb", "stateflow.gemspec"]
   s.homepage = %q{http://github.com/ryanza/stateflow}
   s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Stateflow", "--main", "README.rdoc"]
   s.require_paths = ["lib"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: stateflow
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Ryan Oberholzer
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-22 00:00:00 +02:00
+date: 2010-03-24 00:00:00 +02:00
 default_executable: 
 dependencies: []
 
@@ -24,6 +24,9 @@ extra_rdoc_files:
 - lib/stateflow.rb
 - lib/stateflow/event.rb
 - lib/stateflow/machine.rb
+- lib/stateflow/persistence.rb
+- lib/stateflow/persistence/active_record.rb
+- lib/stateflow/persistence/mongo_mapper.rb
 - lib/stateflow/state.rb
 - lib/stateflow/transition.rb
 files: 
@@ -31,11 +34,15 @@ files:
 - Manifest
 - README.rdoc
 - Rakefile
+- examples/robot.rb
 - examples/test.rb
 - init.rb
 - lib/stateflow.rb
 - lib/stateflow/event.rb
 - lib/stateflow/machine.rb
+- lib/stateflow/persistence.rb
+- lib/stateflow/persistence/active_record.rb
+- lib/stateflow/persistence/mongo_mapper.rb
 - lib/stateflow/state.rb
 - lib/stateflow/transition.rb
 - stateflow.gemspec