Get Mongoid working by aliasing update_attribute with update_column o…

…n Revocable models. May want to refactor to a more ActiveRecord/Mongoid neutral approach in the future.
powershift · Jul 28, 2012 · db2f361 · db2f361
1 parent ebc5854
commit db2f361
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/lib/doorkeeper/models/mongoid/access_grant.rb b/lib/doorkeeper/models/mongoid/access_grant.rb
@@ -1,9 +1,11 @@
+require 'doorkeeper/models/mongoid/revocable'
 require 'doorkeeper/models/mongoid/scopes'
 
 module Doorkeeper
   class AccessGrant
     include Mongoid::Document
     include Mongoid::Timestamps
+    include Doorkeeper::Models::Mongoid::Revocable
     include Doorkeeper::Models::Mongoid::Scopes
 
     self.store_in :oauth_access_grants

diff --git a/lib/doorkeeper/models/mongoid/access_token.rb b/lib/doorkeeper/models/mongoid/access_token.rb
@@ -1,9 +1,11 @@
+require 'doorkeeper/models/mongoid/revocable'
 require 'doorkeeper/models/mongoid/scopes'
 
 module Doorkeeper
   class AccessToken
     include Mongoid::Document
     include Mongoid::Timestamps
+    include Doorkeeper::Models::Mongoid::Revocable
     include Doorkeeper::Models::Mongoid::Scopes
 
     self.store_in :oauth_access_tokens

diff --git a/lib/doorkeeper/models/mongoid/revocable.rb b/lib/doorkeeper/models/mongoid/revocable.rb
@@ -0,0 +1,15 @@
+module Doorkeeper
+  module Models
+    module Mongoid
+      module Revocable
+        def self.included(base)
+          base.class_eval do
+            def update_column(attr, val)
+              update_attribute attr, val
+            end
+          end
+        end
+      end
+    end
+  end
+end