mirror of https://github.com/astral-sh/uv
122 lines
4.4 KiB
Rust
122 lines
4.4 KiB
Rust
use pypi_types::RequirementSource;
|
|
use rustc_hash::FxHashSet;
|
|
|
|
use pep508_rs::MarkerEnvironment;
|
|
use uv_normalize::PackageName;
|
|
|
|
use crate::{DependencyMode, Manifest};
|
|
|
|
#[derive(Debug, Default, Clone, Copy, PartialEq, Eq, serde::Deserialize)]
|
|
#[serde(deny_unknown_fields, rename_all = "kebab-case")]
|
|
#[cfg_attr(feature = "clap", derive(clap::ValueEnum))]
|
|
#[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,
|
|
}
|
|
|
|
impl std::fmt::Display for PreReleaseMode {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
Self::Disallow => write!(f, "disallow"),
|
|
Self::Allow => write!(f, "allow"),
|
|
Self::IfNecessary => write!(f, "if-necessary"),
|
|
Self::Explicit => write!(f, "explicit"),
|
|
Self::IfNecessaryOrExplicit => write!(f, "if-necessary-or-explicit"),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// 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: Option<&MarkerEnvironment>,
|
|
dependencies: DependencyMode,
|
|
) -> Self {
|
|
match mode {
|
|
PreReleaseMode::Disallow => Self::Disallow,
|
|
PreReleaseMode::Allow => Self::Allow,
|
|
PreReleaseMode::IfNecessary => Self::IfNecessary,
|
|
PreReleaseMode::Explicit => Self::Explicit(
|
|
manifest
|
|
.requirements(markers, dependencies)
|
|
.filter(|requirement| {
|
|
let RequirementSource::Registry { specifier, .. } = &requirement.source
|
|
else {
|
|
return false;
|
|
};
|
|
specifier
|
|
.iter()
|
|
.any(pep440_rs::VersionSpecifier::any_prerelease)
|
|
})
|
|
.map(|requirement| requirement.name.clone())
|
|
.collect(),
|
|
),
|
|
PreReleaseMode::IfNecessaryOrExplicit => Self::IfNecessaryOrExplicit(
|
|
manifest
|
|
.requirements(markers, dependencies)
|
|
.filter(|requirement| {
|
|
let RequirementSource::Registry { specifier, .. } = &requirement.source
|
|
else {
|
|
return false;
|
|
};
|
|
specifier
|
|
.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),
|
|
}
|
|
}
|
|
}
|