Implemented basic inheritance.

boof · Feb 16, 2009 · 0e47332 · 0e47332
1 parent 569be26
commit 0e47332
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/shadows.rb b/lib/shadows.rb
@@ -1,9 +1,9 @@
 module Shadows
 
-  def self.shadow_class(base)
+  def self.shadow_class(base, fallback = nil)
     "#{ base.name }Shadow".constantize
   rescue NameError => e
-    Class.new Shadows::Base do
+    fallback or Class.new Shadows::Base do
       helper = "#{ base.name.pluralize }Helper".constantize rescue nil
       include helper if helper
     end
@@ -43,6 +43,10 @@ def self.extended(base)
       base.instance_variable_set :@shadow, Shadows.shadow_class(base)
     end
     attr_reader :shadow
+    def inherited(base)
+      super
+      base.instance_variable_set :@shadow, Shadows.shadow_class(base, @shadow)
+    end
 
   end
   module InstanceMethods