refresh a corpus repositories by updating all of them.

Author: Byron

gitoxide-core/src/corpus/mod.rs

@@ -39,14 +39,34 @@ pub mod engine {
 
         /// Run on the existing set of repositories we have already seen or obtain them from `path` if there is none yet.
         pub fn run(&mut self, corpus_path: PathBuf) -> anyhow::Result<()> {
-            let corpus_path = gix::path::realpath(corpus_path)?;
-            let corpus_id = self.corpus_id_or_insert(&corpus_path)?;
+            let (corpus_path, corpus_id) = self.prepare_corpus_path(corpus_path)?;
             let _gitoxide_id = self.gitoxide_version_id_or_insert()?;
             let _runner_id = self.runner_id_or_insert()?;
             let _repos = self.find_repos_or_insert(&corpus_path, corpus_id)?;
             todo!("do run on repos")
         }
 
+        pub fn refresh(&mut self, corpus_path: PathBuf) -> anyhow::Result<()> {
+            let (corpus_path, corpus_id) = self.prepare_corpus_path(corpus_path)?;
+            let repos = self.refresh_repos(&corpus_path, corpus_id)?;
+            self.progress.info(format!(
+                "Added or updated {} repositories under {corpus_path:?}",
+                repos.len()
+            ));
+            Ok(())
+        }
+    }
+
+    impl<P> Engine<P>
+    where
+        P: gix::Progress,
+    {
+        fn prepare_corpus_path(&self, corpus_path: PathBuf) -> anyhow::Result<(PathBuf, Id)> {
+            let corpus_path = gix::path::realpath(corpus_path)?;
+            let corpus_id = self.corpus_id_or_insert(&corpus_path)?;
+            Ok((corpus_path, corpus_id))
+        }
+
         fn find_repos(&mut self, corpus_id: Id) -> anyhow::Result<Vec<db::Repo>> {
             self.progress.set_name("query db-repos");
             self.progress.init(None, gix::progress::count("repos"));

src/plumbing/main.rs

@@ -139,6 +139,7 @@ pub fn main() -> Result<()> {
                 let mut engine = core::corpus::Engine::open_or_create(db, env!("GITOXIDE_VERSION").into(), progress)?;
                 match cmd {
                     crate::plumbing::options::corpus::SubCommands::Run => engine.run(path),
+                    crate::plumbing::options::corpus::SubCommands::Refresh => engine.refresh(path),
                 }
             },
         ),

src/plumbing/options/mod.rs

@@ -145,6 +145,8 @@ pub mod corpus {
     pub enum SubCommands {
         /// Perform a corpus run on all registered repositories.
         Run,
+        /// Re-read all repositories under the corpus directory, and add or update them.
+        Refresh,
     }
 }