provekit_nargo_cli 1.0.0-beta.20

use std::{fs::DirEntry, path::Path};

use clap::Args;
use nargo::{
    insert_all_files_for_workspace_into_file_manager, ops::report_errors, workspace::Workspace,
};
use nargo_toml::PackageSelection;
use noirc_errors::CustomDiagnostic;
use noirc_frontend::{
    hir::{ParsedFiles, def_map::parse_file},
    parser::ParserError,
};

use crate::errors::CliError;

use super::{LockType, PackageOptions, WorkspaceCommand};

/// Format the Noir files in a workspace
#[derive(Debug, Clone, Args)]
pub(crate) struct FormatCommand {
    /// Run noirfmt in check mode
    #[arg(long)]
    check: bool,

    #[clap(flatten)]
    pub(super) package_options: PackageOptions,
}

impl WorkspaceCommand for FormatCommand {
    fn package_selection(&self) -> PackageSelection {
        match self.package_options.package_selection() {
            PackageSelection::DefaultOrAll => PackageSelection::All,
            other => other,
        }
    }

    fn lock_type(&self) -> LockType {
        // Writes source files, but doesn't touch compilation artifacts.
        LockType::None
    }
}

pub(crate) fn run(args: FormatCommand, workspace: Workspace) -> Result<(), CliError> {
    let check_mode = args.check;
    let mut workspace_file_manager = workspace.new_file_manager();
    insert_all_files_for_workspace_into_file_manager(&workspace, &mut workspace_file_manager);

    let config = nargo_fmt::Config::read(&workspace.root_dir)
        .map_err(|err| CliError::Generic(err.to_string()))?;

    let mut check_exit_code_one = false;

    for package in &workspace {
        visit_noir_files(&package.root_dir.join("src"), &mut |entry| {
            let file_id =   workspace_file_manager.name_to_id(entry.path()).expect("The file should exist since we added all files in the package into the file manager");

            let (parsed_module, errors) = parse_file(&workspace_file_manager, file_id);

            let is_all_warnings =  errors.iter().all(ParserError::is_warning);
            if !is_all_warnings {
                let diagnostics = errors
                    .iter()
                    .map(CustomDiagnostic::from)
                    .collect();

                let mut parsed_files = ParsedFiles::new();
                parsed_files.insert(file_id, (parsed_module, errors));

                let _ = report_errors::<()>(
                    Err(diagnostics),
                    &workspace_file_manager,
                    &parsed_files,
                    false,
                    false,
                );
                return Ok(());
            }

            let original = workspace_file_manager.fetch_file(file_id).expect("The file should exist since we added all files in the package into the file manager");
            let formatted = nargo_fmt::format(original, parsed_module, &config);

            if check_mode {
                let diff = similar_asserts::SimpleDiff::from_str(
                    original,
                    &formatted,
                    "original",
                    "formatted",
                )
                .to_string();

                if !diff.lines().next().is_some_and(|line| line.contains("Invisible differences")) {
                    if !check_exit_code_one {
                        check_exit_code_one = true;
                    }

                    println!("{diff}");
                }

                Ok(())
            } else {
                std::fs::write(entry.path(), formatted)
            }
        })
        .map_err(|error| CliError::Generic(error.to_string()))?;
    }

    if check_exit_code_one {
        std::process::exit(1);
    } else if check_mode {
        println!("No formatting changes were detected");
    }

    Ok(())
}

fn visit_noir_files(
    dir: &Path,
    cb: &mut dyn FnMut(&DirEntry) -> std::io::Result<()>,
) -> std::io::Result<()> {
    if dir.is_dir() {
        for entry in std::fs::read_dir(dir)? {
            let entry = entry?;
            let path = entry.path();
            if path.is_dir() {
                visit_noir_files(&path, cb)?;
            } else if entry.path().extension().is_some_and(|extension| extension == "nr") {
                cb(&entry)?;
            }
        }
    }
    Ok(())
}