#!/bin/bash
# Configure API Keys for DevMatrix
# Supports: Kimi (API key) and Antigravity/Gemini (OAuth)

echo "🔐 DevMatrix API Key Configuration"
echo "==================================="
echo ""

# ============================================
# KIMI API KEY
# ============================================
echo "📍 Kimi Configuration (Primary AI)"
echo "Get your API key from: https://kimi.moonshot.cn/"
echo ""
echo -n "Enter your Kimi API Key: "
read -s KIMI_KEY
echo ""

if [ -n "$KIMI_KEY" ]; then
    mkdir -p ~/.openclaw/config
    
    cat > ~/.openclaw/config/kimi.yaml << EOF
provider: kimi
name: "Kimi K2.5"
auth:
  type: api_key
  key: "$KIMI_KEY"
models:
  primary:
    id: "k2p5"
    max_tokens: 128000
  fallback:
    id: "k1.5"
    max_tokens: 64000
routing:
  priority: 1
EOF
    
    chmod 600 ~/.openclaw/config/kimi.yaml
    echo "✅ Kimi configured"
else
    echo "⚠️  Kimi key not provided (you can add later)"
fi

echo ""

# ============================================
# ANTIGRAVITY / GEMINI (Optional)
# ============================================
echo "📍 Antigravity / Gemini Configuration (Optional Fallback)"
echo ""
echo "Antigravity and Gemini use OAuth (not API keys)."
echo ""
echo "Choose an option:"
echo "  1) Set up Gemini CLI (free, recommended)"
echo "  2) Set up Antigravity (if you have access)"
echo "  3) Skip for now (Kimi only)"
echo ""
echo -n "Choice (1/2/3): "
read CHOICE

case $CHOICE in
    1)
        echo ""
        echo "Setting up Gemini CLI..."
        echo "This will open a browser for OAuth login."
        echo ""
        openclaw plugins enable google-gemini-cli-auth 2>/dev/null || true
        openclaw models auth login --provider google-gemini-cli --set-default
        echo "✅ Gemini CLI configured"
        ;;
    2)
        echo ""
        echo "Setting up Antigravity..."
        echo "This will open a browser for OAuth login."
        echo ""
        openclaw plugins enable google-antigravity-auth 2>/dev/null || true
        openclaw models auth login --provider google-antigravity --set-default
        echo "✅ Antigravity configured"
        ;;
    3|*)
        echo "⚠️  Skipping fallback AI (Kimi only)"
        ;;
esac

echo ""
echo "==================================="
echo "✅ Configuration complete!"
echo ""
echo "Test with: openclaw models list"
echo ""