#!/usr/bin/env bash
# pgEdge repository smoke test. Run as root inside a fresh container.
#   PG     Postgres major version                    (default 18)
#   SCOPE  postgres | all | none  (meta-package)      (default postgres)
#   EXTRA  individual packages, space separated       (optional)
set -euo pipefail
PG="${PG:-18}"
SCOPE="${SCOPE:-postgres}"
EXTRA="${EXTRA:-}"
. /etc/os-release

# >>> apt-helpers (identical in every script that has it; the site build checks)
# OS repositories: tolerate errors. End-of-life releases lose their security and
# updates suites, or move to archive.debian.org, and apt-get update then fails
# even though everything needed is still there.
apt_update_base() {
  local err=/tmp/apt-update.err f
  apt-get update -qq 2> "$err" && return 0
  apt-get update -qq -o Acquire::Check-Valid-Until=false 2> "$err" && return 0
  if [ "${ID:-}" = debian ]; then
    echo "note: ${VERSION_CODENAME:-this release} is no longer on the regular Debian mirrors; using archive.debian.org"
    for f in /etc/apt/sources.list /etc/apt/sources.list.d/*.list /etc/apt/sources.list.d/*.sources; do
      [ -f "$f" ] && sed -i -E 's#https?://(deb|security)\.debian\.org#http://archive.debian.org#g' "$f"
    done
    apt-get update -qq -o Acquire::Check-Valid-Until=false 2> "$err" && return 0
  fi
  echo "warning: apt-get update reported errors for the OS repositories; continuing:" >&2
  sed 's/^/    /' "$err" >&2
  APT_BASE_NOTE="apt-get update had failed: $(grep -m1 -E '^(E|W): ' "$err" | cut -c1-160)"
  return 0
}
# The pgEdge repository alone, so OS repository errors can't hide pgEdge errors (these are fatal).
# Extra arguments are passed to apt-get update.
apt_update_pgedge() {
  local src
  src=$(ls /etc/apt/sources.list.d/*pgedge* 2> /dev/null | head -1 || true)
  [ -n "$src" ] || { echo "error: pgedge-release did not add an apt source" >&2; return 1; }
  # apt-get update exits 0 even when a repository can't be fetched ("W: Failed to fetch ...
  # old ones used instead"), so check its messages too
  local out
  out=$(apt-get update -o Dir::Etc::sourcelist="$src" -o Dir::Etc::sourceparts=/dev/null -o APT::Get::List-Cleanup=0 "$@" 2>&1) \
    || { printf '%s\n' "$out" >&2; return 1; }
  if printf '%s\n' "$out" | grep -qE '^(E:|Err:|W: Failed to fetch|W: Some index files failed)'; then
    printf '%s\n' "$out" | grep -E '^(E:|Err:|W:)' >&2
    echo "error: could not refresh the pgEdge repository" >&2
    return 1
  fi
}
# <<< apt-helpers
echo "==> $PRETTY_NAME ($(uname -m)): Postgres $PG, meta-package: $SCOPE"

case "$ID" in
  rocky|almalinux|ol|rhel)
    MAJOR="${VERSION_ID%%.*}"
    dnf install -y -q dnf-plugins-core
    case "$ID" in
      ol)
        dnf install -y -q "https://dl.fedoraproject.org/pub/epel/epel-release-latest-${MAJOR}.noarch.rpm"
        dnf config-manager --set-enabled "ol${MAJOR}_codeready_builder" ;;
      rhel)
        dnf install -y -q "https://dl.fedoraproject.org/pub/epel/epel-release-latest-${MAJOR}.noarch.rpm"
        subscription-manager repos --enable "codeready-builder-for-rhel-${MAJOR}-$(uname -m)-rpms" ;;
      *)
        dnf install -y -q epel-release
        dnf config-manager --set-enabled crb ;;
    esac
    dnf install -y -q https://dnf.pgedge.com/reporpm/pgedge-release-latest.noarch.rpm
    PKGS="$EXTRA"
    [ "$SCOPE" != "none" ] && PKGS="pgedge-enterprise-${SCOPE}_${PG} $PKGS"
    # shellcheck disable=SC2086
    dnf install -y $PKGS
    BIN="/usr/pgsql-${PG}/bin"
    DATA="/var/lib/pgsql/${PG}/data"
    if [ -x "$BIN/initdb" ] && [ -x "$BIN/postgres" ]; then
      # No systemd in containers: initdb and pg_ctl directly
      su postgres -c "$BIN/initdb -D $DATA" > /dev/null
      su postgres -c "$BIN/pg_ctl -D $DATA -l /tmp/pg.log -w start"
      PSQL="$BIN/psql"
    fi
    ;;
  debian|ubuntu)
    export DEBIAN_FRONTEND=noninteractive
    apt_update_base
    apt-get install -y -qq curl gnupg2 lsb-release ca-certificates > /dev/null
    curl -fsSL -o /tmp/pgedge-release.deb \
      https://apt.pgedge.com/repodeb/pgedge-release_latest_all.deb
    apt-get install -y -qq /tmp/pgedge-release.deb > /dev/null
    apt_update_pgedge
    PKGS="$EXTRA"
    [ "$SCOPE" != "none" ] && PKGS="pgedge-enterprise-${SCOPE}-${PG} $PKGS"
    # shellcheck disable=SC2086
    apt-get install -y $PKGS
    if [ -x "/usr/lib/postgresql/${PG}/bin/postgres" ]; then
      # The server package creates the "main" cluster; start it (no systemd here)
      pg_ctlcluster "$PG" main start 2>/dev/null || true
      pg_lsclusters
      PSQL="psql"
    fi
    ;;
  *)
    echo "Unsupported distribution: $ID" >&2; exit 2 ;;
esac

echo "==> Installed pgEdge packages"
if command -v rpm > /dev/null; then rpm -qa 'pgedge-*' | sort; else dpkg -l 'pgedge-*' | awk '/^ii/{print $2" "$3}'; fi

if [ -n "${PSQL:-}" ]; then
  echo "==> Server check"
  su postgres -c "$PSQL -Atc 'SELECT version();'"
  su postgres -c "$PSQL -Atc \"SELECT 'available extensions: ' || coalesce(string_agg(name, ', ' ORDER BY name), 'none')
    FROM pg_available_extensions
    WHERE name IN ('spock','snowflake','lolor','vector','postgis','pgaudit','vectorizer');\""
else
  echo "==> No Postgres server package selected; skipped the database check"
fi
echo "==> PASS"

# Installs pgEdge packages from dnf.pgedge.com / apt.pgedge.com
# in clean containers on every supported distro, arch and Postgres version.
name: pgEdge repo smoke test

on:
  workflow_dispatch:
  pull_request:
  schedule:
    - cron: "0 3 * * *"   # nightly, catches repo regressions early

jobs:
  smoke:
    name: ${{ matrix.image }} / ${{ matrix.arch }} / PG ${{ matrix.pg }}
    runs-on: ${{ matrix.arch == 'arm64' && 'ubuntu-24.04-arm' || 'ubuntu-24.04' }}
    timeout-minutes: 30
    strategy:
      fail-fast: false
      matrix:
        arch: [amd64, arm64]
        pg: [16, 17, 18]
        image:
          - rockylinux/rockylinux:9
          - rockylinux/rockylinux:10
          - almalinux:9
          - almalinux:10
          - oraclelinux:9
          - oraclelinux:10
          - debian:bookworm
          - debian:trixie
          - ubuntu:22.04
          - ubuntu:24.04
          - ubuntu:26.04
    steps:
      - uses: actions/checkout@v4

      - name: Install from the pgEdge repo and start Postgres
        run: |
          docker run --rm \
            -e PG=${{ matrix.pg }} \
            -e SCOPE=postgres \
            -v "$PWD/ci/pgedge-smoke.sh:/pgedge-smoke.sh:ro" \
            ${{ matrix.image }} bash /pgedge-smoke.sh

  # Heavier check: the full "all components" meta-package on the newest PG.
  full:
    runs-on: ubuntu-24.04
    strategy:
      fail-fast: false
      matrix:
        image: [rockylinux/rockylinux:9, debian:bookworm, ubuntu:24.04]
    steps:
      - uses: actions/checkout@v4
      - run: |
          docker run --rm -e PG=18 -e SCOPE=all \
            -v "$PWD/ci/pgedge-smoke.sh:/pgedge-smoke.sh:ro" \
            ${{ matrix.image }} bash /pgedge-smoke.sh