feat: add a global setup command to preinstall executors

Author: art049

src/app.rs

@@ -1,5 +1,6 @@
 use crate::{
     api_client::CodSpeedAPIClient, auth, local_logger::CODSPEED_U8_COLOR_CODE, prelude::*, run,
+    setup,
 };
 use clap::{
     builder::{styling, Styles},
@@ -37,8 +38,10 @@ pub struct Cli {
 enum Commands {
     /// Run the bench command and upload the results to CodSpeed
     Run(run::RunArgs),
-    /// Commands related to authentication with CodSpeed
+    /// Manage the CLI authentication state
     Auth(auth::AuthArgs),
+    /// Pre-install the codspeed executors
+    Setup,
 }
 
 pub async fn run() -> Result<()> {
@@ -48,6 +51,7 @@ pub async fn run() -> Result<()> {
     match cli.command {
         Commands::Run(args) => run::run(args, &api_client).await?,
         Commands::Auth(args) => auth::run(args, &api_client).await?,
+        Commands::Setup => setup::setup().await?,
     }
     Ok(())
 }

src/main.rs

@@ -7,6 +7,7 @@ mod logger;
 mod prelude;
 mod request_client;
 mod run;
+mod setup;
 
 use console::style;
 use lazy_static::lazy_static;

…runner/valgrind/helpers/download_file.rs src/run/helpers/download_file.rssrc/run/runner/valgrind/helpers/download_file.rs renamed to src/run/helpers/download_file.rs src/run/runner/valgrind/helpers/download_file.rs renamed to src/run/helpers/download_file.rs

@@ -1,7 +1,9 @@
+mod download_file;
 mod find_repository_root;
 mod get_env_var;
 mod parse_git_remote;
 
+pub use download_file::download_file;
 pub use find_repository_root::find_repository_root;
 pub use get_env_var::get_env_variable;
 pub use parse_git_remote::*;

src/run/helpers/mod.rs

@@ -1,4 +1,5 @@
 use std::{
+    env,
     io::Read,
     path::{Path, PathBuf},
     process::{Child, Command, Stdio},
@@ -10,8 +11,8 @@ use reqwest::Client;
 use tokio::fs;
 use url::Url;
 
-use crate::prelude::*;
-use crate::run::helpers::get_env_variable;
+use crate::{prelude::*, run::helpers::download_file};
+use crate::{run::helpers::get_env_variable, MONGODB_TRACER_VERSION};
 
 use super::MongoDBConfig;
 
@@ -228,6 +229,31 @@ impl MongoTracer {
     }
 }
 
+pub async fn install_mongodb_tracer() -> Result<()> {
+    debug!("Installing mongodb-tracer");
+    // TODO: release the tracer and update this url
+    let installer_url = format!("https://codspeed-public-assets.s3.eu-west-1.amazonaws.com/mongo-tracer/{MONGODB_TRACER_VERSION}/cs-mongo-tracer-installer.sh");
+    let installer_path = env::temp_dir().join("cs-mongo-tracer-installer.sh");
+    download_file(
+        &Url::parse(installer_url.as_str()).unwrap(),
+        &installer_path,
+    )
+    .await?;
+
+    let output = Command::new("bash")
+        .arg(installer_path.to_str().unwrap())
+        .stdout(Stdio::piped())
+        .output()
+        .map_err(|_| anyhow!("Failed to install mongo-tracer"))?;
+
+    if !output.status.success() {
+        info!("stdout: {}", String::from_utf8_lossy(&output.stdout));
+        error!("stderr: {}", String::from_utf8_lossy(&output.stderr));
+        bail!("Failed to install mongo-tracer");
+    }
+    Ok(())
+}
+
 #[cfg(test)]
 mod tests {
     use std::ffi::OsStr;

src/run/instruments/mongo_tracer.rs

@@ -5,16 +5,16 @@ use crate::run::{config::Config, logger::Logger};
 use crate::VERSION;
 use check_system::SystemInfo;
 use clap::Args;
-use instruments::mongo_tracer::MongoTracer;
+use instruments::mongo_tracer::{install_mongodb_tracer, MongoTracer};
 use run_environment::interfaces::RunEnvironment;
 use runner::get_run_data;
 
-mod check_system;
-mod helpers;
+pub mod check_system;
+pub mod helpers;
 mod instruments;
 mod poll_results;
 pub mod run_environment;
-mod runner;
+pub mod runner;
 mod uploader;
 
 pub mod config;
@@ -120,14 +120,19 @@ pub async fn run(args: RunArgs, api_client: &CodSpeedAPIClient) -> Result<()> {
     let mode = runner::get_mode()?;
     let executor = runner::get_executor_from_mode(mode);
 
-    let run_data = get_run_data()?;
-
     if !config.skip_setup {
         start_group!("Preparing the environment");
-        executor.setup(&config, &system_info, &run_data).await?;
+        executor.setup(&system_info).await?;
+        // TODO: refactor and move directly in the Instruments struct as a `setup` method
+        if config.instruments.is_mongodb_enabled() {
+            install_mongodb_tracer().await?;
+        }
+        info!("Environment ready");
         end_group!();
     }
 
+    let run_data = get_run_data()?;
+
     start_opened_group!("Running the benchmarks");
 
     // TODO: refactor and move directly in the Instruments struct as a `start` method

src/run/mod.rs

@@ -8,12 +8,9 @@ use async_trait::async_trait;
 pub trait Executor {
     fn name(&self) -> ExecutorName;
 
-    async fn setup(
-        &self,
-        config: &Config,
-        system_info: &SystemInfo,
-        run_data: &RunData,
-    ) -> Result<()>;
+    async fn setup(&self, _system_info: &SystemInfo) -> Result<()> {
+        Ok(())
+    }
 
     /// Runs the executor
     async fn run(

src/run/runner/executor.rs

@@ -60,6 +60,10 @@ pub fn get_executor_from_mode(mode: RunnerMode) -> Box<dyn Executor> {
     }
 }
 
+pub fn get_all_executors() -> Vec<Box<dyn Executor>> {
+    vec![Box::new(ValgrindExecutor), Box::new(WallTimeExecutor)]
+}
+
 pub fn get_run_data() -> Result<RunData> {
     let profile_folder = create_profile_folder()?;
     Ok(RunData { profile_folder })

src/run/runner/mod.rs

@@ -6,7 +6,8 @@ use crate::run::runner::executor::Executor;
 use crate::run::runner::{ExecutorName, RunData};
 use crate::run::{check_system::SystemInfo, config::Config};
 
-use super::{helpers::perf_maps::harvest_perf_maps, measure, setup::setup};
+use super::setup::install_valgrind;
+use super::{helpers::perf_maps::harvest_perf_maps, measure};
 
 pub struct ValgrindExecutor;
 
@@ -16,14 +17,8 @@ impl Executor for ValgrindExecutor {
         ExecutorName::Valgrind
     }
 
-    async fn setup(
-        &self,
-        config: &Config,
-        system_info: &SystemInfo,
-        _run_data: &RunData,
-    ) -> Result<()> {
-        setup(system_info, config).await?;
-
+    async fn setup(&self, system_info: &SystemInfo) -> Result<()> {
+        install_valgrind(system_info).await?;
         Ok(())
     }
 

src/run/runner/valgrind/executor.rs

@@ -1,4 +1,3 @@
-pub mod download_file;
 pub mod ignored_objects_path;
 pub mod introspected_nodejs;
 pub mod perf_maps;