mockforge-core 0.3.116

//! Import utilities for detecting and parsing various API formats
//!
//! This module provides functionality to detect and parse imports from:
//! - Postman collections (JSON)
//! - Postman environments (JSON)
//! - Insomnia exports (JSON/YAML)
//! - Curl commands (text)

use crate::import::postman_environment::is_postman_environment_json;
use serde_json::Value;
use std::path::Path;

/// Import format types supported by MockForge
#[derive(Debug, Clone, PartialEq)]
pub enum ImportFormat {
    /// Postman Collection format (JSON)
    Postman,
    /// Postman Environment format (JSON)
    PostmanEnvironment,
    /// Insomnia export format (JSON/YAML)
    Insomnia,
    /// cURL command format (text)
    Curl,
    /// HAR (HTTP Archive) format (JSON)
    Har,
    /// OpenAPI 3.x specification (JSON/YAML)
    OpenApi,
    /// Swagger 2.0 specification (JSON/YAML)
    Swagger,
    /// Format could not be determined
    Unknown,
}

/// Result of format detection with confidence score
#[derive(Debug)]
pub struct FormatDetection {
    /// Detected import format
    pub format: ImportFormat,
    /// Confidence score from 0.0 to 1.0 (1.0 = very confident)
    pub confidence: f64,
    /// Human-readable description of the detection
    pub details: String,
}

/// Detect the format of an import source
pub fn detect_format(content: &str, file_path: Option<&Path>) -> FormatDetection {
    // Check file extension first
    if let Some(path) = file_path {
        if let Some(ext) = path.extension().and_then(|s| s.to_str()) {
            match ext.to_lowercase().as_str() {
                "json" => {
                    // Could be Postman, Insomnia, or OpenAPI
                    return detect_json_format(content);
                }
                "yaml" | "yml" => {
                    // Could be Insomnia or OpenAPI
                    if is_insomnia_yaml(content) {
                        return FormatDetection {
                            format: ImportFormat::Insomnia,
                            confidence: 0.9,
                            details: "YAML file with Insomnia structure detected".to_string(),
                        };
                    }
                    // Check for OpenAPI/Swagger in YAML
                    if is_openapi_yaml(content) {
                        return FormatDetection {
                            format: ImportFormat::OpenApi,
                            confidence: 0.95,
                            details: "YAML with OpenAPI specification detected".to_string(),
                        };
                    }
                }
                "txt" | "sh" | "bash" => {
                    // Could be curl commands
                    if is_curl_content(content) {
                        return FormatDetection {
                            format: ImportFormat::Curl,
                            confidence: 0.8,
                            details: "File contains curl commands".to_string(),
                        };
                    }
                }
                _ => {}
            }
        }
    }

    // Content-based detection
    if is_curl_content(content) {
        return FormatDetection {
            format: ImportFormat::Curl,
            confidence: 0.9,
            details: "Content contains curl commands".to_string(),
        };
    }

    if is_openapi_json(content) {
        return FormatDetection {
            format: ImportFormat::OpenApi,
            confidence: 0.95,
            details: "JSON with OpenAPI specification detected".to_string(),
        };
    }

    if is_swagger_json(content) {
        return FormatDetection {
            format: ImportFormat::Swagger,
            confidence: 0.95,
            details: "JSON with Swagger specification detected".to_string(),
        };
    }

    if is_postman_json(content) {
        return FormatDetection {
            format: ImportFormat::Postman,
            confidence: 0.95,
            details: "JSON with Postman collection structure detected".to_string(),
        };
    }

    if is_postman_environment_json(content) {
        return FormatDetection {
            format: ImportFormat::PostmanEnvironment,
            confidence: 0.95,
            details: "JSON with Postman environment structure detected".to_string(),
        };
    }

    if is_insomnia_json(content) {
        return FormatDetection {
            format: ImportFormat::Insomnia,
            confidence: 0.95,
            details: "JSON with Insomnia export structure detected".to_string(),
        };
    }

    if is_har_json(content) {
        return FormatDetection {
            format: ImportFormat::Har,
            confidence: 0.95,
            details: "JSON with HAR (HTTP Archive) structure detected".to_string(),
        };
    }

    FormatDetection {
        format: ImportFormat::Unknown,
        confidence: 0.0,
        details: "Could not determine import format".to_string(),
    }
}

/// Detect format for JSON content (Postman vs Insomnia vs HAR)
fn detect_json_format(content: &str) -> FormatDetection {
    if is_postman_json(content) {
        FormatDetection {
            format: ImportFormat::Postman,
            confidence: 0.95,
            details: "Postman collection JSON detected".to_string(),
        }
    } else if is_postman_environment_json(content) {
        FormatDetection {
            format: ImportFormat::PostmanEnvironment,
            confidence: 0.95,
            details: "Postman environment JSON detected".to_string(),
        }
    } else if is_insomnia_json(content) {
        FormatDetection {
            format: ImportFormat::Insomnia,
            confidence: 0.95,
            details: "Insomnia export JSON detected".to_string(),
        }
    } else if is_har_json(content) {
        FormatDetection {
            format: ImportFormat::Har,
            confidence: 0.95,
            details: "HAR (HTTP Archive) JSON detected".to_string(),
        }
    } else {
        FormatDetection {
            format: ImportFormat::Unknown,
            confidence: 0.0,
            details: "JSON format not recognized".to_string(),
        }
    }
}

/// Check if content is a Postman collection
fn is_postman_json(content: &str) -> bool {
    if let Ok(json) = serde_json::from_str::<Value>(content) {
        // Postman collections have specific structure
        if let Some(obj) = json.as_object() {
            // Check for Postman v2.1 structure
            if obj.contains_key("item") && obj.contains_key("info") {
                if let Some(info) = obj.get("info").and_then(|i| i.as_object()) {
                    // Check for Postman schema
                    if let Some(schema) = info.get("schema") {
                        if let Some(schema_str) = schema.as_str() {
                            return schema_str.contains("postman");
                        }
                    }
                    // Alternative: check for _postman_id
                    if info.contains_key("_postman_id") {
                        return true;
                    }
                }
            }
        }
    }
    false
}

/// Check if content is an Insomnia export
fn is_insomnia_json(content: &str) -> bool {
    if let Ok(json) = serde_json::from_str::<Value>(content) {
        if let Some(obj) = json.as_object() {
            // Insomnia exports have __export_format or _type field
            if obj.contains_key("__export_format") {
                if let Some(format_val) = obj.get("__export_format") {
                    if let Some(format_num) = format_val.as_i64() {
                        return format_num >= 3; // Insomnia v3+ format
                    }
                }
            }

            // Check for Insomnia-specific fields
            if obj.contains_key("_type") && obj.contains_key("resources") {
                return true;
            }
        }
    }
    false
}

/// Check if content is a HAR (HTTP Archive) file
fn is_har_json(content: &str) -> bool {
    if let Ok(json) = serde_json::from_str::<Value>(content) {
        if let Some(obj) = json.as_object() {
            // HAR files have a log object with entries array
            if let Some(log) = obj.get("log") {
                if let Some(log_obj) = log.as_object() {
                    // Check for HAR-specific fields
                    if log_obj.contains_key("entries") && log_obj.contains_key("version") {
                        // Verify entries is an array
                        if let Some(entries) = log_obj.get("entries") {
                            if entries.is_array() {
                                return true;
                            }
                        }
                    }
                }
            }
        }
    }
    false
}

/// Check if content is Insomnia YAML
fn is_insomnia_yaml(content: &str) -> bool {
    // Simple check for YAML structure with Insomnia patterns
    content.contains("__export_format:") || content.contains("_type:")
}

/// Check if content contains curl commands
fn is_curl_content(content: &str) -> bool {
    let lines: Vec<&str> = content.lines().collect();

    // Check if any line starts with curl
    for line in &lines {
        let trimmed = line.trim();
        if trimmed.starts_with("curl ") || trimmed.starts_with("curl\t") {
            return true;
        }
        // Also check for curl in scripts
        if trimmed.contains("curl ") && (trimmed.contains("http") || trimmed.contains("https")) {
            return true;
        }
    }

    false
}

/// Check if content is an OpenAPI specification (JSON)
fn is_openapi_json(content: &str) -> bool {
    if let Ok(json) = serde_json::from_str::<Value>(content) {
        if let Some(obj) = json.as_object() {
            // Check for OpenAPI 3.x fields
            if obj.contains_key("openapi") && obj.contains_key("info") && obj.contains_key("paths")
            {
                if let Some(openapi_version) = obj.get("openapi") {
                    if let Some(version_str) = openapi_version.as_str() {
                        // OpenAPI 3.0, 3.0.x, 3.1, 3.1.x etc.
                        return version_str.starts_with("3.");
                    }
                }
            }
        }
    }
    false
}

/// Check if content is a Swagger 2.0 specification (JSON)
fn is_swagger_json(content: &str) -> bool {
    if let Ok(json) = serde_json::from_str::<Value>(content) {
        if let Some(obj) = json.as_object() {
            // Check for Swagger 2.0 fields
            if obj.contains_key("swagger") && obj.contains_key("info") && obj.contains_key("paths")
            {
                if let Some(swagger_version) = obj.get("swagger") {
                    if let Some(version_str) = swagger_version.as_str() {
                        return version_str.starts_with("2.0");
                    }
                }
            }
        }
    }
    false
}

/// Check if content is an OpenAPI specification (YAML)
fn is_openapi_yaml(content: &str) -> bool {
    // Simple checks for YAML structure with OpenAPI patterns
    content.contains("openapi:") || content.contains("swagger:")
}

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

    #[test]
    fn test_detect_postman_format() {
        let postman_json = r#"{
            "info": {
                "_postman_id": "12345",
                "name": "Test Collection",
                "schema": "https://schema.getpostman.com/json/collection/v2.1.0/collection.json"
            },
            "item": []
        }"#;

        let detection = detect_format(postman_json, None);
        assert_eq!(detection.format, ImportFormat::Postman);
        assert!(detection.confidence > 0.9);
    }

    #[test]
    fn test_detect_insomnia_format() {
        let insomnia_json = r#"{
            "__export_format": 4,
            "_type": "export",
            "resources": []
        }"#;

        let detection = detect_format(insomnia_json, None);
        assert_eq!(detection.format, ImportFormat::Insomnia);
        assert!(detection.confidence > 0.9);
    }

    #[test]
    fn test_detect_curl_format() {
        let curl_content = "curl -X GET https://api.example.com/users";

        let detection = detect_format(curl_content, None);
        assert_eq!(detection.format, ImportFormat::Curl);
        assert!(detection.confidence > 0.8);
    }

    #[test]
    fn test_detect_har_format() {
        let har_json = r#"{
            "log": {
                "version": "1.2",
                "creator": {"name": "Test", "version": "1.0"},
                "entries": [{"request": {"method": "GET", "url": "https://example.com"}, "response": {"status": 200}}]
            }
        }"#;

        let detection = detect_format(har_json, None);
        assert_eq!(detection.format, ImportFormat::Har);
        assert!(detection.confidence > 0.9);
    }

    #[test]
    fn test_detect_unknown_format() {
        let unknown_content = "some random text";

        let detection = detect_format(unknown_content, None);
        assert_eq!(detection.format, ImportFormat::Unknown);
        assert_eq!(detection.confidence, 0.0);
    }
}