#!/bin/bash

# Nacos Setup Installation Script
# This script downloads and installs nacos-setup from remote repository

set -e

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

print_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

print_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

print_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# ============================================================================
# Configuration
# ============================================================================

DOWNLOAD_BASE_URL="https://download.nacos.io"

INSTALL_BASE_DIR="$HOME/.nacos/nacos-setup"
CURRENT_LINK="nacos-setup"
BIN_DIR="$HOME/.nacos/bin"
SCRIPT_NAME="nacos-setup"
TEMP_DIR="/tmp/nacos-setup-install-$$"
CACHE_DIR="${HOME}/.nacos/cache"  # 缓存目录

# ============================================================================
# Version Management (embedded for standalone operation)
# ============================================================================

DOWNLOAD_BASE_URL="https://download.nacos.io"
VERSIONS_URL="${DOWNLOAD_BASE_URL}/versions"

# Fallback Versions
FALLBACK_NACOS_CLI_VERSION="1.0.0"
FALLBACK_NACOS_SETUP_VERSION="1.0.2"
FALLBACK_NACOS_SERVER_VERSION="3.2.1-2026.03.30"

# Cached versions
_CACHED_CLI_VERSION=""
_CACHED_SETUP_VERSION=""
_CACHED_SERVER_VERSION=""
_VERSIONS_FETCHED=false

# Helper function for version warnings
_versions_print_warn() {
    print_warn "$1"
}

# Fetch versions from remote URL
fetch_versions() {
    local timeout=${1:-1}
    
    if [ "$_VERSIONS_FETCHED" = true ]; then
        return 0
    fi
    
    _VERSIONS_FETCHED=true
    
    # Try to fetch versions file
    local versions_content
    if versions_content=$(curl -fSL --max-time "$timeout" "$VERSIONS_URL" 2>/dev/null); then
        # Parse the content
        while IFS='=' read -r key value; do
            case "$key" in
                NACOS_CLI_VERSION) _CACHED_CLI_VERSION="$value" ;;
                NACOS_SETUP_VERSION) _CACHED_SETUP_VERSION="$value" ;;
                NACOS_SERVER_VERSION) _CACHED_SERVER_VERSION="$value" ;;
            esac
        done <<< "$versions_content"
    else
        _versions_print_warn "Failed to fetch versions from $VERSIONS_URL, using fallback versions"
    fi
    
    # Use fallback if cache is empty
    if [ -z "$_CACHED_CLI_VERSION" ]; then
        _CACHED_CLI_VERSION="$FALLBACK_NACOS_CLI_VERSION"
    fi
    if [ -z "$_CACHED_SETUP_VERSION" ]; then
        _CACHED_SETUP_VERSION="$FALLBACK_NACOS_SETUP_VERSION"
    fi
    if [ -z "$_CACHED_SERVER_VERSION" ]; then
        _CACHED_SERVER_VERSION="$FALLBACK_NACOS_SERVER_VERSION"
    fi
}

# Get all versions at once
get_all_versions() {
    local timeout=${1:-5}
    
    # Check environment variables first
    if [ -n "$NACOS_CLI_VERSION" ]; then
        _CACHED_CLI_VERSION="$NACOS_CLI_VERSION"
    fi
    if [ -n "$NACOS_SETUP_VERSION" ]; then
        _CACHED_SETUP_VERSION="$NACOS_SETUP_VERSION"
    fi
    if [ -n "$NACOS_SERVER_VERSION" ]; then
        _CACHED_SERVER_VERSION="$NACOS_SERVER_VERSION"
    fi
    
    # Fetch from remote if not set by env
    if [ -z "$NACOS_CLI_VERSION" ] || [ -z "$NACOS_SETUP_VERSION" ] || [ -z "$NACOS_SERVER_VERSION" ]; then
        fetch_versions "$timeout"
    fi
    
    # Export for use
    NACOS_CLI_VERSION="$_CACHED_CLI_VERSION"
    NACOS_SETUP_VERSION="$_CACHED_SETUP_VERSION"
    NACOS_SERVER_VERSION="$_CACHED_SERVER_VERSION"
}

# ============================================================================
# Check Requirements
# ============================================================================

check_requirements() {
    print_info "Checking system requirements..."
    
    # Check if running on macOS or Linux
    if [[ "$OSTYPE" != "darwin"* ]] && [[ "$OSTYPE" != "linux-gnu"* ]]; then
        print_error "Unsupported OS: $OSTYPE"
        print_error "This script only supports macOS and Linux"
        exit 1
    fi
    
    # Check for required commands
    local missing_commands=()
    
    if ! command -v curl >/dev/null 2>&1 && ! command -v wget >/dev/null 2>&1; then
        missing_commands+=("curl or wget")
    fi
    
    if ! command -v unzip >/dev/null 2>&1; then
        missing_commands+=("unzip")
    fi
    
    if [ ${#missing_commands[@]} -gt 0 ]; then
        print_error "Missing required commands: ${missing_commands[*]}"
        echo ""
        print_info "Please install missing commands:"
        if [[ "$OSTYPE" == "darwin"* ]]; then
            echo "  brew install curl unzip"
        else
            echo "  sudo apt-get install curl unzip  # Debian/Ubuntu"
            echo "  sudo yum install curl unzip      # CentOS/RHEL"
        fi
        return 1
    fi
    
    # Check if we have write permission to install directory
    # For user-level paths, attempt to create directories first since they may not exist yet
    local mode="${1:-full}"
    if [[ "$mode" == "onlycli" ]]; then
        mkdir -p "$BIN_DIR" 2>/dev/null
        if [ ! -w "$BIN_DIR" ]; then
            print_warn "No write permission to $BIN_DIR"
            print_warn "Please check directory permissions or create the directory manually: mkdir -p $BIN_DIR"
            return 1
        fi
    else
        mkdir -p "$INSTALL_BASE_DIR" "$BIN_DIR" 2>/dev/null
        if [ ! -w "$INSTALL_BASE_DIR" ]; then
            print_warn "No write permission to $INSTALL_BASE_DIR"
            print_warn "Please check directory permissions or create the directory manually: mkdir -p $INSTALL_BASE_DIR"
            return 1
        fi
    fi
    
    return 0
}

# ============================================================================
# Download
# ============================================================================

download_file() {
    local url=$1
    local output=$2
    
    print_info "Downloading from $url..." >&2
    
    # Try curl first, then wget
    if command -v curl >/dev/null 2>&1; then
        if curl -fSL --progress-bar "$url" -o "$output"; then
            return 0
        else
            print_error "Download failed with curl" >&2
            return 1
        fi
    elif command -v wget >/dev/null 2>&1; then
        if wget -q --show-progress "$url" -O "$output"; then
            return 0
        else
            print_error "Download failed with wget" >&2
            return 1
        fi
    else
        print_error "Neither curl nor wget is available" >&2
        return 1
    fi
}

# Download nacos-setup package with caching support
# Parameters: version
# Returns: path to zip file (in cache or temp) or empty on error
download_nacos_setup() {
    local version=$1
    local zip_filename="nacos-setup-${version}.zip"
    local download_url="${DOWNLOAD_BASE_URL}/nacos-setup-${version}.zip"
    local cached_file="$CACHE_DIR/$zip_filename"
    
    # Create cache directory
    mkdir -p "$CACHE_DIR" 2>/dev/null
    
    # Check if cached file exists and is valid
    if [ -f "$cached_file" ] && [ -s "$cached_file" ]; then
        # Verify the cached zip file is valid
        if unzip -t "$cached_file" >/dev/null 2>&1; then
            print_info "Found cached package: $cached_file" >&2
            print_info "Skipping download, using cached file" >&2
            echo "" >&2
            echo "$cached_file"
            return 0
        else
            print_warn "Cached file is corrupted, re-downloading..." >&2
            rm -f "$cached_file"
        fi
    fi
    
    # Download the file to cache
    print_info "Downloading nacos-setup version: $version" >&2
    echo "" >&2
    
    if ! download_file "$download_url" "$cached_file"; then
        print_error "Failed to download nacos-setup" >&2
        rm -f "$cached_file"
        return 1
    fi
    
    echo "" >&2
    
    # Verify downloaded file is a valid zip
    if ! unzip -t "$cached_file" >/dev/null 2>&1; then
        print_error "Downloaded file is corrupted or invalid" >&2
        rm -f "$cached_file"
        return 1
    fi
    
    print_info "Download completed: $zip_filename" >&2
    echo "$cached_file"
    return 0
}

# Download nacos-cli package with caching support
# Parameters: version, os, arch
# Returns: path to zip file (in cache) or empty on error
download_nacos_cli() {
    local version=$1
    local os=$2
    local arch=$3
    local zip_filename="nacos-cli-${version}-${os}-${arch}.zip"
    local download_url="${DOWNLOAD_BASE_URL}/nacos-cli/${zip_filename}"
    local cached_file="$CACHE_DIR/$zip_filename"
    
    # Create cache directory
    mkdir -p "$CACHE_DIR" 2>/dev/null
    
    # Check if cached file exists and is valid
    if [ -f "$cached_file" ] && [ -s "$cached_file" ]; then
        # Verify the cached zip file is valid
        if unzip -t "$cached_file" >/dev/null 2>&1; then
            print_info "Found cached package: $cached_file" >&2
            print_info "Skipping download, using cached file" >&2
            echo "" >&2
            echo "$cached_file"
            return 0
        else
            print_warn "Cached file is corrupted, re-downloading..." >&2
            rm -f "$cached_file"
        fi
    fi
    
    # Download the file to cache
    print_info "Downloading nacos-cli version: $version" >&2
    echo "" >&2
    
    if ! download_file "$download_url" "$cached_file"; then
        print_error "Failed to download nacos-cli" >&2
        rm -f "$cached_file"
        return 1
    fi
    
    echo "" >&2
    
    # Verify downloaded file is a valid zip
    if ! unzip -t "$cached_file" >/dev/null 2>&1; then
        print_error "Downloaded file is corrupted or invalid" >&2
        rm -f "$cached_file"
        return 1
    fi
    
    print_info "Download completed: $zip_filename" >&2
    echo "$cached_file"
    return 0
}

# ============================================================================
# Installation
# ============================================================================

install_nacos_setup() {
    print_info "Installing nacos-setup..."
    echo ""
    
    # Get version from parameter, environment variable, or use default
    local setup_version="${1:-${NACOS_SETUP_VERSION}}"
    
    print_info "Target version: $setup_version"
    
    # Ensure installation directories exist
    mkdir -p "$INSTALL_BASE_DIR"
    mkdir -p "$BIN_DIR"
    
    # Download nacos-setup (with caching)
    # If cached version exists, it will be used directly
    # If not, download from remote and save to cache
    local zip_file=$(download_nacos_setup "$setup_version")
    
    if [ -z "$zip_file" ]; then
        print_error "Failed to download nacos-setup"
        exit 1
    fi
    
    print_success "Package ready: $zip_file"
    echo ""
    
    # Create temporary directory for extraction
    mkdir -p "$TEMP_DIR"
    
    # Extract zip file
    print_info "Extracting nacos-setup..."
    if ! unzip -q "$zip_file" -d "$TEMP_DIR"; then
        print_error "Failed to extract zip file"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    # Find extracted directory (should be nacos-setup-VERSION or similar)
    local extracted_dir=$(find "$TEMP_DIR" -maxdepth 1 -type d ! -path "$TEMP_DIR" | head -1)
    
    if [ -z "$extracted_dir" ] || [ ! -d "$extracted_dir" ]; then
        print_error "Failed to find extracted directory"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    # Verify required files
    if [ ! -f "$extracted_dir/nacos-setup.sh" ]; then
        print_error "nacos-setup.sh not found in package"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    if [ ! -d "$extracted_dir/lib" ]; then
        print_error "lib directory not found in package"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    # Prepare versioned installation directory
    local INSTALL_DIR="$INSTALL_BASE_DIR/${CURRENT_LINK}-$setup_version"

    # Remove old installation for this version if exists
    if [ -d "$INSTALL_DIR" ]; then
        print_info "Removing old installation..."
        rm -rf "$INSTALL_DIR"
    fi
    
    # Create installation directory
    print_info "Creating installation directory: $INSTALL_DIR"
    mkdir -p "$INSTALL_DIR"
    
    # Copy nacos-setup.sh to bin directory
    print_info "Installing nacos-setup command..."
    mkdir -p "$INSTALL_DIR/bin"
    cp "$extracted_dir/nacos-setup.sh" "$INSTALL_DIR/bin/$SCRIPT_NAME"
    chmod +x "$INSTALL_DIR/bin/$SCRIPT_NAME"
    
    # Copy lib directory
    print_info "Installing libraries..."
    cp -r "$extracted_dir/lib" "$INSTALL_DIR/"
    
    # Make all lib scripts executable
    chmod +x "$INSTALL_DIR/lib"/*.sh
    
    # Create or update current symlink and global command
    print_info "Updating active version symlink: $INSTALL_BASE_DIR/$CURRENT_LINK -> nacos-setup-$setup_version"
    if [ -L "$INSTALL_BASE_DIR/$CURRENT_LINK" ] || [ -e "$INSTALL_BASE_DIR/$CURRENT_LINK" ]; then
        rm -f "$INSTALL_BASE_DIR/$CURRENT_LINK"
    fi
    ln -s "nacos-setup-$setup_version" "$INSTALL_BASE_DIR/$CURRENT_LINK"

    print_info "Creating global command..."
    # Ensure bin directory exists
    mkdir -p "$BIN_DIR"
    
    # Remove old symlink if exists
    if [ -L "$BIN_DIR/$SCRIPT_NAME" ] || [ -f "$BIN_DIR/$SCRIPT_NAME" ]; then
        rm -f "$BIN_DIR/$SCRIPT_NAME"
    fi
    
    # Create symlink with absolute path
    local target_script="$INSTALL_BASE_DIR/$CURRENT_LINK/bin/$SCRIPT_NAME"
    
    # Verify target exists before creating symlink
    if [ ! -f "$target_script" ]; then
        print_error "Target script not found: $target_script"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    ln -s "$target_script" "$BIN_DIR/$SCRIPT_NAME"
    
    # Verify symlink was created successfully
    if [ ! -L "$BIN_DIR/$SCRIPT_NAME" ]; then
        print_error "Failed to create symlink at $BIN_DIR/$SCRIPT_NAME"
        rm -rf "$TEMP_DIR"
        exit 1
    fi
    
    print_info "Global command created: $BIN_DIR/$SCRIPT_NAME -> $target_script"
    
    # Cleanup temporary directory
    rm -rf "$TEMP_DIR"
    
    # Store version info
    echo "$setup_version" > "$INSTALL_DIR/.version"
    
    print_success "Installation completed!"
    echo ""
    
    # Export version for later use
    INSTALLED_VERSION="$setup_version"

    _nacos_try_link_nacos_setup_system_symlink "$BIN_DIR/$SCRIPT_NAME"
}

# If /usr/local/bin is writable, symlink nacos-setup there so it is available
# immediately even when the installer was invoked via sudo (which changes $HOME).
_nacos_try_link_nacos_setup_system_symlink() {
    local src="$1"
    local dst="/usr/local/bin/nacos-setup"
    [ -n "$src" ] && [ -f "$src" ] || return 0
    if [ ! -d /usr/local/bin ]; then
        mkdir -p /usr/local/bin 2>/dev/null || return 0
    fi
    if [ ! -w /usr/local/bin ]; then
        return 0
    fi
    if ln -sf "$src" "$dst" 2>/dev/null; then
        print_success "Linked nacos-setup to $dst (available without source ~/.bashrc)"
    fi
}

# ============================================================================
# nacos-cli Installation
# ============================================================================

# If /usr/local/bin is writable, symlink nacos-cli there so typical PATH (e.g. root)
# includes the command immediately — subprocess installers cannot export PATH to the parent shell.
_nacos_try_link_nacos_cli_system_symlink() {
    local src="$1"
    local dst="/usr/local/bin/nacos-cli"
    [ -n "$src" ] && [ -f "$src" ] || return 0
    if [ ! -d /usr/local/bin ]; then
        mkdir -p /usr/local/bin 2>/dev/null || return 0
    fi
    if [ ! -w /usr/local/bin ]; then
        return 0
    fi
    if ln -sf "$src" "$dst" 2>/dev/null; then
        print_success "Linked nacos-cli to $dst (available in this shell without source ~/.bashrc)"
    fi
}

install_nacos_cli() {
    local version="${NACOS_CLI_VERSION}"

    print_info "Preparing to install nacos-cli version $version..."

    local os=""
    if [[ "$OSTYPE" == "darwin"* ]]; then
        os="darwin"
    elif [[ "$OSTYPE" == "linux-gnu"* ]] || [[ "$OSTYPE" == "linux"* ]]; then
        os="linux"
    else
        local uname_os
        uname_os=$(uname -s 2>/dev/null || echo "")
        if [[ "$uname_os" == "Darwin" ]]; then
            os="darwin"
        elif [[ "$uname_os" == "Linux" ]]; then
            os="linux"
        else
            print_warn "Unsupported OS for nacos-cli: $OSTYPE (uname: $uname_os)"
            return 1
        fi
    fi

    # Detect architecture
    local arch=""
    local uname_arch
    uname_arch=$(uname -m)
    case "$uname_arch" in
        x86_64|amd64)
            arch="amd64"
            ;;
        arm64|aarch64)
            arch="arm64"
            ;;
        *)
            print_warn "Unsupported architecture for nacos-cli: $uname_arch"
            return 1
            ;;
    esac
    # Download nacos-cli (with caching)
    local zip_file=$(download_nacos_cli "$version" "$os" "$arch")
    
    if [ -z "$zip_file" ]; then
        print_error "Failed to download nacos-cli"
        return 1
    fi
    
    print_success "Package ready: $zip_file"
    echo ""
    
    # Create temporary directory for extraction
    local tmp_dir
    tmp_dir=$(mktemp -d "/tmp/nacos-cli-extract-$$.XXXXXX") || {
        print_error "Failed to create temp directory for nacos-cli extraction"
        return 1
    }

    # Extract zip file
    print_info "Extracting nacos-cli..."
    if ! unzip -q "$zip_file" -d "$tmp_dir"; then
        print_error "Failed to extract zip file"
        rm -rf "$tmp_dir"
        return 1
    fi

    # New packages contain a stable binary name; keep old names as fallback.
    local expected_binary_name="nacos-cli"
    local legacy_binary_name="nacos-cli-${version}-${os}-${arch}"
    local legacy_binary_name_exe="${legacy_binary_name}.exe"
    local binary_path
    binary_path=$(find "$tmp_dir" -name "$expected_binary_name" -type f | head -1)
    if [ -z "$binary_path" ]; then
        binary_path=$(find "$tmp_dir" -name "$legacy_binary_name" -type f | head -1)
    fi
    if [ -z "$binary_path" ]; then
        binary_path=$(find "$tmp_dir" -name "$legacy_binary_name_exe" -type f | head -1)
    fi

    if [ -z "$binary_path" ] || [ ! -f "$binary_path" ]; then
        local expected_names="$expected_binary_name, $legacy_binary_name, or $legacy_binary_name_exe"
        print_error "Binary file not found in package. Expected: $expected_names"
        print_info "Available files in package:"
        find "$tmp_dir" -type f | sed 's|^|  |'
        rm -rf "$tmp_dir"
        return 1
    fi

    # Ensure bin dir exists
    mkdir -p "$BIN_DIR"

    local target_binary_name="nacos-cli"

    # Install binary
    if ! cp "$binary_path" "$BIN_DIR/$target_binary_name"; then
        print_error "Failed to copy nacos-cli to $BIN_DIR (permission denied?)"
        rm -rf "$tmp_dir"
        return 1
    fi

    if ! chmod +x "$BIN_DIR/$target_binary_name" 2>/dev/null; then
        print_warn "Failed to mark nacos-cli as executable: $BIN_DIR/$target_binary_name"
    fi

    _nacos_try_link_nacos_cli_system_symlink "$BIN_DIR/$target_binary_name"

    # On macOS, add ad-hoc signature to avoid Gatekeeper killing the binary
    if [[ "$os" == "darwin" ]]; then
        if command -v codesign >/dev/null 2>&1; then
            if ! codesign --force --deep --sign - "$BIN_DIR/$target_binary_name" >/dev/null 2>&1; then
                print_warn "Failed to codesign nacos-cli (may be blocked by Gatekeeper): $BIN_DIR/$target_binary_name"
            fi
        else
            print_warn "codesign not found; nacos-cli may be blocked by Gatekeeper"
        fi
    fi

    # Cleanup
    rm -rf "$tmp_dir"

    print_success "nacos-cli $version installed to $BIN_DIR/$target_binary_name"
}

# ============================================================================
# PATH: ensure ~/.nacos/bin is on PATH (shell rc + optional current session)
# ============================================================================

# Resolve which rc file to append the PATH line to (macOS/Linux differ for bash).
_resolve_shell_rc_for_path() {
    local shell_config=""
    if [ -n "$SHELL" ]; then
        case "$SHELL" in
            */zsh)
                shell_config="$HOME/.zshrc"
                ;;
            */bash)
                if [ -f "$HOME/.bashrc" ]; then
                    shell_config="$HOME/.bashrc"
                elif [ -f "$HOME/.bash_profile" ]; then
                    shell_config="$HOME/.bash_profile"
                elif [ -f "$HOME/.profile" ]; then
                    shell_config="$HOME/.profile"
                else
                    shell_config="$HOME/.bashrc"
                fi
                ;;
        esac
    fi

    if [ -z "$shell_config" ]; then
        if [ -f "$HOME/.zshrc" ]; then
            shell_config="$HOME/.zshrc"
        elif [ -f "$HOME/.bashrc" ]; then
            shell_config="$HOME/.bashrc"
        elif [ -f "$HOME/.bash_profile" ]; then
            shell_config="$HOME/.bash_profile"
        elif [ -f "$HOME/.profile" ]; then
            shell_config="$HOME/.profile"
        else
            shell_config="$HOME/.bashrc"
        fi
    fi
    printf '%s' "$shell_config"
}

# True only when this script was sourced (e.g. source nacos-installer.sh), so export
# affects the caller's shell. When run as "bash nacos-installer.sh", this is false:
# a child process cannot change the parent interactive shell's environment.
_nacos_installer_can_affect_calling_shell() {
    [[ -n "${BASH_VERSION:-}" ]] && [[ "${BASH_SOURCE[0]}" != "${0}" ]]
}

ensure_nacos_bin_in_path() {
    # Check if BIN_DIR is already in PATH
    case ":$PATH:" in
        *":$BIN_DIR:"*)
            print_info "$BIN_DIR is already in PATH"
            ;;
        *)
            print_info "Configuring PATH automatically..."

            local shell_config
            shell_config=$(_resolve_shell_rc_for_path)

            # Check if the export line already exists in the config file (idempotent)
            local path_export_line='export PATH="$HOME/.nacos/bin:$PATH"'
            if grep -qF "$path_export_line" "$shell_config" 2>/dev/null; then
                print_info "PATH already configured in $shell_config"
            else
                echo "" >> "$shell_config"
                echo "# Added by nacos-setup installer" >> "$shell_config"
                echo "$path_export_line" >> "$shell_config"
                print_success "PATH configured in $shell_config"
            fi

            # Ask user whether to show / apply PATH for the current terminal
            # With set -e, read must not fail the script on EOF (e.g. curl | bash).
            REPLY=""
            read -r -p "Show command to use nacos-cli in this terminal now? (Y/n): " REPLY || REPLY=y
            echo ""
            if [[ "$REPLY" =~ ^[Nn]$ ]]; then
                print_info "To use nacos-cli later in this shell, run: source $shell_config"
                print_info "Or open a new terminal (login shells load the file above)."
            else
                if _nacos_installer_can_affect_calling_shell; then
                    export PATH="$HOME/.nacos/bin:$PATH"
                    print_success "PATH has been activated in the current shell."
                else
                    print_info "This installer runs as a subprocess; it cannot change your interactive shell's PATH."
                    print_info "Run one of the following in this terminal, then use nacos-cli:"
                    echo "  source $shell_config"
                    echo "  export PATH=\"\$HOME/.nacos/bin:\$PATH\""
                fi
            fi
            ;;
    esac
}

# ============================================================================
# Verification
# ============================================================================

verify_installation() {
    print_info "Verifying installation..."
    
    # Check if the symlink or file exists (use -e for both files and symlinks)
    if [ ! -e "$BIN_DIR/$SCRIPT_NAME" ]; then
        print_error "Installation failed: $BIN_DIR/$SCRIPT_NAME not found"
        return 1
    fi
    
    # Check if the symlink target exists (resolve and check the actual target)
    if [ -L "$BIN_DIR/$SCRIPT_NAME" ]; then
        local link_path="$BIN_DIR/$SCRIPT_NAME"
        # Follow the symlink to check if target is accessible
        if [ ! -e "$link_path" ]; then
            local target=$(readlink "$link_path")
            print_error "Installation failed: Broken symlink at $link_path"
            print_error "Target does not exist: $target"
            return 1
        fi
    fi
    
    ensure_nacos_bin_in_path
    
    print_success "Installation verified successfully!"
    echo ""
    
    return 0
}


# ============================================================================
# Post-installation Info
# ============================================================================

print_usage_info() {
    local version="${INSTALLED_VERSION:-unknown}"
    local install_location="unknown"
    if [ -L "$INSTALL_BASE_DIR/$CURRENT_LINK" ]; then
        install_location="$INSTALL_BASE_DIR/$(readlink "$INSTALL_BASE_DIR/$CURRENT_LINK")"
    fi

    local cli_status="not installed"
    if [ -x "$BIN_DIR/nacos-cli" ]; then
        cli_status="installed"
    fi

    echo "========================================"
    echo "  Nacos Setup Installation Complete"
    echo "========================================"
    echo ""
    echo "nacos-setup version: $version"
    echo "nacos-cli: $cli_status"
    echo "Installation location: $install_location"
    echo ""
    echo "Quick Start:"
    echo ""
    echo "  # Show help"
    echo "  $SCRIPT_NAME --help"
    echo ""
    echo "  # Install Nacos standalone"
    echo "  $SCRIPT_NAME -v 3.1.1"
    echo ""
    echo "  # Install Nacos cluster"
    echo "  $SCRIPT_NAME -c prod -n 3"
    echo ""
    echo "  # Configure datasource"
    echo "  $SCRIPT_NAME --datasource-conf"
    echo ""
    echo "Documentation: https://nacos.io"
    echo ""
    echo "========================================"
}

# ============================================================================
# Version Check
# ============================================================================

check_installed_version() {
    # Read active version from current symlink
    if [ -L "$INSTALL_BASE_DIR/$CURRENT_LINK" ]; then
        local target=$(readlink "$INSTALL_BASE_DIR/$CURRENT_LINK")
        local active_dir="$INSTALL_BASE_DIR/$target"
        if [ -f "$active_dir/.version" ]; then
            local version=$(cat "$active_dir/.version")
            print_info "Installed nacos-setup version: $version"
            print_info "Installation location: $active_dir"
            return 0
        fi
    fi

    print_warn "nacos-setup is not installed or version information not found"
    return 1
}

# ============================================================================
# Uninstallation
# ============================================================================

uninstall_nacos_setup() {
    print_info "Uninstalling nacos-setup (active version)..."

    # If current symlink exists, remove the target directory
    if [ -L "$INSTALL_BASE_DIR/$CURRENT_LINK" ]; then
        local target=$(readlink "$INSTALL_BASE_DIR/$CURRENT_LINK")
        local target_dir="$INSTALL_BASE_DIR/$target"
        if [ -d "$target_dir" ]; then
            rm -rf "$target_dir"
            print_success "Removed $target_dir"
        fi

        # Remove current symlink
        rm -f "$INSTALL_BASE_DIR/$CURRENT_LINK"
        print_success "Removed $INSTALL_BASE_DIR/$CURRENT_LINK"
    else
        print_warn "No active installation found at $INSTALL_BASE_DIR/$CURRENT_LINK"
    fi

    # Remove global command (nacos-setup symlink)
    if [ -L "$BIN_DIR/$SCRIPT_NAME" ] || [ -f "$BIN_DIR/$SCRIPT_NAME" ]; then
        rm -f "$BIN_DIR/$SCRIPT_NAME"
        print_success "Removed $BIN_DIR/$SCRIPT_NAME"
    fi

    # Remove installer symlink in /usr/local/bin if it points to our nacos-setup
    if [ -L /usr/local/bin/nacos-setup ]; then
        local link_tgt
        link_tgt=$(readlink /usr/local/bin/nacos-setup 2>/dev/null || true)
        if [[ "$link_tgt" == "$BIN_DIR/$SCRIPT_NAME" ]]; then
            rm -f /usr/local/bin/nacos-setup
            print_success "Removed /usr/local/bin/nacos-setup"
        fi
    fi

    # Remove installer symlink in /usr/local/bin if it points to our nacos-cli
    if [ -L /usr/local/bin/nacos-cli ]; then
        local link_tgt
        link_tgt=$(readlink /usr/local/bin/nacos-cli 2>/dev/null || true)
        if [[ "$link_tgt" == "$BIN_DIR/nacos-cli" ]]; then
            rm -f /usr/local/bin/nacos-cli
            print_success "Removed /usr/local/bin/nacos-cli"
        fi
    fi

    # Remove nacos-cli binary
    if [ -f "$BIN_DIR/nacos-cli" ]; then
        rm -f "$BIN_DIR/nacos-cli"
        print_success "Removed $BIN_DIR/nacos-cli"
    fi

    # Note: ~/.nacos parent directory is intentionally preserved
    # as it may contain cache and other user data

    print_success "Uninstallation completed!"
    echo ""
}

# ============================================================================
# Main
# ============================================================================

main() {
    echo ""
    echo "========================================"
    echo "  Nacos Setup Installer"
    echo "========================================"
    echo ""
    echo "    curl -fsSL https://nacos.io/nacos-installer.sh | bash"
    echo ""
    echo "========================================"
    echo ""

    # Initialize versions (fetch from remote or use fallback)
    get_all_versions 1
    print_info "Versions: CLI=$NACOS_CLI_VERSION, Setup=$NACOS_SETUP_VERSION, Server=$NACOS_SERVER_VERSION" >&2
    echo ""

    # Parse arguments
    local only_cli=false
    local cli_version=""
    local setup_version=""

    while [[ $# -gt 0 ]]; do
        case "$1" in
            version|--version)
                check_installed_version
                exit $?
                ;;
            uninstall|--uninstall|-u)
                uninstall_nacos_setup
                exit 0
                ;;
            --cli)
                only_cli=true
                shift
                ;;
            -v|--version)
                if [ -z "$2" ] || [[ "$2" == -* ]]; then
                    print_error "Option $1 requires a version number"
                    echo ""
                    print_info "Usage: ./nacos-installer.sh -v <version>"
                    print_info "        ./nacos-installer.sh --cli -v <version>"
                    exit 1
                fi
                # 根据模式决定版本类型
                if [[ "$only_cli" == true ]]; then
                    cli_version="$2"
                else
                    setup_version="$2"
                fi
                shift 2
                ;;
            --help|-h)
                echo "Install nacos-setup and nacos-cli tools for managing Nacos instances."
                echo ""
                echo "Usage:"
                echo "    curl -fsSL https://nacos.io/nacos-installer.sh | bash"
                echo ""
                echo "Options:"
                echo "  (none)              Install nacos-setup + nacos-cli (default)"
                echo "  -v, --version       Specify version (nacos-setup or nacos-cli with --cli)"
                echo "  --cli               Install nacos-cli only"
                echo "  version             Show installed version"
                echo "  uninstall, -u       Uninstall nacos-setup"
                echo "  --help, -h          Show this help message"
                echo ""
                echo "Examples:"
                echo "  ./nacos-installer.sh                    Install nacos-setup + nacos-cli"
                echo "  ./nacos-installer.sh -v 0.0.3           Install nacos-setup v0.0.3 + nacos-cli"
                echo "  ./nacos-installer.sh --cli              Install nacos-cli only"
                echo "  ./nacos-installer.sh --cli -v 0.0.3     Install nacos-cli v0.0.3 only"
                echo ""
                echo "After installation, use 'nacos-setup' command to manage Nacos:"
                echo "  nacos-setup --help              Show nacos-setup help"
                echo "  nacos-setup -v 3.1.1            Install Nacos standalone"
                echo "  nacos-setup -c prod -n 3        Install Nacos cluster"
                echo ""
                exit 0
                ;;
            *)
                print_error "Unknown option: $1"
                echo ""
                print_info "Use --help for usage information"
                exit 1
                ;;
        esac
    done

    # Apply CLI version if specified
    if [ -n "$cli_version" ]; then
        NACOS_CLI_VERSION="$cli_version"
    fi
    
    # Check requirements
    # If only installing CLI, only check bin directory
    # Otherwise check full installation requirements
    local check_mode="full"
    if [[ "$only_cli" == true ]]; then
        check_mode="onlycli"
    fi
    
    if ! check_requirements "$check_mode"; then
        print_error "Requirements check failed"
        print_info "Please install the missing dependencies and try again."
        exit 1
    fi

    if [[ "$only_cli" == true ]]; then
        echo ""
        if ! install_nacos_cli; then
            exit 1
        fi
        echo ""
        ensure_nacos_bin_in_path
        print_info "After PATH is loaded in your shell (see above), run: nacos-cli --help"
        exit 0
    fi

    # Install nacos-setup
    install_nacos_setup "$setup_version"

    # Verify nacos-setup installation
    if ! verify_installation; then
        print_error "Installation verification failed"
        exit 1
    fi

    # Install nacos-cli (bundled by default)
    echo ""
    if ! install_nacos_cli; then
        print_warn "nacos-cli installation failed, but nacos-setup is ready"
    fi

    # Print usage info after all installations
    print_usage_info

    # Always offer to install Nacos Server
    echo ""
    # Use server version from versions file or fallback
    detected_default_version="${NACOS_SERVER_VERSION:-$FALLBACK_NACOS_SERVER_VERSION}"

    REPLY=""
    read -r -p "Do you want to install Nacos $detected_default_version now? (Y/n): " REPLY || REPLY=y
    echo ""
    if [[ "$REPLY" =~ ^[Yy]?$ ]] || [[ -z "$REPLY" ]]; then
        print_info "Installing Nacos $detected_default_version..."
        # Always use absolute path to ensure it works even if PATH is not yet loaded
        "$BIN_DIR/$SCRIPT_NAME" -v "$detected_default_version"
    else
        print_info "Skipping Nacos installation."
        print_info "To install later, run: $SCRIPT_NAME -v $detected_default_version"
        print_info "Or use absolute path: $BIN_DIR/$SCRIPT_NAME -v $detected_default_version"
    fi

    exit 0
}

# Run main
main "$@"