mirror of
https://github.com/astral-sh/uv.git
synced 2025-11-03 05:03:46 +00:00
121 lines
4.5 KiB
Rust
121 lines
4.5 KiB
Rust
use rustc_hash::FxHashSet;
|
|
|
|
use pep508_rs::{MarkerEnvironment, VersionOrUrl};
|
|
use uv_normalize::PackageName;
|
|
|
|
use crate::Manifest;
|
|
|
|
#[derive(Debug, Default, Clone, Copy, PartialEq, Eq)]
|
|
#[cfg_attr(feature = "clap", derive(clap::ValueEnum))]
|
|
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
|
|
#[cfg_attr(
|
|
feature = "serde",
|
|
serde(deny_unknown_fields, rename_all = "kebab-case")
|
|
)]
|
|
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
|
|
pub enum PreReleaseMode {
|
|
/// Disallow all pre-release versions.
|
|
Disallow,
|
|
|
|
/// Allow all pre-release versions.
|
|
Allow,
|
|
|
|
/// Allow pre-release versions if all versions of a package are pre-release.
|
|
IfNecessary,
|
|
|
|
/// Allow pre-release versions for first-party packages with explicit pre-release markers in
|
|
/// their version requirements.
|
|
Explicit,
|
|
|
|
/// Allow pre-release versions if all versions of a package are pre-release, or if the package
|
|
/// has an explicit pre-release marker in its version requirements.
|
|
#[default]
|
|
IfNecessaryOrExplicit,
|
|
}
|
|
|
|
/// Like [`PreReleaseMode`], but with any additional information required to select a candidate,
|
|
/// like the set of direct dependencies.
|
|
#[derive(Debug, Clone)]
|
|
pub(crate) enum PreReleaseStrategy {
|
|
/// Disallow all pre-release versions.
|
|
Disallow,
|
|
|
|
/// Allow all pre-release versions.
|
|
Allow,
|
|
|
|
/// Allow pre-release versions if all versions of a package are pre-release.
|
|
IfNecessary,
|
|
|
|
/// Allow pre-release versions for first-party packages with explicit pre-release markers in
|
|
/// their version requirements.
|
|
Explicit(FxHashSet<PackageName>),
|
|
|
|
/// Allow pre-release versions if all versions of a package are pre-release, or if the package
|
|
/// has an explicit pre-release marker in its version requirements.
|
|
IfNecessaryOrExplicit(FxHashSet<PackageName>),
|
|
}
|
|
|
|
impl PreReleaseStrategy {
|
|
pub(crate) fn from_mode(
|
|
mode: PreReleaseMode,
|
|
manifest: &Manifest,
|
|
markers: &MarkerEnvironment,
|
|
) -> Self {
|
|
match mode {
|
|
PreReleaseMode::Disallow => Self::Disallow,
|
|
PreReleaseMode::Allow => Self::Allow,
|
|
PreReleaseMode::IfNecessary => Self::IfNecessary,
|
|
PreReleaseMode::Explicit => Self::Explicit(
|
|
manifest
|
|
.requirements(markers)
|
|
.filter(|requirement| {
|
|
let Some(version_or_url) = &requirement.version_or_url else {
|
|
return false;
|
|
};
|
|
let version_specifiers = match version_or_url {
|
|
VersionOrUrl::VersionSpecifier(version_specifiers) => {
|
|
version_specifiers
|
|
}
|
|
VersionOrUrl::Url(_) => return false,
|
|
};
|
|
version_specifiers
|
|
.iter()
|
|
.any(pep440_rs::VersionSpecifier::any_prerelease)
|
|
})
|
|
.map(|requirement| requirement.name.clone())
|
|
.collect(),
|
|
),
|
|
PreReleaseMode::IfNecessaryOrExplicit => Self::IfNecessaryOrExplicit(
|
|
manifest
|
|
.requirements(markers)
|
|
.filter(|requirement| {
|
|
let Some(version_or_url) = &requirement.version_or_url else {
|
|
return false;
|
|
};
|
|
let version_specifiers = match version_or_url {
|
|
VersionOrUrl::VersionSpecifier(version_specifiers) => {
|
|
version_specifiers
|
|
}
|
|
VersionOrUrl::Url(_) => return false,
|
|
};
|
|
version_specifiers
|
|
.iter()
|
|
.any(pep440_rs::VersionSpecifier::any_prerelease)
|
|
})
|
|
.map(|requirement| requirement.name.clone())
|
|
.collect(),
|
|
),
|
|
}
|
|
}
|
|
|
|
/// Returns `true` if a [`PackageName`] is allowed to have pre-release versions.
|
|
pub(crate) fn allows(&self, package: &PackageName) -> bool {
|
|
match self {
|
|
Self::Disallow => false,
|
|
Self::Allow => true,
|
|
Self::IfNecessary => false,
|
|
Self::Explicit(packages) => packages.contains(package),
|
|
Self::IfNecessaryOrExplicit(packages) => packages.contains(package),
|
|
}
|
|
}
|
|
}
|