From 7ec16a8d53363ace979e5585323311ab1bd1641d Mon Sep 17 00:00:00 2001
From: laszlojau <49835454+laszlojau@users.noreply.github.com>
Date: Sat, 6 Apr 2024 07:24:57 +1030
Subject: [PATCH] Keep service backups under /etc/systemd/system (#324)

Signed-off-by: laszlojau <49835454+laszlojau@users.noreply.github.com>
---
 roles/k3s_upgrade/tasks/main.yml | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/roles/k3s_upgrade/tasks/main.yml b/roles/k3s_upgrade/tasks/main.yml
index fb66633..6c1d82d 100644
--- a/roles/k3s_upgrade/tasks/main.yml
+++ b/roles/k3s_upgrade/tasks/main.yml
@@ -15,10 +15,20 @@
 - name: Update node only if needed
   when: installed_k3s_version is version(k3s_version, '<')
   block:
+    - name: Find K3s service files
+      ansible.builtin.find:
+        paths: "{{ systemd_dir }}"
+        patterns: "k3s*.service"
+      register: k3s_service_files
+
     - name: Save current K3s service
-      ansible.builtin.shell:
-        cmd: "cp {{ systemd_dir }}/k3s*.service /tmp/"
-      changed_when: true
+      ansible.builtin.copy:
+        src: "{{ item.path }}"
+        dest: "{{ item.path }}.bak"
+        remote_src: true
+        mode: preserve
+        force: true
+      loop: "{{ k3s_service_files.files }}"
 
     - name: Install new K3s Version
       ansible.builtin.command:
@@ -29,9 +39,19 @@
       changed_when: true
 
     - name: Restore K3s service
-      ansible.builtin.shell:
-        cmd: "mv /tmp/k3s*.service {{ systemd_dir }}/"
-      changed_when: true
+      ansible.builtin.copy:
+        src: "{{ item.path }}.bak"
+        dest: "{{ item.path }}"
+        remote_src: true
+        mode: preserve
+        force: true
+      loop: "{{ k3s_service_files.files }}"
+
+    - name: Clean up temporary K3s service backups
+      ansible.builtin.file:
+        path: "{{ item.path }}.bak"
+        state: absent
+      loop: "{{ k3s_service_files.files }}"
 
     - name: Restart K3s service [server]
       when: "'server' in group_names"