rusty_commit/providers/

registry.rs

//! Provider Registry - Central registry for all AI providers
//!
//! This module provides a extensible registry pattern for AI providers.
//! New providers can be added by implementing the `ProviderBuilder` trait
//! and registering them with the `ProviderRegistry`.

use crate::config::Config;
use anyhow::{Context, Result};
use std::collections::HashMap;
use std::sync::RwLock;

/// Lock error type for registry operations
#[derive(thiserror::Error, Debug)]
#[error("Registry lock error")]
pub struct LockError;

macro_rules! read_lock {
    ($lock:expr, $field:ident) => {
        $lock.read().map_err(|_| {
            tracing::error!("{} lock is poisoned", stringify!($field));
            LockError
        })
    };
}

macro_rules! write_lock {
    ($lock:expr, $field:ident) => {
        $lock.write().map_err(|_| {
            tracing::error!("{} lock is poisoned", stringify!($field));
            LockError
        })
    };
}

/// Trait for building AI provider instances
pub trait ProviderBuilder: Send + Sync {
    /// The provider name/identifier
    fn name(&self) -> &'static str;

    /// Alternative names for this provider (aliases)
    fn aliases(&self) -> Vec<&'static str> {
        vec![]
    }

    /// Provider category for documentation
    fn category(&self) -> ProviderCategory {
        ProviderCategory::Standard
    }

    /// Create a provider instance from config
    fn create(&self, config: &Config) -> Result<Box<dyn super::AIProvider>>;

    /// Whether this provider requires an API key
    fn requires_api_key(&self) -> bool {
        true
    }

    /// Default model for this provider (if applicable)
    fn default_model(&self) -> Option<&'static str> {
        None
    }
}

/// Provider categories for organization
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[allow(dead_code)]
pub enum ProviderCategory {
    /// Direct API providers (OpenAI, Anthropic, etc.)
    Standard,
    /// OpenAI-compatible API providers
    OpenAICompatible,
    /// Self-hosted/local providers
    Local,
    /// Cloud marketplace providers
    Cloud,
}

/// Registry entry for a provider (metadata only, no builder)
#[derive(Clone)]
pub struct ProviderEntry {
    pub name: &'static str,
    pub aliases: Vec<&'static str>,
    pub category: ProviderCategory,
    #[allow(dead_code)]
    pub requires_api_key: bool,
    #[allow(dead_code)]
    pub default_model: Option<&'static str>,
}

impl ProviderEntry {
    pub fn from_builder(builder: &dyn ProviderBuilder) -> Self {
        Self {
            name: builder.name(),
            aliases: builder.aliases(),
            category: builder.category(),
            requires_api_key: builder.requires_api_key(),
            default_model: builder.default_model(),
        }
    }

    /// Check if this entry matches a provider name
    #[allow(dead_code)]
    pub fn matches(&self, provider: &str) -> bool {
        let lower = provider.to_lowercase();
        self.name.eq_ignore_ascii_case(&lower)
            || self.aliases.iter().any(|&a| a.eq_ignore_ascii_case(&lower))
    }
}

/// The provider registry - a thread-safe registry of all available providers
pub struct ProviderRegistry {
    entries: RwLock<HashMap<&'static str, ProviderEntry>>,
    builders: RwLock<HashMap<&'static str, Box<dyn ProviderBuilder>>>,
    by_alias: RwLock<HashMap<&'static str, &'static str>>,
}

impl ProviderRegistry {
    /// Create a new empty registry
    pub fn new() -> Self {
        Self {
            entries: RwLock::new(HashMap::new()),
            builders: RwLock::new(HashMap::new()),
            by_alias: RwLock::new(HashMap::new()),
        }
    }

    /// Register a provider builder
    pub fn register(&self, builder: Box<dyn ProviderBuilder>) -> Result<()> {
        let name = builder.name();
        let entry = ProviderEntry::from_builder(&*builder);

        // Register primary name
        write_lock!(self.entries, entries)?.insert(name, entry.clone());
        write_lock!(self.builders, builders)?.insert(name, builder);

        // Register aliases
        for &alias in &entry.aliases {
            write_lock!(self.by_alias, by_alias)?.insert(alias, name);
        }

        Ok(())
    }

    /// Get a provider entry by name or alias
    #[allow(dead_code)]
    pub fn get(&self, provider: &str) -> Option<ProviderEntry> {
        let lower = provider.to_lowercase();

        // Try direct lookup
        let entries = read_lock!(self.entries, entries).ok()?;
        if let Some(entry) = entries.get(lower.as_str()) {
            return Some(entry.clone());
        }

        // Try alias lookup
        let by_alias = read_lock!(self.by_alias, by_alias).ok()?;
        if let Some(&primary) = by_alias.get(lower.as_str()) {
            return entries.get(primary).cloned();
        }

        None
    }

    /// Get all registered providers
    pub fn all(&self) -> Option<Vec<ProviderEntry>> {
        let entries = read_lock!(self.entries, entries).ok()?;
        Some(entries.values().cloned().collect())
    }

    /// Get providers by category
    pub fn by_category(&self, category: ProviderCategory) -> Option<Vec<ProviderEntry>> {
        let entries = read_lock!(self.entries, entries).ok()?;
        Some(
            entries
                .values()
                .filter(|e| e.category == category)
                .cloned()
                .collect(),
        )
    }

    /// Check if any providers are registered
    #[allow(dead_code)]
    pub fn is_empty(&self) -> bool {
        match read_lock!(self.entries, entries) {
            Ok(entries) => entries.is_empty(),
            Err(_) => true,
        }
    }

    /// Get count of registered providers
    #[allow(dead_code)]
    pub fn len(&self) -> usize {
        match read_lock!(self.entries, entries) {
            Ok(entries) => entries.len(),
            Err(_) => 0,
        }
    }

    /// Create a provider instance
    pub fn create(
        &self,
        name: &str,
        config: &Config,
    ) -> Result<Option<Box<dyn super::AIProvider>>> {
        let lower = name.to_lowercase();

        let builders = read_lock!(self.builders, builders).context("Failed to read builders")?;
        let by_alias = read_lock!(self.by_alias, by_alias).context("Failed to read aliases")?;

        // Try direct lookup first
        if let Some(builder) = builders.get(lower.as_str()) {
            return Ok(Some(builder.create(config)?));
        }

        // Try alias lookup
        if let Some(&primary) = by_alias.get(lower.as_str()) {
            if let Some(builder) = builders.get(primary) {
                return Ok(Some(builder.create(config)?));
            }
        }

        Ok(None)
    }
}

impl Default for ProviderRegistry {
    fn default() -> Self {
        Self::new()
    }
}