miden-debug-engine 0.10.1

use std::path::PathBuf;

use miden_assembly_syntax::diagnostics::Report;

use crate::profiling::{instrument::Instrument, instrument_from_name};

/// Profiler options parsed from the command line.
#[derive(Default, Clone, Debug)]
#[cfg_attr(feature = "tui", derive(clap::Args))]
pub struct ProfilerCliArgs {
    /// Enables profiling and sets the output dir for reports. Profiling
    /// instruments need to be enabled separately.
    #[cfg_attr(
        feature = "tui",
        arg(long = "profiling-reports-dir", value_name = "DIRECTORY")
    )]
    pub reports_dir: Option<PathBuf>,
    #[cfg_attr(
        feature = "tui",
        arg(
            long = "profiling-instruments",
            value_name = "VALUE",
            value_delimiter = ','
        )
    )]
    pub instruments: Vec<String>,
}

#[derive(Default)]
pub struct ProfilerConfig {
    /// The active instrumentations.
    pub instruments: Vec<Box<dyn Instrument>>,
    /// The directory where profiling reports are written.
    pub reports_dir: Option<PathBuf>,
}

impl TryFrom<ProfilerCliArgs> for ProfilerConfig {
    type Error = Report;

    fn try_from(mut args: ProfilerCliArgs) -> Result<Self, Self::Error> {
        let mut config = ProfilerConfig::default();

        if let Some(ref path) = args.reports_dir
            && path.exists()
            && !path.is_dir()
        {
            return Err(Report::msg(format!(
                "invalid profiling reports directory '{}': not a directory",
                path.display()
            )));
        }

        if args.reports_dir.is_some() && args.instruments.is_empty() {
            return Err(Report::msg(
                "profiling requires at least one instrument set with --profiling-instruments",
            ));
        }

        if args.reports_dir.is_none() && !args.instruments.is_empty() {
            return Err(Report::msg(
                "profiling instruments require --profiling-reports-dir to be set",
            ));
        }

        config.reports_dir = args.reports_dir;

        args.instruments.sort();
        args.instruments.dedup();
        for name in &args.instruments {
            let instrument = instrument_from_name(name, &config).map_err(Report::msg)?;
            config.instruments.push(instrument);
        }

        Ok(config)
    }
}

impl std::fmt::Debug for ProfilerConfig {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let instrument_names: Vec<&'static str> =
            self.instruments.iter().map(|i| i.name()).collect();
        f.debug_struct("ProfilerConfig")
            .field("instruments", &instrument_names)
            .field("reports_dir", &self.reports_dir)
            .finish()
    }
}