base.rb in inkwell-1.5.1

- old
+ new

@@ -1,32 +1,42 @@
 module Inkwell
   module Common
     def get_item_type(obj)
-      post_class = Object.const_get ::Inkwell::Engine::config.post_table.to_s.singularize.capitalize
       case obj
         when ::Inkwell::Comment
           result = ::Inkwell::Constants::ItemTypes::COMMENT
         when post_class
           result = ::Inkwell::Constants::ItemTypes::POST
         else
-          raise "obj should be Comment or #{post_class.class}"
+          raise "obj should be Comment or #{post_class.to_s}"
       end
       result
     end
 
     def get_class_for_item_type(type)
-      post_class = Object.const_get ::Inkwell::Engine::config.post_table.to_s.singularize.capitalize
       case type
         when ::Inkwell::Constants::ItemTypes::COMMENT
           result = ::Inkwell::Comment
         when ::Inkwell::Constants::ItemTypes::POST
           result = post_class
         else raise "obj type #{type} is unknown"
       end
       result
     end
 
+    def get_owner_type(obj)
+      case obj
+        when user_class
+          result = ::Inkwell::Constants::OwnerTypes::USER
+        when community_class
+          result = ::Inkwell::Constants::OwnerTypes::COMMUNITY
+        else
+          raise "obj should be #{user_class.to_s} or #{community_class.to_s}"
+      end
+      result
+    end
+
     def check_user(obj)
       user_class = Object.const_get ::Inkwell::Engine::config.user_table.to_s.singularize.capitalize
       raise "user should be a #{user_class.to_s}" unless obj.is_a? user_class
     end
 
@@ -51,9 +61,13 @@
       "#{::Inkwell::Engine::config.community_table.to_s.singularize}_id"
     end
 
     def community_class
       Object.const_get ::Inkwell::Engine::config.community_table.to_s.singularize.capitalize
+    end
+
+    def category_class
+      Object.const_get ::Inkwell::Engine::config.category_table.to_s.singularize.capitalize
     end
   end
 
   module Constants
 
\ No newline at end of file