78 lines
1.5 KiB
Rust
78 lines
1.5 KiB
Rust
use crate::common::spans::Span;
|
|
use crate::frontends::pbs::types::PbsType;
|
|
use std::collections::HashMap;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub enum Visibility {
|
|
FilePrivate,
|
|
Mod,
|
|
Pub,
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub enum SymbolKind {
|
|
Function,
|
|
Service,
|
|
Struct,
|
|
Contract,
|
|
ErrorType,
|
|
Local,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub enum Namespace {
|
|
Type,
|
|
Value,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct Symbol {
|
|
pub name: String,
|
|
pub kind: SymbolKind,
|
|
pub namespace: Namespace,
|
|
pub visibility: Visibility,
|
|
pub ty: Option<PbsType>,
|
|
pub is_host: bool,
|
|
pub span: Span,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct SymbolTable {
|
|
pub symbols: HashMap<String, Symbol>,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct ModuleSymbols {
|
|
pub type_symbols: SymbolTable,
|
|
pub value_symbols: SymbolTable,
|
|
}
|
|
|
|
impl ModuleSymbols {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
type_symbols: SymbolTable::new(),
|
|
value_symbols: SymbolTable::new(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl SymbolTable {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
symbols: HashMap::new(),
|
|
}
|
|
}
|
|
|
|
pub fn insert(&mut self, symbol: Symbol) -> Result<(), Symbol> {
|
|
if let Some(existing) = self.symbols.get(&symbol.name) {
|
|
return Err(existing.clone());
|
|
}
|
|
self.symbols.insert(symbol.name.clone(), symbol);
|
|
Ok(())
|
|
}
|
|
|
|
pub fn get(&self, name: &str) -> Option<&Symbol> {
|
|
self.symbols.get(name)
|
|
}
|
|
}
|