-
Notifications
You must be signed in to change notification settings - Fork 38
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move offline provider into a separate module (#283)
- Loading branch information
Showing
3 changed files
with
125 additions
and
116 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,120 @@ | ||
| use std::cmp::Reverse; | ||
| use std::collections::BTreeMap; | ||
| use std::convert::Infallible; | ||
|
|
||
| use crate::{Dependencies, DependencyConstraints, DependencyProvider, Map, Package, VersionSet}; | ||
|
|
||
| /// A basic implementation of [DependencyProvider]. | ||
| #[derive(Debug, Clone, Default)] | ||
| #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))] | ||
| #[cfg_attr( | ||
| feature = "serde", | ||
| serde(bound( | ||
| serialize = "VS::V: serde::Serialize, VS: serde::Serialize, P: serde::Serialize", | ||
| deserialize = "VS::V: serde::Deserialize<'de>, VS: serde::Deserialize<'de>, P: serde::Deserialize<'de>" | ||
| )) | ||
| )] | ||
| #[cfg_attr(feature = "serde", serde(transparent))] | ||
| pub struct OfflineDependencyProvider<P: Package, VS: VersionSet> { | ||
| dependencies: Map<P, BTreeMap<VS::V, DependencyConstraints<P, VS>>>, | ||
| } | ||
|
|
||
| impl<P: Package, VS: VersionSet> OfflineDependencyProvider<P, VS> { | ||
| /// Creates an empty OfflineDependencyProvider with no dependencies. | ||
| pub fn new() -> Self { | ||
| Self { | ||
| dependencies: Map::default(), | ||
| } | ||
| } | ||
|
|
||
| /// Registers the dependencies of a package and version pair. | ||
| /// Dependencies must be added with a single call to | ||
| /// [add_dependencies](OfflineDependencyProvider::add_dependencies). | ||
| /// All subsequent calls to | ||
| /// [add_dependencies](OfflineDependencyProvider::add_dependencies) for a given | ||
| /// package version pair will replace the dependencies by the new ones. | ||
| /// | ||
| /// The API does not allow to add dependencies one at a time to uphold an assumption that | ||
| /// [OfflineDependencyProvider.get_dependencies(p, v)](OfflineDependencyProvider::get_dependencies) | ||
| /// provides all dependencies of a given package (p) and version (v) pair. | ||
| pub fn add_dependencies<I: IntoIterator<Item = (P, VS)>>( | ||
| &mut self, | ||
| package: P, | ||
| version: impl Into<VS::V>, | ||
| dependencies: I, | ||
| ) { | ||
| let package_deps = dependencies.into_iter().collect(); | ||
| let v = version.into(); | ||
| *self | ||
| .dependencies | ||
| .entry(package) | ||
| .or_default() | ||
| .entry(v) | ||
| .or_default() = package_deps; | ||
| } | ||
|
|
||
| /// Lists packages that have been saved. | ||
| pub fn packages(&self) -> impl Iterator<Item = &P> { | ||
| self.dependencies.keys() | ||
| } | ||
|
|
||
| /// Lists versions of saved packages in sorted order. | ||
| /// Returns [None] if no information is available regarding that package. | ||
| pub fn versions(&self, package: &P) -> Option<impl Iterator<Item = &VS::V>> { | ||
| self.dependencies.get(package).map(|k| k.keys()) | ||
| } | ||
|
|
||
| /// Lists dependencies of a given package and version. | ||
| /// Returns [None] if no information is available regarding that package and version pair. | ||
| fn dependencies(&self, package: &P, version: &VS::V) -> Option<DependencyConstraints<P, VS>> { | ||
| self.dependencies.get(package)?.get(version).cloned() | ||
| } | ||
| } | ||
|
|
||
| /// An implementation of [DependencyProvider] that | ||
| /// contains all dependency information available in memory. | ||
| /// Currently packages are picked with the fewest versions contained in the constraints first. | ||
| /// But, that may change in new versions if better heuristics are found. | ||
| /// Versions are picked with the newest versions first. | ||
| impl<P: Package, VS: VersionSet> DependencyProvider for OfflineDependencyProvider<P, VS> { | ||
| type P = P; | ||
| type V = VS::V; | ||
| type VS = VS; | ||
| type M = String; | ||
|
|
||
| type Err = Infallible; | ||
|
|
||
| #[inline] | ||
| fn choose_version(&self, package: &P, range: &VS) -> Result<Option<VS::V>, Infallible> { | ||
| Ok(self | ||
| .dependencies | ||
| .get(package) | ||
| .and_then(|versions| versions.keys().rev().find(|v| range.contains(v)).cloned())) | ||
| } | ||
|
|
||
| type Priority = Reverse<usize>; | ||
|
|
||
| #[inline] | ||
| fn prioritize(&self, package: &P, range: &VS) -> Self::Priority { | ||
| Reverse( | ||
| self.dependencies | ||
| .get(package) | ||
| .map(|versions| versions.keys().filter(|v| range.contains(v)).count()) | ||
| .unwrap_or(0), | ||
| ) | ||
| } | ||
|
|
||
| #[inline] | ||
| fn get_dependencies( | ||
| &self, | ||
| package: &P, | ||
| version: &VS::V, | ||
| ) -> Result<Dependencies<P, VS, Self::M>, Infallible> { | ||
| Ok(match self.dependencies(package, version) { | ||
| None => { | ||
| Dependencies::Unavailable("its dependencies could not be determined".to_string()) | ||
| } | ||
| Some(dependencies) => Dependencies::Available(dependencies), | ||
| }) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters