mod.rs

use std::{env::args, process::exit, str::FromStr};

use anyhow::{bail, Error, Result};
use tracing::level_filters::LevelFilter;

use rokit::{
    discovery::{discover_non_rokit_tool, discover_tool_spec},
    storage::Home,
    system::{current_exe_name, run_interruptible},
    tool::ToolAlias,
};

use crate::util::init_tracing;

mod info;

use self::info::inform_user_about_potential_fixes;

#[derive(Debug, Clone)]
pub struct Runner {
    exe_name: String,
}

impl Runner {
    pub fn new() -> Self {
        Self {
            exe_name: current_exe_name(),
        }
    }

    pub fn should_run(&self) -> bool {
        self.exe_name != env!("CARGO_BIN_NAME")
    }

    pub async fn run(&self) -> Result<()> {
        // Always log at INFO level when running a managed program
        // unless the user has explicitly set a different level
        // using the RUST_LOG environment variable.
        init_tracing(LevelFilter::INFO);

        let alias = ToolAlias::from_str(&self.exe_name)?;

        let home = Home::load_from_env().await?;
        let spec = discover_tool_spec(&alias, false, false).await;

        let program_args = args().skip(1).collect::<Vec<_>>();
        let program_path = match spec {
            // TODO: Prompt for trust and install tool if not already installed
            Some(spec) => home.tool_storage().tool_path(&spec),
            // FUTURE: Maybe we should add some kind of "fall-through" setting in
            // Rokit manifests instead of always falling through to non-rokit tools?
            None => match discover_non_rokit_tool(&home, &alias).await {
                Some(path) => path,
                None => bail!(
                    "Failed to find tool '{alias}' in any project manifest file.\
                    \nAdd the tool to a project using 'rokit add' before running it."
                ),
            },
        };

        let code = run_interruptible(&program_path, &program_args)
            .await
            .map_err(Error::from)
            .inspect_err(|e| inform_user_about_potential_fixes(&alias, e))?;

        exit(code);
    }
}

impl Default for Runner {
    fn default() -> Self {
        Self::new()
    }
}