Databricks MCP Server

#!/bin/bash set -e # Check for --auto-close flag and temp file AUTO_CLOSE=false TEMP_FILE="" if [ "$1" = "--auto-close" ]; then AUTO_CLOSE=true TEMP_FILE="$2" fi # Function to signal completion on exit cleanup() { if [ "$AUTO_CLOSE" = true ] && [ -n "$TEMP_FILE" ]; then echo "setup_complete" > "$TEMP_FILE" fi } # Set trap to call cleanup on exit if [ "$AUTO_CLOSE" = true ]; then trap cleanup EXIT fi echo "🚀 Databricks App Template Setup" echo "=================================" # Function to prompt for input with default value prompt_with_default() { local prompt="$1" local default="$2" local var_name="$3" read -p "$prompt [$default]: " input if [ -z "$input" ]; then input="$default" fi # Set the variable dynamically eval "$var_name='$input'" } # Function to update or add a value in .env.local update_env_value() { local key="$1" local value="$2" local comment="$3" if grep -q "^${key}=" .env.local 2>/dev/null; then # Update existing value if [[ "$OSTYPE" == "darwin"* ]]; then # macOS sed sed -i '' "s|^${key}=.*|${key}=${value}|" .env.local else # Linux sed sed -i "s|^${key}=.*|${key}=${value}|" .env.local fi else # Add new value with comment if provided if [ -n "$comment" ]; then echo "" >> .env.local echo "# $comment" >> .env.local fi echo "${key}=${value}" >> .env.local fi } # Function to test databricks connection test_databricks_connection() { local profile="$1" echo "🔍 Testing Databricks connection..." # Ensure environment variables are exported for databricks CLI if [ -n "$DATABRICKS_HOST" ] && [ -n "$DATABRICKS_TOKEN" ]; then export DATABRICKS_HOST export DATABRICKS_TOKEN fi if [ -n "$profile" ]; then if databricks current-user me --profile "$profile" >/dev/null 2>&1; then echo "✅ Successfully connected to Databricks with profile '$profile'" return 0 else echo "❌ Failed to connect to Databricks with profile '$profile'" return 1 fi else if databricks current-user me >/dev/null 2>&1; then echo "✅ Successfully connected to Databricks" return 0 else echo "❌ Failed to connect to Databricks" return 1 fi fi } # Check if .env.local already exists if [ -f ".env.local" ]; then echo "📋 Found existing .env.local file." read -p "Do you want to update it? (y/N): " update_env if [[ ! "$update_env" =~ ^[Yy]$ ]]; then echo "Skipping environment configuration." skip_env=true fi fi if [ "$skip_env" != "true" ]; then echo "⚙️ Setting up environment variables..." # Load existing values if they exist if [ -f ".env.local" ]; then source .env.local 2>/dev/null || true fi # Initialize .env.local file if it doesn't exist if [ ! -f ".env.local" ]; then echo "# Databricks App Configuration" > .env.local echo "# Generated by setup script on $(date)" >> .env.local echo "" >> .env.local fi # Databricks Authentication Configuration echo "" echo "🔐 Databricks Authentication" echo "-----------------------------" echo "Choose authentication method:" echo "1. Personal Access Token (PAT)" echo "2. Configuration Profile" echo "" # Pre-select based on existing configuration if [ "$DATABRICKS_AUTH_TYPE" = "pat" ]; then default_choice="1" elif [ "$DATABRICKS_AUTH_TYPE" = "profile" ]; then default_choice="2" else default_choice="" fi if [ -n "$default_choice" ]; then prompt_with_default "Select option" "$default_choice" "auth_choice" else read -p "Select option (1 or 2): " auth_choice fi if [ "$auth_choice" = "1" ]; then # PAT Authentication echo "" echo "📝 Personal Access Token Setup" echo "-------------------------------" # Update auth type in .env.local update_env_value "DATABRICKS_AUTH_TYPE" "pat" "Databricks Authentication Type" if [ -z "$DATABRICKS_HOST" ]; then prompt_with_default "Databricks Host" "https://your-workspace.cloud.databricks.com" "DATABRICKS_HOST" else prompt_with_default "Databricks Host" "$DATABRICKS_HOST" "DATABRICKS_HOST" fi # Update host in .env.local update_env_value "DATABRICKS_HOST" "$DATABRICKS_HOST" "Databricks Configuration (PAT mode)" if [ -n "$DATABRICKS_TOKEN" ]; then echo "Found existing token: ${DATABRICKS_TOKEN:0:10}... (truncated)" read -p "Use existing token? (y/N): " use_existing if [[ ! "$use_existing" =~ ^[Yy]$ ]]; then DATABRICKS_TOKEN="" fi fi if [ -z "$DATABRICKS_TOKEN" ]; then echo "" echo "You can create a Personal Access Token here:" echo "📖 $DATABRICKS_HOST/settings/user/developer/access-tokens" echo "" read -s -p "Databricks Personal Access Token: " DATABRICKS_TOKEN echo "" fi # Update token in .env.local update_env_value "DATABRICKS_TOKEN" "$DATABRICKS_TOKEN" # Set empty profile to indicate PAT mode DATABRICKS_CONFIG_PROFILE="" DATABRICKS_AUTH_TYPE="pat" # Test PAT authentication echo "🔍 Testing PAT authentication..." export DATABRICKS_HOST="$DATABRICKS_HOST" export DATABRICKS_TOKEN="$DATABRICKS_TOKEN" echo "Attempting to connect with:" echo "Host: $DATABRICKS_HOST" echo "Token: ${DATABRICKS_TOKEN:0:10}... (truncated)" echo "" # Test connection and show output for debugging echo "Running: databricks current-user me" # Export for databricks CLI export DATABRICKS_HOST="$DATABRICKS_HOST" export DATABRICKS_TOKEN="$DATABRICKS_TOKEN" if databricks current-user me >/dev/null 2>&1; then echo "✅ Successfully connected to Databricks with PAT" else echo "" echo "❌ PAT authentication failed." echo "Please check your host URL and token are correct." echo "You can test manually with:" echo "DATABRICKS_HOST='$DATABRICKS_HOST' DATABRICKS_TOKEN='$DATABRICKS_TOKEN' databricks current-user me" exit 1 fi elif [ "$auth_choice" = "2" ]; then # Profile Authentication echo "" echo "📋 Configuration Profile Setup" echo "-------------------------------" # Update auth type in .env.local update_env_value "DATABRICKS_AUTH_TYPE" "profile" "Databricks Authentication Type" # List existing profiles echo "Available profiles:" if [ -f "$HOME/.databrickscfg" ]; then grep '^\[' "$HOME/.databrickscfg" | sed 's/\[//g' | sed 's/\]//g' | sed 's/^/ - /' else echo " No existing profiles found" fi echo "" prompt_with_default "Databricks Config Profile" "${DATABRICKS_CONFIG_PROFILE:-DEFAULT}" "DATABRICKS_CONFIG_PROFILE" # Update profile in .env.local update_env_value "DATABRICKS_CONFIG_PROFILE" "$DATABRICKS_CONFIG_PROFILE" "Databricks Configuration (Profile mode)" # Clear PAT credentials when using profile update_env_value "DATABRICKS_HOST" "" update_env_value "DATABRICKS_TOKEN" "" DATABRICKS_HOST="" DATABRICKS_TOKEN="" DATABRICKS_AUTH_TYPE="profile" # Test profile authentication if ! test_databricks_connection "$DATABRICKS_CONFIG_PROFILE"; then echo "" echo "Profile '$DATABRICKS_CONFIG_PROFILE' not found or invalid." echo "Would you like to configure it now? (y/N)" read -p "> " configure_profile if [[ "$configure_profile" =~ ^[Yy]$ ]]; then echo "Running 'databricks configure --profile $DATABRICKS_CONFIG_PROFILE'..." databricks configure --profile "$DATABRICKS_CONFIG_PROFILE" # Test again after configuration if ! test_databricks_connection "$DATABRICKS_CONFIG_PROFILE"; then echo "❌ Profile configuration failed. Please check your settings." exit 1 fi else echo "❌ Valid Databricks authentication is required for deployment." exit 1 fi fi else echo "❌ Invalid option. Please run setup again." exit 1 fi # Get current user information (only if we don't have existing DBA_SOURCE_CODE_PATH) if [ -z "$DBA_SOURCE_CODE_PATH" ]; then echo "" echo "🔍 Getting user information..." if [ "$DATABRICKS_AUTH_TYPE" = "profile" ]; then DATABRICKS_USER=$(databricks current-user me --profile "$DATABRICKS_CONFIG_PROFILE" --output json 2>/dev/null | grep -o '"userName":"[^"]*"' | cut -d'"' -f4) else DATABRICKS_USER=$(databricks current-user me --output json 2>/dev/null | grep -o '"userName":"[^"]*"' | cut -d'"' -f4) fi if [ -n "$DATABRICKS_USER" ]; then echo "✅ Detected user: $DATABRICKS_USER" else echo "⚠️ Could not detect user, will use default email" fi else echo "" echo "✅ Using existing configuration from .env.local" fi # App Configuration echo "" echo "🚀 App Configuration" echo "--------------------" echo "If you haven't created a Databricks App yet, you can create a custom app from the UI:" if [ "$DATABRICKS_AUTH_TYPE" = "profile" ]; then WORKSPACE_HOST=$(databricks current-user me --profile "$DATABRICKS_CONFIG_PROFILE" --output json 2>/dev/null | grep -o '"workspaceUrl":"[^"]*"' | cut -d'"' -f4) else WORKSPACE_HOST="$DATABRICKS_HOST" fi if [ -n "$WORKSPACE_HOST" ]; then echo "📖 $WORKSPACE_HOST/apps/create" else echo "📖 https://your-workspace.cloud.databricks.com/apps/create" fi echo "" prompt_with_default "App Name for Deployment" "${DATABRICKS_APP_NAME:-my-databricks-app}" "DATABRICKS_APP_NAME" # Update the default source path to use the chosen app name if [ -z "$DBA_SOURCE_CODE_PATH" ]; then if [ -n "$DATABRICKS_USER" ]; then DEFAULT_SOURCE_PATH="/Workspace/Users/$DATABRICKS_USER/$DATABRICKS_APP_NAME" else DEFAULT_SOURCE_PATH="/Workspace/Users/<your-email@company.com>/$DATABRICKS_APP_NAME" fi else DEFAULT_SOURCE_PATH="$DBA_SOURCE_CODE_PATH" fi prompt_with_default "Source Code Path for Deployment" "$DEFAULT_SOURCE_PATH" "DBA_SOURCE_CODE_PATH" # Update app configuration in .env.local update_env_value "DATABRICKS_APP_NAME" "$DATABRICKS_APP_NAME" "Databricks App Configuration" update_env_value "DBA_SOURCE_CODE_PATH" "$DBA_SOURCE_CODE_PATH" # MCP Server Configuration echo "" echo "🔧 MCP Server Configuration" echo "---------------------------" # Load current servername from config.yaml if it exists CURRENT_SERVERNAME="" if [ -f "config.yaml" ]; then CURRENT_SERVERNAME=$(grep "servername:" config.yaml 2>/dev/null | cut -d':' -f2 | sed 's/^ *//' | sed 's/ *$//') fi if [ -z "$CURRENT_SERVERNAME" ]; then CURRENT_SERVERNAME="databricks-mcp" fi prompt_with_default "MCP Server Name" "$CURRENT_SERVERNAME" "SERVERNAME" # Create or update config.yaml echo "# MCP Server Configuration" > config.yaml echo "servername: $SERVERNAME" >> config.yaml echo "✅ MCP server configuration saved to config.yaml" echo "" echo "✅ Environment configuration saved to .env.local" fi # Check for required tools echo "" echo "🔧 Checking dependencies..." # Check for databricks CLI if ! command -v databricks &> /dev/null; then echo "❌ Databricks CLI not found. Please install it first:" echo " Visit: https://docs.databricks.com/dev-tools/cli/install.html" echo " Or run: pip install databricks-cli" exit 1 fi # Check for uv if ! command -v uv &> /dev/null; then echo "❌ uv not found. Installing uv..." curl -LsSf https://astral.sh/uv/install.sh | sh source $HOME/.local/bin/env fi # Check for bun if ! command -v bun &> /dev/null; then echo "❌ bun not found. Installing bun..." curl -fsSL https://bun.sh/install | bash source ~/.bashrc fi echo "✅ All required tools are available" # Install Claude MCP Playwright (if Claude Code is available) echo "" echo "🎭 Installing Claude MCP Playwright..." if command -v claude &> /dev/null; then claude mcp add playwright npx '@playwright/mcp@latest' echo "✅ Claude MCP Playwright installed" else echo "⚠️ Claude Code CLI not found, skipping MCP Playwright installation" echo "💡 You can install it later with: claude mcp add playwright npx '@playwright/mcp@latest'" fi # Install Python dependencies echo "📦 Installing Python dependencies..." uv sync --dev # Install frontend dependencies echo "📦 Installing frontend dependencies..." cd client bun install cd .. echo "" echo "🎉 Setup complete!" echo "" echo "⚠️ IMPORTANT: Please restart Claude Code to enable MCP Playwright integration" echo "" echo "Next steps:" echo "1. Restart Claude Code (close and reopen the application)" echo "2. Run './watch.sh' to start the development servers" echo "3. Open http://localhost:3000 to view the app" echo "4. Open http://localhost:8000/docs to view the API documentation" echo "" echo "Optional:" echo "- Run './fix.sh' to format your code" echo "- Edit .env.local to update configuration" # Auto-close terminal if flag is set if [ "$AUTO_CLOSE" = true ]; then echo "" echo "Press Enter to close this terminal..." read # Close appropriate terminal app if [ -d "/Applications/iTerm.app" ]; then # For iTerm, close the current window osascript -e 'tell application "iTerm" to close current window' else # For Terminal, close windows containing setup.sh osascript -e 'tell application "Terminal" to close (every window whose name contains "setup.sh")' fi fi