From b283617f517f47651d3be3bff6a598c0f3f81864 Mon Sep 17 00:00:00 2001
From: Dominik Dancs <dominik@dancs.sk>
Date: Sat, 18 May 2024 18:50:42 +0200
Subject: [PATCH] Reworked entrypoint script

---
 docker-entrypoint.sh | 89 ++++++++++++++------------------------------
 1 file changed, 28 insertions(+), 61 deletions(-)

diff --git a/docker-entrypoint.sh b/docker-entrypoint.sh
index 45a3848..b7c33c8 100644
--- a/docker-entrypoint.sh
+++ b/docker-entrypoint.sh
@@ -1,82 +1,49 @@
-#!/bin/sh
+#! /bin/sh
 
 set -euo pipefail
 
-# Configure mysql env vars
-: "${PDNS_gmysql_host:=${MYSQL_ENV_MYSQL_HOST:-mysql}}"
-: "${PDNS_gmysql_port:=${MYSQL_ENV_MYSQL_PORT:-3306}}"
-: "${PDNS_gmysql_user:=${MYSQL_ENV_MYSQL_USER:-root}}"
-if [ "${PDNS_gmysql_user}" = 'root' ]; then
-    : "${PDNS_gmysql_password:=${MYSQL_ENV_MYSQL_ROOT_PASSWORD:-}}"
-fi
-: "${PDNS_gmysql_password:=${MYSQL_ENV_MYSQL_PASSWORD:-powerdns}}"
-: "${PDNS_gmysql_dbname:=${MYSQL_ENV_MYSQL_DATABASE:-powerdns}}"
-
-# use first part of node name as database name suffix
-if [ "${NODE_NAME:-}" ]; then
-    NODE_NAME=$(echo ${NODE_NAME} | sed -e 's/\..*//' -e 's/-//')
-    PDNS_gmysql_dbname="${PDNS_gmysql_dbname}${NODE_NAME}"
-fi
+if [ "${PDNS_gmysql_host:-}" != "" ]; then
+    echo "Running migrations on the database..."
 
+    SQL_COMMAND="mysql -h ${PDNS_gmysql_host} -P ${PDNS_gmysql_port} -u ${PDNS_gmysql_user} -p${PDNS_gmysql_password} -D ${PDNS_gmysql_dbname}"
 
-export PDNS_gmysql_host PDNS_gmysql_port PDNS_gmysql_user PDNS_gmysql_password PDNS_gmysql_dbname
+    # Fix DB schema
+    sed -i 's/CREATE TABLE/CREATE TABLE IF NOT EXISTS/g ; s/CREATE UNIQUE INDEX/CREATE UNIQUE INDEX IF NOT EXISTS/g ; s/CREATE INDEX/CREATE INDEX IF NOT EXISTS/g' /usr/share/doc/pdns/schema.mysql.sql
 
-EXTRA=""
+    # Initialize DB if needed
+    $SQL_COMMAND </usr/share/doc/pdns/schema.mysql.sql
 
-# Password Auth
-if [ "${PDNS_gmysql_password}" != "" ]; then
-    EXTRA="${EXTRA} -p${PDNS_gmysql_password}"
+    echo "Done."
 fi
 
-# Allow socket connections
-if [ "${PDNS_gmysql_socket:-}" != "" ]; then
-    export PDNS_gmysql_host="localhost"
-    EXTRA="${EXTRA} --socket=${PDNS_gmysql_socket}"
-fi
+if [ "${PDNS_gpgsql_host:-}" != "" ]; then
+    echo "Running migrations on the database..."
 
-MYSQL_COMMAND="mysql -h ${PDNS_gmysql_host} -P ${PDNS_gmysql_port} -u ${PDNS_gmysql_user}${EXTRA}"
+    export PGPASSWORD=${PDNS_gpgsql_password}
+    SQL_COMMAND="psql -h ${PDNS_gpgsql_host} -p ${PDNS_gpgsql_port} -U ${PDNS_gpgsql_user} -w ${PDNS_gpgsql_dbname}"
 
-# Wait for MySQL to respond
-until $MYSQL_COMMAND -e ';' ; do
-    >&2 echo 'MySQL is unavailable - sleeping'
-    sleep 3
-done
+    # Fix DB schema
+    sed -i 's/CREATE TABLE/CREATE TABLE IF NOT EXISTS/g ; s/CREATE UNIQUE INDEX/CREATE UNIQUE INDEX IF NOT EXISTS/g ; s/CREATE INDEX/CREATE INDEX IF NOT EXISTS/g' /usr/share/doc/pdns/schema.pgsql.sql
 
-# Initialize DB if needed
-if [ "${SKIP_DB_CREATE:-false}" != 'true' ]; then
-    $MYSQL_COMMAND -e "CREATE DATABASE IF NOT EXISTS ${PDNS_gmysql_dbname}"
-fi
+    # Initialize DB
+    $SQL_COMMAND </usr/share/doc/pdns/schema.pgsql.sql
 
-MYSQL_CHECK_IF_HAS_TABLE="SELECT COUNT(DISTINCT table_name) FROM information_schema.columns WHERE table_schema = '${PDNS_gmysql_dbname}';"
-MYSQL_NUM_TABLE=$($MYSQL_COMMAND --batch --skip-column-names -e "$MYSQL_CHECK_IF_HAS_TABLE")
-if [ "$MYSQL_NUM_TABLE" -eq 0 ]; then
-    $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" < /usr/share/doc/pdns/schema.mysql.sql
+    echo "Done."
 fi
 
-if [ "${PDNS_superslave:-no}" == "yes" ]; then
-    # Configure supermasters if needed
-    if [ "${SUPERMASTER_IPS:-}" ]; then
-        $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" -e "TRUNCATE supermasters;"
-        MYSQL_INSERT_SUPERMASTERS=''
-        if [ "${SUPERMASTER_COUNT:-0}" == "0" ]; then
-            SUPERMASTER_COUNT=10
-        fi
-        i=1; while [ $i -le ${SUPERMASTER_COUNT} ]; do
-            SUPERMASTER_HOST=$(echo ${SUPERMASTER_HOSTS:-} | awk -v col="$i" '{ print $col }')
-            SUPERMASTER_IP=$(echo ${SUPERMASTER_IPS} | awk -v col="$i" '{ print $col }')
-            if [ -z "${SUPERMASTER_HOST:-}" ]; then
-                SUPERMASTER_HOST=$(hostname -f)
-            fi
-            if [ "${SUPERMASTER_IP:-}" ]; then
-                MYSQL_INSERT_SUPERMASTERS="${MYSQL_INSERT_SUPERMASTERS} INSERT INTO supermasters VALUES('${SUPERMASTER_IP}', '${SUPERMASTER_HOST}', 'admin');"
-            fi
-            i=$(( i + 1 ))
-        done
-        $MYSQL_COMMAND -D "$PDNS_gmysql_dbname" -e "$MYSQL_INSERT_SUPERMASTERS"
+if [ "${PDNS_autosecondary:-no}" == "yes" ]; then
+    # Configure masters for secondary if needed
+    if [ "${SUPERMASTER_IPS:-}" != "" ]; then
+        echo "Inserting masters for autosecondary..."
+        echo "TRUNCATE supermasters;" | $SQL_COMMAND
+
+        SUPERMASTERS=$($SUPERMASTER_IPS | sed 's/,/"), ("/g ; s/^/("/ ; s/$/")/ ; s/:/","/g')
+        echo "INSERT INTO supermasters VALUES $SUPERMASTERS;" | $SQL_COMMAND
+        echo "Done."
     fi
 fi
 
 # Create config file from template
-envtpl < /pdns.conf.tpl > /etc/pdns/pdns.conf
+envtpl </pdns.conf.tpl >/etc/pdns/pdns.conf
 
 exec "$@"