mirror of
https://github.com/dredx/prole.git
synced 2026-09-23 10:13:58 +00:00
299 lines
9.9 KiB
Bash
Executable File
299 lines
9.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
# init_postgrest.sh
|
|
# Purpose:
|
|
# - Deploy PostgREST (postgrest/postgrest) into the knoe-db namespace
|
|
# - Creates the knoe-db-postgrest-secrets k8s secret (postgres password + JWT secret)
|
|
# - Applies the postgrest deployment and service manifests
|
|
# - Provides start/stop/status/restart actions
|
|
#
|
|
# Usage:
|
|
# ./init_postgrest.sh [--mode MODE] [-n NAMESPACE] <start|stop|status|restart>
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
# shellcheck disable=SC1090
|
|
source "$SCRIPT_DIR/knoe_cfg.sh"
|
|
|
|
if [[ "${1:-}" == "--mode" || "${1:-}" == "-m" ]]; then
|
|
knoe_set_mode "${2:-}"
|
|
shift 2
|
|
elif [[ "${1:-}" == --mode=* || "${1:-}" == -m=* ]]; then
|
|
knoe_set_mode "${1#*=}"
|
|
shift
|
|
fi
|
|
|
|
NAMESPACE="${PROLE_NAMESPACE}"
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-n|--namespace)
|
|
shift
|
|
if [[ -z "${1:-}" ]]; then
|
|
echo "ERROR: -n/--namespace requires a value" >&2
|
|
exit 2
|
|
fi
|
|
NAMESPACE="$1"
|
|
;;
|
|
-n=*|--namespace=*)
|
|
NAMESPACE="${1#*=}"
|
|
;;
|
|
start|stop|status|restart)
|
|
ACTION="$1"
|
|
;;
|
|
-h|--help)
|
|
usage
|
|
;;
|
|
*)
|
|
echo "ERROR: Unknown argument: $1" >&2
|
|
exit 2
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
ACTION="${ACTION:-start}"
|
|
KNOE_HOME=${KNOE_HOME:-$(cd "$SCRIPT_DIR/.." && pwd)}
|
|
POSTGREST_IMAGE="${POSTGREST_IMAGE:-postgrest/postgrest:v14.5}"
|
|
POSTGREST_NAME="${POSTGREST_NAME:-knoe-db-postgrest}"
|
|
POSTGREST_PORT="${POSTGREST_PORT:-3000}"
|
|
KNOE_DB_SERVICE="${KNOE_DB_SERVICE:-knoe-db-postgres}"
|
|
POSTGREST_SECRET_NAME="${POSTGREST_SECRET_NAME:-knoe-db-postgrest-secrets}"
|
|
|
|
usage() {
|
|
cat <<USAGE
|
|
Usage: $0 [--mode MODE] [-n NAMESPACE] <start|stop|status|restart>
|
|
|
|
Actions:
|
|
start Create secrets and deploy PostgREST to the knoe-db namespace
|
|
stop Remove PostgREST deployment and secrets
|
|
status Show PostgREST pod/service status
|
|
restart Restart PostgREST pods
|
|
USAGE
|
|
exit 1
|
|
}
|
|
|
|
ensure_tools() {
|
|
for t in kubectl; do
|
|
command -v "$t" >/dev/null || { echo "Missing required tool: $t" >&2; exit 1; }
|
|
done
|
|
}
|
|
|
|
ensure_namespace() {
|
|
if ! kubectl get namespace "$NAMESPACE" >/dev/null 2>&1; then
|
|
echo "Creating namespace '$NAMESPACE' ..."
|
|
kubectl create namespace "$NAMESPACE" >/dev/null 2>&1 || true
|
|
fi
|
|
}
|
|
|
|
resolve_postgres_password() {
|
|
# Try to read the postgres password from the CloudNativePG superuser secret
|
|
local cnpg_secret="knoe-db-superuser"
|
|
local pw=""
|
|
|
|
if kubectl get secret "$cnpg_secret" -n "$NAMESPACE" >/dev/null 2>&1; then
|
|
pw=$(kubectl get secret "$cnpg_secret" -n "$NAMESPACE" -o jsonpath='{.data.password}' 2>/dev/null | base64 -d 2>/dev/null || true)
|
|
fi
|
|
|
|
if [[ -z "$pw" ]]; then
|
|
# Fallback: check for POSTGRES_PASSWORD env var or knoe.cfg
|
|
pw="${POSTGRES_PASSWORD:-}"
|
|
fi
|
|
|
|
if [[ -z "$pw" ]]; then
|
|
echo "ERROR: Cannot resolve postgres password. Ensure knoe-db-superuser secret exists in namespace '$NAMESPACE' or set POSTGRES_PASSWORD." >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo "$pw"
|
|
}
|
|
|
|
resolve_jwt_secret() {
|
|
# Try to read JWT secret from supabase namespace secrets
|
|
local jwt=""
|
|
|
|
if kubectl get secret supabase-jwt -n supabase >/dev/null 2>&1; then
|
|
jwt=$(kubectl get secret supabase-jwt -n supabase -o jsonpath='{.data.jwt-secret}' 2>/dev/null | base64 -d 2>/dev/null || true)
|
|
fi
|
|
|
|
if [[ -z "$jwt" ]]; then
|
|
# Fallback: try reading from supabase .env file
|
|
local env_file="${DEV_HOME:-$HOME/dev}/supabase/docker/.env"
|
|
if [[ -f "$env_file" ]]; then
|
|
jwt=$(grep -E '^JWT_SECRET=' "$env_file" | head -1 | cut -d= -f2- | tr -d "'\"" || true)
|
|
fi
|
|
fi
|
|
|
|
if [[ -z "$jwt" ]]; then
|
|
jwt="${JWT_SECRET:-}"
|
|
fi
|
|
|
|
if [[ -z "$jwt" ]]; then
|
|
echo "ERROR: Cannot resolve JWT secret. Set JWT_SECRET or ensure supabase-jwt secret exists." >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo "$jwt"
|
|
}
|
|
|
|
setup_database_roles() {
|
|
echo "Ensuring PostgREST database roles and schemas exist ..."
|
|
local pg_password
|
|
pg_password=$(resolve_postgres_password)
|
|
|
|
# Find primary pod
|
|
local primary
|
|
primary=$(kubectl -n "$NAMESPACE" get pods -l "cnpg.io/cluster=knoe-db,cnpg.io/instanceRole=primary" -o jsonpath='{.items[0].metadata.name}' 2>/dev/null || true)
|
|
if [[ -z "$primary" ]]; then
|
|
primary=$(kubectl -n "$NAMESPACE" get pods -l "cnpg.io/cluster=knoe-db" -o jsonpath='{.items[0].metadata.name}' 2>/dev/null || true)
|
|
fi
|
|
if [[ -z "$primary" ]]; then
|
|
echo "WARNING: No knoe-db pod found in namespace '$NAMESPACE'; skipping role setup." >&2
|
|
return 0
|
|
fi
|
|
|
|
kubectl -n "$NAMESPACE" exec "$primary" -c postgres -- psql -U postgres -d postgres -c "
|
|
CREATE SCHEMA IF NOT EXISTS storage;
|
|
CREATE SCHEMA IF NOT EXISTS graphql_public;
|
|
DO \$\$ BEGIN IF NOT EXISTS (SELECT FROM pg_roles WHERE rolname='anon') THEN CREATE ROLE anon NOLOGIN; END IF; END \$\$;
|
|
DO \$\$ BEGIN IF NOT EXISTS (SELECT FROM pg_roles WHERE rolname='authenticator') THEN CREATE ROLE authenticator LOGIN PASSWORD '${pg_password}'; END IF; END \$\$;
|
|
GRANT USAGE ON SCHEMA public TO anon;
|
|
GRANT USAGE ON SCHEMA storage TO anon;
|
|
GRANT USAGE ON SCHEMA graphql_public TO anon;
|
|
GRANT anon TO authenticator;
|
|
" || echo "WARNING: Could not set up PostgREST roles (cluster may not be ready yet)." >&2
|
|
echo "PostgREST database roles and schemas ready."
|
|
}
|
|
|
|
create_secrets() {
|
|
echo "Resolving secrets for PostgREST ..."
|
|
local pg_password
|
|
local jwt_secret
|
|
pg_password=$(resolve_postgres_password)
|
|
jwt_secret=$(resolve_jwt_secret)
|
|
|
|
echo "Creating/updating secret '$POSTGREST_SECRET_NAME' in namespace '$NAMESPACE' ..."
|
|
kubectl create secret generic "$POSTGREST_SECRET_NAME" \
|
|
--namespace="$NAMESPACE" \
|
|
--from-literal=postgres-password="$pg_password" \
|
|
--from-literal=jwt-secret="$jwt_secret" \
|
|
--dry-run=client -o yaml | kubectl apply -f -
|
|
|
|
echo "Secret '$POSTGREST_SECRET_NAME' ready."
|
|
}
|
|
|
|
check_and_repair_pods() {
|
|
echo "Checking for non-functioning $POSTGREST_NAME pods ..."
|
|
local bad_pods
|
|
bad_pods=$(kubectl get pods -n "$NAMESPACE" -l app="$POSTGREST_NAME" \
|
|
--field-selector='status.phase!=Running' -o jsonpath='{.items[*].metadata.name}' 2>/dev/null || true)
|
|
|
|
# Also check Running pods that have containers not ready (CrashLoopBackOff, Error, etc.)
|
|
local not_ready_pods
|
|
not_ready_pods=$(kubectl get pods -n "$NAMESPACE" -l app="$POSTGREST_NAME" \
|
|
-o jsonpath='{range .items[*]}{.metadata.name}{" "}{range .status.containerStatuses[*]}{.ready}{" "}{end}{"\n"}{end}' 2>/dev/null | \
|
|
grep -v '^$' | grep 'false' | awk '{print $1}' || true)
|
|
|
|
local all_bad
|
|
all_bad=$(echo -e "${bad_pods}\n${not_ready_pods}" | sort -u | xargs)
|
|
|
|
if [[ -n "$all_bad" ]]; then
|
|
echo "Found non-functioning pods: $all_bad"
|
|
echo "Deleting non-functioning pods to allow redeployment ..."
|
|
for pod in $all_bad; do
|
|
kubectl delete pod "$pod" -n "$NAMESPACE" --grace-period=0 --force 2>/dev/null || true
|
|
done
|
|
echo "Non-functioning pods removed."
|
|
else
|
|
echo "No non-functioning pods found."
|
|
fi
|
|
|
|
# If the deployment exists but is in a bad state, delete it so we can recreate cleanly
|
|
if kubectl get deployment "$POSTGREST_NAME" -n "$NAMESPACE" >/dev/null 2>&1; then
|
|
local available
|
|
available=$(kubectl get deployment "$POSTGREST_NAME" -n "$NAMESPACE" \
|
|
-o jsonpath='{.status.availableReplicas}' 2>/dev/null || echo "0")
|
|
if [[ "${available:-0}" == "0" ]]; then
|
|
echo "Deployment '$POSTGREST_NAME' has no available replicas; deleting for clean redeployment ..."
|
|
kubectl delete deployment "$POSTGREST_NAME" -n "$NAMESPACE" --ignore-not-found=true
|
|
sleep 2
|
|
fi
|
|
fi
|
|
}
|
|
|
|
deploy() {
|
|
echo "Deploying $POSTGREST_NAME to namespace '$NAMESPACE' ..."
|
|
local manifests_dir="$KNOE_HOME/deploy/opentofu/k3s/manifests/knoe"
|
|
|
|
kubectl apply -f "$manifests_dir/postgrest-deployment.yaml" -n "$NAMESPACE"
|
|
kubectl apply -f "$manifests_dir/postgrest-service.yaml" -n "$NAMESPACE"
|
|
|
|
# Check if the backing database service has endpoints before waiting
|
|
local db_endpoints
|
|
db_endpoints=$(kubectl get endpoints "$KNOE_DB_SERVICE" -n "$NAMESPACE" -o jsonpath='{.subsets[*].addresses[*].ip}' 2>/dev/null || true)
|
|
|
|
if [[ -z "$db_endpoints" ]]; then
|
|
echo "WARNING: Database service '$KNOE_DB_SERVICE' has no ready endpoints."
|
|
echo "PostgREST is deployed but will not become ready until the database is available."
|
|
echo "PostgREST will automatically connect once the database is running."
|
|
echo "$POSTGREST_NAME deployment applied (waiting for database)."
|
|
return 0
|
|
fi
|
|
|
|
echo "Waiting for $POSTGREST_NAME rollout ..."
|
|
if kubectl rollout status deployment/"$POSTGREST_NAME" -n "$NAMESPACE" --timeout=120s; then
|
|
echo "$POSTGREST_NAME deployed successfully."
|
|
else
|
|
echo "WARNING: $POSTGREST_NAME rollout did not complete within timeout."
|
|
echo "The pod may still be waiting for the database to become available."
|
|
echo "Check status with: $0 status"
|
|
fi
|
|
}
|
|
|
|
stop() {
|
|
echo "Removing $POSTGREST_NAME from namespace '$NAMESPACE' ..."
|
|
kubectl delete deployment "$POSTGREST_NAME" -n "$NAMESPACE" --ignore-not-found=true
|
|
kubectl delete service "$POSTGREST_NAME" -n "$NAMESPACE" --ignore-not-found=true
|
|
kubectl delete secret "$POSTGREST_SECRET_NAME" -n "$NAMESPACE" --ignore-not-found=true
|
|
echo "$POSTGREST_NAME removed."
|
|
}
|
|
|
|
status() {
|
|
echo "=== $POSTGREST_NAME pods ==="
|
|
kubectl get pods -n "$NAMESPACE" -l app="$POSTGREST_NAME" 2>/dev/null || echo "No pods found"
|
|
echo ""
|
|
echo "=== $POSTGREST_NAME service ==="
|
|
kubectl get svc "$POSTGREST_NAME" -n "$NAMESPACE" 2>/dev/null || echo "No service found"
|
|
}
|
|
|
|
restart() {
|
|
echo "Restarting $POSTGREST_NAME ..."
|
|
kubectl rollout restart deployment/"$POSTGREST_NAME" -n "$NAMESPACE"
|
|
kubectl rollout status deployment/"$POSTGREST_NAME" -n "$NAMESPACE" --timeout=120s
|
|
echo "$POSTGREST_NAME restarted."
|
|
}
|
|
|
|
case "$ACTION" in
|
|
start)
|
|
ensure_tools
|
|
ensure_namespace
|
|
check_and_repair_pods
|
|
setup_database_roles
|
|
create_secrets
|
|
deploy
|
|
;;
|
|
stop)
|
|
ensure_tools
|
|
stop
|
|
;;
|
|
status)
|
|
ensure_tools
|
|
status
|
|
;;
|
|
restart)
|
|
ensure_tools
|
|
restart
|
|
;;
|
|
*)
|
|
usage
|
|
;;
|
|
esac
|