mockforge-contracts 0.3.135

//! Consumer mapping for tracking endpoint to SDK method to consuming app relationships
//!
//! This module provides functionality for mapping API endpoints to SDK methods and
//! tracking which applications consume those methods. This enables consumer-focused
//! drift insights that show which apps will be affected by contract changes.
//!
//! The data types are re-exported from `mockforge-foundation::contract_drift_types`
//! so `mockforge-core` and `mockforge-contracts` share the same underlying types.

pub use mockforge_foundation::contract_drift_types::{
    AppType, ConsumerImpact, ConsumerMapping, ConsumingApp, SDKMethod,
};
use std::collections::{HashMap, HashSet};

/// Registry for managing consumer mappings
#[derive(Debug, Clone)]
pub struct ConsumerMappingRegistry {
    /// Mappings keyed by "{method} {endpoint}"
    mappings: HashMap<String, ConsumerMapping>,
}

impl ConsumerMappingRegistry {
    /// Create a new consumer mapping registry
    pub fn new() -> Self {
        Self {
            mappings: HashMap::new(),
        }
    }

    /// Add or update a consumer mapping
    pub fn add_mapping(&mut self, mapping: ConsumerMapping) {
        let key = format!("{} {}", mapping.method, mapping.endpoint);
        self.mappings.insert(key, mapping);
    }

    /// Get mapping for a specific endpoint
    pub fn get_mapping(&self, endpoint: &str, method: &str) -> Option<&ConsumerMapping> {
        let key = format!("{} {}", method, endpoint);
        self.mappings.get(&key)
    }

    /// List all mappings
    pub fn list_mappings(&self) -> Vec<&ConsumerMapping> {
        self.mappings.values().collect()
    }

    /// Remove a mapping
    pub fn remove_mapping(&mut self, endpoint: &str, method: &str) -> Option<ConsumerMapping> {
        let key = format!("{} {}", method, endpoint);
        self.mappings.remove(&key)
    }

    /// Add an SDK method to an endpoint mapping
    pub fn add_sdk_method(&mut self, endpoint: &str, method: &str, sdk_method: SDKMethod) {
        let key = format!("{} {}", method, endpoint);
        let mapping = self.mappings.entry(key).or_insert_with(|| ConsumerMapping {
            endpoint: endpoint.to_string(),
            method: method.to_string(),
            sdk_methods: Vec::new(),
            created_at: chrono::Utc::now().timestamp(),
            updated_at: chrono::Utc::now().timestamp(),
        });

        // Check if SDK method already exists
        if let Some(existing) = mapping
            .sdk_methods
            .iter_mut()
            .find(|m| m.sdk_name == sdk_method.sdk_name && m.method_name == sdk_method.method_name)
        {
            // Merge consuming apps (avoid duplicates)
            let mut existing_app_ids: HashSet<String> =
                existing.consuming_apps.iter().map(|a| a.app_id.clone()).collect();

            for app in sdk_method.consuming_apps {
                if !existing_app_ids.contains(&app.app_id) {
                    let app_id = app.app_id.clone();
                    existing.consuming_apps.push(app);
                    existing_app_ids.insert(app_id);
                }
            }
        } else {
            mapping.sdk_methods.push(sdk_method);
        }

        mapping.updated_at = chrono::Utc::now().timestamp();
    }

    /// Add a consuming app to an SDK method
    pub fn add_consuming_app(
        &mut self,
        endpoint: &str,
        method: &str,
        sdk_name: &str,
        method_name: &str,
        app: ConsumingApp,
    ) {
        let key = format!("{} {}", method, endpoint);
        if let Some(mapping) = self.mappings.get_mut(&key) {
            if let Some(sdk_method) = mapping
                .sdk_methods
                .iter_mut()
                .find(|m| m.sdk_name == sdk_name && m.method_name == method_name)
            {
                // Check if app already exists
                if !sdk_method.consuming_apps.iter().any(|a| a.app_id == app.app_id) {
                    sdk_method.consuming_apps.push(app);
                    mapping.updated_at = chrono::Utc::now().timestamp();
                }
            }
        }
    }
}

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

/// Analyzer for determining consumer impact from drift
#[derive(Debug)]
pub struct ConsumerImpactAnalyzer {
    registry: ConsumerMappingRegistry,
}

impl ConsumerImpactAnalyzer {
    /// Create a new consumer impact analyzer
    pub fn new(registry: ConsumerMappingRegistry) -> Self {
        Self { registry }
    }

    /// Analyze impact of drift on a specific endpoint
    pub fn analyze_impact(&self, endpoint: &str, method: &str) -> Option<ConsumerImpact> {
        self.analyze_impact_with_operation_id(endpoint, method, None)
    }

    /// Analyze impact with an optional operation_id for more flexible matching
    pub fn analyze_impact_with_operation_id(
        &self,
        endpoint: &str,
        method: &str,
        operation_id: Option<&str>,
    ) -> Option<ConsumerImpact> {
        // Try direct lookup first
        let mut mapping = self.registry.get_mapping(endpoint, method);

        // If operation_id is provided and different from endpoint, try it
        if mapping.is_none() {
            if let Some(op_id) = operation_id {
                if op_id != endpoint {
                    mapping = self.registry.get_mapping(op_id, method);
                }
            }
        }

        // If not found and this looks like a protocol-specific operation, try alternative formats
        if mapping.is_none() {
            mapping = self.try_protocol_specific_lookup(endpoint, method, operation_id);
        }

        let mapping = mapping?;

        // Collect all affected SDK methods and apps
        let affected_sdk_methods = mapping.sdk_methods.clone();
        let mut affected_apps: Vec<ConsumingApp> = Vec::new();
        let mut seen_app_ids: HashSet<String> = HashSet::new();

        for sdk_method in &affected_sdk_methods {
            for app in &sdk_method.consuming_apps {
                if !seen_app_ids.contains(&app.app_id) {
                    affected_apps.push(app.clone());
                    seen_app_ids.insert(app.app_id.clone());
                }
            }
        }

        // Generate impact summary
        let app_names: Vec<String> =
            affected_apps.iter().map(|app| app.app_type.to_string()).collect();

        let impact_summary = if app_names.is_empty() {
            format!("No known consumers for {} {}", method, endpoint)
        } else {
            format!("This change may break: {}", app_names.join(", "))
        };

        Some(ConsumerImpact {
            endpoint: endpoint.to_string(),
            method: method.to_string(),
            affected_sdk_methods,
            affected_apps,
            impact_summary,
        })
    }

    /// Analyze impact for multiple endpoints
    pub fn analyze_impact_multiple(&self, endpoints: &[(String, String)]) -> Vec<ConsumerImpact> {
        endpoints
            .iter()
            .filter_map(|(endpoint, method)| self.analyze_impact(endpoint, method))
            .collect()
    }

    /// Get the registry
    pub fn registry(&self) -> &ConsumerMappingRegistry {
        &self.registry
    }

    /// Get mutable access to the registry
    pub fn registry_mut(&mut self) -> &mut ConsumerMappingRegistry {
        &mut self.registry
    }

    /// Try protocol-specific lookup strategies
    fn try_protocol_specific_lookup<'a>(
        &'a self,
        endpoint: &str,
        method: &str,
        operation_id: Option<&str>,
    ) -> Option<&'a ConsumerMapping> {
        // For gRPC
        if method == "grpc"
            || (method.contains("grpc")
                && (endpoint.contains('.')
                    || operation_id.map(|id| id.contains('.')).unwrap_or(false)))
        {
            return self.try_grpc_lookup(endpoint, operation_id);
        }
        // For WebSocket
        if method == "websocket" || method.contains("websocket") {
            return self.try_websocket_lookup(endpoint, operation_id);
        }
        // For MQTT/Kafka
        if method == "mqtt"
            || method == "kafka"
            || method.contains("mqtt")
            || method.contains("kafka")
        {
            let protocol = if method.contains("mqtt") {
                "mqtt"
            } else {
                "kafka"
            };
            return self.try_topic_lookup(endpoint, protocol, operation_id);
        }

        None
    }

    /// Try gRPC-specific lookup strategies
    fn try_grpc_lookup<'a>(
        &'a self,
        endpoint: &str,
        operation_id: Option<&str>,
    ) -> Option<&'a ConsumerMapping> {
        let mut mapping = self.registry.get_mapping(endpoint, "grpc");

        if mapping.is_none() {
            if let Some(op_id) = operation_id {
                if op_id != endpoint {
                    mapping = self.registry.get_mapping(op_id, "grpc");
                }
            }
        }

        if mapping.is_none() {
            let lookup_key = operation_id.unwrap_or(endpoint);
            if lookup_key.contains('.') {
                let parts: Vec<&str> = lookup_key.split('.').collect();
                if parts.len() >= 2 {
                    let service_pattern = format!("{}.*", parts[0..parts.len() - 1].join("."));
                    mapping = self.try_wildcard_match(&service_pattern, "grpc");

                    if mapping.is_none() {
                        let service_name = parts[0..parts.len() - 1].join(".");
                        mapping = self.registry.get_mapping(&service_name, "grpc");
                    }
                }
            }
        }

        mapping
    }

    /// Try WebSocket-specific lookup strategies
    fn try_websocket_lookup<'a>(
        &'a self,
        endpoint: &str,
        operation_id: Option<&str>,
    ) -> Option<&'a ConsumerMapping> {
        let mut mapping = self.registry.get_mapping(endpoint, "websocket");

        if mapping.is_none() {
            if let Some(op_id) = operation_id {
                if op_id != endpoint {
                    mapping = self.registry.get_mapping(op_id, "websocket");
                }
            }
        }

        if mapping.is_none() && endpoint.contains(':') {
            if let Some(message_type) = endpoint.split(':').nth(1) {
                mapping = self.registry.get_mapping(message_type, "websocket");
            }
        }

        if mapping.is_none() {
            if let Some(op_id) = operation_id {
                if op_id.contains(':') {
                    if let Some(message_type) = op_id.split(':').nth(1) {
                        mapping = self.registry.get_mapping(message_type, "websocket");
                    }
                }
            }
        }

        mapping
    }

    /// Try topic-based lookup for MQTT/Kafka
    fn try_topic_lookup<'a>(
        &'a self,
        endpoint: &str,
        protocol: &str,
        operation_id: Option<&str>,
    ) -> Option<&'a ConsumerMapping> {
        let mut mapping = self.registry.get_mapping(endpoint, protocol);

        if mapping.is_none() {
            if let Some(op_id) = operation_id {
                if op_id != endpoint {
                    mapping = self.registry.get_mapping(op_id, protocol);
                }
            }
        }

        if mapping.is_none() {
            let lookup_key = operation_id.unwrap_or(endpoint);
            mapping = self.try_topic_wildcard_match(lookup_key, protocol);
        }

        mapping
    }

    /// Try to find a mapping using wildcard pattern matching
    fn try_wildcard_match(&self, pattern: &str, method: &str) -> Option<&ConsumerMapping> {
        let base_pattern = pattern.strip_suffix(".*").unwrap_or(pattern);

        for (key, mapping) in &self.registry.mappings {
            if key.starts_with(&format!("{} ", method))
                && mapping.endpoint.starts_with(base_pattern)
            {
                return Some(mapping);
            }
        }

        None
    }

    /// Try to find a mapping using topic wildcard matching
    fn try_topic_wildcard_match(&self, topic: &str, method: &str) -> Option<&ConsumerMapping> {
        for (key, mapping) in &self.registry.mappings {
            if key.starts_with(&format!("{} ", method))
                && Self::matches_topic_pattern(topic, &mapping.endpoint)
            {
                return Some(mapping);
            }
        }

        None
    }

    /// Check if a topic matches a pattern with MQTT wildcards
    fn matches_topic_pattern(topic: &str, pattern: &str) -> bool {
        if topic == pattern {
            return true;
        }

        if pattern.contains('+') || pattern.contains('#') {
            let topic_parts: Vec<&str> = topic.split('/').collect();
            let pattern_parts: Vec<&str> = pattern.split('/').collect();

            if let Some(base_pattern) = pattern.strip_suffix("/#") {
                return topic.starts_with(base_pattern);
            }

            if topic_parts.len() == pattern_parts.len() {
                for (topic_part, pattern_part) in topic_parts.iter().zip(pattern_parts.iter()) {
                    if *pattern_part != "+" && *pattern_part != "#" && topic_part != pattern_part {
                        return false;
                    }
                }
                return true;
            }
        }

        false
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_consumer_mapping_registry() {
        let mut registry = ConsumerMappingRegistry::new();

        let app1 = ConsumingApp {
            app_id: "web-app-1".to_string(),
            app_name: "Web App".to_string(),
            app_type: AppType::Web,
            repository_url: Some("https://github.com/org/web-app".to_string()),
            last_updated: Some(chrono::Utc::now().timestamp()),
            description: None,
        };

        let sdk_method = SDKMethod {
            sdk_name: "typescript-sdk".to_string(),
            method_name: "getUser".to_string(),
            consuming_apps: vec![app1],
        };

        registry.add_sdk_method("/api/users", "GET", sdk_method);

        let mapping = registry.get_mapping("/api/users", "GET");
        assert!(mapping.is_some());
        assert_eq!(mapping.unwrap().sdk_methods.len(), 1);
    }

    #[test]
    fn test_consumer_impact_analyzer() {
        let mut registry = ConsumerMappingRegistry::new();

        let app1 = ConsumingApp {
            app_id: "web-app-1".to_string(),
            app_name: "Web App".to_string(),
            app_type: AppType::Web,
            repository_url: None,
            last_updated: None,
            description: None,
        };

        let app2 = ConsumingApp {
            app_id: "mobile-android-1".to_string(),
            app_name: "Mobile App".to_string(),
            app_type: AppType::MobileAndroid,
            repository_url: None,
            last_updated: None,
            description: None,
        };

        let sdk_method = SDKMethod {
            sdk_name: "typescript-sdk".to_string(),
            method_name: "getUser".to_string(),
            consuming_apps: vec![app1, app2],
        };

        registry.add_sdk_method("/api/users", "GET", sdk_method);

        let analyzer = ConsumerImpactAnalyzer::new(registry);
        let impact = analyzer.analyze_impact("/api/users", "GET");

        assert!(impact.is_some());
        let impact = impact.unwrap();
        assert_eq!(impact.affected_apps.len(), 2);
        assert!(impact.impact_summary.contains("Web App"));
        assert!(impact.impact_summary.contains("Mobile App (Android)"));
    }
}