154 lines
4.2 KiB
Rust
154 lines
4.2 KiB
Rust
use std::ffi::OsString;
|
|
use std::path::PathBuf;
|
|
|
|
use anyhow::anyhow;
|
|
use anyhow::Context as _;
|
|
|
|
use radicle::prelude::*;
|
|
use radicle::storage::WriteStorage;
|
|
|
|
use crate::project;
|
|
use crate::terminal as term;
|
|
use crate::terminal::args::{Args, Error, Help};
|
|
|
|
pub const HELP: Help = Help {
|
|
name: "checkout",
|
|
description: "Checkout a project into the local directory",
|
|
version: env!("CARGO_PKG_VERSION"),
|
|
usage: r#"
|
|
Usage
|
|
|
|
rad checkout <id> [<option>...]
|
|
|
|
Options
|
|
|
|
--no-confirm Don't ask for confirmation during checkout
|
|
--help Print help
|
|
"#,
|
|
};
|
|
|
|
pub struct Options {
|
|
pub id: Id,
|
|
}
|
|
|
|
impl Args for Options {
|
|
fn from_args(args: Vec<OsString>) -> anyhow::Result<(Self, Vec<OsString>)> {
|
|
use lexopt::prelude::*;
|
|
use std::str::FromStr;
|
|
|
|
let mut parser = lexopt::Parser::from_args(args);
|
|
let mut id = None;
|
|
|
|
while let Some(arg) = parser.next()? {
|
|
match arg {
|
|
Long("no-confirm") => {
|
|
// Ignored for now.
|
|
}
|
|
Long("help") => return Err(Error::Help.into()),
|
|
Value(val) if id.is_none() => {
|
|
let val = val.to_string_lossy();
|
|
let val = Id::from_str(&val).context(format!("invalid id '{}'", val))?;
|
|
|
|
id = Some(val);
|
|
}
|
|
_ => return Err(anyhow::anyhow!(arg.unexpected())),
|
|
}
|
|
}
|
|
|
|
Ok((
|
|
Options {
|
|
id: id.ok_or_else(|| anyhow!("a project id to checkout must be provided"))?,
|
|
},
|
|
vec![],
|
|
))
|
|
}
|
|
}
|
|
|
|
pub fn run(options: Options, ctx: impl term::Context) -> anyhow::Result<()> {
|
|
let profile = ctx.profile()?;
|
|
let path = execute(options, &profile)?;
|
|
|
|
term::headline(&format!(
|
|
"🌱 Project checkout successful under ./{}",
|
|
term::format::highlight(path.file_name().unwrap_or_default().to_string_lossy())
|
|
));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn execute(options: Options, profile: &Profile) -> anyhow::Result<PathBuf> {
|
|
let id = options.id;
|
|
let storage = &profile.storage;
|
|
let doc = storage
|
|
.repository(id)?
|
|
.identity_of(profile.id())
|
|
.context("project could not be found in local storage")?;
|
|
let payload = doc.project()?;
|
|
let path = PathBuf::from(payload.name().clone());
|
|
|
|
if path.exists() {
|
|
anyhow::bail!("the local path {:?} already exists", path.as_path());
|
|
}
|
|
|
|
term::headline(&format!(
|
|
"Initializing local checkout for 🌱 {} ({})",
|
|
term::format::highlight(options.id),
|
|
payload.name(),
|
|
));
|
|
|
|
let spinner = term::spinner("Performing checkout...");
|
|
let repo = match radicle::rad::checkout(options.id, profile.id(), path.clone(), &storage) {
|
|
Ok(repo) => repo,
|
|
Err(err) => {
|
|
spinner.failed();
|
|
term::blank();
|
|
|
|
return Err(err.into());
|
|
}
|
|
};
|
|
spinner.finish();
|
|
|
|
let remotes = doc
|
|
.delegates
|
|
.into_iter()
|
|
.map(|did| *did)
|
|
.filter(|id| id != profile.id())
|
|
.collect::<Vec<_>>();
|
|
|
|
// Setup remote tracking branches for project delegates.
|
|
setup_remotes(
|
|
project::SetupRemote {
|
|
project: id,
|
|
default_branch: payload.default_branch().clone(),
|
|
repo: &repo,
|
|
fetch: true,
|
|
tracking: true,
|
|
},
|
|
&remotes,
|
|
)?;
|
|
|
|
Ok(path)
|
|
}
|
|
|
|
/// Setup a remote and tracking branch for each given remote.
|
|
pub fn setup_remotes(setup: project::SetupRemote, remotes: &[NodeId]) -> anyhow::Result<()> {
|
|
let mut spinner = term::spinner("Setting up remotes...");
|
|
for remote_id in remotes {
|
|
spinner.message(format!("Setting up remote {remote_id}.."));
|
|
|
|
if let Some((remote, branch)) = setup.run(*remote_id)? {
|
|
let remote = remote.name().unwrap(); // Only valid UTF-8 is used.
|
|
|
|
term::success!("Remote {} set", term::format::highlight(remote));
|
|
term::success!(
|
|
"Remote-tracking branch {} created for {}",
|
|
term::format::highlight(branch),
|
|
term::format::tertiary(term::format::node(remote_id))
|
|
);
|
|
}
|
|
}
|
|
spinner.finish();
|
|
|
|
Ok(())
|
|
}
|