create and test derived classes too

2016-12-18 17:17:58 +02:00
parent 170d453a36
commit af9987659f
2 changed files with 21 additions and 12 deletions
--- a/lib/melon/compiler.rb
+++ b/lib/melon/compiler.rb
@ -9,10 +9,14 @@ module Melon
      compiler.process ast
    end

+    def get_name( statement )
+      return nil unless statement
+      statement.children[1]
+    end
+
    def on_class statement
-      name , _ , _ = *statement
-      clazz_name = name.children[1]
-      Parfait::Space.object_space.create_class(clazz_name , nil )
+      name , sup , _ = *statement
+      Parfait::Space.object_space.create_class(get_name(name) , get_name(sup) )
    end
  end
 end