run.sh

#!/bin/bash
# ==============================================================================
# run.sh — AgenticXRAG Full-Stack Runner
# ==============================================================================
#
# DESCRIPTION:
#   Master entry point that brings up the complete AgenticXRAG stack by
#   orchestrating two sub-scripts in the correct dependency order:
#
#     ┌─────────────────────────────────────────────────────────┐
#     │  Startup order  (start / restart / smart-restart)       │
#     │                                                         │
#     │  1. platform.sh        →  Infrastructure + Services layer  │
#     │       • RabbitMQ        (message broker)                │
#     │       • Qdrant          (vector database)               │
#     │       • FastEmbed       (sparse + ColBERT embeddings)   │
#     │       • HF              (dense HF embeddings)           │
#     │                                                          │
#     │  2. agenticxrag.sh  →  Application (Core) layer        │
#     │       • AgenticXRAG API  (FastAPI + Uvicorn)            │
#     └─────────────────────────────────────────────────────────┘
#
#     Shutdown order is reversed — Core stops before Platform:
#     stop / clean:  agenticxrag.sh → platform.sh
#
# USAGE:
#   ./run.sh [ACTION]
#
# ACTIONS:
#   start           Start the full stack (Platform → Core).
#                   Safe to call when services are already running;
#                   skips containers that do not need restarting.
#
#   stop            Gracefully stop the full stack (Core → Platform).
#                   Containers are stopped but NOT removed.
#                   Volumes and images are preserved.
#
#   restart         Full stop → start cycle for the entire stack.
#                   Equivalent to `stop` followed by `start`.
#
#   smart-restart   (DEFAULT) Intelligent restart — only restarts
#                   containers that are currently running; starts
#                   stopped ones directly. Faster than a full restart.
#
#   status          Show the live status of every container in
#                   all layers (Platform + Core).
#                   Read-only; no containers are modified.
#
#   clean           ⚠️  DESTRUCTIVE — tear down all containers,
#                   volumes, and images across every layer.
#                   Runs in reverse order (Core → Platform).
#                   All persisted data (Qdrant, RabbitMQ) will be lost.
#
# EXAMPLES:
#   ./run.sh                 # smart-restart (default — safest choice)
#   ./run.sh start           # cold-start the entire stack
#   ./run.sh stop            # gracefully stop everything
#   ./run.sh restart         # full cycle restart
#   ./run.sh status          # view container health
#   ./run.sh clean           # ⚠️  full teardown (data loss)
#
# LOGS:
#   Master log  →  logs/agenticxrag.log
#   Each sub-script appends its own output to the same log file.
#
# AUTHOR:
#   Bajrang Chapola
# ==============================================================================

BASE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
LOG_DIR="$BASE_DIR/logs"
LOG_FILE="$LOG_DIR/agenticxrag.log"

mkdir -p "$LOG_DIR"
> "$LOG_FILE"

timestamp() {
  date "+%Y-%m-%d %H:%M:%S"
}

log() {
  echo -e "[$(timestamp)] $1" | tee -a "$LOG_FILE"
}

info()    { log "\033[1;34m[INFO]\033[0m $1"; }
success() { log "\033[1;32m[SUCCESS]\033[0m $1"; }
error()   { log "\033[1;31m[ERROR]\033[0m $1"; }

ACTION=$1

if [ -z "$ACTION" ]; then
  ACTION="smart-restart"
  info "No action provided → Smart Restart Full AgenticXRAG"
else
  info "AgenticXRAG started with action: $ACTION"
fi

echo "---------------------------------------" | tee -a "$LOG_FILE"

PLATFORM_SCRIPT="$BASE_DIR/platform.sh"
CORE_SCRIPT="$BASE_DIR/agenticxrag.sh"

for SCRIPT in "$PLATFORM_SCRIPT" "$CORE_SCRIPT"
do
  if [ ! -f "$SCRIPT" ]; then
    error "Missing required script: $SCRIPT"
    exit 1
  fi
done

run_script() {
  NAME=$1
  SCRIPT=$2

  info "Running $NAME..."

  bash "$SCRIPT" "$ACTION" >> "$LOG_FILE" 2>&1
  EXIT_CODE=$?

  if [ $EXIT_CODE -ne 0 ]; then
    error "$NAME failed (exit code $EXIT_CODE)"
    exit 1
  fi

  success "$NAME completed successfully"
}

# ---------------------------
# Action Routing Logic
# ---------------------------

case $ACTION in

  start|restart|smart-restart|status)
    # Start platform first, then core
    run_script "AgenticXRAG Platform" "$PLATFORM_SCRIPT"
    run_script "AgenticXRAG Core" "$CORE_SCRIPT"

    if [[ "$ACTION" != "status" ]]; then
      echo "---------------------------------------" | tee -a "$LOG_FILE"
      success "🦅 Full AgenticXRAG Platform is UP and Running"
    fi
    ;;

  stop|clean)
    # Stop/clean core first, then platform
    run_script "AgenticXRAG Core" "$CORE_SCRIPT"
    run_script "AgenticXRAG Platform" "$PLATFORM_SCRIPT"

    echo "---------------------------------------" | tee -a "$LOG_FILE"
    success "🧹 Full AgenticXRAG Platform cleaned/stopped successfully"
    ;;

  *)
    error "Invalid action: $ACTION"
    echo "Usage: ./agenticxrag.sh [start|stop|restart|status|clean]"
    exit 1
    ;;

esac