setup.sh

#!/bin/bash
set -euo pipefail

PROJECT_DIR="$(cd "$(dirname "$0")" && pwd)"
ENV_FILE="$PROJECT_DIR/.env"

if [ ! -f "$ENV_FILE" ]; then
    echo "No .env file found. Let's set one up."
    echo ""

    read -rp "Your quay fork URL (e.g. git@github.com:youruser/quay.git): " QUAY_REPO_URL
    read -rp "GCP Vertex AI project ID: " ANTHROPIC_VERTEX_PROJECT_ID
    read -rp "GCP region [global]: " CLOUD_ML_REGION
    CLOUD_ML_REGION="${CLOUD_ML_REGION:-global}"

    cat > "$ENV_FILE" <<EOF
QUAY_REPO_URL=${QUAY_REPO_URL}
ANTHROPIC_VERTEX_PROJECT_ID=${ANTHROPIC_VERTEX_PROJECT_ID}
CLOUD_ML_REGION=${CLOUD_ML_REGION}
EOF

    echo ""
    echo "Saved to .env"
fi

source "$ENV_FILE"

for var in QUAY_REPO_URL ANTHROPIC_VERTEX_PROJECT_ID CLOUD_ML_REGION; do
    if [ -z "${!var:-}" ]; then
        echo "Error: $var is not set in .env"
        exit 1
    fi
done

SSH_DIR="$PROJECT_DIR/.quaydev-ssh"
SSH_KEY="$SSH_DIR/id_ed25519"

if [ ! -f "$SSH_KEY" ]; then
    read -rp "Generate SSH key pair for the VM? [Y/n]: " gen_key
    if [[ "${gen_key:-Y}" =~ ^[Yy]$ ]]; then
        mkdir -p "$SSH_DIR"
        ssh-keygen -t ed25519 -f "$SSH_KEY" -C "quaydev-vm" -N ""
        echo ""
        echo "Add this public key to https://github.com/settings/keys:"
        echo ""
        cat "${SSH_KEY}.pub"
        echo ""
    fi
fi

sed \
    -e "s|__QUAYDEV_DIR__|${PROJECT_DIR}|g" \
    -e "s|__HOME__|${HOME}|g" \
    -e "s|__QUAY_REPO_URL__|${QUAY_REPO_URL}|g" \
    -e "s|__VERTEX_PROJECT_ID__|${ANTHROPIC_VERTEX_PROJECT_ID}|g" \
    -e "s|__CLOUD_ML_REGION__|${CLOUD_ML_REGION}|g" \
    "$PROJECT_DIR/quaydev.yaml.template" > "$PROJECT_DIR/quaydev.yaml"

ORANGE='\033[38;5;208m'
BOLD='\033[1m'
RESET='\033[0m'

echo "Generated quaydev.yaml"
echo ""
echo -e "${BOLD}${ORANGE}>>> Add your SSH public key to https://github.com/settings/keys before starting the VM <<<${RESET}"
echo ""
echo "Next steps:"
echo "  limactl create --name=quaydev quaydev.yaml"
echo "  limactl start quaydev"