lib/parole.rb in parole-0.1 vs lib/parole.rb in parole-0.1.1

@@ -6,10 +6,12 @@
 require 'parole/commentable'
 require 'parole/comment'
 
 class ActiveRecord::Base
   def self.acts_as_commentable(options = {})
+    Parole.commentable_classes << self
+
     class_attribute :commentable_options, :acts_as_commentable
     self.acts_as_commentable = true
     self.commentable_options = options.reverse_merge(roles: [])
     self.commentable_options[:roles] = commentable_options[:roles].map(&:to_s)
 
@@ -20,7 +22,13 @@
     !!self.acts_as_commentable
   end
 
   def self.acts_as_comment(*args)
     include Parole::Comment
+  end
+end
+
+module Parole
+  def self.commentable_classes
+    @commentable_classes ||= []
   end
 end