From 29f418de57d3d2104152f6bbbe94e00a233a0608 Mon Sep 17 00:00:00 2001
From: Manuel Romero <mjromper@gmail.com>
Date: Thu, 11 Jul 2024 12:37:12 +0200
Subject: [PATCH] restart linux after rename

---
 vm-fromsnapshot-linux/main.tf                        |  5 +++--
 .../provision-scripts/rename-machine.sh              | 12 ++++++++++--
 vm-fromsnapshot-linux/variables.tf                   |  4 ++++
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/vm-fromsnapshot-linux/main.tf b/vm-fromsnapshot-linux/main.tf
index 920e40f..c8a9216 100644
--- a/vm-fromsnapshot-linux/main.tf
+++ b/vm-fromsnapshot-linux/main.tf
@@ -25,6 +25,7 @@ locals {
 
   snaploc = (var.location == "westeurope")? "weu" : (var.location == "eastus")? "" : "sea"
   storage_account_id = "/subscriptions/62ebff8f-c40b-41be-9239-252d6c0c8ad9/resourceGroups/QMI-Machines/providers/Microsoft.Storage/storageAccounts/machinesnapshots${local.snaploc}"
+  restartAfter = (var.restartAfterRename == true)? "YES" : "NO"
 }
 
 resource "azurerm_managed_disk" "md-import" {
@@ -154,11 +155,11 @@ resource "null_resource" "post-vm-fromsnapshot-linux" {
 
     inline = [
         "echo ${var.initial_password} | sudo -S chmod a+x /home/${local.admin_username}/provision-scripts/*.sh",
-        "sudo /home/${local.admin_username}/provision-scripts/rename-machine.sh ${local.virtual_machine_name}",
         "sudo /home/${local.admin_username}/provision-scripts/setnewpassword.sh ${local.admin_username} ${local.admin_password}",
+        "sudo /home/${local.admin_username}/provision-scripts/rename-machine.sh ${local.virtual_machine_name} ${local.restartAfter}",
     ]
   }
-  
+    
 }
 
 
diff --git a/vm-fromsnapshot-linux/provision-scripts/rename-machine.sh b/vm-fromsnapshot-linux/provision-scripts/rename-machine.sh
index faf7fd6..86be87d 100644
--- a/vm-fromsnapshot-linux/provision-scripts/rename-machine.sh
+++ b/vm-fromsnapshot-linux/provision-scripts/rename-machine.sh
@@ -1,4 +1,12 @@
 #!/bin/bash
 
-echo "Renaming machine to $1"
-hostnamectl set-hostname $1
\ No newline at end of file
+echo "--- Renaming machine to $1"
+echo "--- Restart after renamed: $2"
+hostnamectl set-hostname $1
+
+if [ "$2" = "YES" ]; then 
+    echo "--- Linux machine will restart in 1 minute...." 
+    shutdown -r +1
+else
+    echo "--- No need to restart linux machine."  
+fi;
\ No newline at end of file
diff --git a/vm-fromsnapshot-linux/variables.tf b/vm-fromsnapshot-linux/variables.tf
index e2e1e3f..0b930dd 100644
--- a/vm-fromsnapshot-linux/variables.tf
+++ b/vm-fromsnapshot-linux/variables.tf
@@ -69,4 +69,8 @@ variable "startupTime"{
   default = null
 }
 
+variable "restartAfterRename" {
+  default = false
+}
+