erg/crates/erg_compiler/module/global.rs
2023-10-08 18:16:31 +09:00

87 lines
3.1 KiB
Rust

use erg_common::config::ErgConfig;
use erg_common::pathutil::NormalizedPathBuf;
use crate::context::Context;
use super::cache::SharedModuleCache;
use super::errors::{SharedCompileErrors, SharedCompileWarnings};
use super::graph::SharedModuleGraph;
use super::impls::SharedTraitImpls;
use super::index::SharedModuleIndex;
use super::promise::SharedPromises;
#[derive(Debug, Clone, Default)]
pub struct SharedCompilerResource {
pub mod_cache: SharedModuleCache,
pub py_mod_cache: SharedModuleCache,
pub index: SharedModuleIndex,
pub graph: SharedModuleGraph,
/// K: name of a trait, V: (type, monomorphised trait that the type implements)
/// K: トレイトの名前, V: (型, その型が実装する単相化トレイト)
/// e.g. { "Named": [(Type, Named), (Func, Named), ...], "Add": [(Nat, Add(Nat)), (Int, Add(Int)), ...], ... }
pub trait_impls: SharedTraitImpls,
pub promises: SharedPromises,
pub errors: SharedCompileErrors,
pub warns: SharedCompileWarnings,
}
impl SharedCompilerResource {
/// Initialize the shared compiler resource.
/// This API is normally called only once throughout the compilation phase.
pub fn new(cfg: ErgConfig) -> Self {
let graph = SharedModuleGraph::new();
let self_ = Self {
mod_cache: SharedModuleCache::new(),
py_mod_cache: SharedModuleCache::new(),
index: SharedModuleIndex::new(),
graph: graph.clone(),
trait_impls: SharedTraitImpls::new(),
promises: SharedPromises::new(graph, NormalizedPathBuf::from(cfg.input.path())),
errors: SharedCompileErrors::new(),
warns: SharedCompileWarnings::new(),
};
Context::init_builtins(cfg, self_.clone());
self_
}
pub fn inherit<P: Into<NormalizedPathBuf>>(&self, path: P) -> Self {
let mut _self = self.clone();
_self.promises.path = path.into();
_self
}
/// Clear all but builtin modules
pub fn clear_all(&self) {
self.mod_cache.initialize();
self.py_mod_cache.initialize();
self.index.initialize();
self.graph.initialize();
// self.trait_impls.initialize();
self.promises.initialize();
self.errors.clear();
self.warns.clear();
}
/// Clear all information about the module.
/// Graph information is not cleared (due to ELS).
pub fn clear(&self, path: &NormalizedPathBuf) {
for child in self.graph.children(path) {
self.clear(&child);
}
self.mod_cache.remove(path);
self.py_mod_cache.remove(path);
self.index.remove_path(path);
// self.graph.remove(path);
self.promises.remove(path);
self.errors.remove(path);
self.warns.remove(path);
}
pub fn rename_path(&self, old: &NormalizedPathBuf, new: NormalizedPathBuf) {
self.mod_cache.rename_path(old, new.clone());
self.py_mod_cache.rename_path(old, new.clone());
self.index.rename_path(old, new.clone());
self.graph.rename_path(old, new.clone());
self.promises.rename(old, new);
}
}