147 lines
3.8 KiB
Rust
147 lines
3.8 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use parser::ast::TopLevel;
|
|
|
|
use parser::ast::{ParamList, TypeExpr};
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct TypeSymbol(pub String);
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct VarSymbol {
|
|
pub var_type: TypeSymbol,
|
|
pub name: String,
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct FunctionSymbol {
|
|
pub export: bool,
|
|
pub name: String,
|
|
pub params: HashMap<String, VarSymbol>,
|
|
pub return_type: Option<TypeSymbol>,
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct StructSymbol {
|
|
pub export: bool,
|
|
pub name: String,
|
|
pub fields: HashMap<String, VarSymbol>,
|
|
pub methods: HashMap<String, FunctionSymbol>,
|
|
}
|
|
|
|
impl TypeSymbol {
|
|
pub fn from_ast(expr: TypeExpr) -> Self {
|
|
TypeSymbol(match expr {
|
|
TypeExpr::Identifier(i) => i.to_string(),
|
|
})
|
|
}
|
|
}
|
|
|
|
impl FunctionSymbol {
|
|
pub fn from_ast(
|
|
export: bool,
|
|
name: String,
|
|
params: ParamList,
|
|
return_type: Option<TypeExpr>,
|
|
) -> Self {
|
|
Self {
|
|
export: export,
|
|
name: name.clone(),
|
|
params: params
|
|
.0
|
|
.iter()
|
|
.map(|(name, (_, v))| {
|
|
(
|
|
name.clone(),
|
|
VarSymbol {
|
|
name: name.clone(),
|
|
var_type: TypeSymbol(match v {
|
|
TypeExpr::Identifier(i) => i.to_string(),
|
|
}),
|
|
},
|
|
)
|
|
})
|
|
.collect(),
|
|
return_type: return_type.map(TypeSymbol::from_ast),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl StructSymbol {
|
|
pub fn from_ast(export: bool, name: String, fields: ParamList) -> Self {
|
|
Self {
|
|
export,
|
|
name,
|
|
fields: fields
|
|
.0
|
|
.iter()
|
|
.map(|(name, (_, v))| {
|
|
(
|
|
name.clone(),
|
|
VarSymbol {
|
|
name: name.clone(),
|
|
var_type: TypeSymbol(match v {
|
|
TypeExpr::Identifier(i) => i.to_string(),
|
|
}),
|
|
},
|
|
)
|
|
})
|
|
.collect(),
|
|
// TODO - parse struct methods
|
|
methods: HashMap::new(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct TopLevelSymbolScope {
|
|
pub structs: HashMap<String, StructSymbol>,
|
|
pub functions: HashMap<String, FunctionSymbol>,
|
|
}
|
|
|
|
impl TopLevelSymbolScope {
|
|
pub fn from(top_level: &Vec<TopLevel>) -> Self {
|
|
let mut scope = TopLevelSymbolScope {
|
|
structs: HashMap::new(),
|
|
functions: HashMap::new(),
|
|
};
|
|
|
|
for top in top_level {
|
|
match top {
|
|
TopLevel::Import(_) => unimplemented!(),
|
|
|
|
TopLevel::FunctionDecl {
|
|
export,
|
|
name,
|
|
params,
|
|
return_type,
|
|
..
|
|
} => {
|
|
scope.functions.insert(
|
|
name.clone(),
|
|
FunctionSymbol::from_ast(
|
|
*export,
|
|
name.clone(),
|
|
params.clone(),
|
|
return_type.clone(),
|
|
),
|
|
);
|
|
}
|
|
|
|
TopLevel::StructDecl {
|
|
export,
|
|
name,
|
|
fields,
|
|
} => {
|
|
scope.structs.insert(
|
|
name.clone(),
|
|
StructSymbol::from_ast(*export, name.clone(), fields.clone()),
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
scope
|
|
}
|
|
}
|