mirror of https://github.com/astral-sh/uv
48 lines
1.6 KiB
Rust
48 lines
1.6 KiB
Rust
use rustc_hash::{FxHashMap, FxHashSet};
|
|
|
|
use pep440_rs::Version;
|
|
use pep508_rs::Requirement;
|
|
use puffin_normalize::PackageName;
|
|
|
|
/// A set of package versions that are permitted, even if they're marked as yanked by the
|
|
/// relevant index.
|
|
#[derive(Debug, Default)]
|
|
pub struct AllowedYanks(FxHashMap<PackageName, FxHashSet<Version>>);
|
|
|
|
impl AllowedYanks {
|
|
/// Returns `true` if the given package version is allowed, even if it's marked as yanked by
|
|
/// the relevant index.
|
|
pub(crate) fn allowed(&self, package_name: &PackageName, version: &Version) -> bool {
|
|
self.0
|
|
.get(package_name)
|
|
.map(|allowed_yanks| allowed_yanks.contains(version))
|
|
.unwrap_or_default()
|
|
}
|
|
}
|
|
|
|
impl<'a> FromIterator<&'a Requirement> for AllowedYanks {
|
|
fn from_iter<T: IntoIterator<Item = &'a Requirement>>(iter: T) -> Self {
|
|
let mut allowed_yanks = FxHashMap::<PackageName, FxHashSet<Version>>::default();
|
|
for requirement in iter {
|
|
let Some(pep508_rs::VersionOrUrl::VersionSpecifier(specifiers)) =
|
|
&requirement.version_or_url
|
|
else {
|
|
continue;
|
|
};
|
|
let [specifier] = specifiers.as_ref() else {
|
|
continue;
|
|
};
|
|
if matches!(
|
|
specifier.operator(),
|
|
pep440_rs::Operator::Equal | pep440_rs::Operator::ExactEqual
|
|
) {
|
|
allowed_yanks
|
|
.entry(requirement.name.clone())
|
|
.or_default()
|
|
.insert(specifier.version().clone());
|
|
}
|
|
}
|
|
Self(allowed_yanks)
|
|
}
|
|
}
|