provekit_nargo_cli 1.0.0-beta.20-alpha.1

use clap::{Args, Parser, Subcommand};
use const_format::formatcp;
use nargo::workspace::Workspace;
use nargo_toml::{
    ManifestError, NargoToml, PackageConfig, PackageMetadata, PackageSelection,
    get_package_manifest, resolve_workspace_from_fixed_toml, resolve_workspace_from_toml,
};
use noir_artifact_cli::commands::parse_and_normalize_path;
use noirc_driver::{CrateName, NOIR_ARTIFACT_VERSION_STRING};
use std::{
    collections::BTreeMap,
    fs::File,
    path::{Path, PathBuf},
    str::FromStr,
};

use color_eyre::eyre;

use crate::errors::CliError;

mod check_cmd;
pub mod compile_cmd;
mod dap_cmd;
mod debug_cmd;
mod doc_cmd;
mod execute_cmd;
mod expand_cmd;
mod export_cmd;
mod fmt_cmd;
mod fuzz_cmd;
mod generate_completion_script_cmd;
mod info_cmd;
mod init_cmd;
mod interpret_cmd;
mod lsp_cmd;
mod new_cmd;
mod test_cmd;

const GIT_HASH: &str = env!("GIT_COMMIT");
const IS_DIRTY: &str = env!("GIT_DIRTY");
const NARGO_VERSION: &str = env!("CARGO_PKG_VERSION");

static VERSION_STRING: &str = formatcp!(
    "version = {}\nnoirc version = {}\n(git version hash: {}, is dirty: {})",
    NARGO_VERSION,
    NOIR_ARTIFACT_VERSION_STRING,
    GIT_HASH,
    IS_DIRTY
);

#[derive(Parser, Debug)]
#[command(name="nargo", author, version=VERSION_STRING, about, long_about = None)]
struct NargoCli {
    #[command(subcommand)]
    command: NargoCommand,

    #[clap(flatten)]
    config: NargoConfig,
}

#[non_exhaustive]
#[derive(Args, Clone, Debug)]
pub struct NargoConfig {
    // REMINDER: Also change this flag in the LSP test lens if renamed
    #[arg(long, hide = true, global = true, default_value = "./", value_parser = parse_and_normalize_path)]
    program_dir: PathBuf,

    /// Override the default target directory.
    #[arg(long, hide = true, global = true, value_parser = parse_and_normalize_path)]
    target_dir: Option<PathBuf>,
}

/// Options for commands that work on either workspace or package scope.
#[derive(Args, Clone, Debug, Default)]
pub(crate) struct PackageOptions {
    /// The name of the package to run the command on.
    /// By default run on the first one found moving up along the ancestors of the current directory.
    #[clap(long, conflicts_with = "workspace")]
    package: Option<CrateName>,

    /// Run on all packages in the workspace
    #[clap(long, conflicts_with = "package")]
    workspace: bool,
}

impl PackageOptions {
    /// Decide which package to run the command on:
    /// * `package` if non-empty
    /// * all packages if `workspace` is `true`
    /// * otherwise the default package
    pub(crate) fn package_selection(&self) -> PackageSelection {
        let default_selection =
            if self.workspace { PackageSelection::All } else { PackageSelection::DefaultOrAll };

        self.package.clone().map_or(default_selection, PackageSelection::Selected)
    }
}

#[non_exhaustive]
#[derive(Subcommand, Clone, Debug)]
enum NargoCommand {
    Check(check_cmd::CheckCommand),
    Fmt(fmt_cmd::FormatCommand),
    #[command(alias = "build")]
    Compile(compile_cmd::CompileCommand),
    #[command(hide = true)]
    Interpret(interpret_cmd::InterpretCommand),
    New(new_cmd::NewCommand),
    Init(init_cmd::InitCommand),
    Execute(execute_cmd::ExecuteCommand),
    Export(export_cmd::ExportCommand),
    Debug(debug_cmd::DebugCommand),
    Test(test_cmd::TestCommand),
    Fuzz(fuzz_cmd::FuzzCommand),
    Info(info_cmd::InfoCommand),
    Lsp(lsp_cmd::LspCommand),
    #[command(hide = true)]
    Dap(dap_cmd::DapCommand),
    Expand(expand_cmd::ExpandCommand),
    Doc(doc_cmd::DocCommand),
    GenerateCompletionScript(generate_completion_script_cmd::GenerateCompletionScriptCommand),
}

/// Commands that can execute on the workspace level, or be limited to a selected package.
trait WorkspaceCommand {
    /// Indicate which package the command will be applied to.
    fn package_selection(&self) -> PackageSelection;
    /// The kind of lock the command needs to take out on the selected packages.
    fn lock_type(&self) -> LockType;
}

/// What kind of lock to take out on the (selected) workspace members.
#[derive(Clone, Debug, PartialEq, Eq)]
#[allow(dead_code)] // Not using `Shared` at the moment, e.g. while we `debug` we can `compile` a different version.
enum LockType {
    /// For commands that write artifacts.
    Exclusive,
    /// For commands that read artifacts, but never write them.
    Shared,
    /// For commands that cannot interfere with others.
    None,
}

#[cfg(not(feature = "codegen-docs"))]
#[tracing::instrument(level = "trace")]
pub(crate) fn start_cli() -> eyre::Result<()> {
    let NargoCli { command, config } = NargoCli::parse();

    match command {
        NargoCommand::New(args) => new_cmd::run(args, config),
        NargoCommand::Init(args) => init_cmd::run(args, config),
        NargoCommand::Check(args) => with_workspace(args, config, check_cmd::run),
        NargoCommand::Compile(args) => compile_with_maybe_dummy_workspace(args, config),
        NargoCommand::Interpret(args) => with_workspace(args, config, interpret_cmd::run),
        NargoCommand::Debug(args) => with_workspace(args, config, debug_cmd::run),
        NargoCommand::Execute(args) => with_workspace(args, config, execute_cmd::run),
        NargoCommand::Export(args) => with_workspace(args, config, export_cmd::run),
        NargoCommand::Test(args) => with_workspace(args, config, test_cmd::run),
        NargoCommand::Fuzz(args) => with_workspace(args, config, fuzz_cmd::run),
        NargoCommand::Info(args) => with_workspace(args, config, info_cmd::run),
        NargoCommand::Lsp(_) => lsp_cmd::run(),
        NargoCommand::Dap(args) => dap_cmd::run(args),
        NargoCommand::Fmt(args) => with_workspace(args, config, fmt_cmd::run),
        NargoCommand::Expand(args) => with_workspace(args, config, expand_cmd::run),
        NargoCommand::Doc(args) => with_workspace(args, config, doc_cmd::run),
        NargoCommand::GenerateCompletionScript(args) => generate_completion_script_cmd::run(args),
    }?;

    Ok(())
}

#[cfg(feature = "codegen-docs")]
pub(crate) fn start_cli() -> eyre::Result<()> {
    let markdown: String = clap_markdown::help_markdown::<NargoCli>();
    println!("{markdown}");
    Ok(())
}

/// Read a given program directory into a workspace.
fn read_workspace(
    program_dir: &Path,
    selection: PackageSelection,
) -> Result<Workspace, ManifestError> {
    let toml_path = get_package_manifest(program_dir)?;

    let workspace = resolve_workspace_from_toml(
        &toml_path,
        selection,
        Some(NOIR_ARTIFACT_VERSION_STRING.to_owned()),
    )?;

    Ok(workspace)
}

/// "with_workspace", but use a dummy workspace when 'debug_compile_stdin' is enabled
#[allow(clippy::field_reassign_with_default)]
fn compile_with_maybe_dummy_workspace(
    cmd: compile_cmd::CompileCommand,
    config: NargoConfig,
) -> Result<(), CliError> {
    if cmd.compile_options.debug_compile_stdin {
        let package_name = "debug_compile_stdin".to_string();

        // dummy root dir
        let root_dir = PathBuf::new();
        // This `PackageMetadata::default()` is leading to a clippy error but the suggested solution
        // is invalid because the fields are private
        let mut package = PackageMetadata::default();
        package.name = package_name.clone();
        package.package_type = Some("bin".into());
        let dependencies = BTreeMap::new();
        let package_config = PackageConfig { package, dependencies };
        let config = nargo_toml::Config::Package { package_config };
        let nargo_toml = NargoToml { root_dir, config };
        let package_name =
            CrateName::from_str(&package_name).expect("package_name to be a valid CrateName");
        let selection = PackageSelection::Selected(package_name);

        let assume_default_entry = true;
        let workspace = resolve_workspace_from_fixed_toml(
            nargo_toml,
            selection,
            Some(NOIR_ARTIFACT_VERSION_STRING.to_owned()),
            assume_default_entry,
        )?;
        compile_cmd::run(cmd, workspace)
    } else {
        with_workspace(cmd, config, compile_cmd::run)
    }
}

/// Find the root directory, parse the workspace, lock the packages, then execute the command.
fn with_workspace<C, R>(cmd: C, config: NargoConfig, run: R) -> Result<(), CliError>
where
    C: WorkspaceCommand,
    R: FnOnce(C, Workspace) -> Result<(), CliError>,
{
    if !config.program_dir.exists() {
        return Err(CliError::ProgramDirDoesNotExist(config.program_dir));
    }
    if !config.program_dir.is_dir() {
        return Err(CliError::ProgramDirIsNotADirectory(config.program_dir));
    }

    // All commands need to run on the workspace level, because that's where the `target` directory is.
    let workspace_dir = nargo_toml::find_root(&config.program_dir, true)?;
    let package_dir = nargo_toml::find_root(&config.program_dir, false)?;
    // Check if we're running inside the directory of a package, without having selected the entire workspace
    // or a specific package; if that's the case then parse the package name to select it in the workspace.
    let selection = match cmd.package_selection() {
        PackageSelection::DefaultOrAll if workspace_dir != package_dir => {
            let package = read_workspace(&package_dir, PackageSelection::DefaultOrAll)?;
            let package = package.into_iter().next().expect("there should be exactly 1 package");
            PackageSelection::Selected(package.name.clone())
        }
        other => other,
    };
    // Parse the top level workspace with the member selected.
    let mut workspace = read_workspace(&workspace_dir, selection)?;
    // Optionally override the target directory. It's only done here because most commands like the LSP and DAP
    // don't read or write artifacts, so they don't use the target directory.
    workspace.target_dir = config.target_dir;
    // Lock manifests if the command needs it.
    let _locks = match cmd.lock_type() {
        LockType::None => None,
        typ => Some(lock_workspace(&workspace, typ == LockType::Exclusive)?),
    };
    run(cmd, workspace)
}

/// Lock the (selected) packages in the workspace.
/// The lock taken can be shared for commands that only read the artifacts,
/// or exclusive for the ones that (might) write artifacts as well.
fn lock_workspace(
    workspace: &Workspace,
    exclusive: bool,
) -> Result<Vec<impl Drop + use<>>, CliError> {
    struct LockedFile(File);

    impl Drop for LockedFile {
        fn drop(&mut self) {
            let _ = fs2::FileExt::unlock(&self.0);
        }
    }

    let mut locks = Vec::new();
    for pkg in workspace {
        let toml_path = get_package_manifest(&pkg.root_dir)?;
        let path_display = toml_path.display();

        let file = File::open(&toml_path)
            .unwrap_or_else(|e| panic!("Expected {path_display} to exist: {e}"));

        if exclusive {
            if fs2::FileExt::try_lock_exclusive(&file).is_err() {
                eprintln!("Waiting for lock on {path_display}...");
            }
            fs2::FileExt::lock_exclusive(&file)
                .unwrap_or_else(|e| panic!("Failed to lock {path_display}: {e}"));
        } else {
            if fs2::FileExt::try_lock_shared(&file).is_err() {
                eprintln!("Waiting for lock on {path_display}...",);
            }
            fs2::FileExt::lock_shared(&file)
                .unwrap_or_else(|e| panic!("Failed to lock {path_display}: {e}"));
        }

        locks.push(LockedFile(file));
    }
    Ok(locks)
}

#[cfg(test)]
mod tests {
    use super::NargoCli;
    use clap::Parser;

    #[test]
    fn test_parse_target_dir() {
        let cmd = "nargo --program-dir . --target-dir ../foo/bar execute";
        let cli = NargoCli::try_parse_from(cmd.split_ascii_whitespace()).expect("should parse");

        let target_dir = cli.config.target_dir.expect("should parse target dir");
        assert!(target_dir.is_absolute(), "should be made absolute");
        assert!(target_dir.ends_with("foo/bar"));

        let cmd = "nargo --program-dir . execute";
        let cli = NargoCli::try_parse_from(cmd.split_ascii_whitespace()).expect("should parse");
        assert!(cli.config.target_dir.is_none());
    }
}