drasi-bootstrap-http 0.1.4

// Copyright 2025 The Drasi Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! Configuration types for the HTTP bootstrap provider.

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

/// Top-level configuration for the HTTP bootstrap provider.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpBootstrapConfig {
    /// List of endpoint configurations to fetch data from.
    pub endpoints: Vec<EndpointConfig>,

    /// HTTP request timeout in seconds (default: 30).
    #[serde(default = "default_timeout_seconds")]
    pub timeout_seconds: u64,

    /// Maximum number of retries on failure (default: 3).
    #[serde(default = "default_max_retries")]
    pub max_retries: u32,

    /// Delay between retries in milliseconds (default: 1000).
    #[serde(default = "default_retry_delay_ms")]
    pub retry_delay_ms: u64,

    /// Maximum number of pages to fetch per endpoint (default: 10,000).
    /// Set a higher value for very large initial loads.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub max_pages: Option<u64>,
}

fn default_timeout_seconds() -> u64 {
    30
}

fn default_max_retries() -> u32 {
    3
}

fn default_retry_delay_ms() -> u64 {
    1000
}

/// Configuration for a single HTTP endpoint to bootstrap from.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct EndpointConfig {
    /// The URL to fetch data from.
    pub url: String,

    /// HTTP method (default: GET).
    #[serde(default = "default_method")]
    pub method: HttpMethod,

    /// Additional HTTP headers to include in requests.
    #[serde(default, skip_serializing_if = "HashMap::is_empty")]
    pub headers: HashMap<String, String>,

    /// Optional request body (for POST/PUT methods).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub body: Option<serde_json::Value>,

    /// Authentication configuration.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub auth: Option<AuthConfig>,

    /// Pagination configuration.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub pagination: Option<PaginationConfig>,

    /// Response parsing and element mapping configuration.
    pub response: ResponseConfig,
}

fn default_method() -> HttpMethod {
    HttpMethod::Get
}

/// HTTP methods supported for bootstrap requests.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "UPPERCASE")]
pub enum HttpMethod {
    Get,
    Post,
    Put,
}

/// Authentication configuration.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", rename_all = "kebab-case")]
pub enum AuthConfig {
    /// Bearer token authentication.
    Bearer {
        /// Environment variable containing the token.
        token_env: String,
    },
    /// API key authentication (in header or query parameter).
    ApiKey {
        /// Where to send the API key.
        location: ApiKeyLocation,
        /// Header name or query parameter name.
        name: String,
        /// Environment variable containing the API key value.
        value_env: String,
    },
    /// HTTP Basic authentication.
    Basic {
        /// Environment variable containing the username.
        username_env: String,
        /// Environment variable containing the password (optional, can be empty).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        password_env: Option<String>,
    },
    /// OAuth2 Client Credentials flow.
    #[serde(rename = "oauth2-client-credentials")]
    OAuth2ClientCredentials {
        /// Token endpoint URL.
        token_url: String,
        /// Environment variable containing the client ID.
        client_id_env: String,
        /// Environment variable containing the client secret.
        client_secret_env: String,
        /// Optional scopes to request.
        #[serde(default, skip_serializing_if = "Vec::is_empty")]
        scopes: Vec<String>,
    },
}

/// Where to place an API key.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "kebab-case")]
pub enum ApiKeyLocation {
    Header,
    Query,
}

/// Pagination configuration.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", rename_all = "kebab-case")]
pub enum PaginationConfig {
    /// Offset/limit pagination.
    OffsetLimit {
        /// Query parameter name for offset (default: "offset").
        #[serde(default = "default_offset_param")]
        offset_param: String,
        /// Query parameter name for limit/page size (default: "limit").
        #[serde(default = "default_limit_param")]
        limit_param: String,
        /// Number of items per page.
        page_size: u64,
        /// JSONPath to extract total count from response (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        total_path: Option<String>,
    },
    /// Page number pagination.
    PageNumber {
        /// Query parameter name for page number (default: "page").
        #[serde(default = "default_page_param")]
        page_param: String,
        /// Query parameter name for page size (default: "per_page").
        #[serde(default = "default_per_page_param")]
        page_size_param: String,
        /// Number of items per page.
        page_size: u64,
        /// JSONPath to extract total pages from response (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        total_pages_path: Option<String>,
    },
    /// Cursor-based pagination (e.g., Stripe's `starting_after`).
    Cursor {
        /// Query parameter name to send the cursor value.
        cursor_param: String,
        /// JSONPath to extract the next cursor value from the response.
        cursor_path: String,
        /// JSONPath to a boolean `has_more` field (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        has_more_path: Option<String>,
        /// Query parameter name for page size (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        page_size_param: Option<String>,
        /// Number of items per page (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        page_size: Option<u64>,
    },
    /// Link header pagination (RFC 5988).
    LinkHeader {
        /// Query parameter name for page size (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        page_size_param: Option<String>,
        /// Number of items per page (optional).
        #[serde(default, skip_serializing_if = "Option::is_none")]
        page_size: Option<u64>,
    },
    /// Next URL from response body (e.g., Salesforce `nextRecordsUrl`).
    NextUrl {
        /// JSONPath to extract the next URL from the response body.
        next_url_path: String,
        /// Base URL to prepend if the extracted URL is relative.
        #[serde(default, skip_serializing_if = "Option::is_none")]
        base_url: Option<String>,
    },
}

impl HttpBootstrapConfig {
    /// Validate the configuration and return an error if invalid.
    pub fn validate(&self) -> anyhow::Result<()> {
        if self.endpoints.is_empty() {
            return Err(anyhow::anyhow!(
                "Validation error: at least one endpoint must be configured"
            ));
        }
        if self.timeout_seconds == 0 {
            return Err(anyhow::anyhow!(
                "Validation error: timeoutSeconds must be greater than 0"
            ));
        }
        for (i, endpoint) in self.endpoints.iter().enumerate() {
            endpoint.validate(i)?;
        }
        Ok(())
    }
}

impl EndpointConfig {
    fn validate(&self, index: usize) -> anyhow::Result<()> {
        if self.url.is_empty() {
            return Err(anyhow::anyhow!(
                "Validation error: endpoint[{index}].url cannot be empty"
            ));
        }
        if !self.url.starts_with("http://") && !self.url.starts_with("https://") {
            return Err(anyhow::anyhow!(
                "Validation error: endpoint[{index}].url must start with http:// or https://"
            ));
        }
        if self.response.mappings.is_empty() {
            return Err(anyhow::anyhow!(
                "Validation error: endpoint[{index}].response.mappings must have at least one mapping"
            ));
        }
        if let Some(ref pagination) = self.pagination {
            pagination.validate(index)?;
        }
        for (j, mapping) in self.response.mappings.iter().enumerate() {
            mapping.validate(index, j)?;
        }
        Ok(())
    }
}

impl PaginationConfig {
    fn validate(&self, endpoint_index: usize) -> anyhow::Result<()> {
        match self {
            PaginationConfig::OffsetLimit { page_size, .. } => {
                if *page_size == 0 {
                    return Err(anyhow::anyhow!(
                        "Validation error: endpoint[{endpoint_index}].pagination.page_size must be greater than 0"
                    ));
                }
            }
            PaginationConfig::PageNumber { page_size, .. } => {
                if *page_size == 0 {
                    return Err(anyhow::anyhow!(
                        "Validation error: endpoint[{endpoint_index}].pagination.page_size must be greater than 0"
                    ));
                }
            }
            PaginationConfig::Cursor {
                cursor_param,
                cursor_path,
                ..
            } => {
                if cursor_param.is_empty() {
                    return Err(anyhow::anyhow!(
                        "Validation error: endpoint[{endpoint_index}].pagination.cursor_param cannot be empty"
                    ));
                }
                if cursor_path.is_empty() {
                    return Err(anyhow::anyhow!(
                        "Validation error: endpoint[{endpoint_index}].pagination.cursor_path cannot be empty"
                    ));
                }
            }
            PaginationConfig::NextUrl { next_url_path, .. } => {
                if next_url_path.is_empty() {
                    return Err(anyhow::anyhow!(
                        "Validation error: endpoint[{endpoint_index}].pagination.next_url_path cannot be empty"
                    ));
                }
            }
            PaginationConfig::LinkHeader { .. } => {}
        }
        Ok(())
    }
}

impl ElementMappingConfig {
    fn validate(&self, endpoint_index: usize, mapping_index: usize) -> anyhow::Result<()> {
        if self.template.id.is_empty() {
            return Err(anyhow::anyhow!(
                "Validation error: endpoint[{endpoint_index}].mappings[{mapping_index}].template.id cannot be empty"
            ));
        }
        if self.template.labels.is_empty() {
            return Err(anyhow::anyhow!(
                "Validation error: endpoint[{endpoint_index}].mappings[{mapping_index}].template.labels must have at least one label"
            ));
        }
        if self.element_type == ElementType::Relation {
            if self.template.from.is_none() {
                return Err(anyhow::anyhow!(
                    "Validation error: endpoint[{endpoint_index}].mappings[{mapping_index}].template.from is required for relation mappings"
                ));
            }
            if self.template.to.is_none() {
                return Err(anyhow::anyhow!(
                    "Validation error: endpoint[{endpoint_index}].mappings[{mapping_index}].template.to is required for relation mappings"
                ));
            }
        }
        Ok(())
    }
}

fn default_offset_param() -> String {
    "offset".to_string()
}

fn default_limit_param() -> String {
    "limit".to_string()
}

fn default_page_param() -> String {
    "page".to_string()
}

fn default_per_page_param() -> String {
    "per_page".to_string()
}

/// Response parsing configuration.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ResponseConfig {
    /// JSONPath expression to locate the array of items in the response.
    /// Use "$" if the response is a top-level array.
    #[serde(default = "default_items_path")]
    pub items_path: String,

    /// Content type override (auto-detected from Content-Type header if not set).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub content_type: Option<ContentTypeOverride>,

    /// Element mapping configurations.
    pub mappings: Vec<ElementMappingConfig>,
}

fn default_items_path() -> String {
    "$".to_string()
}

/// Override for response content type.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "lowercase")]
pub enum ContentTypeOverride {
    Json,
    Xml,
    Yaml,
}

/// Mapping configuration from response items to Drasi graph elements.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ElementMappingConfig {
    /// Type of element to create.
    pub element_type: ElementType,

    /// Template for element creation.
    pub template: ElementTemplate,
}

/// Element type.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "lowercase")]
pub enum ElementType {
    Node,
    Relation,
}

/// Template for element creation using Handlebars expressions.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ElementTemplate {
    /// Handlebars template for element ID.
    pub id: String,

    /// Handlebars templates for element labels.
    pub labels: Vec<String>,

    /// Properties mapping (each value is a Handlebars template or literal).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub properties: Option<serde_json::Value>,

    /// Template for relation source node ID (relations only).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub from: Option<String>,

    /// Template for relation target node ID (relations only).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub to: Option<String>,
}

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

    #[test]
    fn test_deserialize_full_config() {
        let json = r#"{
            "endpoints": [{
                "url": "https://api.example.com/users",
                "method": "GET",
                "auth": {
                    "type": "bearer",
                    "token_env": "API_TOKEN"
                },
                "pagination": {
                    "type": "offset-limit",
                    "offset_param": "offset",
                    "limit_param": "limit",
                    "page_size": 100
                },
                "response": {
                    "itemsPath": "$.data",
                    "mappings": [{
                        "elementType": "node",
                        "template": {
                            "id": "{{item.id}}",
                            "labels": ["User"],
                            "properties": {
                                "name": "{{item.name}}"
                            }
                        }
                    }]
                }
            }],
            "timeoutSeconds": 30,
            "maxRetries": 3,
            "retryDelayMs": 1000
        }"#;

        let config: HttpBootstrapConfig = serde_json::from_str(json).unwrap();
        assert_eq!(config.endpoints.len(), 1);
        assert_eq!(config.timeout_seconds, 30);
        assert_eq!(config.endpoints[0].url, "https://api.example.com/users");
    }

    #[test]
    fn test_deserialize_cursor_pagination() {
        let json = r#"{
            "type": "cursor",
            "cursor_param": "starting_after",
            "cursor_path": "$.data[-1].id",
            "has_more_path": "$.has_more",
            "page_size_param": "limit",
            "page_size": 100
        }"#;

        let config: PaginationConfig = serde_json::from_str(json).unwrap();
        match config {
            PaginationConfig::Cursor {
                cursor_param,
                cursor_path,
                has_more_path,
                ..
            } => {
                assert_eq!(cursor_param, "starting_after");
                assert_eq!(cursor_path, "$.data[-1].id");
                assert_eq!(has_more_path, Some("$.has_more".to_string()));
            }
            _ => panic!("Expected Cursor pagination"),
        }
    }

    #[test]
    fn test_deserialize_oauth2_auth() {
        let json = r#"{
            "type": "oauth2-client-credentials",
            "token_url": "https://auth.example.com/token",
            "client_id_env": "CLIENT_ID",
            "client_secret_env": "CLIENT_SECRET",
            "scopes": ["read", "write"]
        }"#;

        let config: AuthConfig = serde_json::from_str(json).unwrap();
        match config {
            AuthConfig::OAuth2ClientCredentials {
                token_url, scopes, ..
            } => {
                assert_eq!(token_url, "https://auth.example.com/token");
                assert_eq!(scopes, vec!["read", "write"]);
            }
            _ => panic!("Expected OAuth2ClientCredentials"),
        }
    }

    #[test]
    fn test_deserialize_next_url_pagination() {
        let json = r#"{
            "type": "next-url",
            "next_url_path": "$.nextRecordsUrl",
            "base_url": "https://instance.salesforce.com"
        }"#;

        let config: PaginationConfig = serde_json::from_str(json).unwrap();
        match config {
            PaginationConfig::NextUrl {
                next_url_path,
                base_url,
            } => {
                assert_eq!(next_url_path, "$.nextRecordsUrl");
                assert_eq!(
                    base_url,
                    Some("https://instance.salesforce.com".to_string())
                );
            }
            _ => panic!("Expected NextUrl pagination"),
        }
    }

    fn make_valid_config() -> HttpBootstrapConfig {
        HttpBootstrapConfig {
            endpoints: vec![EndpointConfig {
                url: "https://api.example.com/users".to_string(),
                method: HttpMethod::Get,
                headers: HashMap::new(),
                body: None,
                auth: None,
                pagination: None,
                response: ResponseConfig {
                    items_path: "$".to_string(),
                    content_type: None,
                    mappings: vec![ElementMappingConfig {
                        element_type: ElementType::Node,
                        template: ElementTemplate {
                            id: "{{item.id}}".to_string(),
                            labels: vec!["User".to_string()],
                            properties: None,
                            from: None,
                            to: None,
                        },
                    }],
                },
            }],
            timeout_seconds: 30,
            max_retries: 3,
            retry_delay_ms: 1000,
            max_pages: None,
        }
    }

    #[test]
    fn test_validate_valid_config() {
        let config = make_valid_config();
        assert!(config.validate().is_ok());
    }

    #[test]
    fn test_validate_no_endpoints() {
        let mut config = make_valid_config();
        config.endpoints.clear();
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("at least one endpoint"));
    }

    #[test]
    fn test_validate_empty_url() {
        let mut config = make_valid_config();
        config.endpoints[0].url = String::new();
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("url cannot be empty"));
    }

    #[test]
    fn test_validate_no_mappings() {
        let mut config = make_valid_config();
        config.endpoints[0].response.mappings.clear();
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("at least one mapping"));
    }

    #[test]
    fn test_validate_zero_page_size() {
        let mut config = make_valid_config();
        config.endpoints[0].pagination = Some(PaginationConfig::OffsetLimit {
            offset_param: "offset".to_string(),
            limit_param: "limit".to_string(),
            page_size: 0,
            total_path: None,
        });
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("page_size must be greater than 0"));
    }

    #[test]
    fn test_validate_zero_timeout() {
        let mut config = make_valid_config();
        config.timeout_seconds = 0;
        let err = config.validate().unwrap_err();
        assert!(err
            .to_string()
            .contains("timeoutSeconds must be greater than 0"));
    }

    #[test]
    fn test_validate_relation_missing_from() {
        let mut config = make_valid_config();
        config.endpoints[0].response.mappings[0].element_type = ElementType::Relation;
        // from is None
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("from is required"));
    }

    #[test]
    fn test_validate_empty_labels() {
        let mut config = make_valid_config();
        config.endpoints[0].response.mappings[0]
            .template
            .labels
            .clear();
        let err = config.validate().unwrap_err();
        assert!(err.to_string().contains("at least one label"));
    }
}