From cbaa6168d04c8fbeaebd619cb9ba2663ec62428e Mon Sep 17 00:00:00 2001
From: Ilia Ross <ilia@ross.gdn>
Date: Sun, 20 Aug 2023 17:18:22 +0300
Subject: [PATCH] Add support for cloud-init to preserve the hostname after
 reboots

---
 slib.sh | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/slib.sh b/slib.sh
index 66a5866..d35c145 100644
--- a/slib.sh
+++ b/slib.sh
@@ -471,6 +471,17 @@ detect_ip () {
   fi
 }
 
+# Set the hostname in cloud-init
+set_hostname_cloud () {
+  # If cloud-init is installed, preserve the hostname Virtualmin sets
+  if [ -f "/etc/cloud/cloud.cfg" ]; then
+    if grep "^preserve_hostname: false" /etc/cloud/cloud.cfg >/dev/null; then
+      log_debug "Setting preserve_hostname to true in /etc/cloud/cloud.cfg"
+      sed -i "s/^preserve_hostname: false/preserve_hostname: true/" /etc/cloud/cloud.cfg
+    fi
+  fi
+}
+
 # Set the hostname
 set_hostname () {
   local i=0
@@ -501,6 +512,7 @@ set_hostname () {
       hostname "$line"
       echo "$line" > /etc/hostname
       hostnamectl set-hostname "$line" 1>/dev/null 2>&1
+      set_hostname_cloud
       detect_ip
       shortname=$(echo "$line" | cut -d"." -f1)
       if grep "^$address" /etc/hosts >/dev/null; then
@@ -526,6 +538,10 @@ is_fully_qualified () {
       log_warning "Hostname cannot be *.localdomain."
       return 1
       ;;
+    *.internal)
+      log_warning "Hostname cannot be *.internal."
+      return 1
+      ;;
     *.*)
       log_debug "Hostname is fully qualified as $1"
       return 0