use std::fs::create_dir_all;
use std::path::PathBuf;
use duct::Expression;
use miette::{IntoDiagnostic, Result};
use crate::cmd;
use crate::file::touch_dir;
pub struct Git {
pub dir: PathBuf,
}
macro_rules! git_cmd {
( $dir:expr $(, $arg:expr )* $(,)? ) => {
{
let safe = format!("safe.directory={}", $dir.display());
cmd!("git", "-C", $dir, "-c", safe $(, $arg)*)
}
}
}
impl Git {
pub fn new(dir: PathBuf) -> Self {
Self { dir }
}
pub fn is_repo(&self) -> bool {
self.dir.join(".git").is_dir()
}
pub fn update(&self, gitref: Option<String>) -> Result<(String, String)> {
let gitref = gitref.map_or_else(|| self.current_branch(), Ok)?;
debug!("updating {} to {}", self.dir.display(), gitref);
let exec = |cmd: Expression| match cmd.stderr_to_stdout().stdout_capture().unchecked().run()
{
Ok(res) => {
if res.status.success() {
Ok(())
} else {
Err(miette!(
"git failed: {cmd:?} {}",
String::from_utf8(res.stdout).unwrap()
))
}
}
Err(err) => Err(miette!("git failed: {cmd:?} {err:#}")),
};
exec(git_cmd!(
&self.dir,
"fetch",
"--prune",
"--update-head-ok",
"origin",
&format!("{}:{}", gitref, gitref),
))?;
let prev_rev = self.current_sha()?;
exec(git_cmd!(
&self.dir,
"-c",
"advice.detachedHead=false",
"-c",
"advice.objectNameWarning=false",
"checkout",
"--force",
&gitref
))?;
let post_rev = self.current_sha()?;
touch_dir(&self.dir)?;
Ok((prev_rev, post_rev))
}
pub fn clone(&self, url: &str) -> Result<()> {
debug!("cloning {} to {}", url, self.dir.display());
if let Some(parent) = self.dir.parent() {
create_dir_all(parent).into_diagnostic()?;
}
match get_git_version() {
Ok(version) => trace!("git version: {}", version),
Err(err) => warn!(
"failed to get git version: {:#}\n Git is required to use mise.",
err
),
}
cmd!("git", "clone", "-q", "--depth", "1", url, &self.dir)
.run()
.into_diagnostic()?;
Ok(())
}
pub fn current_branch(&self) -> Result<String> {
let branch = git_cmd!(&self.dir, "branch", "--show-current")
.read()
.into_diagnostic()?;
debug!("current branch for {}: {}", self.dir.display(), &branch);
Ok(branch)
}
pub fn current_sha(&self) -> Result<String> {
let sha = git_cmd!(&self.dir, "rev-parse", "HEAD")
.read()
.into_diagnostic()?;
debug!("current sha for {}: {}", self.dir.display(), &sha);
Ok(sha)
}
pub fn current_sha_short(&self) -> Result<String> {
let sha = git_cmd!(&self.dir, "rev-parse", "--short", "HEAD")
.read()
.into_diagnostic()?;
debug!("current sha for {}: {}", self.dir.display(), &sha);
Ok(sha)
}
pub fn current_abbrev_ref(&self) -> Result<String> {
let aref = git_cmd!(&self.dir, "rev-parse", "--abbrev-ref", "HEAD")
.read()
.into_diagnostic()?;
debug!("current abbrev ref for {}: {}", self.dir.display(), &aref);
Ok(aref)
}
pub fn get_remote_url(&self) -> Option<String> {
if !self.dir.exists() {
return None;
}
let res = git_cmd!(&self.dir, "config", "--get", "remote.origin.url").read();
match res {
Ok(url) => {
debug!("remote url for {}: {}", self.dir.display(), &url);
Some(url)
}
Err(err) => {
warn!(
"failed to get remote url for {}: {:#}",
self.dir.display(),
err
);
None
}
}
}
pub fn split_url_and_ref(url: &str) -> (String, Option<String>) {
match url.split_once('#') {
Some((url, _ref)) => (url.to_string(), Some(_ref.to_string())),
None => (url.to_string(), None),
}
}
}
fn get_git_version() -> Result<String> {
let version = cmd!("git", "--version").read().into_diagnostic()?;
Ok(version.trim().into())
}