Add DerivationTree.packages() -> HashSet<&P> (#11)

* Add `DeriviationTree.packages -> HashSet<&P>` * Fixup `main` * Use FxHashSet
astral-sh · Dec 8, 2023 · c30acff · c30acff
1 parent a1d584a
commit c30acff
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/examples/unsat_root_message_no_version.rs b/examples/unsat_root_message_no_version.rs
@@ -80,6 +80,13 @@ impl ReportFormatter<Package, Range<SemanticVersion>> for CustomReportFormatter
                     format!("dependencies of {package} at version {set} are unavailable")
                 }
             }
+            External::UnusableDependencies(package, set, ..) => {
+                if set == &Range::full() {
+                    format!("dependencies of {package} are unusable")
+                } else {
+                    format!("dependencies of {package} at version {set} are unusable")
+                }
+            }
             External::FromDependencyOf(package, package_set, dependency, dependency_set) => {
                 if package_set == &Range::full() && dependency_set == &Range::full() {
                     format!("{package} depends on {dependency}")

diff --git a/src/report.rs b/src/report.rs
@@ -6,6 +6,8 @@
 use std::fmt;
 use std::ops::{Deref, DerefMut};
 
+use rustc_hash::FxHashSet;
+
 use crate::package::Package;
 use crate::term::Term;
 use crate::type_aliases::Map;
@@ -72,6 +74,31 @@ pub struct Derived<P: Package, VS: VersionSet> {
 }
 
 impl<P: Package, VS: VersionSet> DerivationTree<P, VS> {
+    /// Get all [Package]s referred to in the deriviation tree.
+    pub fn packages(&self) -> FxHashSet<&P> {
+        let mut packages = FxHashSet::default();
+        match self {
+            Self::External(external) => match external {
+                External::FromDependencyOf(p, _, p2, _) => {
+                    packages.insert(p);
+                    packages.insert(p2);
+                }
+                External::NoVersions(p, _)
+                | External::NotRoot(p, _)
+                | External::UnavailableDependencies(p, _)
+                | External::UnusableDependencies(p, ..) => {
+                    packages.insert(p);
+                }
+            },
+            Self::Derived(derived) => {
+                packages.extend(derived.terms.keys());
+                packages.extend(derived.cause1.packages().iter());
+                packages.extend(derived.cause2.packages().iter());
+            }
+        }
+        packages
+    }
+
     /// Merge the [NoVersions](External::NoVersions) external incompatibilities
     /// with the other one they are matched with
     /// in a derived incompatibility.