main
Alpha Chen 2 years ago
parent 5850036689
commit c697f9480b

@ -2,14 +2,14 @@ require_relative "environment"
module Lox
class Function
def initialize(decl, closure)
@decl, @closure = decl, closure
def initialize(decl, closure, is_initializer)
@decl, @closure, @is_initializer = decl, closure, is_initializer
end
def bind(instance)
env = Environment.new(@closure)
env.define("this", instance)
Function.new(@decl, env)
Function.new(@decl, env, @is_initializer)
end
def arity = @decl.params.size
@ -20,9 +20,13 @@ module Lox
env.define(name, value)
end
catch(:return) {
return_value = catch(:return) {
interpreter.execute_block(@decl.body, env)
}
return @closure.get_at(0, "this") if @is_initializer
return_value
end
def to_s = "<fn #{@decl.name.lexeme}>"

@ -48,7 +48,7 @@ module Lox
@env.define(stmt.name.lexeme, nil)
methods = stmt.methods.to_h {|method|
[method.name.lexeme, Function.new(method, @env)]
[method.name.lexeme, Function.new(method, @env, method.name.lexeme == "init")]
}
klass = LoxClass.new(stmt.name.lexeme, methods)
@ -73,7 +73,7 @@ module Lox
end
def visit_function(stmt)
function = Function.new(stmt, @env)
function = Function.new(stmt, @env, false)
@env.define(stmt.name.lexeme, function)
nil
end

@ -416,6 +416,23 @@ class TestInterpreter < Lox::Test
end
end
def test_init
assert_interpreted <<~EXPECTED.chomp, <<~SRC
Foo instance
Foo instance
Foo instance
EXPECTED
class Foo {
init() {
print this;
}
}
var foo = Foo();
print foo.init();
SRC
end
private
def assert_interpreted(expected, src)

Loading…
Cancel
Save