use std::os::unix::process::ExitStatusExt; use std::path::{Path, PathBuf}; use std::process::Command; use std::str; use anyhow::bail; use log::info; use crate::gitea::Repository; pub(crate) struct Job { repo: Repository, local_path: Option, } // TODO: implement git operations with git2-rs where possible impl Job { pub(crate) fn new(repo: Repository) -> Self { Self { repo, local_path: None, } } fn repo_exists(&self) -> bool { self.local_path .as_ref() .map(|p| p.is_dir()) .unwrap_or(false) } fn mirror_repo(&self) -> anyhow::Result<()> { info!("Cloning repo {}...", self.repo.full_name); let output = Command::new("git") .arg("clone") .arg("--mirror") .arg(&self.repo.clone_url) .arg(format!("{}", self.local_path.as_ref().unwrap().display())) .output()?; if !output.status.success() { let error = str::from_utf8(&output.stderr)?; let code = output .status .code() .unwrap_or_else(|| output.status.signal().unwrap()); bail!( "couldn't mirror repo: exit code {}, stderr:\n{}", code, error ); } // TODO: handle git LFS mirroring: // https://github.com/git-lfs/git-lfs/issues/2342#issuecomment-310323647 Ok(()) } fn update_repo(&self) -> anyhow::Result<()> { info!("Updating repo {}...", self.repo.full_name); let output = Command::new("git") .arg("-C") .arg(format!("{}", self.local_path.as_ref().unwrap().display())) .arg("remote") .arg("update") .arg("origin") // otherwise deleted tags and branches aren't updated on local copy .arg("--prune") .output()?; if !output.status.success() { let error = str::from_utf8(&output.stderr)?; let code = output .status .code() .unwrap_or_else(|| output.status.signal().unwrap()); bail!( "couldn't update origin remote: exit code {}, stderr:\n{}", code, error ); } Ok(()) } fn get_remotes(&self) -> anyhow::Result { let output = Command::new("git") .arg("-C") .arg(format!("{}", self.local_path.as_ref().unwrap().display())) .arg("show") .arg("HEAD:.lohr") .output()?; if !output.status.success() { let error = str::from_utf8(&output.stderr)?; let code = output .status .code() .unwrap_or_else(|| output.status.signal().unwrap()); bail!( "couldn't read .lohr file from repo {}: exit code {}, stderr:\n{}", self.repo.full_name, code, error ); } Ok(String::from_utf8(output.stdout)?) } fn update_mirrors(&self) -> anyhow::Result<()> { for remote in self.get_remotes()?.lines() { info!("Updating mirror {}:{}...", remote, self.repo.full_name); let output = Command::new("git") .arg("-C") .arg(format!("{}", self.local_path.as_ref().unwrap().display())) .arg("push") .arg("--mirror") .arg(remote) .output()?; if !output.status.success() { let error = str::from_utf8(&output.stderr)?; let code = output .status .code() .unwrap_or_else(|| output.status.signal().unwrap()); bail!( "couldn't update remote {}: exit code {}, stderr:\n{}", remote, code, error ); } } Ok(()) } pub(crate) fn run(&mut self, homedir: &Path) -> anyhow::Result<()> { let local_path = homedir.join(&self.repo.full_name); assert!(local_path.is_absolute()); self.local_path = Some(local_path); if !self.repo_exists() { self.mirror_repo()?; } else { self.update_repo()?; } self.update_mirrors() } }