Charlie Marsh
53 lines
2.0 KiB
Rust
53 lines
2.0 KiB
Rust
use std::path::PathBuf;
|
|
|
|
use anyhow::{Error, Result};
|
|
use url::Url;
|
|
|
|
use puffin_distribution::RemoteDistributionRef;
|
|
use puffin_git::Git;
|
|
|
|
/// The source of a distribution.
|
|
#[derive(Debug)]
|
|
pub(crate) enum Source<'a> {
|
|
/// The distribution is available at a URL in a registry, like PyPI.
|
|
RegistryUrl(Url),
|
|
/// The distribution is available at an arbitrary remote URL, like a GitHub Release.
|
|
RemoteUrl(&'a Url, Option<PathBuf>),
|
|
/// The distribution is available in a remote Git repository.
|
|
Git(Git, Option<PathBuf>),
|
|
}
|
|
|
|
impl<'a> TryFrom<&'a RemoteDistributionRef<'_>> for Source<'a> {
|
|
type Error = Error;
|
|
|
|
fn try_from(value: &'a RemoteDistributionRef<'_>) -> Result<Self, Self::Error> {
|
|
match value {
|
|
// If a distribution is hosted on a registry, it must be available at a URL.
|
|
RemoteDistributionRef::Registry(_, _, file) => {
|
|
Ok(Self::RegistryUrl(Url::parse(&file.url)?))
|
|
}
|
|
|
|
// If a distribution is specified via a direct URL, it could be a URL to a hosted file,
|
|
// or a URL to a Git repository.
|
|
RemoteDistributionRef::Url(_, url) => {
|
|
// If the URL points to a subdirectory, extract it, as in:
|
|
// `https://git.example.com/MyProject.git@v1.0#subdirectory=pkg_dir`
|
|
// `https://git.example.com/MyProject.git@v1.0#egg=pkg&subdirectory=pkg_dir`
|
|
let subdirectory = url.fragment().and_then(|fragment| {
|
|
fragment.split('&').find_map(|fragment| {
|
|
fragment.strip_prefix("subdirectory=").map(PathBuf::from)
|
|
})
|
|
});
|
|
|
|
if let Some(url) = url.as_str().strip_prefix("git+") {
|
|
let url = Url::parse(url)?;
|
|
let git = Git::try_from(url)?;
|
|
Ok(Self::Git(git, subdirectory))
|
|
} else {
|
|
Ok(Self::RemoteUrl(url, subdirectory))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|