wsb 0.196.4203

use crate::prelude::*;
use crate::fleet::config::GitModules;
use crate::fleet::error::{Result, SuperGitError};
use crate::fleet::git::{GitOps, SubmoduleOps};
use crate::fleet::output::Formatter;
use crate::git_backend;
use dialoguer::Confirm;

pub fn execute<P: AsRef<Path>>(
    repo_root: P,
    branch: &str,
    create: bool,
    dry_run: bool,
    verbose: bool,
) -> Result<()> {
    let gitmodules = GitModules::parse(&repo_root)?;

    Formatter::print_info(&format!(
        "Checking out '{}' in superproject + {} submodule(s){}",
        branch,
        gitmodules.submodules.len(),
        if create { " (creating if needed)" } else { "" }
    ));

    if dry_run {
        Formatter::print_warning("DRY RUN: No changes will be made");
    }

    let mut successful = 0;
    let mut failed = 0;
    let mut skipped = 0;

    // Checkout in superproject first
    if verbose {
        Formatter::print_submodule_header("[SUPERPROJECT]");
    }

    let super_repo = GitOps::open_repo(&repo_root)?;
    let super_is_clean = GitOps::is_clean(&super_repo)?;

    if !super_is_clean && !dry_run {
        Formatter::print_warning("Superproject has uncommitted changes");
        let should_continue = Confirm::new()
            .with_prompt("Do you want to continue? (changes may be lost)")
            .default(false)
            .interact()
            .unwrap_or(false);

        if !should_continue {
            return Err(SuperGitError::Cancelled);
        }
    }

    let super_branch_exists = GitOps::branch_exists(&super_repo, branch)?;

    if !super_branch_exists && !create {
        Formatter::print_error(&format!(
            "Branch '{}' does not exist in superproject. Use --create to create it.",
            branch
        ));
        return Err(SuperGitError::Other(
            "Branch does not exist in superproject".to_string(),
        ));
    }

    if !dry_run {
        let result = if !super_branch_exists && create {
            git_backend::create_branch(&super_repo, branch, None)
        } else {
            git_backend::checkout_branch(&super_repo, branch)
        };
        if let Err(err) = result {
            Formatter::print_error(&format!("Failed to checkout '{}' in superproject: {}", branch, err));
            return Err(SuperGitError::Other("Superproject checkout failed".to_string()));
        }

        if super_branch_exists {
            Formatter::print_success(&format!("Checked out '{}' in superproject", branch));
        } else {
            Formatter::print_success(&format!(
                "Created and checked out '{}' in superproject",
                branch
            ));
        }
        successful += 1;
    } else {
        if super_branch_exists {
            Formatter::print_info(&format!("Would checkout '{}'", branch));
        } else {
            Formatter::print_info(&format!("Would create and checkout '{}'", branch));
        }
        successful += 1;
    }

    for submodule in &gitmodules.submodules {
        if verbose {
            Formatter::print_submodule_header(&submodule.name);
        }

        let submodule_path = repo_root.as_ref().join(&submodule.path);

        // Check if submodule exists
        if !submodule_path.exists() {
            Formatter::print_warning(&format!(
                "Submodule '{}' not initialized, skipping",
                submodule.name
            ));
            skipped += 1;
            continue;
        }

        // Check for uncommitted changes
        let repo = match GitOps::open_repo(&submodule_path) {
            Ok(r) => r,
            Err(e) => {
                Formatter::print_error(&format!("Failed to open '{}': {}", submodule.name, e));
                failed += 1;
                continue;
            }
        };

        let is_clean = match GitOps::is_clean(&repo) {
            Ok(c) => c,
            Err(e) => {
                Formatter::print_error(&format!(
                    "Failed to check status of '{}': {}",
                    submodule.name, e
                ));
                failed += 1;
                continue;
            }
        };

        if !is_clean && !dry_run {
            Formatter::print_warning(&format!(
                "Submodule '{}' has uncommitted changes",
                submodule.name
            ));

            let should_continue = Confirm::new()
                .with_prompt("Do you want to continue? (changes may be lost)")
                .default(false)
                .interact()
                .unwrap_or(false);

            if !should_continue {
                Formatter::print_info("Skipping");
                skipped += 1;
                continue;
            }
        }

        // Check if branch exists
        let branch_exists = match GitOps::branch_exists(&repo, branch) {
            Ok(exists) => exists,
            Err(e) => {
                Formatter::print_error(&format!(
                    "Failed to check branch in '{}': {}",
                    submodule.name, e
                ));
                failed += 1;
                continue;
            }
        };

        if !branch_exists && !create {
            Formatter::print_error(&format!(
                "Branch '{}' does not exist in '{}'. Use --create to create it.",
                branch, submodule.name
            ));
            failed += 1;
            continue;
        }

        if dry_run {
            if branch_exists {
                Formatter::print_info(&format!("Would checkout '{}'", branch));
            } else {
                Formatter::print_info(&format!("Would create and checkout '{}'", branch));
            }
            successful += 1;
            continue;
        }

        // Perform checkout
        match SubmoduleOps::checkout(&repo_root, submodule, branch, !branch_exists && create) {
            Ok(_) => {
                if branch_exists {
                    Formatter::print_success(&format!(
                        "Checked out '{}' in '{}'",
                        branch, submodule.name
                    ));
                } else {
                    Formatter::print_success(&format!(
                        "Created and checked out '{}' in '{}'",
                        branch, submodule.name
                    ));
                }
                successful += 1;
            }
            Err(e) => {
                Formatter::print_error(&format!(
                    "Failed to checkout '{}' in '{}': {}",
                    branch, submodule.name, e
                ));
                failed += 1;
            }
        }
    }

    if !dry_run {
        Formatter::print_summary(successful, failed, skipped);
    }

    if failed > 0 {
        return Err(SuperGitError::Other(format!(
            "{} submodule(s) failed to checkout",
            failed
        )));
    }

    Ok(())
}