git_iris/

common.rs

use crate::config::Config;
use crate::instruction_presets::{PresetType, get_instruction_preset_library};
use crate::providers::{Provider, ProviderConfig};
use anyhow::Result;
use clap::Args;

#[allow(clippy::struct_excessive_bools)]
#[derive(Args, Clone, Default, Debug)]
pub struct CommonParams {
    /// Override default LLM provider
    #[arg(long, help = "Override default LLM provider", value_parser = available_providers_parser)]
    pub provider: Option<String>,

    /// Override model for this operation
    #[arg(long, help = "Override model for this operation")]
    pub model: Option<String>,

    /// Custom instructions for this operation
    #[arg(short, long, help = "Custom instructions for this operation")]
    pub instructions: Option<String>,

    /// Select an instruction preset
    #[arg(
        long,
        help = "Select an instruction preset (use 'git-iris list-presets' to see available presets for commits and reviews)"
    )]
    pub preset: Option<String>,

    /// Enable Gitmoji (default: true)
    #[arg(
        long = "gitmoji",
        help = "Enable Gitmoji",
        conflicts_with = "no_gitmoji",
        action = clap::ArgAction::SetTrue
    )]
    pub gitmoji_flag: bool,

    /// Disable Gitmoji
    #[arg(long = "no-gitmoji", help = "Disable Gitmoji", action = clap::ArgAction::SetTrue)]
    pub no_gitmoji: bool,

    /// Enable critic verification pass
    #[arg(
        long = "critic",
        help = "Enable critic verification pass",
        conflicts_with = "no_critic",
        action = clap::ArgAction::SetTrue
    )]
    pub critic_flag: bool,

    /// Disable critic verification pass
    #[arg(long = "no-critic", help = "Disable critic verification pass", action = clap::ArgAction::SetTrue)]
    pub no_critic: bool,

    /// Internal: resolved gitmoji value (Some(true), Some(false), or None)
    #[arg(skip)]
    pub gitmoji: Option<bool>,

    /// Internal: resolved critic value (Some(true), Some(false), or None)
    #[arg(skip)]
    pub critic: Option<bool>,

    /// Repository URL to use instead of local repository
    #[arg(
        short = 'r',
        long = "repo",
        help = "Repository URL to use instead of local repository"
    )]
    pub repository_url: Option<String>,
}

impl CommonParams {
    /// Get the resolved gitmoji value from CLI flags
    /// Returns Some(true) for --gitmoji, Some(false) for --no-gitmoji, None if neither specified
    #[must_use]
    pub fn resolved_gitmoji(&self) -> Option<bool> {
        if self.gitmoji_flag {
            Some(true)
        } else if self.no_gitmoji {
            Some(false)
        } else {
            self.gitmoji // fallback to any programmatically set value
        }
    }

    /// Get the resolved critic value from CLI flags
    /// Returns Some(true) for --critic, Some(false) for --no-critic, None if neither specified
    #[must_use]
    pub fn resolved_critic(&self) -> Option<bool> {
        if self.critic_flag {
            Some(true)
        } else if self.no_critic {
            Some(false)
        } else {
            self.critic
        }
    }

    ///
    /// # Errors
    ///
    /// Returns an error when the requested provider is invalid.
    pub fn apply_to_config(&self, config: &mut Config) -> Result<bool> {
        let mut changes_made = false;

        if let Some(provider_str) = &self.provider {
            // Parse and validate provider
            let provider: Provider = provider_str.parse()?;
            let provider_name = provider.name().to_string();

            // Check if we need to update the default provider
            if config.default_provider != provider_name {
                // Ensure the provider exists in the providers HashMap
                if !config.providers.contains_key(&provider_name) {
                    config.providers.insert(
                        provider_name.clone(),
                        ProviderConfig::with_defaults(provider),
                    );
                }

                config.default_provider = provider_name;
                changes_made = true;
            }
        }

        // Apply model override if specified
        if let Some(model) = &self.model {
            let provider_name = config.default_provider.clone();
            if let Some(provider_config) = config.providers.get_mut(&provider_name) {
                provider_config.model.clone_from(model);
                changes_made = true;
            }
        }

        if let Some(instructions) = &self.instructions {
            config.set_temp_instructions(Some(instructions.clone()));
        }

        if let Some(preset) = &self.preset {
            config.set_temp_preset(Some(preset.clone()));
        }

        // Track whether gitmoji was explicitly set via CLI (for style detection)
        if let Some(use_gitmoji) = self.resolved_gitmoji() {
            config.gitmoji_override = Some(use_gitmoji);
            if config.use_gitmoji != use_gitmoji {
                config.use_gitmoji = use_gitmoji;
                changes_made = true;
            }
        }

        if let Some(critic_enabled) = self.resolved_critic()
            && config.critic_enabled != critic_enabled
        {
            config.critic_enabled = critic_enabled;
            changes_made = true;
        }

        Ok(changes_made)
    }

    /// Check if the provided preset is valid for the specified preset type
    #[must_use]
    pub fn is_valid_preset_for_type(&self, preset_type: PresetType) -> bool {
        if let Some(preset_name) = &self.preset {
            let library = get_instruction_preset_library();
            let valid_presets = library.list_valid_presets_for_command(preset_type);
            return valid_presets.iter().any(|(key, _)| *key == preset_name);
        }
        true // No preset specified is always valid
    }
}

/// Validates that a provider name is available in the system
///
/// # Errors
///
/// Returns an error string when the provider name is not supported.
pub fn available_providers_parser(s: &str) -> Result<String, String> {
    match s.parse::<Provider>() {
        Ok(provider) => Ok(provider.name().to_string()),
        Err(_) => Err(format!(
            "Invalid provider '{}'. Available providers: {}",
            s,
            Provider::all_names().join(", ")
        )),
    }
}