tauri-typegen 0.5.1

use clap::Parser;
use std::fs;
use std::path::PathBuf;
use tauri_typegen::analysis::CommandAnalyzer;
use tauri_typegen::build::GenerationCache;
use tauri_typegen::generators::create_generator;
use tauri_typegen::interface::{
    print_dependency_visualization_info, print_usage_info, CargoCli, CargoSubcommands,
    GenerateConfig, Logger, ProgressReporter, TypegenCommands,
};

fn main() {
    let args = CargoCli::parse();

    match args.command {
        CargoSubcommands::TauriTypegen(typegen_args) => {
            // Handle --version flag
            if typegen_args.version {
                println!("tauri-typegen {}", env!("CARGO_PKG_VERSION"));
                return;
            }

            // If no subcommand provided, show error
            let Some(command) = typegen_args.command else {
                eprintln!("Error: No subcommand provided. Use 'generate' or 'init'.");
                eprintln!("Run 'cargo tauri-typegen --help' for more information.");
                std::process::exit(1);
            };

            match command {
                TypegenCommands::Generate {
                    project_path,
                    output_path,
                    validation_library,
                    verbose,
                    visualize_deps,
                    config_file,
                    force,
                } => {
                    if let Err(e) = run_generate(
                        project_path,
                        output_path,
                        validation_library,
                        verbose,
                        visualize_deps,
                        config_file,
                        force,
                    ) {
                        eprintln!("Error: {}", e);
                        std::process::exit(1);
                    }
                }
                TypegenCommands::Init {
                    project_path,
                    generated_path,
                    output_path,
                    validation_library,
                    verbose,
                    visualize_deps,
                    force,
                } => {
                    if let Err(e) = run_init(
                        project_path,
                        generated_path,
                        output_path,
                        validation_library,
                        verbose,
                        visualize_deps,
                        force,
                    ) {
                        eprintln!("Error: {}", e);
                        std::process::exit(1);
                    }
                }
            }
        }
    }
}

fn run_generate(
    project_path: Option<PathBuf>,
    output_path: Option<PathBuf>,
    validation_library: Option<String>,
    verbose: bool,
    visualize_deps: bool,
    config_file: Option<PathBuf>,
    force: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    let logger = Logger::new(verbose, false);
    let mut reporter = ProgressReporter::new(logger, 4);

    // Load configuration
    reporter.start_step("Loading configuration");
    let mut config = if let Some(config_path) = config_file {
        // Explicit config file specified
        if config_path.exists() {
            GenerateConfig::from_file(config_path)?
        } else {
            return Err(format!("Configuration file not found: {}", config_path.display()).into());
        }
    } else {
        // Try to find tauri.conf.json in common locations
        let possible_paths = vec![
            PathBuf::from("tauri.conf.json"),           // Current directory
            PathBuf::from("src-tauri/tauri.conf.json"), // Common Tauri structure
            PathBuf::from("../tauri.conf.json"),        // If running from src-tauri
        ];

        let mut config_loaded = false;
        let mut config = GenerateConfig::default();

        for path in possible_paths {
            if path.exists() {
                match GenerateConfig::from_tauri_config(&path) {
                    Ok(Some(loaded_config)) => {
                        config = loaded_config;
                        config_loaded = true;
                        break;
                    }
                    Ok(None) => break,
                    Err(_) => continue,
                }
            }
        }

        if !config_loaded {
            // No config file found, use defaults
            config = GenerateConfig::default();
        }

        config
    };

    // CLI arguments override config file settings only when explicitly provided
    if let Some(path) = project_path {
        config.project_path = path.to_string_lossy().to_string();
    }
    if let Some(path) = output_path {
        config.output_path = path.to_string_lossy().to_string();
    }
    if let Some(validation) = validation_library {
        config.validation_library = validation;
    }
    // For boolean flags: only override if flag was present (true)
    if verbose {
        config.verbose = Some(true);
    }
    if visualize_deps {
        config.visualize_deps = Some(true);
    }
    // CLI --force flag overrides config
    if force {
        config.force = Some(true);
    }

    reporter.complete_step(Some(&format!(
        "Using {} validation",
        config.validation_library
    )));

    // Validate paths and configuration
    reporter.start_step("Validating project structure");
    config.validate()?;
    reporter.complete_step(None);

    // Analyze and generate
    reporter.start_step("Analyzing Tauri commands");
    let mut analyzer = CommandAnalyzer::new();

    // Apply custom type mappings from configuration
    if let Some(ref mappings) = config.type_mappings {
        analyzer.add_type_mappings(mappings);
        if config.is_verbose() {
            reporter.update_progress(&format!("Applied {} custom type mappings", mappings.len()));
            for (rust_type, ts_type) in mappings {
                reporter.update_progress(&format!("  {} → {}", rust_type, ts_type));
            }
        }
    }

    let commands =
        analyzer.analyze_project_with_verbose(&config.project_path, config.is_verbose())?;

    if config.is_verbose() {
        reporter.update_progress(&format!("Found {} Tauri commands", commands.len()));
        commands.iter().for_each(|cmd| {
            reporter.update_progress(&format!("  - {} ({})", cmd.name, cmd.file_path));
        });

        let discovered_structs = analyzer.get_discovered_structs();
        reporter.update_progress(&format!(
            "Found {} struct definitions",
            discovered_structs.len()
        ));
        discovered_structs.iter().for_each(|(name, struct_info)| {
            let struct_type = if struct_info.is_enum {
                "enum"
            } else {
                "struct"
            };
            reporter.update_progress(&format!(
                "  - {} ({}) with {} fields",
                name,
                struct_type,
                struct_info.fields.len()
            ));
        });
    }
    reporter.complete_step(Some(&format!("Found {} commands", commands.len())));

    if commands.is_empty() {
        println!("⚠️  No Tauri commands found. Make sure your project contains functions with #[tauri::command] attributes.");
        return Ok(());
    }

    // Check cache to see if regeneration is needed (unless force is set)
    let discovered_structs = analyzer.get_discovered_structs();
    let discovered_events = analyzer.get_discovered_events();
    let needs_regeneration = if config.should_force() {
        if config.is_verbose() {
            println!("🔄 Force flag set, regenerating bindings");
        }
        true
    } else {
        GenerationCache::needs_regeneration(
            &config.output_path,
            &commands,
            discovered_structs,
            discovered_events,
            &config,
        )
        .unwrap_or(true) // On error, assume regeneration is needed
    };

    if !needs_regeneration {
        if config.is_verbose() {
            println!("✨ Cache hit - no changes detected, skipping generation");
        }
        println!("✅ TypeScript bindings are up to date");
        return Ok(());
    }

    if config.is_verbose() && !config.should_force() {
        println!("🔄 Changes detected, regenerating bindings");
    }

    // Generate bindings
    reporter.start_step("Generating TypeScript bindings");
    let validation = match config.validation_library.as_str() {
        "zod" | "none" => Some(config.validation_library.clone()),
        _ => return Err("Invalid validation library. Use 'zod' or 'none'".into()),
    };

    let mut generator = create_generator(validation);
    let generated_files = generator.generate_models(
        &commands,
        discovered_structs,
        &config.output_path,
        &analyzer,
        &config,
    )?;
    reporter.complete_step(Some(&format!("Generated {} files", generated_files.len())));

    // Generate dependency visualization if requested
    if config.should_visualize_deps() {
        let text_viz = analyzer.visualize_dependencies(&commands);
        let viz_file_path = PathBuf::from(&config.output_path).join("dependency-graph.txt");
        fs::write(&viz_file_path, text_viz)?;

        let dot_viz = analyzer.generate_dot_graph(&commands);
        let dot_file_path = PathBuf::from(&config.output_path).join("dependency-graph.dot");
        fs::write(&dot_file_path, dot_viz)?;

        print_dependency_visualization_info(&config.output_path);
    }

    // Save cache after successful generation
    let cache = GenerationCache::new(&commands, discovered_structs, discovered_events, &config)?;
    if let Err(e) = cache.save(&config.output_path) {
        eprintln!("Warning: Failed to save generation cache: {}", e);
    }

    // Print summary
    reporter.finish("Generation complete");
    print_usage_info(&config.output_path, &generated_files, commands.len());

    Ok(())
}

fn run_init(
    project_path: Option<PathBuf>,
    generated_path: Option<PathBuf>,
    output_path: Option<PathBuf>,
    validation_library: Option<String>,
    verbose: bool,
    visualize_deps: bool,
    force: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    let logger = Logger::new(verbose, false);

    logger.info("🚀 Initializing Tauri TypeScript generation configuration");

    // Resolve paths with defaults
    let project_path = project_path.unwrap_or_else(|| PathBuf::from("./src-tauri"));
    let generated_path = generated_path.unwrap_or_else(|| PathBuf::from("./src/generated"));
    let mut output_path = output_path.unwrap_or_else(|| PathBuf::from("tauri.conf.json"));
    let validation_library = validation_library.unwrap_or_else(|| "none".to_string());

    // If output path is just "tauri.conf.json" (default), place it in the project path
    let has_no_meaningful_parent = output_path
        .parent()
        .map(|p| p.as_os_str().is_empty())
        .unwrap_or(true);

    if output_path.file_name().and_then(|n| n.to_str()) == Some("tauri.conf.json")
        && has_no_meaningful_parent
    {
        output_path = project_path.join("tauri.conf.json");
    }

    let is_tauri_config =
        output_path.file_name().and_then(|n| n.to_str()) == Some("tauri.conf.json");

    // For tauri.conf.json, we always update/merge (no force required)
    // For custom config files, require force if they exist
    if !is_tauri_config && output_path.exists() && !force {
        return Err(format!(
            "Configuration file already exists at {}. Use --force to overwrite.",
            output_path.display()
        )
        .into());
    }

    // Create configuration
    let config = GenerateConfig {
        project_path: project_path.to_string_lossy().to_string(),
        output_path: generated_path.to_string_lossy().to_string(),
        validation_library,
        verbose: Some(verbose),
        visualize_deps: Some(visualize_deps),
        ..Default::default()
    };

    // Determine file format and save
    if is_tauri_config {
        // For tauri.conf.json, require it to exist
        if !output_path.exists() {
            return Err(format!(
                "tauri.conf.json not found at {}.\n\
                 Please ensure you have a Tauri project initialized.\n\
                 Run 'cargo tauri init' or use --output to specify a different config file.",
                output_path.display()
            )
            .into());
        }

        config.save_to_tauri_config(&output_path)?;
        logger.info(&format!(
            "✅ Updated typegen configuration in {}",
            output_path.display()
        ));
    } else {
        config.save_to_file(&output_path)?;
        logger.info(&format!(
            "✅ Created configuration file: {}",
            output_path.display()
        ));
    }

    // Print configuration summary
    logger.info("📋 Configuration summary:");
    logger.info(&format!("  • Project path: {}", config.project_path));
    logger.info(&format!(
        "  • Generated files output path: {}",
        config.output_path
    ));
    logger.info(&format!(
        "  • Validation library: {}",
        config.validation_library
    ));

    // Now run initial generation
    logger.info("");
    logger.info("🔄 Running initial generation...");

    run_generate(
        Some(project_path),
        Some(generated_path),
        Some(config.validation_library.clone()),
        verbose,
        visualize_deps,
        None,  // No config file since we just created one
        false, // Respect cache behavior
    )?;

    logger.info("");
    logger.info(
        "✨ Initialization complete! Your Tauri project is now set up for TypeScript generation.",
    );
    logger.info("💡 You can run 'cargo tauri-typegen generate' anytime to regenerate bindings.");

    Ok(())
}