leax_cli

#!/bin/bash

CONFIG_FILE="$HOME/.leaxrc"
OS="$(uname -s)"

# ANSI color codes
GREEN="\033[38;5;49m"
YELLOW="\033[38;5;214m"
BLUE="\033[38;5;31m"
RED="\033[38;5;196m"
RESET="\033[0m"

# Handle 'leax configure' command
if [ "$1" = "configure" ]; then
    echo ""
    read -p "$(echo -e "${YELLOW} - ${RESET}Enter your Mistral API key: ")" api_key

    if [ -z "$api_key" ]; then
        echo -e " ${RED}×${RESET} Error: API key cannot be empty"
        exit 1
    fi

    echo "MISTRAL_API_KEY=$api_key" > "$CONFIG_FILE"
    chmod 600 "$CONFIG_FILE"

    echo ""
    echo -e "${GREEN} ✓ ${RESET} Configuration saved to $CONFIG_FILE"
    echo -e "${GREEN} ✓ ${RESET} You can now use leax from anywhere!"
    echo -e "${GREEN} ✓ ${RESET} Usage: leax ./<executable> [args...]"
     echo ""

    exit 0
fi

# Normal leax usage
if [ $# -lt 1 ]; then
    echo "Usage: leax <executable> [args...]"
    echo "       leax configure"
    echo ""
    echo "Examples:"
    echo "  leax ./my_program"
    echo "  leax ./my_program arg1 arg2"
    echo "  leax configure              # Configure Mistral API key"
    exit 1
fi

EXECUTABLE="$1"
shift
ARGS=("$@")

# Verify executable exists
if [ ! -f "$EXECUTABLE" ]; then
    echo "Error: Executable '$EXECUTABLE' not found"
    exit 1
fi

if [ ! -x "$EXECUTABLE" ]; then
    echo "Warning: '$EXECUTABLE' is not executable, attempting to make it executable..."
    chmod +x "$EXECUTABLE"
fi

# Get absolute path of current directory
CURRENT_DIR=$(pwd)

# Find .env file by traversing up the directory tree (fallback)
find_env_file() {
    local dir="$1"
    local levels=0

    while [ "$levels" -lt 5 ]; do
        if [ -f "$dir/.env" ]; then
            if grep -q "MISTRAL_API_KEY" "$dir/.env" 2>/dev/null; then
                echo "$dir/.env"
                return 0
            fi
        fi

        local parent=$(dirname "$dir")
        if [ "$parent" = "$dir" ]; then
            break
        fi
        dir="$parent"
        levels=$((levels + 1))
    done

    return 1
}

# Load API key from config
load_api_key() {
    if [ -f "$CONFIG_FILE" ]; then
        export $(grep -v '^#' "$CONFIG_FILE" | xargs)
    else
        local env_file
        env_file=$(find_env_file "$CURRENT_DIR")
        if [ -n "$env_file" ]; then
            export $(grep -v '^#' "$env_file" | grep 'MISTRAL_API_KEY' | xargs)
        else
            echo "Error: No Mistral API key configured"
            echo ""
            echo "Please run: leax configure"
            echo ""
            echo "Or create a .env file with:"
            echo "  MISTRAL_API_KEY=your_api_key"
            echo ""
            exit 1
        fi
    fi
}

if [ "$OS" = "Linux" ]; then
    # ── Linux: run Python directly ──
    LEAX_SRCS="$HOME/.local/share/leax/srcs"

    if [ ! -f "$LEAX_SRCS/vex.py" ]; then
        echo "Error: Leax sources not found at $LEAX_SRCS"
        echo "Please run 'make install' from the Leax repository"
        exit 1
    fi

    load_api_key
    python3 "$LEAX_SRCS/vex.py" "$EXECUTABLE" "${ARGS[@]}"
else
    # ── macOS: run via Docker ──

    # Determine env source for --env-file
    if [ -f "$CONFIG_FILE" ]; then
        ENV_SOURCE="$CONFIG_FILE"
    else
        ENV_FILE=$(find_env_file "$CURRENT_DIR")
        if [ -n "$ENV_FILE" ]; then
            ENV_SOURCE="$ENV_FILE"
        else
            echo "Error: No Mistral API key configured"
            echo ""
            echo "Please run: leax configure"
            echo ""
            echo "Or create a .env file with:"
            echo "  MISTRAL_API_KEY=your_api_key"
            echo ""
            exit 1
        fi
    fi

    docker run -it --rm \
        --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
        --env-file "$ENV_SOURCE" \
        -v "$CURRENT_DIR:/work" \
        -w /work \
        leax /app/srcs/vex.py "$EXECUTABLE" "${ARGS[@]}" 2>/dev/null
fi