feat(aggregator): add database vacuum clap command

Author: dlachaume

mithril-aggregator/src/commands/database_command.rs

@@ -26,12 +26,16 @@ impl DatabaseCommand {
 pub enum DatabaseSubCommand {
     /// Migrate databases located in the given stores directory
     Migrate(MigrateCommand),
+
+    /// Vacuum the aggregator main database
+    Vacuum(VacuumCommand),
 }
 
 impl DatabaseSubCommand {
     pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
         match self {
             Self::Migrate(cmd) => cmd.execute(root_logger).await,
+            Self::Vacuum(cmd) => cmd.execute(root_logger).await,
         }
     }
 }
@@ -79,3 +83,38 @@ impl MigrateCommand {
         Ok(())
     }
 }
+
+#[derive(Parser, Debug, Clone)]
+pub struct VacuumCommand {
+    /// Stores directory
+    #[clap(long, env = "STORES_DIRECTORY")]
+    stores_directory: PathBuf,
+}
+
+impl VacuumCommand {
+    pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
+        let config = Configuration {
+            environment: ExecutionEnvironment::Production,
+            data_stores_directory: self.stores_directory.clone(),
+            // Temporary solution to avoid the need to provide a full configuration
+            ..Configuration::new_sample(std::env::temp_dir())
+        };
+        debug!(root_logger, "DATABASE VACUUM command"; "config" => format!("{config:?}"));
+        println!(
+            "Vacuuming database from stores directory: {}",
+            self.stores_directory.to_string_lossy()
+        );
+        let mut dependencies_builder =
+            DependenciesBuilder::new(root_logger.clone(), config.clone());
+
+        dependencies_builder
+            .get_upkeep_service()
+            .await
+            .with_context(|| "Dependencies Builder can not get upkeep service")?
+            .vacuum()
+            .await
+            .with_context(|| "Upkeep service can not vacuum")?;
+
+        Ok(())
+    }
+}