feat: fixes for beta support

Signed-off-by: Niccolò Fei <niccolo.fei@enterprisedb.com>

Author: NiccoloFei

Dockerfile

@@ -2,13 +2,13 @@ ARG BASE=debian:12.10-slim
 FROM $BASE AS minimal
 
 ARG PG_VERSION
-ARG PG_MAJOR=${PG_VERSION%%.*}
+ARG PG_MAJOR
 
 ENV PATH=$PATH:/usr/lib/postgresql/$PG_MAJOR/bin
 
 RUN apt-get update && \
     apt-get install -y --no-install-recommends postgresql-common ca-certificates gnupg && \
-    /usr/share/postgresql-common/pgdg/apt.postgresql.org.sh -y && \
+    /usr/share/postgresql-common/pgdg/apt.postgresql.org.sh -y -c "${PG_MAJOR}" && \
     apt-get install -y --no-install-recommends -o Dpkg::::="--force-confdef" -o Dpkg::::="--force-confold" postgresql-common && \
     sed -ri 's/#(create_main_cluster) .*$/\1 = false/' /etc/postgresql-common/createcluster.conf && \
     apt-get install -y --no-install-recommends \
@@ -21,13 +21,10 @@ USER 26
 
 
 FROM minimal AS standard
-
+ARG EXTENSIONS
 USER root
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends locales-all \
-      "postgresql-${PG_MAJOR}-pgaudit" \
-      "postgresql-${PG_MAJOR}-pgvector" \
-      "postgresql-${PG_MAJOR}-pg-failover-slots" && \
+    apt-get install -y --no-install-recommends locales-all "${EXTENSIONS}" && \
     apt-get purge -y --auto-remove -o APT::AutoRemove::RecommendsImportant=false && \
     rm -rf /var/lib/apt/lists/* /var/cache/* /var/log/*
 

docker-bake.hcl

@@ -20,6 +20,12 @@ now = timestamp()
 authors = "The CloudNativePG Contributors"
 url = "https://github.com/cloudnative-pg/postgres-containers"
 
+extensions = [
+  "pgaudit",
+  "pgvector",
+  "pg-failover-slots"
+]
+
 target "default" {
   matrix = {
     tgt = [
@@ -56,7 +62,9 @@ target "default" {
   target = "${tgt}"
   args = {
     PG_VERSION = "${pgVersion}"
+    PG_MAJOR = "${getMajor(pgVersion)}"
     BASE = "${base}"
+    EXTENSIONS = "${getExtensionsString(pgVersion, extensions)}"
   }
   attest = [
     "type=provenance,mode=max",
@@ -113,3 +121,18 @@ function cleanVersion {
     params = [ version ]
     result = replace(version, "~", "")
 }
+
+function isBeta {
+    params = [ version ]
+    result = (length(regexall("[0-9]+~beta.*", version)) > 0) ? true : false
+}
+
+function getMajor {
+    params = [ version ]
+    result = (isBeta(version) == true) ? index(split("~", version),0) : index(split(".", version),0)
+}
+
+function getExtensionsString {
+    params = [ version, extensions ]
+    result = (isBeta(version) == true) ? "" : join(" ", formatlist("postgresql-%s-%s", getMajor(version), extensions))
+}