codes

#!/bin/bash

# --- Function Definitions ---

# Function to initialize and update Git submodules (Initial checkout)
init_submodules() {
    echo "--- Starting Submodule Initialization and Checkout ---"

    # 1. Ensure Git submodule configuration is initialized
    git submodule init

    # 2. Loop through all submodules defined in .gitmodules
    git config -f .gitmodules --get-regexp submodule\..*\.path | while read key path
    do
        echo ""
        echo "--- Attempting to checkout $path ---"

        # Update/clone the submodule using a selective command
        if git submodule update --init -- $path; then
            echo "✅ Successfully checked out $path."
        else
            # Handle Failure (Non-fatal warning)
            echo "🚨 WARNING: Failed to checkout $path due to access issues."
            echo "Skipping installation for this module."
        fi
        echo "-----------------------------------"
    done

    echo "Submodule checkout complete! All accessible codes checked out."
}

# Function to install submodules that are valid Python packages
install_packages() {
    echo "--- Starting Python Package Installation (Editable Mode) ---"

    # 1. Loop through all submodules defined in .gitmodules
    git config -f .gitmodules --get-regexp submodule\..*\.path | while read key path
    do
        # Check if the path exists AND if it contains a pyproject.toml
        if [ -d "$path" ] && [ -f "$path/pyproject.toml" ]; then
            echo ""
            echo "📦 Installing $path in editable mode (pip install -e)."
            # Using -U for upgrade/update in case the package is already installed
            pip install -U -e "$path"
        fi
    done

    echo "Installation complete! Environment is ready to use."
}

# Update all submodules to the latest commit on their tracked branch
update_submodules() {
    echo "--- Starting Submodule Update (Fetching latest remote commits and REBASING) ---"

    # This command fetches the latest remote commit and updates the local submodule
    # using a rebase operation instead of a merge.
    if git submodule update --remote --rebase; then
        echo "✅ All submodules updated successfully using rebase."
    else
        echo "❌ ERROR: Failed to update one or more submodules via rebase."
        echo "Please check permissions and network connectivity, and resolve any rebase conflicts."
    fi
}


# Function to display usage information
show_usage() {
    echo "Usage: $0 <command>"
    echo ""
    echo "Available commands:"
    echo "  init    : Initializes and checks out submodules (for first time setup)."
    echo "  update  : Pulls the latest version from the remote branch for all submodules."
    echo "  install : Installs all submodules that are valid Python packages using 'pip install -U -e'."
    echo "  help    : Show this usage message."
    echo ""
}

# --- Main Logic (Command Parser) ---

# Check if a command was provided
if [ -z "$1" ]; then
    show_usage
    exit 1
fi

COMMAND=$1

case "$COMMAND" in
    init)
        init_submodules
        ;;
    update)
        update_submodules
        ;;
    install)
        install_packages
        ;;
    help)
        show_usage
        ;;
    *)
        echo "❌ Error: Unknown command '$COMMAND'"
        show_usage
        exit 1
        ;;
esac

exit 0