use std::borrow::Borrow;
use std::fmt;
use std::hash::Hash;
use erg_common::dict::Dict;
use erg_common::set::Set;
use erg_common::shared::{
MappedRwLockReadGuard, MappedRwLockWriteGuard, RwLockReadGuard, RwLockWriteGuard, Shared,
};
use erg_common::Str;
use crate::context::TraitImpl;
#[derive(Debug, Default)]
pub struct TraitImpls {
cache: Dict<Str, Set<TraitImpl>>,
}
impl fmt::Display for TraitImpls {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "TraitImpls {{")?;
for (name, impls) in self.cache.iter() {
writeln!(f, "{name}: {impls}, ")?;
}
write!(f, "}}")
}
}
impl TraitImpls {
pub fn new() -> Self {
Self { cache: Dict::new() }
}
pub fn get<P: Eq + Hash + ?Sized>(&self, path: &P) -> Option<&Set<TraitImpl>>
where
Str: Borrow<P>,
{
self.cache.get(path)
}
pub fn get_mut<Q: Eq + Hash + ?Sized>(&mut self, path: &Q) -> Option<&mut Set<TraitImpl>>
where
Str: Borrow<Q>,
{
self.cache.get_mut(path)
}
pub fn register(&mut self, name: Str, impls: Set<TraitImpl>) {
self.cache.insert(name, impls);
}
pub fn remove<Q: Eq + Hash + ?Sized>(&mut self, path: &Q) -> Option<Set<TraitImpl>>
where
Str: Borrow<Q>,
{
self.cache.remove(path)
}
pub fn rename<Q: Eq + Hash + ?Sized>(&mut self, old: &Q, new: Str)
where
Str: Borrow<Q>,
{
if let Some(impls) = self.remove(old) {
self.register(new, impls);
}
}
pub fn initialize(&mut self) {
self.cache.clear();
}
}
#[derive(Debug, Clone, Default)]
pub struct SharedTraitImpls(Shared<TraitImpls>);
impl fmt::Display for SharedTraitImpls {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Shared{}", self.0)
}
}
impl SharedTraitImpls {
pub fn new() -> Self {
Self(Shared::new(TraitImpls::new()))
}
pub fn get<Q: Eq + Hash + ?Sized>(
&self,
path: &Q,
) -> Option<MappedRwLockReadGuard<Set<TraitImpl>>>
where
Str: Borrow<Q>,
{
if self.0.borrow().get(path).is_some() {
Some(RwLockReadGuard::map(self.0.borrow(), |tis| {
tis.get(path).unwrap()
}))
} else {
None
}
}
pub fn get_mut<Q: Eq + Hash + ?Sized>(
&self,
path: &Q,
) -> Option<MappedRwLockWriteGuard<Set<TraitImpl>>>
where
Str: Borrow<Q>,
{
if self.0.borrow().get(path).is_some() {
Some(RwLockWriteGuard::map(self.0.borrow_mut(), |tis| {
tis.get_mut(path).unwrap()
}))
} else {
None
}
}
pub fn register(&self, name: Str, impls: Set<TraitImpl>) {
self.0.borrow_mut().register(name, impls);
}
pub fn remove<Q: Eq + Hash + ?Sized>(&self, path: &Q) -> Option<Set<TraitImpl>>
where
Str: Borrow<Q>,
{
self.0.borrow_mut().remove(path)
}
pub fn rename<Q: Eq + Hash + ?Sized>(&self, old: &Q, new: Str)
where
Str: Borrow<Q>,
{
self.0.borrow_mut().rename(old, new);
}
pub fn ref_inner(&self) -> MappedRwLockReadGuard<Dict<Str, Set<TraitImpl>>> {
RwLockReadGuard::map(self.0.borrow(), |tis| &tis.cache)
}
pub fn initialize(&self) {
self.0.borrow_mut().initialize();
}
}