lib/uploader/asset.rb in rails-uploader-0.2.1 vs lib/uploader/asset.rb in rails-uploader-0.2.2

@@ -1,19 +1,25 @@
 module Uploader
   module Asset
     def self.included(base)
       base.send(:extend, Uploader::Asset::ClassMethods)
       base.send(:include, Uploader::Asset::AssetProcessor)
+
+      base.instance_eval do
+        before_create :generate_public_token
+      end
     end
 
     module Mongoid
       def self.included(base)
         base.send(:extend, Uploader::Asset::ClassMethods)
         base.send(:include, Uploader::Asset::AssetProcessor)
 
         base.instance_eval do
           field :guid, type: String
           field :public_token, type: String
+
+          before_create :generate_public_token
         end
       end
 
       def as_json(options = {})
         json_data = super