use std::collections::BTreeMap;
use callisto_model::{DepEdge, Diagnostic, Package, PackageId};
use crate::identity::IdentityIndex;
pub use callisto_model::DependencyResolver;
pub struct ManifestWalkResolver {
pub(crate) packages: BTreeMap<PackageId, Package>,
pub(crate) edges: Vec<DepEdge>,
pub(crate) out_index: BTreeMap<PackageId, Vec<usize>>,
pub(crate) in_index: BTreeMap<PackageId, Vec<usize>>,
pub(crate) index: IdentityIndex,
pub(crate) diagnostics: Vec<Diagnostic>,
}
impl ManifestWalkResolver {
pub fn diagnostics(&self) -> &[Diagnostic] {
&self.diagnostics
}
pub fn identity(&self) -> &IdentityIndex {
&self.index
}
pub fn get(&self, id: &PackageId) -> Option<&Package> {
self.packages.get(id)
}
pub fn edges(&self) -> &[DepEdge] {
&self.edges
}
}
impl DependencyResolver for ManifestWalkResolver {
fn packages(&self) -> impl Iterator<Item = &Package> {
self.packages.values()
}
fn dependencies_of(&self, id: &PackageId) -> impl Iterator<Item = &DepEdge> {
let edges = &self.edges;
self.out_index
.get(id)
.into_iter()
.flat_map(|v| v.iter())
.map(move |&idx| &edges[idx])
}
fn dependents_of(&self, id: &PackageId) -> impl Iterator<Item = &DepEdge> {
let edges = &self.edges;
self.in_index
.get(id)
.into_iter()
.flat_map(|v| v.iter())
.map(move |&idx| &edges[idx])
}
fn diagnostics(&self) -> &[Diagnostic] {
&self.diagnostics
}
}