Add mongoid support

.gitignore

@@ -5,6 +5,7 @@ html/
 *.rbc
 .bundle
 .config
+.ruby*
 .yardoc
 Gemfile.lock
 InstalledFiles
@@ -18,3 +19,5 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+
+gemfiles/*.lock

.travis.yml

@@ -22,3 +22,6 @@ matrix:
 
     - rvm: 2.1.0
       gemfile: gemfiles/Gemfile.rails-3.x
+
+services:
+  - mongodb

gemfiles/Gemfile.rails-3.x

@@ -4,6 +4,7 @@ group :development do
   gem "rdoc", ">= 3.12"
   gem "bundler", ">= 1.0.0"
   gem "activerecord", "~>3.2"
+  gem "mongoid"
   gem "sqlite3"
   gem "mocha"
   gem "rake"

gemfiles/Gemfile.rails-edge

@@ -3,6 +3,7 @@ source "http://rubygems.org"
 group :development do
   gem "rdoc", ">= 3.12"
   gem "bundler", ">= 1.0.0"
+  gem "mongoid", "~> 4.0.0.beta"
   gem "activerecord"
   gem "sqlite3"
   gem "mocha"

lib/workflow.rb

@@ -3,6 +3,7 @@
 require 'workflow/specification'
 require 'workflow/adapters/active_record'
 require 'workflow/adapters/remodel'
+require 'workflow/adapters/mongoid'
 
 # See also README.markdown for documentation
 module Workflow
@@ -262,16 +263,17 @@ def self.included(klass)
 
     klass.extend ClassMethods
 
-    if Object.const_defined?(:ActiveRecord)
-      if klass < ActiveRecord::Base
-        klass.send :include, Adapter::ActiveRecord::InstanceMethods
-        klass.send :extend, Adapter::ActiveRecord::Scopes
+    if Object.const_defined?(:ActiveRecord) && klass < ActiveRecord::Base
+      klass.send :include, Adapter::ActiveRecord::InstanceMethods
+      klass.send :extend, Adapter::ActiveRecord::Scopes
+      klass.before_validation :write_initial_state
+    elsif Object.const_defined?(:Remodel) && klass < Adapter::Remodel::Entity
+      klass.send :include, Remodel::InstanceMethods
+    elsif Object.const_defined?(:Mongoid) && klass < Mongoid::Document
+      klass.class_eval do
+        klass.send :include, Adapter::Mongoid::InstanceMethods
         klass.before_validation :write_initial_state
       end
-    elsif Object.const_defined?(:Remodel)
-      if klass < Adapter::Remodel::Entity
-        klass.send :include, Remodel::InstanceMethods
-      end
     end
   end
 end

lib/workflow/adapters/mongoid.rb

@@ -0,0 +1,22 @@
+module Workflow
+  module Adapter
+    module Mongoid
+      module InstanceMethods
+        def load_workflow_state
+          send(self.class.workflow_column)
+        end
+
+        def persist_workflow_state(new_value)
+          self.update_attribute(self.class.workflow_column, new_value)
+        end
+
+        private
+
+        def write_initial_state
+          send("#{self.class.workflow_column}=", current_state.to_s) if load_workflow_state.blank?
+        end
+
+      end
+    end
+  end
+end