setup.sh

#!/bin/bash
set -euo pipefail
BASE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$BASE_DIR"

[[ -f .env ]] || cp .env.example .env
source .env

install_uv() {
    if ! command -v uv &> /dev/null; then
        echo "Installing uv..."
        curl -LsSf https://install.python.dev/uv | sh
        export PATH="$HOME/.local/bin:$PATH"
    fi
}

install_deps() {
    sudo apt update
    sudo apt install -y curl git docker.io openjdk-17-jre-headless
    sudo usermod -aG docker $USER
}

start_services() {
    curl -fsSL https://ollama.com/install.sh | sh
    sudo systemctl enable --now ollama
    sudo mkdir -p /etc/systemd/system/ollama.service.d
    sudo tee /etc/systemd/system/ollama.service.d/override.conf <<< "[Service]
Environment=\"OLLAMA_HOST=0.0.0.0:11434\""
    sudo systemctl daemon-reload && sudo systemctl restart ollama

    docker run -d --name neo4j --restart unless-stopped \
        -p 7474:7474 -p 7687:7687 \
        -v "$BASE_DIR/neo4j_data:/data" \
        -e NEO4J_AUTH=neo4j/offline-ai-rules \
        neo4j:5.24.0

    docker run -d --name qdrant --restart unless-stopped \
        -p 6333:6333 -p 6334:6334 \
        -v "$BASE_DIR/qdrant_storage:/qdrant/storage" \
        qdrant/qdrant:latest
}

select_model() {
    if [[ -f .selected-model ]]; then
        cat .selected-model
    else
        RAM=$(free -g | awk '/^Mem:/{print $2}')
        GPU=$(lspci | grep -Ei "nvidia|amd" || true)
        if [[ $RAM -ge 32 && "$GPU" =~ NVIDIA ]]; then
            echo "qwen2.5-coder:32b-q4_K_M"
        elif [[ $RAM -ge 16 ]]; then
            echo "qwen2.5-coder:14b-q4_K_M"
        else
            echo "qwen2.5-coder:7b-q4_K_M"
        fi
    fi
}

case "${1:-}" in
    "")
        install_uv
        install_deps
        start_services
        select_model > .selected-model
        ollama pull $(select_model)

        echo "Setting up uv environment..."
        uv venv .venv
        source .venv/bin/activate
        uv pip install -r template/requirements.txt
        uv pip compile template/requirements.txt -o uv.lock

        echo "Setup complete!"
        echo "Next: ./src/create-project.sh my-app"
        ;;
    "create-project") shift; ./src/create-project.sh "$@" ;;
    "run") ./src/run-project.sh ;;
    *) echo "Usage: $0 [create-project name | run]" ;;
esac