cargo-hatch 0.1.3

Hatch new projects like a chick coming out of its egg.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

use anyhow::{Context, Result};
use camino::Utf8Path;
use git2::{
    build::{CheckoutBuilder, RepoBuilder},
    Cred, FetchOptions, RemoteCallbacks, Repository,
};

pub fn clone_or_update(url: &str, target: &Utf8Path) -> Result<()> {
    if target.exists() && target.join(".git").exists() {
        update(url, target)?
    } else {
        clone(url, target)?
    };

    Ok(())
}

/// Update an already existing repository to the latest changes of the default head branch.
fn update(url: &str, target: &Utf8Path) -> Result<Repository> {
    let repo = Repository::open(target)?;

    {
        let mut remote = repo.remote_anonymous(url)?;
        let mut head = repo.head()?;
        let head_name = head.name().context("repo head is not valid UTF8")?;

        remote.fetch(&[head_name], Some(&mut create_fetch_opts()), None)?;

        let fetch_head = repo.find_reference("FETCH_HEAD")?;
        let fetch_head = fetch_head.resolve()?.peel_to_commit()?.id();

        head.set_target(fetch_head, "")?;
        repo.checkout_head(Some(
            CheckoutBuilder::new()
                .force()
                .remove_ignored(true)
                .remove_untracked(true),
        ))?;
    }

    Ok(repo)
}

/// Clone a new repo to the given output path or fail if it already exists.
fn clone(url: &str, target: &Utf8Path) -> Result<Repository> {
    let mut builder = RepoBuilder::new();
    builder.fetch_options(create_fetch_opts());

    builder.clone(url, target.as_std_path()).map_err(Into::into)
}

fn create_fetch_opts() -> FetchOptions<'static> {
    let callbacks = {
        let mut cb = RemoteCallbacks::new();
        cb.credentials(|_url, username, allowed_types| {
            if allowed_types.is_ssh_key() {
                if let Some(username) = username {
                    Cred::ssh_key_from_agent(username)
                } else {
                    Err(git2::Error::from_str(
                        "need username for SSH authentication",
                    ))
                }
            } else {
                Err(git2::Error::from_str(
                    "only SSH authentication is supported",
                ))
            }
        });
        cb
    };

    let mut fo = FetchOptions::new();
    fo.remote_callbacks(callbacks);
    fo
}

/// Find the full repo name (including its owner) from a typical git URL as used on GitHub, GitLab
/// and other popular hosts.
///
/// The format one of (with the `.git` suffix being optional):
/// - `git@<host>:<owner>/<user>.git`
/// - `https://<host>/<owner>/<user>.git`
/// - `http://<host>/<owner>/<user>.git`
///
/// Therefore, to get the `<owner>/<user>` part, the prefix and suffix must be stripped. The final
/// string part is checked to contain only a single slash (`/`) to further validate the correctness
/// of the extracted name.
#[must_use]
pub fn find_repo_name(url: &str) -> Option<&str> {
    if url.starts_with("git@") {
        let name = url.split_once(':')?.1;
        Some(name.strip_suffix(".git").unwrap_or(name))
    } else if let Some(url) = url
        .strip_prefix("http://")
        .or_else(|| url.strip_prefix("https://"))
    {
        let name = url.split_once('/')?.1;
        Some(name.strip_suffix(".git").unwrap_or(name))
    } else {
        None
    }
    .filter(|name| name.chars().filter(|&c| c == '/').count() == 1)
}

/// Initialize a new Git repository at the given location.
pub fn init(target: &Utf8Path) -> Result<()> {
    Repository::init(target)?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn parse_git_url() {
        for input in &[
            "git@github.com:rust-lang/git2-rs",
            "git@github.com:rust-lang/git2-rs.git",
            "http://github.com/rust-lang/git2-rs",
            "http://github.com/rust-lang/git2-rs.git",
            "https://github.com/rust-lang/git2-rs",
            "https://github.com/rust-lang/git2-rs.git",
        ] {
            assert_eq!(Some("rust-lang/git2-rs"), find_repo_name(input));
        }
    }
}