cargo-clone-core 0.2.4

// Copyright 2015 Jan Likar.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use cargo::util::GlobalContext;
use cargo::{CargoResult, core::SourceId, util::IntoUrl};
use url::Url;

/// Where to clone the crate from.
#[derive(Debug, Default)]
pub struct ClonerSource {
    pub(crate) cargo_source: CargoSource,
}

#[derive(Debug, Default)]
pub(crate) enum CargoSource {
    #[default]
    CratesIo,
    Index(Url),
    LocalRegistry(String),
    Registry(String),
}

impl ClonerSource {
    /// Creates a [`ClonerSource`] from the name of the remote registry.
    pub fn registry(key: impl Into<String>) -> Self {
        Self {
            cargo_source: CargoSource::Registry(key.into()),
        }
    }

    /// Creates a [`ClonerSource`] from a local registry path.
    pub fn local_registry(path: impl Into<String>) -> Self {
        Self {
            cargo_source: CargoSource::LocalRegistry(path.into()),
        }
    }

    /// Creates a [`ClonerSource`] from a remote registry URL.
    pub fn index(index: impl AsRef<str>) -> CargoResult<Self> {
        let index: &str = index.as_ref();
        let cargo_source = CargoSource::Index(index.into_url()?);
        Ok(Self { cargo_source })
    }

    /// Creates a [`ClonerSource`] from a remote registry URL.
    pub fn index_from_url(url: Url) -> Self {
        let cargo_source = CargoSource::Index(url);
        Self { cargo_source }
    }

    /// Creates a [`ClonerSource`] from [crates.io](https://crates.io/).
    pub fn crates_io() -> Self {
        Self {
            cargo_source: CargoSource::CratesIo,
        }
    }
}

impl CargoSource {
    pub(crate) fn to_source_id(&self, context: &GlobalContext) -> CargoResult<SourceId> {
        match self {
            CargoSource::CratesIo => SourceId::crates_io(context),
            CargoSource::Index(url) => SourceId::for_registry(url),
            CargoSource::LocalRegistry(path) => {
                SourceId::for_local_registry(&context.cwd().join(path))
            }
            CargoSource::Registry(key) => SourceId::alt_registry(context, key),
        }
    }
}