use sway_error::error::CompileError;
use sway_types::Spanned;
use crate::{
error::*,
language::{ty, CallPath, Visibility},
CompileResult, Engines, Ident,
};
use super::{module::Module, namespace::Namespace, Path};
#[derive(Clone, Debug)]
pub struct Root {
pub(crate) module: Module,
}
impl Root {
pub(crate) fn resolve_call_path(
&self,
mod_path: &Path,
call_path: &CallPath,
) -> CompileResult<&ty::TyDeclaration> {
let symbol_path: Vec<_> = mod_path
.iter()
.chain(&call_path.prefixes)
.cloned()
.collect();
self.resolve_symbol(&symbol_path, &call_path.suffix)
}
pub(crate) fn resolve_call_path_with_visibility_check(
&self,
engines: Engines<'_>,
mod_path: &Path,
call_path: &CallPath,
) -> CompileResult<&ty::TyDeclaration> {
let mut warnings = vec![];
let mut errors = vec![];
let result = self.resolve_call_path(mod_path, call_path);
if call_path.prefixes.is_empty() {
return result;
}
if let CompileResult {
value: Some(decl), ..
} = result
{
let visibility = check!(
decl.visibility(engines.de()),
return err(warnings, errors),
warnings,
errors
);
if visibility != Visibility::Public {
errors.push(CompileError::ImportPrivateSymbol {
name: call_path.suffix.clone(),
span: call_path.suffix.span(),
});
return err(warnings, errors);
}
}
result
}
pub(crate) fn resolve_symbol(
&self,
mod_path: &Path,
symbol: &Ident,
) -> CompileResult<&ty::TyDeclaration> {
self.check_submodule(mod_path).flat_map(|module| {
let true_symbol = self[mod_path]
.use_aliases
.get(symbol.as_str())
.unwrap_or(symbol);
match module.use_synonyms.get(symbol) {
Some((src_path, _, _)) if mod_path != src_path => {
self.resolve_symbol(src_path, true_symbol)
}
_ => CompileResult::from(module.check_symbol(true_symbol)),
}
})
}
}
impl std::ops::Deref for Root {
type Target = Module;
fn deref(&self) -> &Self::Target {
&self.module
}
}
impl std::ops::DerefMut for Root {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.module
}
}
impl From<Module> for Root {
fn from(module: Module) -> Self {
Root { module }
}
}
impl From<Namespace> for Root {
fn from(namespace: Namespace) -> Self {
namespace.root
}
}