schala/schala-lang/src/symbol_table/resolver.rs

241 lines
10 KiB
Rust
Raw Normal View History

2021-10-19 13:48:00 -07:00
use std::rc::Rc;
use crate::{
ast::*,
2021-11-02 01:20:30 -07:00
symbol_table::{Fqsn, ScopeSegment, SymbolSpec, SymbolTable},
util::ScopeStack,
};
2021-10-19 13:48:00 -07:00
2021-10-24 00:08:26 -07:00
#[derive(Debug)]
enum NameType {
//TODO eventually this needs to support closures
Param(u8),
LocalVariable(ItemId),
2021-11-02 18:07:08 -07:00
LocalFunction(ItemId),
Import(Fqsn),
2021-10-24 00:08:26 -07:00
}
#[derive(Debug)]
enum ScopeType {
Function { name: Rc<String> },
2021-10-24 19:05:41 -07:00
Lambda,
2021-10-26 11:37:43 -07:00
PatternMatch,
2021-10-24 00:08:26 -07:00
//TODO add some notion of a let-like scope?
}
pub struct ScopeResolver<'a> {
2021-10-21 14:46:42 -07:00
symbol_table: &'a mut super::SymbolTable,
2021-10-26 11:37:43 -07:00
//TODO maybe this shouldn't be a scope stack, b/c the recursion behavior comes from multiple
//instances of ScopeResolver
lexical_scopes: ScopeStack<'a, Rc<String>, NameType, ScopeType>,
2021-10-19 13:48:00 -07:00
}
impl<'a> ScopeResolver<'a> {
2021-10-21 14:46:42 -07:00
pub fn new(symbol_table: &'a mut SymbolTable) -> Self {
2021-10-24 00:08:26 -07:00
let lexical_scopes = ScopeStack::new(None);
Self { symbol_table, lexical_scopes }
2021-10-21 14:46:42 -07:00
}
2021-10-24 00:08:26 -07:00
2021-10-21 14:46:42 -07:00
pub fn resolve(&mut self, ast: &AST) {
walk_ast(self, ast);
}
2021-10-19 13:48:00 -07:00
/// This method correctly modifies the id_to_def table (ItemId) to have the appropriate
2021-10-24 00:08:26 -07:00
/// mappings.
fn lookup_name_in_scope(&mut self, name: &QualifiedName) {
let QualifiedName { id, components } = name;
2021-11-01 21:34:45 -07:00
let local_name = components.first().unwrap().clone();
let name_type = self.lexical_scopes.lookup(&local_name);
2021-11-02 01:20:30 -07:00
let fqsn = Fqsn { scopes: components.iter().map(|name| ScopeSegment::Name(name.clone())).collect() };
let def_id = self.symbol_table.symbol_trie.lookup(&fqsn);
2021-11-01 21:34:45 -07:00
2021-10-26 13:02:40 -07:00
//TODO handle a "partial" qualified name, and also handle it down in the pattern-matching
//section
if components.len() == 1 {
match name_type {
Some(NameType::Import(fqsn)) => {
2021-11-14 03:27:30 -08:00
let def_id = self.symbol_table.symbol_trie.lookup(fqsn);
if let Some(def_id) = def_id {
self.symbol_table.id_to_def.insert(*id, def_id);
}
}
Some(NameType::Param(n)) => {
let spec = SymbolSpec::FunctionParam(*n);
//TODO need to come up with a better solution for local variable FQSNs
2021-11-02 01:20:30 -07:00
let lscope = ScopeSegment::Name(Rc::new("<local-param>".to_string()));
let fqsn = Fqsn { scopes: vec![lscope, ScopeSegment::Name(local_name.clone())] };
self.symbol_table.add_symbol(id, fqsn, spec);
}
2021-11-02 18:07:08 -07:00
Some(NameType::LocalFunction(item_id)) => {
let def_id = self.symbol_table.id_to_def.get(item_id);
if let Some(def_id) = def_id {
2021-11-14 03:27:30 -08:00
let def_id = *def_id;
2021-11-02 18:07:08 -07:00
self.symbol_table.id_to_def.insert(*id, def_id);
}
}
Some(NameType::LocalVariable(item_id)) => {
let def_id = self.symbol_table.id_to_def.get(item_id);
if let Some(def_id) = def_id {
2021-11-14 03:27:30 -08:00
let def_id = *def_id;
self.symbol_table.id_to_def.insert(*id, def_id);
}
}
2021-11-01 21:34:45 -07:00
None =>
if let Some(def_id) = def_id {
self.symbol_table.id_to_def.insert(*id, def_id);
2021-11-01 21:34:45 -07:00
},
}
2021-11-14 03:27:30 -08:00
} else if let Some(def_id) = def_id {
self.symbol_table.id_to_def.insert(*id, def_id);
2021-10-21 14:46:42 -07:00
}
2021-10-19 13:48:00 -07:00
}
}
impl<'a> ASTVisitor for ScopeResolver<'a> {
2021-10-24 00:08:26 -07:00
// Import statements bring in a bunch of local names that all map to a specific FQSN.
// FQSNs map to a Symbol (or this is an error), Symbols have a DefId. So for every
// name we import, we map a local name (a string) to a NameType::ImportedDefinition(DefId).
fn import(&mut self, import_spec: &ImportSpecifier) -> Recursion {
let ImportSpecifier { ref path_components, ref imported_names, .. } = &import_spec;
2021-10-21 14:46:42 -07:00
match imported_names {
ImportedNames::All => {
let prefix =
2021-11-02 01:20:30 -07:00
Fqsn { scopes: path_components.iter().map(|c| ScopeSegment::Name(c.clone())).collect() };
2021-10-21 14:46:42 -07:00
let members = self.symbol_table.symbol_trie.get_children(&prefix);
for fqsn in members.into_iter() {
2021-11-02 16:56:12 -07:00
self.lexical_scopes.insert(fqsn.last_elem(), NameType::Import(fqsn));
2021-10-21 14:46:42 -07:00
}
}
ImportedNames::LastOfPath => {
2021-11-02 01:20:30 -07:00
let fqsn =
Fqsn { scopes: path_components.iter().map(|c| ScopeSegment::Name(c.clone())).collect() };
2021-11-02 16:56:12 -07:00
self.lexical_scopes.insert(fqsn.last_elem(), NameType::Import(fqsn));
2021-10-21 14:46:42 -07:00
}
ImportedNames::List(ref names) => {
2021-11-02 01:20:30 -07:00
let fqsn_prefix: Vec<ScopeSegment> =
path_components.iter().map(|c| ScopeSegment::Name(c.clone())).collect();
2021-10-21 14:46:42 -07:00
for name in names.iter() {
let mut scopes = fqsn_prefix.clone();
2021-11-02 01:20:30 -07:00
scopes.push(ScopeSegment::Name(name.clone()));
let fqsn = Fqsn { scopes };
2021-11-02 16:56:12 -07:00
self.lexical_scopes.insert(fqsn.last_elem(), NameType::Import(fqsn));
2021-10-21 14:46:42 -07:00
}
}
2021-10-19 13:48:00 -07:00
};
2021-10-24 00:08:26 -07:00
Recursion::Continue
}
fn declaration(&mut self, declaration: &Declaration, id: &ItemId) -> Recursion {
let cur_function_name = match self.lexical_scopes.get_name() {
//TODO this needs to be a fqsn
Some(ScopeType::Function { name }) => Some(name.clone()),
_ => None,
};
match declaration {
Declaration::FuncDecl(signature, block) => {
let param_names = signature.params.iter().map(|param| param.name.clone());
//TODO I'm 90% sure this is right, until I get to closures
//let mut new_scope = self.lexical_scopes.new_scope(Some(ScopeType::Function { name: signature.name.clone() }));
let mut new_scope =
ScopeStack::new(Some(ScopeType::Function { name: signature.name.clone() }));
2021-10-30 00:01:59 -07:00
for (n, param) in param_names.enumerate() {
new_scope.insert(param, NameType::Param(n as u8));
}
2021-10-24 00:08:26 -07:00
2021-11-02 18:07:08 -07:00
self.lexical_scopes.insert(signature.name.clone(), NameType::LocalFunction(*id));
let mut new_resolver =
ScopeResolver { symbol_table: self.symbol_table, lexical_scopes: new_scope };
walk_block(&mut new_resolver, block);
Recursion::Stop
2021-10-24 00:08:26 -07:00
}
Declaration::Binding { name, .. } => {
if let Some(fn_name) = cur_function_name {
// We are within a function scope
2021-11-02 01:20:30 -07:00
let fqsn =
Fqsn { scopes: vec![ScopeSegment::Name(fn_name), ScopeSegment::Name(name.clone())] };
self.symbol_table.add_symbol(id, fqsn, SymbolSpec::LocalVariable);
2021-10-27 01:17:53 -07:00
self.lexical_scopes.insert(name.clone(), NameType::LocalVariable(*id));
}
Recursion::Continue
}
_ => Recursion::Continue,
2021-10-24 00:08:26 -07:00
}
2021-10-21 14:46:42 -07:00
}
2021-10-19 13:48:00 -07:00
2021-10-24 00:08:26 -07:00
fn expression(&mut self, expression: &Expression) -> Recursion {
use ExpressionKind::*;
match &expression.kind {
Value(name) => {
self.lookup_name_in_scope(name);
}
NamedStruct { name, fields: _ } => {
self.lookup_name_in_scope(name);
}
2021-10-24 19:05:41 -07:00
Lambda { params, body, .. } => {
let param_names = params.iter().map(|param| param.name.clone());
//TODO need to properly handle closure scope, this is currently broken
//let mut new_scope = self.lexical_scopes.new_scope(Some(ScopeType::Function { name: signature.name.clone() }));
let mut new_scope = ScopeStack::new(Some(ScopeType::Lambda));
2021-10-30 00:01:59 -07:00
for (n, param) in param_names.enumerate() {
2021-10-24 19:05:41 -07:00
new_scope.insert(param, NameType::Param(n as u8));
}
let mut new_resolver =
ScopeResolver { symbol_table: self.symbol_table, lexical_scopes: new_scope };
2021-10-24 19:05:41 -07:00
walk_block(&mut new_resolver, body);
return Recursion::Stop;
}
2021-10-26 11:37:43 -07:00
IfExpression { discriminator, body } => {
if let Some(d) = discriminator.as_ref() {
2021-10-26 13:37:03 -07:00
walk_expression(self, d);
2021-10-26 11:37:43 -07:00
}
let mut resolver = ScopeResolver {
lexical_scopes: self.lexical_scopes.new_scope(Some(ScopeType::PatternMatch)),
symbol_table: self.symbol_table,
2021-10-26 11:37:43 -07:00
};
2021-11-02 20:23:08 -07:00
walk_if_expr_body(&mut resolver, body);
2021-10-26 11:37:43 -07:00
return Recursion::Stop;
}
_ => (),
2021-10-21 21:55:21 -07:00
}
2021-10-24 00:08:26 -07:00
Recursion::Continue
2021-10-21 14:46:42 -07:00
}
2021-10-19 13:48:00 -07:00
2021-10-24 00:08:26 -07:00
fn pattern(&mut self, pat: &Pattern) -> Recursion {
2021-10-21 14:46:42 -07:00
use Pattern::*;
2021-10-21 14:46:42 -07:00
match pat {
2021-10-26 11:37:43 -07:00
Literal(..) | Ignored | TuplePattern(..) => (),
TupleStruct(name, _) | Record(name, _) => {
self.lookup_name_in_scope(name);
2021-10-21 14:46:42 -07:00
}
2021-10-26 11:37:43 -07:00
//TODO this isn't really the right syntax for a VarOrName
2021-10-26 13:12:24 -07:00
VarOrName(QualifiedName { id, components }) => {
2021-10-26 13:02:40 -07:00
if components.len() == 1 {
//TODO need a better way to construct a FQSN from a QualifiedName
let local_name: Rc<String> = components[0].clone();
2021-11-02 01:20:30 -07:00
let lscope = ScopeSegment::Name(Rc::new("<local-case-match>".to_string()));
let fqsn = Fqsn { scopes: vec![lscope, ScopeSegment::Name(local_name.clone())] };
2021-10-26 13:02:40 -07:00
self.symbol_table.add_symbol(id, fqsn, SymbolSpec::LocalVariable);
2021-10-27 01:17:53 -07:00
self.lexical_scopes.insert(local_name, NameType::LocalVariable(*id));
2021-10-26 13:02:40 -07:00
} else {
2021-11-02 01:20:30 -07:00
let fqsn = Fqsn {
scopes: components.iter().map(|name| ScopeSegment::Name(name.clone())).collect(),
};
let def_id = self.symbol_table.symbol_trie.lookup(&fqsn);
if let Some(def_id) = def_id {
self.symbol_table.id_to_def.insert(*id, def_id);
2021-10-26 13:02:40 -07:00
}
}
}
2021-10-21 14:46:42 -07:00
};
2021-10-24 00:08:26 -07:00
Recursion::Continue
2021-10-21 14:46:42 -07:00
}
2021-10-19 13:48:00 -07:00
}