mirror of https://github.com/astral-sh/uv
39 lines
1.3 KiB
Rust
39 lines
1.3 KiB
Rust
use rustc_hash::FxHashMap;
|
|
use uv_distribution_types::IndexMetadata;
|
|
use uv_normalize::PackageName;
|
|
|
|
use crate::ResolveError;
|
|
use crate::resolver::ResolverEnvironment;
|
|
|
|
/// See [`crate::resolver::ForkState`].
|
|
#[derive(Default, Debug, Clone)]
|
|
pub(crate) struct ForkIndexes(FxHashMap<PackageName, IndexMetadata>);
|
|
|
|
impl ForkIndexes {
|
|
/// Get the [`Index`] previously used for a package in this fork.
|
|
pub(crate) fn get(&self, package_name: &PackageName) -> Option<&IndexMetadata> {
|
|
self.0.get(package_name)
|
|
}
|
|
|
|
/// Check that this is the only [`Index`] used for this package in this fork.
|
|
pub(crate) fn insert(
|
|
&mut self,
|
|
package_name: &PackageName,
|
|
index: &IndexMetadata,
|
|
env: &ResolverEnvironment,
|
|
) -> Result<(), ResolveError> {
|
|
if let Some(previous) = self.0.insert(package_name.clone(), index.clone()) {
|
|
if &previous != index {
|
|
let mut conflicts = vec![previous.url, index.url.clone()];
|
|
conflicts.sort();
|
|
return Err(ResolveError::ConflictingIndexesForEnvironment {
|
|
package_name: package_name.clone(),
|
|
indexes: conflicts,
|
|
env: env.clone(),
|
|
});
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|