extracting methods for the compiler to handle

lib/melon/compiler.rb

@@ -1,4 +1,6 @@
-require_relative "type_collector"
+require_relative "compiler/type_collector"
+require_relative "compiler/method_collector"
+require_relative "compiler/ruby_method"
 
 
 module Melon
@@ -19,7 +21,17 @@ module Melon
       name , sup , body = *statement
       clazz = Parfait::Space.object_space.create_class(get_name(name) , get_name(sup) )
       ivar_hash = TypeCollector.new.collect(body)
-      clazz.set_instance_type( Parfait::Type.for_hash( clazz ,  ivar_hash ) ) 
+      clazz.set_instance_type( Parfait::Type.for_hash( clazz ,  ivar_hash ) )
+
+      MethodCollector.new.collect(body)
+
     end
+
+    def handler_missing(node)
+      node.children.each do |kid |
+        process(kid) if kid.is_a?(AST::Node)
+      end
+    end
+
   end
 end

lib/melon/compiler/method_collector.rb

@@ -0,0 +1,35 @@
+module Melon
+
+  class MethodCollector < AST::Processor
+
+    def initialize
+      @methods = []
+    end
+
+    def collect(statement)
+      process statement
+      @methods
+    end
+
+    def on_def(statement)
+      name , args , body = *statement
+      args_type = make_type(args)
+      @methods << RubyMethod.new(name , args_type , body )
+    end
+
+    def make_type( statement )
+      type = Parfait::Space.object_space.get_class_by_name(:Message ).instance_type
+      statement.children.each do |arg|
+        type = type.add_instance_variable( arg.children[0] , :Object )
+      end
+      type
+    end
+
+    def handler_missing(node)
+      node.children.each do |kid |
+        process(kid) if kid.is_a?(AST::Node)
+      end
+    end
+
+  end
+end