You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

118 lines
2.4 KiB

2 years ago
require_relative "error"
2 years ago
module Lox
class Resolver
def initialize(interpreter)
@interpreter = interpreter
@scopes = []
end
def resolve(*resolvees)
resolvees.each do |resolvee|
resolvee.accept(self)
2 years ago
end
nil
2 years ago
end
def visit_block(stmt)
with_scope do
2 years ago
resolve(*stmt.stmts)
2 years ago
end
nil
end
def visit_expr(stmt) = resolve(stmt.expr)
2 years ago
end
def visit_function(stmt)
declare(stmt.name)
define(stmt.name)
resolve_function(stmt)
nil
end
def visit_if(stmt)
resolve(stmt.condition, stmt.then)
2 years ago
resolve(stmt.else) if stmt.else
end
def visit_print(stmt) = resolve(stmt.expr)
def visit_return(stmt) = resolve(stmt.value) if stmt.value
2 years ago
2 years ago
def visit_var(stmt)
declare(stmt.name)
resolve(stmt.initializer) if stmt.initializer
define(stmt.name)
nil
end
def visit_while(stmt) = resolve(stmt.condition, stmt.body)
2 years ago
2 years ago
def visit_variable(expr)
if !@scopes.empty? && @scopes.last[expr.name.lexeme] == false
raise ResolverError.new(expr.name, "Can't read local variable in its own initializer.")
end
resolve_local(expr, expr.name)
nil
end
2 years ago
def visit_assign(expr)
resolve(expr.value)
resolve_local(expr, expr.name)
nil
end
def visit_binary(expr) = resolve(expr.left, expr.right)
def visit_call(expr) = resolve(expr.callee, *expr.args)
def visit_grouping(expr) = resolve(expr.expr)
def visit_literal(expr) = nil
def visit_logical(expr) = resolve(expr.left, expr.right)
def visit_unary(expr) = resolve(expr.right)
2 years ago
2 years ago
private
def with_block
2 years ago
@scopes.unshift({})
2 years ago
yield
2 years ago
@scopes.shift
2 years ago
end
2 years ago
def declare(name)
scope = @scopes.last
return if scope.nil?
scope[name.lexeme] = false
end
def define(name)
scope = @scopes.last
return if scope.nil?
scopes[name.lexeme] = true
end
2 years ago
2 years ago
def resolve_local(expr, name)
scope_and_depth = @scopes.each.with_index.find {|scope, depth| scope.has_key?(name.lexeme) }
return unless scope_and_depth
scope, depth = scope_and_depth
@interpreter.resolve(expr, depth)
end
2 years ago
def resolve_function(fn)
with_scope do
fn.params.each do |param|
declare(param)
define(param)
end
resolve(fn.body)
end
end
2 years ago
end
end