leo-bot/start_dev.sh at main · USPA-Technology/leo-bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/env bash
set -euo pipefail

# ------------------------------------------------------------------------------
# LEO BOT — Parallel Development Startup Script
# ------------------------------------------------------------------------------
# Fast, fault-tolerant startup sequence:
#   1. Checks PostgreSQL (5432) and Keycloak (via Nginx HTTPS realm endpoint).
#   2. Starts any missing services in parallel.
#   3. Updates code, activates venv, and runs FastAPI app.
# ------------------------------------------------------------------------------

PG_PORT=5432
KEYCLOAK_REALM="master"
KEYCLOAK_URL="https://leoid.example.com"
KEYCLOAK_HEALTHCHECK="${KEYCLOAK_URL}/realms/${KEYCLOAK_REALM}/.well-known/openid-configuration"
VENV_PATH="env/bin/activate"
FASTAPI_APP="main_app:leobot"
FASTAPI_PORT=8888

# Colors for readable logs
GREEN="\033[1;32m"
YELLOW="\033[1;33m"
RED="\033[1;31m"
NC="\033[0m"

echo -e "${GREEN}🚀 Starting LEO BOT dev environment (parallel mode)...${NC}"

# ------------------------------------------------------------------------------
# Functions
# ------------------------------------------------------------------------------

check_postgres() {
  if nc -z localhost "$PG_PORT" 2>/dev/null; then
    echo -e "${GREEN}✅ PostgreSQL already running on port $PG_PORT.${NC}"
  else
    echo -e "${YELLOW}⚙️  Starting PostgreSQL (pgvector)...${NC}"
    bash ./start_pgsql_pgvector.sh
    for i in {1..5}; do
      sleep 1
      if nc -z localhost "$PG_PORT" 2>/dev/null; then
        echo -e "${GREEN}✅ PostgreSQL is now up.${NC}"
        return
      fi
    done
    echo -e "${RED}❌ PostgreSQL failed to start on port $PG_PORT.${NC}"
    exit 1
  fi
}

check_keycloak() {
  echo -e "${YELLOW}🔍 Checking Keycloak health at:${NC}"
  echo -e "    ${KEYCLOAK_HEALTHCHECK}"
  HTTP_STATUS=$(curl -sk --connect-timeout 3 --max-time 5 -o /dev/null -w "%{http_code}" "$KEYCLOAK_HEALTHCHECK")

  if [[ "$HTTP_STATUS" =~ ^(200)$ ]]; then
    echo -e "${GREEN}✅ Keycloak is healthy (HTTP $HTTP_STATUS).${NC}"
  else
    echo -e "${YELLOW}⚙️  Starting Keycloak (Docker)...${NC}"
    bash ./dockers/keycloak/run-keycloak.sh
    for i in {1..10}; do
      sleep 1
      HTTP_STATUS=$(curl -sk --connect-timeout 3 --max-time 5 -o /dev/null -w "%{http_code}" "$KEYCLOAK_HEALTHCHECK")
      if [[ "$HTTP_STATUS" =~ ^(200)$ ]]; then
        echo -e "${GREEN}✅ Keycloak is now healthy (HTTP $HTTP_STATUS).${NC}"
        return
      fi
    done
    echo -e "${RED}❌ Keycloak failed to respond after startup. Check Docker logs.${NC}"
    exit 1
  fi
}

# ------------------------------------------------------------------------------
# Step 1–2: Parallel checks
# ------------------------------------------------------------------------------
check_postgres &
PID_PG=$!

check_keycloak &
PID_KC=$!

# Wait for both processes
wait $PID_PG
wait $PID_KC

# ------------------------------------------------------------------------------
# Step 3: Update repo
# ------------------------------------------------------------------------------
echo -e "${YELLOW}📦 Updating Git repository...${NC}"
git pull --quiet
echo -e "${GREEN}✅ Repository is up to date.${NC}"

# ------------------------------------------------------------------------------
# Step 4: Activate virtual environment
# ------------------------------------------------------------------------------
if [[ -f "$VENV_PATH" ]]; then
  echo -e "${YELLOW}🐍 Activating Python virtual environment...${NC}"
  source "$VENV_PATH"
else
  echo -e "${RED}❌ Virtual environment not found at $VENV_PATH.${NC}"
  exit 1
fi

# ------------------------------------------------------------------------------
# Step 5: Launch FastAPI app
# ------------------------------------------------------------------------------
echo -e "${YELLOW}⚡ Launching FastAPI app (port $FASTAPI_PORT)...${NC}"
uvicorn "$FASTAPI_APP" \
  --reload \
  --env-file .env \
  --host 0.0.0.0 \
  --port $FASTAPI_PORT