mockforge-core 0.3.116

//! Insomnia export import functionality
//!
//! This module handles parsing Insomnia exports and converting them
//! to MockForge routes and configurations.

use serde::{Deserialize, Serialize};
use serde_json::{json, Value};
use std::collections::HashMap;

/// Insomnia export structure
#[derive(Debug, Deserialize)]
pub struct InsomniaExport {
    /// Export format version
    #[serde(rename = "__export_format")]
    pub export_format: Option<i32>,
    /// Export type identifier
    #[serde(rename = "_type")]
    pub export_type: Option<String>,
    /// Array of resources (requests, environments, folders, etc.)
    pub resources: Vec<InsomniaResource>,
}

/// Generic Insomnia resource (request, folder, environment, etc.)
#[derive(Debug, Deserialize)]
pub struct InsomniaResource {
    /// Unique resource identifier
    #[serde(rename = "_id")]
    pub id: String,
    /// Resource type (request, folder, environment, etc.)
    #[serde(rename = "_type")]
    pub resource_type: String,
    /// Parent resource ID (for nested resources)
    pub parent_id: Option<String>,
    /// Resource name
    pub name: Option<String>,
    /// Request URL (for request resources)
    pub url: Option<String>,
    /// HTTP method (for request resources)
    pub method: Option<String>,
    /// Request headers (for request resources)
    pub headers: Option<Vec<InsomniaHeader>>,
    /// Request body (for request resources)
    pub body: Option<InsomniaBody>,
    /// Authentication configuration (for request resources)
    pub authentication: Option<InsomniaAuth>,
    /// Query/form parameters (for request resources)
    pub parameters: Option<Vec<InsomniaParameter>>,
    /// Environment variable data (for environment resources)
    pub data: Option<Value>,
    /// Environment name (for environment resources)
    pub environment: Option<String>,
}

/// Insomnia header entry
#[derive(Debug, Deserialize)]
pub struct InsomniaHeader {
    /// Header name
    pub name: String,
    /// Header value
    pub value: String,
    /// Whether this header is disabled
    pub disabled: Option<bool>,
}

/// Insomnia request body structure
#[derive(Debug, Deserialize)]
pub struct InsomniaBody {
    /// MIME type of the body
    pub mime_type: Option<String>,
    /// Raw body text
    pub text: Option<String>,
    /// Form data parameters (for form-data bodies)
    pub params: Option<Vec<InsomniaParameter>>,
}

/// Insomnia parameter (query params, form data, etc.)
#[derive(Debug, Deserialize)]
pub struct InsomniaParameter {
    /// Parameter name
    pub name: String,
    /// Parameter value
    pub value: String,
    /// Whether this parameter is disabled
    pub disabled: Option<bool>,
}

/// Insomnia authentication configuration
#[derive(Debug, Deserialize)]
pub struct InsomniaAuth {
    /// Authentication type (bearer, basic, apikey, etc.)
    #[serde(rename = "type")]
    pub auth_type: String,
    /// Whether authentication is disabled
    pub disabled: Option<bool>,
    /// Username (for basic auth)
    pub username: Option<String>,
    /// Password (for basic auth)
    pub password: Option<String>,
    /// Bearer token (for bearer auth)
    pub token: Option<String>,
    /// Token prefix (for bearer auth)
    pub prefix: Option<String>,
    /// API key name (for apikey auth)
    pub key: Option<String>,
    /// API key value (for apikey auth)
    pub value: Option<String>,
    /// OAuth2 access token URL
    #[serde(rename = "accessTokenUrl")]
    pub access_token_url: Option<String>,
    /// OAuth2 client ID
    #[serde(rename = "clientId")]
    pub client_id: Option<String>,
    /// OAuth2 grant type
    #[serde(rename = "grantType")]
    pub grant_type: Option<String>,
    /// OAuth2 access token value (stored after successful OAuth flow)
    #[serde(rename = "accessToken")]
    pub access_token: Option<String>,
}

/// MockForge route structure for import
#[derive(Debug, Serialize)]
pub struct MockForgeRoute {
    /// HTTP method
    pub method: String,
    /// Request path
    pub path: String,
    /// Request headers
    pub headers: HashMap<String, String>,
    /// Optional request body
    pub body: Option<String>,
    /// Mock response for this route
    pub response: MockForgeResponse,
}

/// MockForge response structure
#[derive(Debug, Serialize)]
pub struct MockForgeResponse {
    /// HTTP status code
    pub status: u16,
    /// Response headers
    pub headers: HashMap<String, String>,
    /// Response body
    pub body: Value,
}

/// Result of importing an Insomnia export
#[derive(Debug)]
pub struct InsomniaImportResult {
    /// Converted routes from Insomnia requests
    pub routes: Vec<MockForgeRoute>,
    /// Extracted environment variables
    pub variables: HashMap<String, String>,
    /// Warnings encountered during import
    pub warnings: Vec<String>,
}

/// Import an Insomnia export
pub fn import_insomnia_export(
    content: &str,
    environment: Option<&str>,
) -> Result<InsomniaImportResult, String> {
    let export: InsomniaExport = serde_json::from_str(content)
        .map_err(|e| format!("Failed to parse Insomnia export: {}", e))?;

    // Validate export format
    if let Some(format) = export.export_format {
        if format < 3 {
            return Err("Insomnia export format version 3 or higher is required".to_string());
        }
    }

    let mut routes = Vec::new();
    let mut variables = HashMap::new();
    let mut warnings = Vec::new();

    // Extract environment variables if specified
    if let Some(env_name) = environment {
        extract_environment_variables(&export.resources, env_name, &mut variables);
    } else {
        // Try to find default environment
        extract_environment_variables(&export.resources, "Base Environment", &mut variables);
    }

    // Process all resources to find requests
    for resource in &export.resources {
        if resource.resource_type == "request" {
            match convert_insomnia_request_to_route(resource, &variables) {
                Ok(route) => routes.push(route),
                Err(e) => warnings.push(format!(
                    "Failed to convert request '{}': {}",
                    resource.name.as_deref().unwrap_or("unnamed"),
                    e
                )),
            }
        }
    }

    Ok(InsomniaImportResult {
        routes,
        variables,
        warnings,
    })
}

/// Extract variables from specified environment
fn extract_environment_variables(
    resources: &[InsomniaResource],
    env_name: &str,
    variables: &mut HashMap<String, String>,
) {
    for resource in resources {
        if resource.resource_type == "environment" && resource.name.as_deref() == Some(env_name) {
            if let Some(data) = &resource.data {
                if let Some(obj) = data.as_object() {
                    for (key, value) in obj {
                        if let Some(str_value) = value.as_str() {
                            variables.insert(key.clone(), str_value.to_string());
                        } else if let Some(num_value) = value.as_f64() {
                            variables.insert(key.clone(), num_value.to_string());
                        } else if let Some(bool_value) = value.as_bool() {
                            variables.insert(key.clone(), bool_value.to_string());
                        }
                    }
                }
            }
        }
    }
}

/// Convert an Insomnia request to a MockForge route
fn convert_insomnia_request_to_route(
    resource: &InsomniaResource,
    variables: &HashMap<String, String>,
) -> Result<MockForgeRoute, String> {
    let method = resource.method.as_deref().ok_or("Request missing method")?.to_uppercase();

    let raw_url = resource.url.as_deref().ok_or("Request missing URL")?;

    let url = resolve_variables(raw_url, variables);

    // Extract path from URL
    let path = extract_path_from_url(&url)?;

    // Extract headers
    let mut headers = HashMap::new();
    if let Some(resource_headers) = &resource.headers {
        for header in resource_headers {
            if !header.disabled.unwrap_or(false) && !header.name.is_empty() {
                headers.insert(header.name.clone(), resolve_variables(&header.value, variables));
            }
        }
    }

    // Add authentication headers
    if let Some(auth) = &resource.authentication {
        if !auth.disabled.unwrap_or(false) {
            add_auth_headers(auth, &mut headers, variables);
        }
    }

    // Extract body
    let body = extract_request_body(resource, variables);

    // Generate mock response
    let response = generate_mock_response(&method);

    Ok(MockForgeRoute {
        method,
        path,
        headers,
        body,
        response,
    })
}

/// Extract path from URL, handling full URLs and relative paths
fn extract_path_from_url(url: &str) -> Result<String, String> {
    if let Ok(parsed_url) = url::Url::parse(url) {
        Ok(parsed_url.path().to_string())
    } else if url.starts_with('/') {
        Ok(url.to_string())
    } else {
        // Assume it's a relative path
        Ok(format!("/{}", url))
    }
}

/// Add authentication headers based on Insomnia auth configuration
fn add_auth_headers(
    auth: &InsomniaAuth,
    headers: &mut HashMap<String, String>,
    variables: &HashMap<String, String>,
) {
    match auth.auth_type.as_str() {
        "bearer" => {
            if let Some(token) = &auth.token {
                let resolved_token = resolve_variables(token, variables);
                headers.insert("Authorization".to_string(), format!("Bearer {}", resolved_token));
            }
        }
        "basic" => {
            if let (Some(username), Some(password)) = (&auth.username, &auth.password) {
                let user = resolve_variables(username, variables);
                let pass = resolve_variables(password, variables);
                use base64::{engine::general_purpose, Engine as _};
                let credentials = general_purpose::STANDARD.encode(format!("{}:{}", user, pass));
                headers.insert("Authorization".to_string(), format!("Basic {}", credentials));
            }
        }
        "apikey" => {
            if let (Some(key), Some(value)) = (&auth.key, &auth.value) {
                let resolved_key = resolve_variables(key, variables);
                let resolved_value = resolve_variables(value, variables);
                headers.insert(resolved_key, resolved_value);
            }
        }
        "oauth2" => {
            // OAuth2: use the stored access token if available, otherwise fall back to token field
            if let Some(access_token) = &auth.access_token {
                let resolved = resolve_variables(access_token, variables);
                if !resolved.is_empty() {
                    headers.insert("Authorization".to_string(), format!("Bearer {}", resolved));
                }
            } else if let Some(token) = &auth.token {
                let resolved = resolve_variables(token, variables);
                if !resolved.is_empty() {
                    headers.insert("Authorization".to_string(), format!("Bearer {}", resolved));
                }
            }
        }
        _ => {
            // Unsupported auth types are silently skipped
        }
    }
}

/// Extract request body from Insomnia resource
fn extract_request_body(
    resource: &InsomniaResource,
    variables: &HashMap<String, String>,
) -> Option<String> {
    if let Some(body) = &resource.body {
        if let Some(text) = &body.text {
            return Some(resolve_variables(text, variables));
        }
    }
    None
}

/// Resolve variables in a string (similar to Postman)
fn resolve_variables(input: &str, variables: &HashMap<String, String>) -> String {
    let mut result = input.to_string();
    for (key, value) in variables {
        let pattern = format!("{{{{{}}}}}", key);
        result = result.replace(&pattern, value);
    }
    result
}

/// Generate a mock response for the request
fn generate_mock_response(method: &str) -> MockForgeResponse {
    let mut headers = HashMap::new();
    headers.insert("Content-Type".to_string(), "application/json".to_string());

    let body = match method {
        "GET" => json!({"message": "Mock GET response", "method": "GET"}),
        "POST" => json!({"message": "Mock POST response", "method": "POST", "created": true}),
        "PUT" => json!({"message": "Mock PUT response", "method": "PUT", "updated": true}),
        "DELETE" => json!({"message": "Mock DELETE response", "method": "DELETE", "deleted": true}),
        "PATCH" => json!({"message": "Mock PATCH response", "method": "PATCH", "patched": true}),
        _ => json!({"message": "Mock response", "method": method}),
    };

    MockForgeResponse {
        status: 200,
        headers,
        body,
    }
}

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

    #[test]
    fn test_parse_insomnia_export() {
        let export_json = r#"{
            "__export_format": 4,
            "_type": "export",
            "resources": [
                {
                    "_id": "req_1",
                    "_type": "request",
                    "name": "Get Users",
                    "method": "GET",
                    "url": "{{baseUrl}}/users",
                    "headers": [
                        {"name": "Authorization", "value": "Bearer {{token}}"}
                    ],
                    "authentication": {
                        "type": "bearer",
                        "token": "{{token}}"
                    }
                },
                {
                    "_id": "env_1",
                    "_type": "environment",
                    "name": "Base Environment",
                    "data": {
                        "baseUrl": "https://api.example.com",
                        "token": "test-token"
                    }
                }
            ]
        }"#;

        let result = import_insomnia_export(export_json, Some("Base Environment")).unwrap();

        assert_eq!(result.routes.len(), 1);
        assert_eq!(result.routes[0].method, "GET");
        assert_eq!(result.routes[0].path, "/users");
        assert!(result.routes[0].headers.contains_key("Authorization"));
        assert!(result.variables.contains_key("baseUrl"));
    }

    #[test]
    fn test_insomnia_format_validation() {
        let old_format = r#"{
            "__export_format": 2,
            "resources": []
        }"#;

        let result = import_insomnia_export(old_format, None);
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("version 3 or higher"));
    }
}