bat-cli 0.8.10

use crate::batbelt::metadata::{BatMetadata, BatMetadataCommit};
use crate::batbelt::sonar::SonarResultType;

use crate::batbelt::BatEnumerator;
use clap::Subcommand;

use crate::batbelt::git::git_commit::GitCommit;
use crate::batbelt::path::BatFile;
use error_stack::{Result, ResultExt};

use crate::batbelt::sonar::sonar_interactive::BatSonarInteractive;
use crate::batbelt::templates::TemplateGenerator;
use crate::commands::{BatCommandEnumerator, CommandResult};

use super::CommandError;

#[derive(Subcommand, Debug, strum_macros::Display, PartialEq, Clone, strum_macros::EnumIter)]
pub enum SonarCommand {
    /// Gets metadata from the source code
    Run {
        skip_source_code: bool,
        only_context_accounts: bool,
        only_entry_points: bool,
        only_traits: bool,
        only_function_dependencies: bool,
    },
}
impl BatEnumerator for SonarCommand {}

impl BatCommandEnumerator for SonarCommand {
    fn execute_command(&self) -> CommandResult<()> {
        match self {
            SonarCommand::Run {
                skip_source_code,
                only_context_accounts,
                only_entry_points,
                only_traits,
                only_function_dependencies,
            } => self.execute_run(
                *skip_source_code,
                *only_context_accounts,
                *only_entry_points,
                *only_traits,
                *only_function_dependencies,
            ),
        }
    }

    fn check_metadata_is_initialized(&self) -> bool {
        match self {
            SonarCommand::Run { .. } => false,
        }
    }

    fn check_correct_branch(&self) -> bool {
        match self {
            SonarCommand::Run { .. } => true,
        }
    }
}
impl SonarCommand {
    fn execute_run(
        &self,
        skip_source_code: bool,
        only_context_accounts: bool,
        only_entry_points: bool,
        only_traits: bool,
        only_function_dependencies: bool,
    ) -> CommandResult<()> {
        let metadata_bat_file = BatFile::BatMetadataFile;
        // in case the file does not exist, so the BatMetadata can be read
        if !metadata_bat_file
            .file_exists()
            .change_context(CommandError)?
        {
            TemplateGenerator
                .create_metadata_json()
                .change_context(CommandError)?;
        }
        let metadata_bkp_bat_file = BatFile::Generic {
            file_path: "BatMetadata_backup.json".to_string(),
        };

        // if the backup exists, then the previous process didn't finished successfully, so we use it to reload sensitive data
        if metadata_bkp_bat_file
            .file_exists()
            .change_context(CommandError)?
        {
            let metadata_bkp_content = metadata_bkp_bat_file
                .read_content(false)
                .change_context(CommandError)?;
            metadata_bat_file
                .write_content(false, &metadata_bkp_content)
                .change_context(CommandError)?;
        }

        // backup miro metadata
        let mut bat_metadata = BatMetadata::read_metadata().change_context(CommandError)?;
        let miro_metadata = bat_metadata.miro;

        // backup co metadata
        let metadata_content = metadata_bat_file
            .read_content(false)
            .change_context(CommandError)?;
        metadata_bkp_bat_file
            .write_content(false, &metadata_content)
            .change_context(CommandError)?;

        // if not skip source code, it should be run completely
        if !skip_source_code {
            // create new file
            TemplateGenerator
                .create_metadata_json()
                .change_context(CommandError)?;

            // reload miro backup
            let mut bat_metadata = BatMetadata::read_metadata().change_context(CommandError)?;
            bat_metadata.miro = miro_metadata;
            bat_metadata.save_metadata().change_context(CommandError)?;
            BatSonarInteractive::SonarStart {
                sonar_result_type: SonarResultType::Struct,
            }
            .print_interactive()
            .change_context(CommandError)?;
            self.execute_source_code()?;
            BatSonarInteractive::run_post_scan_parallel()
                .change_context(CommandError)?;
        } else if only_context_accounts {
            bat_metadata.context_accounts = vec![];
            self.execute_context_accounts()?;
        } else if only_entry_points {
            bat_metadata.entry_points = vec![];
            self.execute_entry_points()?;
        } else if only_traits {
            bat_metadata.traits = vec![];
            self.execute_traits()?;
        } else if only_function_dependencies {
            bat_metadata.function_dependencies = vec![];
            self.execute_function_dependencies()?;
        } else {
            bat_metadata.context_accounts = vec![];
            bat_metadata.entry_points = vec![];
            bat_metadata.traits = vec![];
            bat_metadata.function_dependencies = vec![];
            BatSonarInteractive::run_post_scan_parallel()
                .change_context(CommandError)?;
        }

        let mut bat_metadata = BatMetadata::read_metadata().change_context(CommandError)?;
        bat_metadata.initialized = true;
        bat_metadata.save_metadata().change_context(CommandError)?;

        // delete metadata backup
        metadata_bkp_bat_file
            .remove_file()
            .change_context(CommandError)?;

        GitCommit::UpdateMetadataJson {
            bat_metadata_commit: BatMetadataCommit::RunSonarMetadataCommit,
        }
        .create_commit(true)
        .change_context(CommandError)?;

        Ok(())
    }

    fn execute_source_code(&self) -> Result<(), CommandError> {
        BatSonarInteractive::GetSourceCodeMetadata
            .print_interactive()
            .change_context(CommandError)?;
        Ok(())
    }

    fn execute_context_accounts(&self) -> Result<(), CommandError> {
        BatSonarInteractive::GetContextAccountsMetadata
            .print_interactive()
            .change_context(CommandError)?;
        Ok(())
    }

    fn execute_entry_points(&self) -> Result<(), CommandError> {
        BatSonarInteractive::GetEntryPointsMetadata
            .print_interactive()
            .change_context(CommandError)?;
        Ok(())
    }

    fn execute_traits(&self) -> Result<(), CommandError> {
        BatSonarInteractive::GetTraitsMetadata
            .print_interactive()
            .change_context(CommandError)?;
        Ok(())
    }

    fn execute_function_dependencies(&self) -> Result<(), CommandError> {
        BatSonarInteractive::GetFunctionDependenciesMetadata
            .print_interactive()
            .change_context(CommandError)?;
        Ok(())
    }
}