You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.3 KiB
78 lines
2.3 KiB
use crate::models::{CargoCrate, Crate, CrateId};
|
|
use std::collections::{BTreeMap, BTreeSet};
|
|
use std::path::PathBuf;
|
|
|
|
/// Dependency graph in a workspace
|
|
pub struct DepGraph {
|
|
/// Mapping of crates in the workspace
|
|
members: BTreeMap<CrateId, Crate>,
|
|
/// Map of crates and the members that depend on them
|
|
dependents: BTreeMap<CrateId, BTreeSet<CrateId>>,
|
|
}
|
|
|
|
impl DepGraph {
|
|
/// Create a new, empty dependency graph
|
|
pub fn new() -> Self {
|
|
Self {
|
|
members: Default::default(),
|
|
dependents: Default::default(),
|
|
}
|
|
}
|
|
|
|
pub fn add_crate(&mut self, cc: CargoCrate) {
|
|
let cc = Crate::new(cc);
|
|
self.members.insert(cc.id, cc);
|
|
}
|
|
|
|
/// Cache the dependents graph for all crates
|
|
pub fn finalise(&mut self) {
|
|
let mut members = self.members.clone();
|
|
members.iter_mut().for_each(|(_, c)| c.process(&self));
|
|
|
|
members.iter().for_each(|(id, _crate)| {
|
|
_crate.dependencies.iter().for_each(|dep_id| {
|
|
self.dependents.entry(*dep_id).or_default().insert(*id);
|
|
});
|
|
});
|
|
let _ = std::mem::replace(&mut self.members, members);
|
|
}
|
|
|
|
/// Get a crate by ID
|
|
pub fn get_crate(&self, id: CrateId) -> &Crate {
|
|
self.members.get(&id).as_ref().unwrap()
|
|
}
|
|
|
|
/// Get mutable access to a crate by ID
|
|
pub fn mut_crate(&mut self, id: CrateId) -> &mut Crate {
|
|
self.members.get_mut(&id).unwrap()
|
|
}
|
|
|
|
/// Find a crate via it's name
|
|
pub fn find_crate(&self, name: &String) -> Option<CrateId> {
|
|
self.members
|
|
.iter()
|
|
.find(|(_, c)| c.name() == name)
|
|
.map(|(id, _)| *id)
|
|
}
|
|
|
|
/// Find a crate by it's path-offset in the workspace
|
|
pub fn find_crate_by_path(&self, name: &String) -> Option<CrateId> {
|
|
self.members
|
|
.iter()
|
|
.find(|(_, c)| c.path() == &PathBuf::new().join(name))
|
|
.map(|(id, _)| *id)
|
|
}
|
|
|
|
/// Get a crate's dependents
|
|
pub fn get_dependents(&self, id: CrateId) -> Vec<CrateId> {
|
|
self.dependents
|
|
.get(&id)
|
|
.as_ref()
|
|
.map(|set| set.iter().cloned().collect())
|
|
.unwrap_or(vec![])
|
|
}
|
|
|
|
pub fn get_all(&self) -> Vec<&Crate> {
|
|
self.members.iter().map(|(_, c)| c).collect()
|
|
}
|
|
}
|
|
|