tauri_typegen/interface/

config.rs

use serde::{Deserialize, Serialize};
use std::fs;
use std::path::Path;
use thiserror::Error;

#[derive(Error, Debug)]
pub enum ConfigError {
    #[error("IO error: {0}")]
    Io(#[from] std::io::Error),
    #[error("JSON parsing error: {0}")]
    Json(#[from] serde_json::Error),
    #[error("Invalid validation library: {0}. Use 'zod' or 'none'")]
    InvalidValidationLibrary(String),
    #[error("Invalid configuration: {0}")]
    InvalidConfig(String),
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct GenerateConfig {
    /// Path to the Tauri project source directory
    #[serde(default = "default_project_path")]
    pub project_path: String,

    /// Output path for generated TypeScript files
    #[serde(default = "default_output_path")]
    pub output_path: String,

    /// Validation library to use ('zod' or 'none')
    #[serde(default = "default_validation_library")]
    pub validation_library: String,

    /// Enable verbose output
    #[serde(default)]
    pub verbose: Option<bool>,

    /// Generate dependency graph visualization
    #[serde(default)]
    pub visualize_deps: Option<bool>,

    /// Include private struct fields in generation
    #[serde(default)]
    pub include_private: Option<bool>,

    /// Custom type mappings
    #[serde(default)]
    pub type_mappings: Option<std::collections::HashMap<String, String>>,

    /// File patterns to exclude from analysis
    #[serde(default)]
    pub exclude_patterns: Option<Vec<String>>,

    /// File patterns to include in analysis (overrides excludes)
    #[serde(default)]
    pub include_patterns: Option<Vec<String>>,

    /// Default naming convention for command parameters when no serde attribute is present
    /// Options: "camelCase", "snake_case", "PascalCase", "SCREAMING_SNAKE_CASE", "kebab-case", "SCREAMING-KEBAB-CASE"
    /// Default: "camelCase" (matches Tauri's default behavior - Tauri converts camelCase from JS to snake_case in Rust)
    #[serde(default = "default_parameter_case")]
    pub default_parameter_case: String,

    /// Default naming convention for struct fields when no serde attribute is present
    /// Options: same as default_parameter_case
    /// Default: "snake_case" (matches serde's default serialization behavior)
    /// Note: Use #[serde(rename_all = "camelCase")] on your structs if you want camelCase in TypeScript
    #[serde(default = "default_field_case")]
    pub default_field_case: String,

    /// Force regeneration, ignoring cache
    #[serde(default)]
    pub force: Option<bool>,
}

fn default_project_path() -> String {
    "./src-tauri".to_string()
}

fn default_output_path() -> String {
    "./src/generated".to_string()
}

fn default_validation_library() -> String {
    "none".to_string()
}

fn default_parameter_case() -> String {
    "camelCase".to_string()
}

fn default_field_case() -> String {
    // Default to snake_case to match serde's default serialization behavior
    // Users should add #[serde(rename_all = "camelCase")] if they want camelCase
    "snake_case".to_string()
}

impl Default for GenerateConfig {
    fn default() -> Self {
        Self {
            project_path: default_project_path(),
            output_path: default_output_path(),
            validation_library: default_validation_library(),
            verbose: Some(false),
            visualize_deps: Some(false),
            include_private: Some(false),
            type_mappings: None,
            exclude_patterns: None,
            include_patterns: None,
            default_parameter_case: default_parameter_case(),
            default_field_case: default_field_case(),
            force: Some(false),
        }
    }
}

impl GenerateConfig {
    /// Create a new configuration with defaults
    pub fn new() -> Self {
        Self::default()
    }

    /// Load configuration from a file
    pub fn from_file<P: AsRef<Path>>(path: P) -> Result<Self, ConfigError> {
        let is_tauri_config = path
            .as_ref()
            .file_name()
            .and_then(|n| n.to_str())
            .map(|n| n == "tauri.conf.json")
            .unwrap_or(false);

        if is_tauri_config {
            Self::from_tauri_config(path).and_then(|opt| {
                opt.ok_or_else(|| {
                    ConfigError::InvalidConfig(
                        "No tauri-typegen plugin configuration found in tauri.conf.json"
                            .to_string(),
                    )
                })
            })
        } else {
            let content = fs::read_to_string(path)?;
            let config: Self = serde_json::from_str(&content)?;
            Ok(config)
        }
    }

    /// Load configuration from Tauri configuration file
    pub fn from_tauri_config<P: AsRef<Path>>(path: P) -> Result<Option<Self>, ConfigError> {
        let content = fs::read_to_string(path)?;
        let tauri_config: serde_json::Value = serde_json::from_str(&content)?;

        // Look for typegen plugin configuration
        if let Some(plugins) = tauri_config.get("plugins") {
            if let Some(typegen) = plugins.get("typegen") {
                let mut config = Self::default();

                let get_string = |keys: &[&str]| {
                    for key in keys {
                        if let Some(val) = typegen.get(*key).and_then(|v| v.as_str()) {
                            return Some(val.to_string());
                        }
                    }
                    None
                };

                let get_bool = |keys: &[&str]| {
                    for key in keys {
                        if let Some(val) = typegen.get(*key).and_then(|v| v.as_bool()) {
                            return Some(val);
                        }
                    }
                    None
                };

                if let Some(p) = get_string(&["projectPath", "project_path"]) {
                    config.project_path = p;
                }
                if let Some(o) = get_string(&[
                    "outputPath",
                    "output_path",
                    "generatedPath",
                    "generated_path",
                ]) {
                    config.output_path = o;
                }
                if let Some(v) = get_string(&["validationLibrary", "validation_library"]) {
                    config.validation_library = v;
                }
                if let Some(v) = get_bool(&["verbose"]) {
                    config.verbose = Some(v);
                }
                if let Some(v) = get_bool(&["visualizeDeps", "visualize_deps"]) {
                    config.visualize_deps = Some(v);
                }
                if let Some(v) = get_bool(&["includePrivate", "include_private"]) {
                    config.include_private = Some(v);
                }
                if let Some(type_mappings) = typegen
                    .get("typeMappings")
                    .or_else(|| typegen.get("type_mappings"))
                {
                    if let Ok(mappings) = serde_json::from_value::<
                        std::collections::HashMap<String, String>,
                    >(type_mappings.clone())
                    {
                        config.type_mappings = Some(mappings);
                    }
                }
                if let Some(exclude_patterns) = typegen
                    .get("excludePatterns")
                    .or_else(|| typegen.get("exclude_patterns"))
                {
                    if let Ok(patterns) =
                        serde_json::from_value::<Vec<String>>(exclude_patterns.clone())
                    {
                        config.exclude_patterns = Some(patterns);
                    }
                }
                if let Some(include_patterns) = typegen
                    .get("includePatterns")
                    .or_else(|| typegen.get("include_patterns"))
                {
                    if let Ok(patterns) =
                        serde_json::from_value::<Vec<String>>(include_patterns.clone())
                    {
                        config.include_patterns = Some(patterns);
                    }
                }
                if let Some(v) = get_bool(&["force"]) {
                    config.force = Some(v);
                }
                if let Some(p) = get_string(&["defaultParameterCase", "default_parameter_case"]) {
                    config.default_parameter_case = p;
                }
                if let Some(f) = get_string(&["defaultFieldCase", "default_field_case"]) {
                    config.default_field_case = f;
                }

                return Ok(Some(config));
            }
        }

        Ok(None)
    }

    /// Save configuration to a file
    pub fn save_to_file<P: AsRef<Path>>(&self, path: P) -> Result<(), ConfigError> {
        let content = serde_json::to_string_pretty(self)?;
        fs::write(path, content)?;
        Ok(())
    }

    /// Save configuration to Tauri configuration file
    pub fn save_to_tauri_config<P: AsRef<Path>>(&self, path: P) -> Result<(), ConfigError> {
        // Read existing tauri.conf.json - we require it to exist
        if !path.as_ref().exists() {
            return Err(ConfigError::InvalidConfig(format!(
                "tauri.conf.json not found at {}. Please ensure you have a Tauri project initialized.",
                path.as_ref().display()
            )));
        }

        let content = fs::read_to_string(&path)?;
        let mut tauri_config = serde_json::from_str::<serde_json::Value>(&content)?;

        // Create typegen plugin configuration
        let typegen_config = serde_json::json!({
            "projectPath": self.project_path,
            "outputPath": self.output_path,
            "validationLibrary": self.validation_library,
            "verbose": self.verbose.unwrap_or(false),
            "visualizeDeps": self.visualize_deps.unwrap_or(false),
            "includePrivate": self.include_private.unwrap_or(false),
            "typeMappings": self.type_mappings,
            "excludePatterns": self.exclude_patterns,
            "includePatterns": self.include_patterns,
            "force": self.force.unwrap_or(false),
        });

        // Ensure plugins section exists and insert typegen configuration
        if !tauri_config.is_object() {
            tauri_config = serde_json::json!({});
        }

        let tauri_obj = tauri_config.as_object_mut().unwrap();

        // Create plugins section if it doesn't exist
        if !tauri_obj.contains_key("plugins") {
            tauri_obj.insert("plugins".to_string(), serde_json::json!({}));
        }

        // Insert typegen configuration into plugins
        if let Some(plugins) = tauri_obj.get_mut("plugins") {
            if let Some(plugins_obj) = plugins.as_object_mut() {
                plugins_obj.insert("typegen".to_string(), typegen_config);
            }
        }

        let content = serde_json::to_string_pretty(&tauri_config)?;
        fs::write(path, content)?;
        Ok(())
    }

    /// Validate the configuration
    pub fn validate(&self) -> Result<(), ConfigError> {
        // Validate validation library
        match self.validation_library.as_str() {
            "zod" | "none" => {}
            _ => {
                return Err(ConfigError::InvalidValidationLibrary(
                    self.validation_library.clone(),
                ));
            }
        }

        // Validate paths exist
        let project_path = Path::new(&self.project_path);
        if !project_path.exists() {
            return Err(ConfigError::InvalidConfig(format!(
                "Project path does not exist: {}",
                self.project_path
            )));
        }

        Ok(())
    }

    /// Merge with another configuration, with other taking precedence
    pub fn merge(&mut self, other: &GenerateConfig) {
        if other.project_path != default_project_path() {
            self.project_path = other.project_path.clone();
        }
        if other.output_path != default_output_path() {
            self.output_path = other.output_path.clone();
        }
        if other.validation_library != default_validation_library() {
            self.validation_library = other.validation_library.clone();
        }
        if other.verbose.is_some() {
            self.verbose = other.verbose;
        }
        if other.visualize_deps.is_some() {
            self.visualize_deps = other.visualize_deps;
        }
        if other.include_private.is_some() {
            self.include_private = other.include_private;
        }
        if other.type_mappings.is_some() {
            self.type_mappings = other.type_mappings.clone();
        }
        if other.exclude_patterns.is_some() {
            self.exclude_patterns = other.exclude_patterns.clone();
        }
        if other.include_patterns.is_some() {
            self.include_patterns = other.include_patterns.clone();
        }
        if other.force.is_some() {
            self.force = other.force;
        }
    }

    /// Get effective verbose setting
    pub fn is_verbose(&self) -> bool {
        self.verbose.unwrap_or(false)
    }

    /// Get effective visualize_deps setting
    pub fn should_visualize_deps(&self) -> bool {
        self.visualize_deps.unwrap_or(false)
    }

    /// Get effective include_private setting
    pub fn should_include_private(&self) -> bool {
        self.include_private.unwrap_or(false)
    }

    /// Get effective force setting
    pub fn should_force(&self) -> bool {
        self.force.unwrap_or(false)
    }
}

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

    #[test]
    fn test_default_config() {
        let config = GenerateConfig::default();
        assert_eq!(config.project_path, "./src-tauri");
        assert_eq!(config.output_path, "./src/generated");
        assert_eq!(config.validation_library, "none");
        assert!(!config.is_verbose());
        assert!(!config.should_visualize_deps());
        assert!(!config.should_include_private());
        assert!(!config.should_force());
    }

    #[test]
    fn test_config_validation() {
        let config = GenerateConfig {
            validation_library: "invalid".to_string(),
            ..Default::default()
        };

        let result = config.validate();
        assert!(result.is_err());
        if let Err(ConfigError::InvalidValidationLibrary(lib)) = result {
            assert_eq!(lib, "invalid");
        } else {
            panic!("Expected InvalidValidationLibrary error");
        }
    }

    #[test]
    fn test_config_merge() {
        let mut base = GenerateConfig::default();
        let override_config = GenerateConfig {
            output_path: "./custom".to_string(),
            verbose: Some(true),
            ..Default::default()
        };

        base.merge(&override_config);
        assert_eq!(base.output_path, "./custom");
        assert!(base.is_verbose());
        assert_eq!(base.validation_library, "none"); // Should remain default
    }

    #[test]
    fn test_save_and_load_config() {
        let temp_dir = tempfile::TempDir::new().unwrap();
        let project_path = temp_dir.path().join("src-tauri");
        std::fs::create_dir_all(&project_path).unwrap();

        let config = GenerateConfig {
            project_path: project_path.to_string_lossy().to_string(),
            output_path: "./test".to_string(),
            verbose: Some(true),
            ..Default::default()
        };

        let temp_file = NamedTempFile::new().unwrap();
        config.save_to_file(temp_file.path()).unwrap();

        let loaded_config = GenerateConfig::from_file(temp_file.path()).unwrap();
        assert_eq!(loaded_config.output_path, "./test");
        assert!(loaded_config.is_verbose());
    }

    #[test]
    fn test_save_to_tauri_config_preserves_existing_content() {
        let temp_dir = tempfile::TempDir::new().unwrap();
        let project_path = temp_dir.path().join("src-tauri");
        std::fs::create_dir_all(&project_path).unwrap();

        let tauri_conf_path = temp_dir.path().join("tauri.conf.json");

        // Create existing tauri.conf.json with some content
        let existing_content = serde_json::json!({
            "package": {
                "productName": "My App",
                "version": "1.0.0"
            },
            "tauri": {
                "allowlist": {
                    "all": false
                }
            },
            "plugins": {
                "shell": {
                    "all": false
                }
            }
        });

        fs::write(
            &tauri_conf_path,
            serde_json::to_string_pretty(&existing_content).unwrap(),
        )
        .unwrap();

        let config = GenerateConfig {
            project_path: project_path.to_string_lossy().to_string(),
            output_path: "./test".to_string(),
            validation_library: "zod".to_string(),
            verbose: Some(true),
            ..Default::default()
        };

        // Save to tauri config - should preserve existing content
        config.save_to_tauri_config(&tauri_conf_path).unwrap();

        // Read back and verify
        let updated_content = fs::read_to_string(&tauri_conf_path).unwrap();
        let updated_json: serde_json::Value = serde_json::from_str(&updated_content).unwrap();

        // Check that existing content is preserved
        assert_eq!(updated_json["package"]["productName"], "My App");
        assert_eq!(updated_json["package"]["version"], "1.0.0");
        assert_eq!(updated_json["tauri"]["allowlist"]["all"], false);
        assert_eq!(updated_json["plugins"]["shell"]["all"], false);

        // Check that typegen config was added
        assert_eq!(updated_json["plugins"]["typegen"]["outputPath"], "./test");
        assert_eq!(
            updated_json["plugins"]["typegen"]["validationLibrary"],
            "zod"
        );
        assert_eq!(updated_json["plugins"]["typegen"]["verbose"], true);
    }

    #[test]
    fn test_save_to_tauri_config_creates_plugins_section() {
        let temp_dir = tempfile::TempDir::new().unwrap();
        let project_path = temp_dir.path().join("src-tauri");
        std::fs::create_dir_all(&project_path).unwrap();

        let tauri_conf_path = temp_dir.path().join("tauri.conf.json");

        // Create existing tauri.conf.json without plugins section
        let existing_content = serde_json::json!({
            "package": {
                "productName": "My App",
                "version": "1.0.0"
            },
            "tauri": {
                "allowlist": {
                    "all": false
                }
            }
        });

        fs::write(
            &tauri_conf_path,
            serde_json::to_string_pretty(&existing_content).unwrap(),
        )
        .unwrap();

        let config = GenerateConfig {
            project_path: project_path.to_string_lossy().to_string(),
            output_path: "./test".to_string(),
            validation_library: "none".to_string(),
            ..Default::default()
        };

        // Save to tauri config - should create plugins section
        config.save_to_tauri_config(&tauri_conf_path).unwrap();

        // Read back and verify
        let updated_content = fs::read_to_string(&tauri_conf_path).unwrap();
        let updated_json: serde_json::Value = serde_json::from_str(&updated_content).unwrap();

        // Check that existing content is preserved
        assert_eq!(updated_json["package"]["productName"], "My App");
        assert_eq!(updated_json["tauri"]["allowlist"]["all"], false);

        // Check that plugins section was created with typegen config
        assert!(updated_json["plugins"].is_object());
        assert_eq!(updated_json["plugins"]["typegen"]["outputPath"], "./test");
        assert_eq!(
            updated_json["plugins"]["typegen"]["validationLibrary"],
            "none"
        );
    }
}