mirror of https://github.com/astral-sh/uv
50 lines
1.6 KiB
Rust
50 lines
1.6 KiB
Rust
use std::hash::BuildHasherDefault;
|
|
|
|
use rustc_hash::FxHashMap;
|
|
|
|
use pep508_rs::Requirement;
|
|
use uv_normalize::PackageName;
|
|
|
|
/// A set of constraints for a set of requirements.
|
|
#[derive(Debug, Default, Clone)]
|
|
pub struct Constraints(FxHashMap<PackageName, Vec<Requirement>>);
|
|
|
|
impl Constraints {
|
|
/// Create a new set of constraints from a set of requirements.
|
|
pub fn from_requirements(requirements: Vec<Requirement>) -> Self {
|
|
let mut constraints: FxHashMap<PackageName, Vec<Requirement>> =
|
|
FxHashMap::with_capacity_and_hasher(requirements.len(), BuildHasherDefault::default());
|
|
for requirement in requirements {
|
|
constraints
|
|
.entry(requirement.name.clone())
|
|
.or_default()
|
|
.push(requirement);
|
|
}
|
|
Self(constraints)
|
|
}
|
|
|
|
/// Return an iterator over all [`Requirement`]s in the constraint set.
|
|
pub fn requirements(&self) -> impl Iterator<Item = &Requirement> {
|
|
self.0.values().flat_map(|requirements| requirements.iter())
|
|
}
|
|
|
|
/// Get the constraints for a package.
|
|
pub fn get(&self, name: &PackageName) -> Option<&Vec<Requirement>> {
|
|
self.0.get(name)
|
|
}
|
|
|
|
/// Apply the constraints to a set of requirements.
|
|
pub fn apply<'a>(
|
|
&'a self,
|
|
requirements: impl IntoIterator<Item = &'a Requirement>,
|
|
) -> impl Iterator<Item = &Requirement> {
|
|
requirements.into_iter().flat_map(|requirement| {
|
|
std::iter::once(requirement).chain(
|
|
self.get(&requirement.name)
|
|
.into_iter()
|
|
.flat_map(|constraints| constraints.iter()),
|
|
)
|
|
})
|
|
}
|
|
}
|