OSVauco/scripts/preflight.sh

162 lines
6.8 KiB
Bash

#!/usr/bin/env bash
# =============================================================================
# preflight.sh — OSVauco Operator Preflight Dashboard
# Provides immediate situational awareness of the OPAX ecosystem on GCP.
# Usage: bash scripts/preflight.sh
# =============================================================================
set -uo pipefail
# Colors
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
CYAN='\033[0;36m'
DIM='\033[2m'
BOLD='\033[1m'
NC='\033[0m'
# Icons
OK="${GREEN}\u2713${NC}"
WARN="${YELLOW}\u26a0\ufe0f${NC}"
FAIL="${RED}\u2718${NC}"
INFO="${CYAN}\u2139${NC}"
# 1. Context Discovery
PROJECT_ID=$(gcloud config get-value project 2>/dev/null | tr -d '\r' || echo "NOT_SET")
ACCOUNT=$(gcloud config get-value account 2>/dev/null | tr -d '\r' || echo "NOT_SET")
# Region fallback if not set in environment
REGION="${RAG_LOCATION:-${REGION:-us-central1}}"
REGION=$(echo "$REGION" | tr -d '\r')
clear
echo -e "${CYAN}${BOLD}==========================================================${NC}"
echo -e "${CYAN}${BOLD} OSVauco OPAX PREFLIGHT DASHBOARD${NC}"
echo -e "${CYAN}${BOLD}==========================================================${NC}"
echo -e "${DIM}Time : $(date '+%Y-%m-%d %H:%M %Z')${NC}"
echo -e "${DIM}Project : ${PROJECT_ID}${NC}"
echo -e "${DIM}Identity: ${ACCOUNT}${NC}"
echo -e "${CYAN}${BOLD}----------------------------------------------------------${NC}"
# --- 1. Infrastructure Status (Cloud Run) ---
echo -e "${BOLD}[1] Infrastructure (Cloud Run)${NC}"
SERVICE_NAME="opax-mcp-core"
# Check if service exists and its state
SERVICE_DATA=$(gcloud run services describe "${SERVICE_NAME}" --region="${REGION}" --project="${PROJECT_ID}" --format="value(status.conditions[0].status,status.url)" 2>/dev/null | tr -d '\r' || echo "NOT_FOUND -")
SERVICE_STATUS=$(echo "$SERVICE_DATA" | awk '{print $1}')
SERVICE_URL=$(echo "$SERVICE_DATA" | awk '{print $2}')
if [ "$SERVICE_STATUS" = "True" ]; then
echo -e " ${OK} ${SERVICE_NAME}: ${GREEN}ACTIVE${NC}"
echo -e " ${DIM}URL: ${SERVICE_URL}${NC}"
elif [ "$SERVICE_STATUS" = "NOT_FOUND" ]; then
echo -e " ${FAIL} ${SERVICE_NAME}: ${RED}NOT DEPLOYED${NC}"
echo -e " ${DIM}Action: bash infrastructure/05-cloudrun-deploy.sh${NC}"
else
echo -e " ${WARN} ${SERVICE_NAME}: ${YELLOW}INACTIVE / UNHEALTHY${NC} (Status: ${SERVICE_STATUS})"
echo -e " ${DIM}Check console for deployment errors${NC}"
fi
echo ""
# --- 2. Knowledge Base Status (RAG Engine) ---
echo -e "${BOLD}[2] Knowledge Base (RAG Engine)${NC}"
CORPUS_DISPLAY_NAME="osvauco-knowledge-base"
# Note: gcloud ai rag-corpora might not be available in all SDK versions.
if gcloud ai rag-corpora --help &>/dev/null; then
RAG_DATA=$(gcloud ai rag-corpora list --project="${PROJECT_ID}" --format="value(name,location)" 2>/dev/null | tr -d '\r' | grep "${CORPUS_DISPLAY_NAME}" || true)
if [ -n "$RAG_DATA" ]; then
FULL_NAME=$(echo "$RAG_DATA" | awk '{print $1}')
RAG_LOC=$(echo "$RAG_DATA" | awk '{print $2}')
# Get count of files
FILE_COUNT=$(gcloud ai rag-files list --corpus="${FULL_NAME}" --project="${PROJECT_ID}" --location="${RAG_LOC}" --format="value(name)" 2>/dev/null | wc -l | tr -dc '0-9')
FILE_COUNT=${FILE_COUNT:-0}
echo -e " ${OK} Corpus: ${GREEN}FOUND${NC} in ${RAG_LOC}"
echo -e " ${DIM}Files: ${FILE_COUNT} documents indexed${NC}"
else
echo -e " ${WARN} Corpus: ${YELLOW}NOT FOUND${NC}"
echo -e " ${DIM}Action: python3 agents/rag/setup_corpus.py${NC}"
fi
else
# Fallback: Try a lightweight python check
echo -e " ${INFO} Corpus check: ${DIM}gcloud ai rag commands missing. Checking via SDK...${NC}"
RAG_CHECK=$(python3 -c "
import os
try:
from vertexai import rag
import vertexai
vertexai.init(project='${PROJECT_ID}')
# Check common regions
for loc in ['${REGION}', 'europe-west4']:
vertexai.init(project='${PROJECT_ID}', location=loc)
for c in rag.list_corpora():
if c.display_name == '${CORPUS_DISPLAY_NAME}':
print(f'FOUND|{loc}|{c.name}')
exit(0)
print('MISSING')
except Exception:
print('ERROR')
" 2>/dev/null | tr -d '\r' || echo "ERROR")
if [[ "$RAG_CHECK" == FOUND* ]]; then
LOC=$(echo "$RAG_CHECK" | cut -d'|' -f2)
echo -e " ${OK} Corpus: ${GREEN}FOUND${NC} in ${LOC}"
elif [ "$RAG_CHECK" = "MISSING" ]; then
echo -e " ${WARN} Corpus: ${YELLOW}NOT FOUND${NC}"
echo -e " ${DIM}Action: python3 agents/rag/setup_corpus.py${NC}"
else
echo -e " ${WARN} Corpus: ${DIM}Unable to verify (Vertex AI SDK issues)${NC}"
fi
fi
echo ""
# --- 3. Resource & Cost Risk Audit ---
echo -e "${BOLD}[3] Resource & Cost Risk Audit${NC}"
# Resources that incur cost even when idle
VM_COUNT=$(gcloud compute instances list --project="${PROJECT_ID}" --filter="status=RUNNING" --format="value(name)" 2>/dev/null | wc -l | tr -dc '0-9')
SQL_COUNT=$(gcloud sql instances list --project="${PROJECT_ID}" --filter="state=RUNNABLE" --format="value(name)" 2>/dev/null | wc -l | tr -dc '0-9')
# Reasoning Engines (Memory Bank) are common cost leaks
RB_COUNT=$(gcloud ai reasoning-engines list --project="${PROJECT_ID}" --format="value(name)" 2>/dev/null | wc -l | tr -dc '0-9')
# Initialize defaults
VM_COUNT=${VM_COUNT:-0}
SQL_COUNT=${SQL_COUNT:-0}
RB_COUNT=${RB_COUNT:-0}
if [ "$VM_COUNT" -eq 0 ] && [ "$SQL_COUNT" -eq 0 ] && [ "$RB_COUNT" -eq 0 ]; then
echo -e " ${OK} No expensive idle resources detected."
else
if [ "$VM_COUNT" -gt 0 ]; then
echo -e " ${RED}${FAIL} Running VMs: ${VM_COUNT}${NC} (Always-on cost!)"
fi
if [ "$SQL_COUNT" -gt 0 ]; then
echo -e " ${RED}${FAIL} Running SQL: ${SQL_COUNT}${NC} (Always-on cost!)"
fi
if [ "$RB_COUNT" -gt 0 ]; then
echo -e " ${RED}${FAIL} Memory Bank: ${RB_COUNT}${NC} (Idle cost!)"
fi
echo -e " ${DIM}Tip: Run bash infrastructure/03-teardown.sh at end of day.${NC}"
fi
echo ""
# --- 4. Security & Integrations ---
echo -e "${BOLD}[4] Security & Secrets${NC}"
SECRETS=("github-token" "webhook-url" "rag-corpus-name")
for s in "${SECRETS[@]}"; do
if gcloud secrets describe "$s" --project="${PROJECT_ID}" &>/dev/null; then
echo -e " ${OK} Secret: ${GREEN}${s}${NC}"
else
if [ "$s" = "rag-corpus-name" ]; then
echo -e " ${INFO} Secret: ${DIM}${s}${NC} (Auto-populated by setup_corpus.py)"
else
echo -e " ${WARN} Secret: ${YELLOW}${s}${NC} (MISSING)"
echo -e " ${DIM}Action: gcloud secrets create ${s} --data-file=...${NC}"
fi
fi
done
echo ""
echo -e "${CYAN}${BOLD}==========================================================${NC}"
echo -e " ${BOLD}Next Step:${NC} Run ${DIM}bash scripts/osvauco-opax-boot.sh${NC} to start OPAX."
echo -e "${CYAN}${BOLD}==========================================================${NC}"