Something went wrong. Try again.
Our Personal Data Server from scratch! tranquil.farm
pds rust database fun oauth atproto
Something went wrong. Try again.
4.1 kB · 127 lines
Shell
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128#!/usr/bin/env bashset -euo pipefailINFRA_FILE="${TMPDIR:-/tmp}/tranquil_pds_test_infra.env"CONTAINER_PREFIX="tranquil-pds-test"command_exists() { command -v "$1" >/dev/null 2>&1}if command_exists podman; then CONTAINER_CMD="podman" if [[ -z "${DOCKER_HOST:-}" ]]; then RUNTIME_DIR="${XDG_RUNTIME_DIR:-/run/user/$(id -u)}" PODMAN_SOCK="$RUNTIME_DIR/podman/podman.sock" if [[ -S "$PODMAN_SOCK" ]]; then export DOCKER_HOST="unix://$PODMAN_SOCK" fi fielif command_exists docker; then CONTAINER_CMD="docker"else echo "Error: Neither podman nor docker found" >&2 exit 1fistart_infra() { echo "Starting test infrastructure..." if [[ -f "$INFRA_FILE" ]]; then source "$INFRA_FILE" if $CONTAINER_CMD ps --format '{{.Names}}' 2>/dev/null | grep -q "^${CONTAINER_PREFIX}-postgres$"; then echo "Infrastructure already running (found $INFRA_FILE)" cat "$INFRA_FILE" return 0 fi echo "Stale infra file found, cleaning up..." rm -f "$INFRA_FILE" fi $CONTAINER_CMD rm -f "${CONTAINER_PREFIX}-postgres" 2>/dev/null || true echo "Starting PostgreSQL..." $CONTAINER_CMD run -d \ --name "${CONTAINER_PREFIX}-postgres" \ -e POSTGRES_PASSWORD=postgres \ -e POSTGRES_USER=postgres \ -e POSTGRES_DB=postgres \ -P \ --label tranquil_pds_test=true \ postgres:18-alpine \ -c max_connections=500 \ -c fsync=off \ -c synchronous_commit=off \ -c full_page_writes=off >/dev/null echo "Waiting for services to be ready..." for i in {1..30}; do if $CONTAINER_CMD exec "${CONTAINER_PREFIX}-postgres" pg_isready -U postgres >/dev/null 2>&1; then break fi echo "Waiting for PostgreSQL... ($i/30)" sleep 1 done PG_PORT=$($CONTAINER_CMD port "${CONTAINER_PREFIX}-postgres" 5432 | head -1 | cut -d: -f2) cat > "$INFRA_FILE" << EOFexport DATABASE_URL="postgres://postgres:postgres@127.0.0.1:${PG_PORT}/postgres"export TEST_DB_PORT="${PG_PORT}"export TRANQUIL_PDS_TEST_INFRA_READY="1"export TRANQUIL_PDS_ALLOW_INSECURE_SECRETS="1"export SKIP_IMPORT_VERIFICATION="true"export DISABLE_RATE_LIMITING="1"export TRANQUIL_LEXICON_OFFLINE="1"EOF echo "" echo "Infrastructure ready!" echo "Config written to: $INFRA_FILE" echo "" cat "$INFRA_FILE"}stop_infra() { echo "Stopping test infrastructure..." $CONTAINER_CMD rm -f "${CONTAINER_PREFIX}-postgres" 2>/dev/null || true rm -f "$INFRA_FILE" rm -rf "${TMPDIR:-/tmp}"/tranquil-pds-test-* 2>/dev/null || true echo "Infrastructure stopped."}status_infra() { echo "Test Infrastructure Status:" echo "============================" if [[ -f "$INFRA_FILE" ]]; then echo "Config file: $INFRA_FILE" source "$INFRA_FILE" echo "Database URL: $DATABASE_URL" else echo "Config file: NOT FOUND" fi echo "" echo "Containers:" $CONTAINER_CMD ps -a --filter "label=tranquil_pds_test=true" --format "table {{.Names}}\t{{.Status}}\t{{.Ports}}" 2>/dev/null || echo " (none)"}case "${1:-}" in start) start_infra ;; stop) stop_infra ;; restart) stop_infra start_infra ;; status) status_infra ;; env) if [[ -f "$INFRA_FILE" ]]; then cat "$INFRA_FILE" else echo "Infrastructure not running. Run: $0 start" >&2 exit 1 fi ;; *) echo "Usage: $0 {start|stop|restart|status|env}" echo "" echo "Commands:" echo " start - Start test infrastructure (Postgres)" echo " stop - Stop and remove test containers" echo " restart - Stop then start infrastructure" echo " status - Show infrastructure status" echo " env - Output environment variables for sourcing" exit 1 ;;esac