usm/installer/main.sh

#!/bin/bash
# main.sh - Main entry point for USM installer
# This file will be the entry point of the compiled installer

# Default configuration
TARGET_DIR="/opt/usm"
ASSUME_YES="false"
VERBOSE="false"
SCRIPT_VERSION="@VERSION@"
PM_TYPE=""
ORIGINAL_ARGS=("$@")

# Parse command line arguments
parse_args() {
    while [[ $# -gt 0 ]]; do
        case "$1" in
            --target|-t)
                TARGET_DIR="$2"
                shift 2
                ;;
            --target=*)
                TARGET_DIR="${1#*=}"
                shift
                ;;
            -y|--yes|--assume-yes)
                ASSUME_YES="true"
                shift
                ;;
            --verbose|-V)
                VERBOSE="true"
                shift
                ;;
            --help|-h)
                show_help
                exit 0
                ;;
            --version|-v)
                echo "USM Installer v${SCRIPT_VERSION}"
                exit 0
                ;;
            *)
                echo "Unknown option: $1"
                show_help
                exit 1
                ;;
        esac
    done
}

# Show help message
show_help() {
    cat << EOF
USM Installer v${SCRIPT_VERSION}

Usage: $(basename "$0") [OPTIONS]

Options:
    -t, --target DIR    Install to DIR instead of /opt/usm
    -y, --assume-yes    Answer yes to all prompts automatically
    -V, --verbose       Show verbose build output
    -h, --help          Show this help message
    -v, --version       Show version information

Examples:
    $(basename "$0")                    # Install to /opt/usm
    $(basename "$0") -t /opt/myusm     # Install to /opt/myusm
    $(basename "$0") -y                # Non-interactive install
    $(basename "$0") -V                # Verbose output

EOF
}

# Check for root privileges and offer to re-run with sudo
check_root() {
    if is_root; then
        return 0
    fi
    
    local sudo_cmd=$(get_sudo)
    
    if [[ -z "$sudo_cmd" ]]; then
        show_error_and_exit "This script requires root privileges. Please run with sudo or as root."
    fi
    
    log_error "This script requires root privileges."
    
    if confirm "Would you like to re-run the script with $sudo_cmd?" "y"; then
        log_info "Re-running with $sudo_cmd..."
        exec $sudo_cmd "$0" "${ORIGINAL_ARGS[@]}"
    else
        show_error_and_exit "Root privileges are required. Installation cancelled."
    fi
}

# Check if target directory exists and handle reinstallation
check_existing_installation() {
    if [[ -d "$TARGET_DIR" ]]; then
        log_warn "Target directory '$TARGET_DIR' already exists."
        
        if [[ "$ASSUME_YES" == "true" ]]; then
            log_info "Removing existing installation (auto-confirmed with -y)..."
            rm -rf "$TARGET_DIR"
            return 0
        fi
        
        if confirm "Do you want to delete it and reinstall?" "n"; then
            log_info "Removing existing installation..."
            rm -rf "$TARGET_DIR"
        else
            echo "Installation cancelled."
            exit 0
        fi
    fi
}

# Install the system package manager helper matching the detected PM from
# the extracted payload to /usr/bin. Runs before config generation so the
# `usm genconfig` call in install_shim detects the system package manager
# and wires the generated usm.config to this helper.
install_spm_shim() {
    local extract_dir="$1"
    local shim_name="$PM_TYPE"
    local source_shim="$extract_dir/spm/$shim_name/usm-spm-$shim_name"
    local shim_path="/usr/bin/usm-spm-$shim_name"
    local sudo=""
    
    if ! is_root; then
        sudo=$(get_sudo)
    fi
    
    if [[ ! -f "$source_shim" ]]; then
        log_error "SPM helper \"$source_shim\" not found in the installer payload"
        return 1
    fi
    
    log_step "Installing SPM helper to $shim_path..."
    
    if [[ -n "$sudo" ]]; then
        $sudo cp "$source_shim" "$shim_path" && $sudo chmod 755 "$shim_path"
    else
        cp "$source_shim" "$shim_path" && chmod 755 "$shim_path"
    fi
}

# Main installation function
run_installation() {
    # Install system dependencies first (before extracting sources)
    log_step "Checking system dependencies..."
    if ! pm_install_missing_deps; then
        show_error_and_exit "Failed to install system dependencies. Installation halted."
    fi
    
    # Set up temporary directory
    setup_temp_dir
    local extract_dir="$TEMP_DIR/sources"
    ensure_dir "$extract_dir"
    
    # Extract bundled sources
    log_step "Extracting bundled sources..."
    if ! extract_payload "$extract_dir"; then
        show_error_and_exit "Failed to extract bundled sources"
    fi
    
    # Build and install all components
    log_step "Building and installing components..."
    if ! build_all "$extract_dir" "$TARGET_DIR"; then
        show_error_and_exit "Failed to build components. Installation halted."
    fi
    
    # Install the SPM helper before the shim step so config generation sees it
    if ! install_spm_shim "$extract_dir"; then
        show_error_and_exit "Failed to install the SPM helper. Installation halted."
    fi
    
    # Install shim
    install_shim "$TARGET_DIR"
    
    # Show completion message
    show_completion "$TARGET_DIR"
}

# Main entry point
main() {
    parse_args "$@"
    
    # Print banner
    print_banner
    
    # Check for root privileges
    check_root
    
    # Detect package manager
    if ! detect_package_manager; then
        show_error_and_exit "Could not detect a supported package manager (apk, apt, dnf or emerge)"
    fi
    
    log_info "Detected package manager: ${PM_TYPE}"
    
    # Get missing dependencies for display
    local missing_deps=$(pm_get_missing_deps)
    local deps_count=$(count_missing_deps)
    
    # Show installation summary and get confirmation
    if ! show_installation_summary "$TARGET_DIR" "$PM_TYPE" "$deps_count" "$missing_deps"; then
        echo "Installation cancelled."
        exit 0
    fi
    
    # Check if target directory exists and handle reinstallation
    check_existing_installation
    
    # Run the installation
    run_installation
}