use std::{ ffi::OsStr, path::{Path, PathBuf},}; use git::{objs::bstr::ByteSlice, progress, Progress};use git_repository as git; #[derive(Copy, Clone, Eq, PartialEq)]pub enum Mode { Execute, Simulate,} impl Default for Mode { fn default() -> Self { Mode::Simulate }} fn find_git_repository_workdirs<P: Progress>( root: impl AsRef<Path>, mut progress: P, debug: bool,) -> impl Iterator<Item = (PathBuf, git::Kind)>where <P as Progress>::SubProgress: Sync,{ progress.init(None, progress::count("filesystem items")); fn is_repository(path: &Path) -> Option<git::Kind> { // Can be git dir or worktree checkout (file) if path.file_name() != Some(OsStr::new(".git")) { return None; } if path.is_dir() { if path.join("HEAD").is_file() && path.join("config").is_file() { git::discover::is_git(path).ok().map(Into::into) } else { None } } else { // git files are always worktrees Some(git::Kind::WorkTree { is_linked: true }) } } fn into_workdir(git_dir: PathBuf) -> PathBuf { if git::discover::is_bare(&git_dir) { git_dir } else { git_dir.parent().expect("git is never in the root").to_owned() } } #[derive(Debug, Default)] struct State { kind: Option<git::Kind>, } let walk = jwalk::WalkDirGeneric::<((), State)>::new(root) .follow_links(false) .sort(true) .skip_hidden(false); // On macos with apple silicon, the IO subsystem is entirely different and one thread can mostly max it out. // Thus using more threads just burns energy unnecessarily. // It's notable that `du` is very fast even on a single core and more power efficient than dua with a single core. // The default of '4' seems related to the amount of performance cores present in the system. #[cfg(all(target_os = "macos", target_arch = "aarch64"))] let walk = walk.parallelism(jwalk::Parallelism::RayonNewPool(4)); walk.process_read_dir(move |_depth, path, _read_dir_state, siblings| { if debug { eprintln!("{}", path.display()); } let mut found_any_repo = false; let mut found_bare_repo = false; for entry in siblings.iter_mut().flatten() { let path = entry.path(); if let Some(kind) = is_repository(&path) { let is_bare = kind.is_bare(); entry.client_state = State { kind: kind.into() }; entry.read_children_path = None; found_any_repo = true; found_bare_repo = is_bare; } } // Only return paths which are repositories are further participating in the traversal // Don't let bare repositories cause siblings to be pruned. if found_any_repo && !found_bare_repo { siblings.retain(|e| e.as_ref().map(|e| e.client_state.kind.is_some()).unwrap_or(false)); } }) .into_iter() .inspect(move |_| progress.inc()) .filter_map(Result::ok) .filter_map(|mut e| e.client_state.kind.take().map(|kind| (into_workdir(e.path()), kind)))} fn find_origin_remote(repo: &Path) -> anyhow::Result<Option<git_url::Url>> { let non_bare = repo.join(".git").join("config"); let local = git::config::Source::Local; let config = git::config::File::from_path_no_includes(non_bare.as_path(), local) .or_else(|_| git::config::File::from_path_no_includes(repo.join("config").as_path(), local))?; Ok(config .string("remote", Some("origin"), "url") .map(|url| git_url::Url::from_bytes(url.as_ref())) .transpose()?)} fn handle( mode: Mode, kind: git::Kind, git_workdir: &Path, canonicalized_destination: &Path, progress: &mut impl Progress,) -> anyhow::Result<()> { if let git::Kind::WorkTree { is_linked: true } = kind { return Ok(()); } fn to_relative(path: PathBuf) -> PathBuf { path.components() .skip_while(|c| c == &std::path::Component::RootDir) .collect() } fn find_parent_repo(mut git_workdir: &Path) -> Option<PathBuf> { while let Some(parent) = git_workdir.parent() { let has_contained_git_folder_or_file = std::fs::read_dir(parent).ok()?.any(|e| { e.ok() .and_then(|e| { e.file_name() .to_str() .map(|name| name == ".git" && e.path() != git_workdir) }) .unwrap_or(false) }); if has_contained_git_folder_or_file { return Some(parent.to_owned()); } git_workdir = parent; } None } if let Some(parent_repo_path) = find_parent_repo(git_workdir) { progress.fail(format!( "Skipping repository at {:?} as it is nested within repository {:?}", git_workdir.display(), parent_repo_path )); return Ok(()); } let url = match find_origin_remote(git_workdir)? { None => { progress.info(format!( "Skipping repository {:?} without 'origin' remote", git_workdir.display() )); return Ok(()); } Some(url) => url, }; if url.path.is_empty() { progress.info(format!( "Skipping repository at {:?} whose remote does not have a path: {:?}", git_workdir.display(), url.to_bstring() )); return Ok(()); } let destination = canonicalized_destination .join( url.host() .ok_or_else(|| anyhow::Error::msg(format!("Remote URLs must have host names: {}", url.to_bstring())))?, ) .join(to_relative({ let mut path = git_url::expand_path(None, url.path.as_bstr())?; match kind { git::Kind::Submodule => { unreachable!("BUG: We should not try to relocated submodules and not find them the first place") } git::Kind::Bare => path, git::Kind::WorkTree { .. } => { if let Some(ext) = path.extension() { if ext == "git" { path.set_extension(""); } } path } } })); if let Ok(destination) = destination.canonicalize() { if git_workdir.canonicalize()? == destination { return Ok(()); } } match mode { Mode::Simulate => progress.info(format!( "WOULD move {} to {}", git_workdir.display(), destination.display() )), Mode::Execute => { std::fs::create_dir_all(destination.parent().expect("repo destination is not the root"))?; progress.done(format!("Moving {} to {}", git_workdir.display(), destination.display())); std::fs::rename(git_workdir, &destination)?; } } Ok(())} /// Find all working directories in the given `source_dir` and print them to `out` while providing `progress`.pub fn discover<P: Progress>( source_dir: impl AsRef<Path>, mut out: impl std::io::Write, mut progress: P, debug: bool,) -> anyhow::Result<()>where <<P as Progress>::SubProgress as Progress>::SubProgress: Sync,{ for (git_workdir, _kind) in find_git_repository_workdirs(source_dir, progress.add_child("Searching repositories"), debug) { writeln!(&mut out, "{}", git_workdir.display())?; } Ok(())} pub fn run<P: Progress>( mode: Mode, source_dir: impl AsRef<Path>, destination: impl AsRef<Path>, mut progress: P,) -> anyhow::Result<()>where <<P as Progress>::SubProgress as Progress>::SubProgress: Sync,{ let mut num_errors = 0usize; let destination = destination.as_ref().canonicalize()?; for (path_to_move, kind) in find_git_repository_workdirs(source_dir, progress.add_child("Searching repositories"), false) { if let Err(err) = handle(mode, kind, &path_to_move, &destination, &mut progress) { progress.fail(format!( "Error when handling directory {:?}: {}", path_to_move.display(), err )); num_errors += 1; } } if num_errors > 0 { anyhow::bail!("Failed to handle {} repositories", num_errors) } else { Ok(()) }}
use std::{
ffi::OsStr,
path::{Path, PathBuf},
};
use git::{objs::bstr::ByteSlice, progress, Progress};
use git_repository as git;
#[derive(Copy, Clone, Eq, PartialEq)]
pub enum Mode {
Execute,
Simulate,
}
impl Default for Mode {
fn default() -> Self {
Mode::Simulate
}
}
fn find_git_repository_workdirs<P: Progress>(
root: impl AsRef<Path>,
mut progress: P,
debug: bool,
) -> impl Iterator<Item = (PathBuf, git::Kind)>
where
<P as Progress>::SubProgress: Sync,
{
progress.init(None, progress::count("filesystem items"));
fn is_repository(path: &Path) -> Option<git::Kind> {
// Can be git dir or worktree checkout (file)
if path.file_name() != Some(OsStr::new(".git")) {
return None;
}
if path.is_dir() {
if path.join("HEAD").is_file() && path.join("config").is_file() {
git::discover::is_git(path).ok().map(Into::into)
} else {
None
}
} else {
// git files are always worktrees
Some(git::Kind::WorkTree { is_linked: true })
}
}
fn into_workdir(git_dir: PathBuf) -> PathBuf {
if git::discover::is_bare(&git_dir) {
git_dir
} else {
git_dir.parent().expect("git is never in the root").to_owned()
}
}
#[derive(Debug, Default)]
struct State {
kind: Option<git::Kind>,
}
let walk = jwalk::WalkDirGeneric::<((), State)>::new(root)
.follow_links(false)
.sort(true)
.skip_hidden(false);
// On macos with apple silicon, the IO subsystem is entirely different and one thread can mostly max it out.
// Thus using more threads just burns energy unnecessarily.
// It's notable that `du` is very fast even on a single core and more power efficient than dua with a single core.
// The default of '4' seems related to the amount of performance cores present in the system.
#[cfg(all(target_os = "macos", target_arch = "aarch64"))]
let walk = walk.parallelism(jwalk::Parallelism::RayonNewPool(4));
walk.process_read_dir(move |_depth, path, _read_dir_state, siblings| {
if debug {
eprintln!("{}", path.display());
}
let mut found_any_repo = false;
let mut found_bare_repo = false;
for entry in siblings.iter_mut().flatten() {
let path = entry.path();
if let Some(kind) = is_repository(&path) {
let is_bare = kind.is_bare();
entry.client_state = State { kind: kind.into() };
entry.read_children_path = None;
found_any_repo = true;
found_bare_repo = is_bare;
}
}
// Only return paths which are repositories are further participating in the traversal
// Don't let bare repositories cause siblings to be pruned.
if found_any_repo && !found_bare_repo {
siblings.retain(|e| e.as_ref().map(|e| e.client_state.kind.is_some()).unwrap_or(false));
}
})
.into_iter()
.inspect(move |_| progress.inc())
.filter_map(Result::ok)
.filter_map(|mut e| e.client_state.kind.take().map(|kind| (into_workdir(e.path()), kind)))
}
fn find_origin_remote(repo: &Path) -> anyhow::Result<Option<git_url::Url>> {
let non_bare = repo.join(".git").join("config");
let local = git::config::Source::Local;
let config = git::config::File::from_path_no_includes(non_bare.as_path(), local)
.or_else(|_| git::config::File::from_path_no_includes(repo.join("config").as_path(), local))?;
Ok(config
.string("remote", Some("origin"), "url")
.map(|url| git_url::Url::from_bytes(url.as_ref()))
.transpose()?)
}
fn handle(
mode: Mode,
kind: git::Kind,
git_workdir: &Path,
canonicalized_destination: &Path,
progress: &mut impl Progress,
) -> anyhow::Result<()> {
if let git::Kind::WorkTree { is_linked: true } = kind {
return Ok(());
}
fn to_relative(path: PathBuf) -> PathBuf {
path.components()
.skip_while(|c| c == &std::path::Component::RootDir)
.collect()
}
fn find_parent_repo(mut git_workdir: &Path) -> Option<PathBuf> {
while let Some(parent) = git_workdir.parent() {
let has_contained_git_folder_or_file = std::fs::read_dir(parent).ok()?.any(|e| {
e.ok()
.and_then(|e| {
e.file_name()
.to_str()
.map(|name| name == ".git" && e.path() != git_workdir)
})
.unwrap_or(false)
});
if has_contained_git_folder_or_file {
return Some(parent.to_owned());
}
git_workdir = parent;
}
None
}
if let Some(parent_repo_path) = find_parent_repo(git_workdir) {
progress.fail(format!(
"Skipping repository at {:?} as it is nested within repository {:?}",
git_workdir.display(),
parent_repo_path
));
return Ok(());
}
let url = match find_origin_remote(git_workdir)? {
None => {
progress.info(format!(
"Skipping repository {:?} without 'origin' remote",
git_workdir.display()
));
return Ok(());
}
Some(url) => url,
};
if url.path.is_empty() {
progress.info(format!(
"Skipping repository at {:?} whose remote does not have a path: {:?}",
git_workdir.display(),
url.to_bstring()
));
return Ok(());
}
let destination = canonicalized_destination
.join(
url.host()
.ok_or_else(|| anyhow::Error::msg(format!("Remote URLs must have host names: {}", url.to_bstring())))?,
)
.join(to_relative({
let mut path = git_url::expand_path(None, url.path.as_bstr())?;
match kind {
git::Kind::Submodule => {
unreachable!("BUG: We should not try to relocated submodules and not find them the first place")
}
git::Kind::Bare => path,
git::Kind::WorkTree { .. } => {
if let Some(ext) = path.extension() {
if ext == "git" {
path.set_extension("");
}
}
path
}
}
}));
if let Ok(destination) = destination.canonicalize() {
if git_workdir.canonicalize()? == destination {
return Ok(());
}
}
match mode {
Mode::Simulate => progress.info(format!(
"WOULD move {} to {}",
git_workdir.display(),
destination.display()
)),
Mode::Execute => {
std::fs::create_dir_all(destination.parent().expect("repo destination is not the root"))?;
progress.done(format!("Moving {} to {}", git_workdir.display(), destination.display()));
std::fs::rename(git_workdir, &destination)?;
}
}
Ok(())
}
/// Find all working directories in the given `source_dir` and print them to `out` while providing `progress`.
pub fn discover<P: Progress>(
source_dir: impl AsRef<Path>,
mut out: impl std::io::Write,
mut progress: P,
debug: bool,
) -> anyhow::Result<()>
where
<<P as Progress>::SubProgress as Progress>::SubProgress: Sync,
{
for (git_workdir, _kind) in
find_git_repository_workdirs(source_dir, progress.add_child("Searching repositories"), debug)
{
writeln!(&mut out, "{}", git_workdir.display())?;
}
Ok(())
}
pub fn run<P: Progress>(
mode: Mode,
source_dir: impl AsRef<Path>,
destination: impl AsRef<Path>,
mut progress: P,
) -> anyhow::Result<()>
where
<<P as Progress>::SubProgress as Progress>::SubProgress: Sync,
{
let mut num_errors = 0usize;
let destination = destination.as_ref().canonicalize()?;
for (path_to_move, kind) in
find_git_repository_workdirs(source_dir, progress.add_child("Searching repositories"), false)
{
if let Err(err) = handle(mode, kind, &path_to_move, &destination, &mut progress) {
progress.fail(format!(
"Error when handling directory {:?}: {}",
path_to_move.display(),
err
));
num_errors += 1;
}
}
if num_errors > 0 {
anyhow::bail!("Failed to handle {} repositories", num_errors)
} else {
Ok(())
}
}