refactored for late-binding of Role class

kristianmandrup · Jan 21, 2011 · 7a2f1dd · 7a2f1dd
1 parent a545312
commit 7a2f1dd
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/lib/roles_generic/generic/role.rb b/lib/roles_generic/generic/role.rb
@@ -6,17 +6,19 @@ def role_class
   end
 
   module ClassMethods 
+    attr_accessor :class_constant
+
     def role_class_name
-      @role_class_name          
-    end
-
-    def role_class class_constant
       const = class_constant.to_s.camelize
       begin
-        @role_class_name = "#{const}".constantize
+        @role_class_name ||= "#{const}".constantize
       rescue
         puts "Role class constant '#{const}' is not defined so it could not be set!"
       end
     end
+
+    def role_class class_constant
+      @class_constant = class_constant  
+    end
   end    
 end