pick up instance variables from the class

Crude first set to creating types
This commit is contained in:
Torsten Ruger 2016-12-18 20:05:11 +02:00
parent e77298f4b1
commit 107e3e6d58
3 changed files with 35 additions and 2 deletions

View File

@ -1,3 +1,4 @@
require_relative "type_collector"
module Melon module Melon
@ -15,8 +16,9 @@ module Melon
end end
def on_class statement def on_class statement
name , sup , _ = *statement name , sup , body = *statement
Parfait::Space.object_space.create_class(get_name(name) , get_name(sup) ) clazz = Parfait::Space.object_space.create_class(get_name(name) , get_name(sup) )
clazz.set_instance_names( TypeCollector.new.collect(body) )
end end
end end
end end

View File

@ -0,0 +1,24 @@
module Melon
class TypeCollector < AST::Processor
def initialize
@ivar_names = []
end
def collect(statement)
process statement
@ivar_names
end
def on_ivar(statement)
@ivar_names.push statement.children[0].to_s[1..-1].to_sym
end
def handler_missing(node)
node.children.each do |kid |
process(kid) if kid.is_a?(AST::Node)
end
end
end
end

View File

@ -6,6 +6,7 @@ module Melon
def setup def setup
Register.machine.boot unless Register.machine.booted Register.machine.boot unless Register.machine.booted
end end
def test_creates_class_without_deriviation def test_creates_class_without_deriviation
Compiler.compile "class Testing ; end" Compiler.compile "class Testing ; end"
assert t = Parfait::Space.object_space.get_class_by_name(:Testing) , "No classes created" assert t = Parfait::Space.object_space.get_class_by_name(:Testing) , "No classes created"
@ -18,5 +19,11 @@ module Melon
assert_equal :List , t.super_class_name assert_equal :List , t.super_class_name
end end
def test_picks_up_an_ivar_name
Compiler.compile "class TestIvar < Object ; def meth; return @ivar;end; end"
itest = Parfait::Space.object_space.get_class_by_name(:TestIvar)
assert itest.instance_names.include?(:ivar) , itest.instance_names.inspect
end
end end
end end