start_dev_web_client.sh

#!/bin/bash

#--------------------------------------------------------------------------
#
# pgEdge AI DBA Workbench Web Client Development Startup Script
#
# Copyright (c) 2025 - 2026, pgEdge, Inc.
# This software is released under The PostgreSQL License
#
# Note: For production deployments, use Docker Compose instead:
#   docker-compose up
#
#--------------------------------------------------------------------------

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
BIN_DIR="$SCRIPT_DIR/bin"
CLIENT_DIR="$SCRIPT_DIR/client"

# Source directories
SERVER_SRC_DIR="$SCRIPT_DIR/server/src"

# Configuration files
SERVER_CONFIG="$BIN_DIR/ai-dba-server.yaml"
EXAMPLE_CONFIG="$SCRIPT_DIR/examples/ai-dba-server.yaml"
SERVER_BIN="$BIN_DIR/ai-dba-server"

# Log files
SERVER_LOG="/tmp/ai-dba-server.log"
VITE_LOG="/tmp/ai-dba-vite.log"

# PID files for cleanup
SERVER_PID=""
CLIENT_PID=""

# Cleanup function
cleanup() {
    echo -e "\n${YELLOW}Shutting down services...${NC}"

    if [ ! -z "$CLIENT_PID" ]; then
        echo "Stopping Vite dev server (PID: $CLIENT_PID)..."
        kill $CLIENT_PID 2>/dev/null || true
    fi

    if [ ! -z "$SERVER_PID" ]; then
        echo "Stopping MCP server (PID: $SERVER_PID)..."
        kill $SERVER_PID 2>/dev/null || true
    fi

    echo -e "${GREEN}Cleanup complete${NC}"
}

# Set up trap for cleanup
trap cleanup EXIT INT TERM

# Create bin directory if needed
mkdir -p "$BIN_DIR"

# Copy example config if no config exists
if [ ! -f "$SERVER_CONFIG" ]; then
    if [ -f "$EXAMPLE_CONFIG" ]; then
        echo -e "${YELLOW}No config found, copying example config to $SERVER_CONFIG${NC}"
        cp "$EXAMPLE_CONFIG" "$SERVER_CONFIG"
        echo -e "${YELLOW}Please edit $SERVER_CONFIG with your settings${NC}"
    else
        echo -e "${RED}Error: Example config not found at $EXAMPLE_CONFIG${NC}"
        exit 1
    fi
fi

# Build or rebuild server binary if needed
if [ ! -f "$SERVER_BIN" ]; then
    echo -e "${BLUE}Building MCP server binary...${NC}"
    cd "$SERVER_SRC_DIR"
    go build -o "$SERVER_BIN" ./cmd/mcp-server
    if [ $? -ne 0 ]; then
        echo -e "${RED}Error: Failed to build MCP server${NC}"
        exit 1
    fi
    cd "$SCRIPT_DIR"
else
    echo -e "${BLUE}Checking if server binary needs rebuilding...${NC}"
    # Check if any Go source files are newer than the binary
    if [ -n "$(find "$SERVER_SRC_DIR" -name "*.go" -newer "$SERVER_BIN" 2>/dev/null)" ]; then
        echo -e "${BLUE}Source files changed, rebuilding MCP server...${NC}"
        cd "$SERVER_SRC_DIR"
        go build -o "$SERVER_BIN" ./cmd/mcp-server
        if [ $? -ne 0 ]; then
            echo -e "${RED}Error: Failed to build MCP server${NC}"
            exit 1
        fi
        cd "$SCRIPT_DIR"
    else
        echo -e "${GREEN}Server binary is up to date${NC}"
    fi
fi

# Check if client dependencies need updating
# Platform marker tracks OS/arch to detect cross-platform node_modules
PLATFORM_MARKER="$CLIENT_DIR/node_modules/.platform_marker"
CURRENT_PLATFORM="$(uname -s)-$(uname -m)"

needs_npm_install() {
    # No node_modules directory
    if [ ! -d "$CLIENT_DIR/node_modules" ]; then
        echo "node_modules directory missing"
        return 0
    fi

    # Platform mismatch (cross-platform development)
    if [ ! -f "$PLATFORM_MARKER" ] || [ "$(cat "$PLATFORM_MARKER" 2>/dev/null)" != "$CURRENT_PLATFORM" ]; then
        echo "platform changed (native modules need rebuild)"
        return 0
    fi

    # Package files newer than node_modules
    if [ "$CLIENT_DIR/package.json" -nt "$CLIENT_DIR/node_modules" ]; then
        echo "package.json changed"
        return 0
    fi

    if [ -f "$CLIENT_DIR/package-lock.json" ] && [ "$CLIENT_DIR/package-lock.json" -nt "$CLIENT_DIR/node_modules" ]; then
        echo "package-lock.json changed"
        return 0
    fi

    return 1
}

echo -e "${BLUE}Checking if client dependencies need updating...${NC}"
INSTALL_REASON=$(needs_npm_install) && NEEDS_INSTALL=true || NEEDS_INSTALL=false
if $NEEDS_INSTALL; then
    echo -e "${BLUE}Installing client dependencies ($INSTALL_REASON)...${NC}"
    cd "$CLIENT_DIR"
    npm install
    # Record the platform for future cross-platform detection
    echo "$CURRENT_PLATFORM" > "$PLATFORM_MARKER"
    cd "$SCRIPT_DIR"
else
    echo -e "${GREEN}Client dependencies are up to date${NC}"
fi

echo -e "${BLUE}+------------------------------------------------------------+${NC}"
echo -e "${BLUE}|     pgEdge AI DBA Workbench Development Startup            |${NC}"
echo -e "${BLUE}+------------------------------------------------------------+${NC}"
echo ""

# Start MCP server
echo -e "${GREEN}[1/2] Starting MCP Server (HTTP mode with auth)...${NC}"
cd "$BIN_DIR"
"$SERVER_BIN" --config "$SERVER_CONFIG" > "$SERVER_LOG" 2>&1 &
SERVER_PID=$!
cd "$SCRIPT_DIR"

echo "      PID: $SERVER_PID"
echo "      Config: $SERVER_CONFIG"
echo "      Log: $SERVER_LOG"

# Wait a moment for process to stabilize
sleep 1

# Check if MCP server process is still running (catch immediate failures like port conflicts)
if ! kill -0 $SERVER_PID 2>/dev/null; then
    echo -e "${RED}Error: MCP Server process exited immediately${NC}"
    echo "This usually means the port is already in use or there's a configuration error."
    echo "Check the log file: $SERVER_LOG"
    tail -20 "$SERVER_LOG"
    exit 1
fi

# Wait for MCP server to be ready
echo -e "${GREEN}Waiting for MCP Server to be ready...${NC}"
MAX_RETRIES=30
RETRY_COUNT=0
while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
    if curl -s http://localhost:8080/health > /dev/null 2>&1; then
        echo "      MCP Server is ready!"
        break
    fi

    # Check if process is still running
    if ! kill -0 $SERVER_PID 2>/dev/null; then
        echo -e "${RED}Error: MCP Server process died during startup${NC}"
        echo "Check the log file: $SERVER_LOG"
        tail -20 "$SERVER_LOG"
        exit 1
    fi

    RETRY_COUNT=$((RETRY_COUNT + 1))
    sleep 1
done

if [ $RETRY_COUNT -eq $MAX_RETRIES ]; then
    echo -e "${RED}Error: MCP Server failed to start within 30 seconds${NC}"
    echo "Check the log file: $SERVER_LOG"
    tail -20 "$SERVER_LOG"
    exit 1
fi

# Start Vite dev server
echo -e "${GREEN}[2/2] Starting Vite Dev Server (Frontend)...${NC}"
cd "$CLIENT_DIR"
npm run dev > "$VITE_LOG" 2>&1 &
CLIENT_PID=$!
cd "$SCRIPT_DIR"

echo "      PID: $CLIENT_PID"
echo "      Port: 5173 (Vite default)"
echo "      Log: $VITE_LOG"

# Wait a moment for process to stabilize
sleep 1

# Check if Vite server process is still running (catch immediate failures like port conflicts)
if ! kill -0 $CLIENT_PID 2>/dev/null; then
    echo -e "${RED}Error: Vite Server process exited immediately${NC}"
    echo "This usually means port 5173 is already in use or there's a configuration error."
    echo "Check the log file: $VITE_LOG"
    tail -20 "$VITE_LOG"
    exit 1
fi

# Wait for Vite server to be ready
echo -e "${GREEN}Waiting for Vite Server to be ready...${NC}"
MAX_RETRIES=30
RETRY_COUNT=0
while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
    if grep -q "Local:" "$VITE_LOG" 2>/dev/null; then
        break
    fi

    # Check if process is still running
    if ! kill -0 $CLIENT_PID 2>/dev/null; then
        echo -e "${RED}Error: Vite Server process died during startup${NC}"
        echo "Check the log file: $VITE_LOG"
        tail -20 "$VITE_LOG"
        exit 1
    fi

    RETRY_COUNT=$((RETRY_COUNT + 1))
    sleep 1
done

if [ $RETRY_COUNT -eq $MAX_RETRIES ]; then
    echo -e "${RED}Error: Vite Server failed to start within 30 seconds${NC}"
    echo "Check the log file: $VITE_LOG"
    tail -20 "$VITE_LOG"
    exit 1
fi

echo ""
echo -e "${GREEN}+------------------------------------------------------------+${NC}"
echo -e "${GREEN}|  AI DBA Workbench Development is now running!              |${NC}"
echo -e "${GREEN}+------------------------------------------------------------+${NC}"
echo ""
echo -e "${BLUE}Services:${NC}"
echo "  - MCP Server:     http://localhost:8080"
echo "  - Web Interface:  http://localhost:5173"
echo ""
echo -e "${BLUE}Logs:${NC}"
echo "  - MCP Server:     $SERVER_LOG"
echo "  - Vite Dev:       $VITE_LOG"
echo ""
echo -e "${BLUE}Login Credentials (for web interface):${NC}"
echo "  - Create users with: ./bin/ai-dba-server add-user"
echo "  - List users with:   ./bin/ai-dba-server list-users"
echo ""
echo -e "${BLUE}Architecture:${NC}"
echo "  - Web client communicates with MCP server via REST API"
echo "  - LLM API keys are stored server-side (configured in ai-dba-server.yaml)"
echo "  - MCP server provides tool/resource access and LLM proxy endpoints"
echo ""
echo -e "${YELLOW}Press Ctrl+C to stop all services${NC}"
echo ""

# Wait for interrupt
wait $CLIENT_PID