schala/schala-lang/language/src/scope_resolution.rs

144 lines
3.7 KiB
Rust
Raw Normal View History

use crate::symbol_table::{ScopeSegment, ScopeSegmentKind, FullyQualifiedSymbolName};
2019-09-03 01:42:28 -07:00
use crate::ast::*;
2019-09-03 02:59:19 -07:00
pub struct ScopeResolver {
2019-09-03 02:19:37 -07:00
}
2019-09-03 02:59:19 -07:00
impl ScopeResolver {
pub fn new() -> ScopeResolver {
ScopeResolver { }
}
pub fn resolve(&mut self, ast: &mut AST) -> Result<(), String> {
println!("Resolving scopes - nothing so far!");
for statement in ast.0.iter_mut() {
match statement.mut_node() {
Statement::Declaration(ref mut decl) => self.decl(decl),
Statement::ExpressionStatement(ref mut expr) => self.expr(expr),
}?;
}
Ok(())
}
fn decl(&mut self, decl: &mut Declaration) -> Result<(), String> {
match decl {
Declaration::Binding { expr, .. } => self.expr(expr),
_ => Ok(()),
}
2019-09-03 02:59:19 -07:00
}
fn expr(&mut self, expr: &mut Meta<Expression>) -> Result<(), String> {
2019-09-03 10:23:38 -07:00
use ExpressionKind::*;
let inner_expr = expr.mut_node();
match &mut inner_expr.kind {
ExpressionKind::Value(qualified_name) => {
2019-09-06 10:03:50 -07:00
let fqsn = lookup_name_in_scope(&qualified_name.node());
expr.fqsn = Some(fqsn);
},
2019-09-03 10:23:38 -07:00
NamedStruct { name, .. } => {
2019-09-06 17:19:41 -07:00
let fqsn = lookup_name_in_scope(&name.node());
2019-09-03 10:23:38 -07:00
expr.fqsn = Some(fqsn);
},
BinExp(_, ref mut lhs, ref mut rhs) => {
self.expr(lhs)?;
self.expr(rhs)?;
},
PrefixExp(_, ref mut arg) => {
self.expr(arg)?;
},
TupleLiteral(exprs) => {
for expr in exprs.iter_mut() {
self.expr(expr)?;
}
},
Call { ref mut f, arguments } => {
self.expr(f)?;
for arg in arguments.iter_mut() {
self.invoc(arg)?;
}
2019-09-08 02:11:15 -07:00
},
IfExpression { ref mut body, .. } => match &mut **body {
IfExpressionBody::SimplePatternMatch(ref mut pat, _, _) => {
self.pattern(pat)?;
},
IfExpressionBody::GuardList(guardarms) => {
for arm in guardarms.iter_mut() {
if let Guard::Pat(ref mut pat) = arm.guard {
self.pattern(pat)?;
}
}
}
_ => ()
},
_ => ()
};
2019-09-03 02:59:19 -07:00
Ok(())
}
2019-09-03 10:23:38 -07:00
fn invoc(&mut self, invoc: &mut InvocationArgument) -> Result<(), String> {
use InvocationArgument::*;
match invoc {
Positional(expr) => self.expr(expr),
Keyword { expr, .. } => self.expr(expr),
_ => Ok(())
}
}
2019-09-08 02:11:15 -07:00
fn pattern(&mut self, pat: &mut Pattern) -> Result<(), String> {
use Pattern::*;
match pat {
Ignored => (),
TuplePattern(patterns) => {
for pat in patterns {
self.pattern(pat)?;
}
},
Literal(_) => (),
TupleStruct(name, patterns) => {
self.qualified_name(name);
for pat in patterns {
self.pattern(pat)?;
}
},
Record(name, key_patterns) => {
self.qualified_name(name);
for (_, pat) in key_patterns {
self.pattern(pat)?;
}
},
VarOrName(name) => {
self.qualified_name(name);
},
};
Ok(())
}
fn qualified_name(&mut self, meta_qualified_name: &mut Meta<QualifiedName>, ) {
let inner_name = meta_qualified_name.node();
let fqsn = lookup_name_in_scope(inner_name);
meta_qualified_name.fqsn = Some(fqsn);
}
}
//TODO this is incomplete
fn lookup_name_in_scope(sym_name: &QualifiedName) -> FullyQualifiedSymbolName {
let QualifiedName(vec) = sym_name;
let len = vec.len();
let new_vec: Vec<ScopeSegment> = vec.iter().enumerate().map(|(i, name)| {
let kind = if i == (len - 1) {
ScopeSegmentKind::Terminal
} else {
ScopeSegmentKind::Type
};
ScopeSegment { name: name.clone(), kind }
}).collect();
FullyQualifiedSymbolName(new_vec)
}
2019-09-03 02:59:19 -07:00
#[cfg(test)]
mod tests {
#[test]
fn basic_scope() {
}
2019-09-03 01:42:28 -07:00
}