setup_max.sh

#!/bin/bash

set -e

echo "╔═══════════════════════════════════════════════════════════════╗"
echo "║              MAX Installation Script for Axion                ║"
echo "╚═══════════════════════════════════════════════════════════════╝"
echo ""

# Configuration
VENV_DIR=".venv"
PYTHON_VERSION="python3"

# Check if Python is available
if ! command -v $PYTHON_VERSION &> /dev/null; then
    echo "❌ Python 3 is not installed. Please install Python 3 first."
    exit 1
fi

echo "✓ Python found: $($PYTHON_VERSION --version)"
echo ""

# Create virtual environment if it doesn't exist
if [ ! -d "$VENV_DIR" ]; then
    echo "📦 Creating Python virtual environment..."
    $PYTHON_VERSION -m venv $VENV_DIR
    echo "✓ Virtual environment created at $VENV_DIR"
else
    echo "✓ Virtual environment already exists at $VENV_DIR"
fi
echo ""

# Activate virtual environment
echo "🔄 Activating virtual environment..."
source $VENV_DIR/bin/activate
echo "✓ Virtual environment activated"
echo ""

# Check if uv is available
if command -v uv &> /dev/null; then
    echo "🚀 Using uv for fast installation..."
    echo "📥 Installing MAX with uv..."
    uv pip install modular \
        --extra-index-url https://modular.gateway.scarf.sh/simple/
else
    echo "⚡ uv not found, using pip instead..."
    echo "💡 Tip: Install uv for faster package installation: pip install uv"
    echo ""
    echo "📥 Installing MAX with pip..."
    pip install --upgrade pip
    pip install modular \
        --extra-index-url https://modular.gateway.scarf.sh/simple/
fi

echo ""
echo "✓ MAX installation complete!"
echo ""

# Verify installation
if command -v max &> /dev/null; then
    echo "✅ MAX is installed and available in PATH"
    max --version
else
    echo "⚠️  MAX binary not found in PATH"
    echo "   You may need to add it to your PATH or restart your shell"
    echo ""
    echo "   Try running: source $VENV_DIR/bin/activate"
fi

echo ""
echo "╔═══════════════════════════════════════════════════════════════╗"
echo "║                    Setup Complete! 🎉                         ║"
echo "╚═══════════════════════════════════════════════════════════════╝"
echo ""
echo "Next steps:"
echo "  1. Activate the virtual environment:"
echo "     source $VENV_DIR/bin/activate"
echo ""
echo "  2. Start Axion server:"
echo "     cargo run --release"
echo ""
echo "  3. MAX will be automatically used when available!"
echo ""