use std::path::Path;
use crate::cli;
pub async fn reset_repo(repo_path: &Path) -> bool {
if !repo_path.join(".git").exists() {
tracing::error!(repo_path = %repo_path.display(), "not the root of a repository");
return false;
}
if let Err(error) = cli::reset_hard(repo_path).await {
tracing::error!(repo_path = %repo_path.display(), error = %error, "failed to reset the repository");
return false;
}
let root = repo_path.to_path_buf();
match tokio::task::spawn_blocking(move || super::fs::remove_symlinks(&root)).await {
Ok(Ok(())) => true,
Ok(Err(error)) => {
tracing::error!(repo_path = %repo_path.display(), error = %error, "failed to remove the symlinks");
false
}
Err(error) => {
tracing::error!(repo_path = %repo_path.display(), error = %error, "failed to remove the symlinks");
false
}
}
}
#[cfg(test)]
mod tests {
use super::reset_repo;
use std::path::Path;
fn git(repo: &Path, args: &[&str]) {
let status = std::process::Command::new("git")
.arg("-C")
.arg(repo)
.args(args)
.status()
.unwrap();
assert!(status.success());
}
#[tokio::test]
async fn only_the_root_of_the_repository_gets_reset() {
let dir = tempfile::tempdir().unwrap();
std::env::set_var("GIT_CONFIG_GLOBAL", dir.path().join("gitconfig"));
let repo = dir.path().join("repo");
std::fs::create_dir_all(repo.join("sub/deep")).unwrap();
git(&repo, &["init", "-q"]);
std::fs::write(repo.join("f.txt"), b"committed").unwrap();
git(&repo, &["add", "-A"]);
git(
&repo,
&[
"-c",
"user.email=dev@fluidattacks.com",
"-c",
"user.name=dev",
"commit",
"-q",
"-m",
"files",
],
);
std::fs::write(repo.join("f.txt"), b"dirty").unwrap();
let deep = repo.join("sub/deep");
assert!(!reset_repo(&deep).await);
assert_eq!(
std::fs::read_to_string(repo.join("f.txt")).unwrap(),
"dirty"
);
assert!(reset_repo(&repo).await);
assert_eq!(
std::fs::read_to_string(repo.join("f.txt")).unwrap(),
"committed"
);
}
#[tokio::test]
async fn the_reset_runs_no_fsmonitor_of_the_repository() {
let dir = tempfile::tempdir().unwrap();
let marker = dir.path().join("pwned");
let payload = dir.path().join("payload.sh");
std::fs::write(&payload, format!("#!/bin/sh\ntouch {}\n", marker.display())).unwrap();
std::fs::set_permissions(
&payload,
std::os::unix::fs::PermissionsExt::from_mode(0o755),
)
.unwrap();
let repo = dir.path().join("repo");
std::fs::create_dir_all(&repo).unwrap();
git(&repo, &["init", "-q"]);
std::fs::write(repo.join("f.txt"), b"committed").unwrap();
git(&repo, &["add", "-A"]);
git(
&repo,
&[
"-c",
"user.email=dev@fluidattacks.com",
"-c",
"user.name=dev",
"commit",
"-q",
"-m",
"files",
],
);
git(
&repo,
&["config", "core.fsmonitor", &payload.to_string_lossy()],
);
assert!(reset_repo(&repo).await);
assert!(!marker.exists());
}
}