summaryrefslogtreecommitdiff
path: root/crates/git-tracker/src/managed_repository.rs
diff options
context:
space:
mode:
authorseth <[email protected]>2024-08-16 22:14:16 -0400
committerGitHub <[email protected]>2024-08-17 02:14:16 +0000
commitbbc00edc6508ea1910c4d9e6a272f7859900950d (patch)
tree538fe23940cb6bca8afa48059e502bfa7d5608cd /crates/git-tracker/src/managed_repository.rs
parent66ea6f8c225cc2273cfbfb1b948604c170a78e8c (diff)
end of summer refactor (#31)
* crates: `bot-*` -> `discord-bot` I didn't really need all these crates to be split :/ * discord-bot: revamp http impl also handles the new errors reported by teawieAPI * crates: split http backend this can be reused easily * git-tracker: short-circuit boolean logic We don't need to check if the commit is a descendant of the HEAD of the branch if it *is* the HEAD * nix: fenix -> nixpkgs * treefmt: add actionlint * nix: use docker arch names for containers * ci: use actions-rust-lang actions * nix: drop ci dev shell * git-tracker: init ManagedRepository this logic can be shared * ci: use nix for clippy scan * discord-bot: better handle unmerged PRs * ci: fix treefmt check * nix: fix clippy check * .env.template: update crate names * git-tracker: use remote name for remote name i was half asleep * discord-bot: handle merged PRs that aren't found in any tracked branches * git-tracker: make collect_statuses_in() return a Vec * discord-bot: add more PR info with response fixes #18
Diffstat (limited to 'crates/git-tracker/src/managed_repository.rs')
-rw-r--r--crates/git-tracker/src/managed_repository.rs95
1 files changed, 95 insertions, 0 deletions
diff --git a/crates/git-tracker/src/managed_repository.rs b/crates/git-tracker/src/managed_repository.rs
new file mode 100644
index 0000000..0a41bd0
--- /dev/null
+++ b/crates/git-tracker/src/managed_repository.rs
@@ -0,0 +1,95 @@
+use git2::{AutotagOption, FetchOptions, RemoteCallbacks, RemoteUpdateFlags, Repository};
+use log::{debug, info, trace, warn};
+use std::{io::Write, path::PathBuf};
+
+// much of this is shamelessly lifted from
+// https://github.com/rust-lang/git2-rs/blob/9a5c9706ff578c936be644dd1e8fe155bdc4d129/examples/pull.rs
+
+#[derive(Debug, thiserror::Error)]
+pub enum Error {
+ #[error("libgit2 error")]
+ Git(#[from] git2::Error),
+}
+
+pub struct ManagedRepository {
+ pub path: PathBuf,
+ pub tracked_branches: Vec<String>,
+ pub upstream_remote_url: String,
+ pub upstream_remote_name: String,
+}
+
+impl ManagedRepository {
+ /// basic set of options for fetching from remotes
+ fn fetch_options<'a>() -> FetchOptions<'a> {
+ let mut remote_callbacks = RemoteCallbacks::new();
+ remote_callbacks.transfer_progress(|progress| {
+ if progress.received_objects() == progress.total_objects() {
+ trace!(
+ "Resolving deltas {}/{}\r",
+ progress.indexed_deltas(),
+ progress.total_deltas()
+ );
+ } else {
+ trace!(
+ "Received {}/{} objects ({}) in {} bytes\r",
+ progress.received_objects(),
+ progress.total_objects(),
+ progress.indexed_objects(),
+ progress.received_bytes()
+ );
+ }
+ std::io::stdout().flush().ok();
+ true
+ });
+
+ let mut fetch_opts = FetchOptions::new();
+ fetch_opts.remote_callbacks(remote_callbacks);
+
+ fetch_opts
+ }
+
+ /// Update the given branches in the [`repository`] using the nixpkgs remote
+ fn update_branches_in(&self, repository: &Repository) -> Result<(), Error> {
+ let mut remote = repository.find_remote(&self.upstream_remote_name)?;
+ // download all the refs
+ remote.download(&self.tracked_branches, Some(&mut Self::fetch_options()))?;
+ remote.disconnect()?;
+ // and (hopefully) update what they refer to for later
+ remote.update_tips(
+ None,
+ RemoteUpdateFlags::UPDATE_FETCHHEAD,
+ AutotagOption::Auto,
+ None,
+ )?;
+
+ Ok(())
+ }
+
+ /// Fetch the repository or update it if it exists
+ ///
+ /// # Errors
+ /// Will return [`Err`] if the repository cannot be opened, cloned, or updated
+ pub fn fetch_or_update(&self) -> Result<(), Error> {
+ // Open our repository or clone it if it doesn't exist
+ let repository = if self.path.exists() {
+ Repository::open(self.path.as_path())?
+ } else {
+ warn!(
+ "Couldn't find repository at {}! Cloning a fresh one from {}",
+ self.path.display(),
+ self.upstream_remote_url
+ );
+ Repository::clone(&self.upstream_remote_url, self.path.as_path())?;
+ info!("Finished cloning to {}", self.path.display());
+
+ // bail early as we already have a fresh copy
+ return Ok(());
+ };
+
+ debug!("Updating repository at {}", self.path.display());
+ self.update_branches_in(&repository)?;
+ debug!("Finished updating!");
+
+ Ok(())
+ }
+}