realitydefender 0.1.8

Reality Defender SDK for Rust - Tools for detecting deepfakes and manipulated media
Documentation
use crate::error::{Error, Result};

/// Default API base URL
pub const DEFAULT_BASE_URL: &str = "https://api.prd.realitydefender.xyz";

/// Configuration for the Reality Defender client
#[derive(Debug, Clone, Default)]
pub struct Config {
    /// API key for authentication
    pub api_key: String,

    /// Base URL for the API
    pub base_url: Option<String>,

    /// Timeout in seconds for HTTP requests
    pub timeout_seconds: Option<u64>,
}

impl Config {
    /// Validate the configuration
    pub fn validate(&self) -> Result<()> {
        if self.api_key.trim().is_empty() {
            return Err(Error::InvalidConfig("API key is required".to_string()));
        }

        if let Some(url) = &self.base_url {
            if url.trim().is_empty() {
                return Err(Error::InvalidConfig("Base URL cannot be empty".to_string()));
            }
        }

        Ok(())
    }

    /// Get the base URL, falling back to the default if not set
    pub fn get_base_url(&self) -> String {
        self.base_url
            .clone()
            .unwrap_or_else(|| DEFAULT_BASE_URL.to_string())
    }

    /// Get the timeout in seconds, falling back to the default if not set
    pub fn get_timeout_seconds(&self) -> u64 {
        self.timeout_seconds.unwrap_or(30)
    }
}

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

    #[test]
    fn test_config_default() {
        let config = Config::default();
        assert_eq!(config.api_key, "".to_string());
        assert_eq!(config.base_url, None);
        assert_eq!(config.timeout_seconds, None);
    }

    #[test]
    fn test_config_with_custom_url() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: Some("https://custom-api.example.com".to_string()),
            timeout_seconds: None,
        };

        assert_eq!(config.api_key, "test_api_key");
        assert_eq!(
            config.base_url,
            Some("https://custom-api.example.com".to_string())
        );
        assert_eq!(config.timeout_seconds, None);
    }

    #[test]
    fn test_config_with_timeout() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: None,
            timeout_seconds: Some(120),
        };

        assert_eq!(config.api_key, "test_api_key");
        assert_eq!(config.base_url, None);
        assert_eq!(config.timeout_seconds, Some(120));
    }

    #[test]
    fn test_config_with_all_options() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: Some("https://custom-api.example.com".to_string()),
            timeout_seconds: Some(120),
        };

        assert_eq!(config.api_key, "test_api_key");
        assert_eq!(
            config.base_url,
            Some("https://custom-api.example.com".to_string())
        );
        assert_eq!(config.timeout_seconds, Some(120));
    }

    #[test]
    fn test_validate_valid_config() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: None,
            timeout_seconds: None,
        };

        let result = config.validate();
        assert!(result.is_ok());
    }

    #[test]
    fn test_validate_empty_api_key() {
        let config = Config::default(); // Empty API key
        let result = config.validate();
        assert!(result.is_err());

        match result {
            Err(Error::InvalidConfig(msg)) => {
                assert!(msg.contains("API key is required"));
            }
            _ => panic!("Expected InvalidConfig error"),
        }
    }

    #[test]
    fn test_validate_empty_base_url() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: Some("".to_string()),
            timeout_seconds: None,
        };

        let result = config.validate();
        assert!(result.is_err());

        match result {
            Err(Error::InvalidConfig(msg)) => {
                assert!(msg.contains("Base URL cannot be empty"));
            }
            _ => panic!("Expected InvalidConfig error"),
        }
    }

    #[test]
    fn test_get_base_url_default() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: None,
            timeout_seconds: None,
        };

        assert_eq!(config.get_base_url(), "https://api.prd.realitydefender.xyz");
    }

    #[test]
    fn test_get_base_url_custom() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: Some("https://custom-api.example.com".to_string()),
            timeout_seconds: None,
        };

        assert_eq!(config.get_base_url(), "https://custom-api.example.com");
    }

    #[test]
    fn test_get_timeout_seconds_default() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: None,
            timeout_seconds: None,
        };

        assert_eq!(config.get_timeout_seconds(), 30); // Default is 30 seconds
    }

    #[test]
    fn test_get_timeout_seconds_custom() {
        let config = Config {
            api_key: "test_api_key".to_string(),
            base_url: None,
            timeout_seconds: Some(120),
        };

        assert_eq!(config.get_timeout_seconds(), 120);
    }
}