From 03f57ae414f56dc8758a7e8fb2b3b544b7ce39a6 Mon Sep 17 00:00:00 2001
From: Francois Andrieu <darknao@fedoraproject.org>
Date: Thu, 10 Aug 2023 23:42:26 +0200
Subject: [PATCH] add btrfs role

---
 playbooks/groups/buildvm.yml           |  5 +++-
 roles/btrfs/defaults/main.yml          |  2 ++
 roles/btrfs/handlers/main.yml          |  5 ++++
 roles/btrfs/tasks/main.yml             | 34 ++++++++++++++++++++++++++
 roles/btrfs/templates/schedule.conf.j2 |  2 ++
 5 files changed, 47 insertions(+), 1 deletion(-)
 create mode 100644 roles/btrfs/defaults/main.yml
 create mode 100644 roles/btrfs/handlers/main.yml
 create mode 100644 roles/btrfs/tasks/main.yml
 create mode 100644 roles/btrfs/templates/schedule.conf.j2

diff --git a/playbooks/groups/buildvm.yml b/playbooks/groups/buildvm.yml
index 14e2cab742..487cf42574 100644
--- a/playbooks/groups/buildvm.yml
+++ b/playbooks/groups/buildvm.yml
@@ -31,7 +31,8 @@
         when: "env == 'staging' or createrepo or 'runroot' in group_names and not inventory_hostname.startswith('buildvm-s390x')",
       }
     - ipa/client
-    - { role: sudo, when: not inventory_hostname.startswith('bkernel') and env == 'production' }
+    - role: sudo
+      when: not inventory_hostname.startswith('bkernel') and env == 'production'
     - koji_builder
     - role: nfs/client
       mnt_dir: '/mnt/koji/ostree'
@@ -85,6 +86,8 @@
       host: "odcs{{ env_suffix }}.fedoraproject.org"
       kt_location: /etc/kojid/secrets/odcs_inner.keytab
       when: env == "staging"
+    - role: btrfs
+      btrfs_balance_period: weekly
 
   tasks:
     - import_tasks: "{{ tasks_path }}/motd.yml"
diff --git a/roles/btrfs/defaults/main.yml b/roles/btrfs/defaults/main.yml
new file mode 100644
index 0000000000..18ebbbd61c
--- /dev/null
+++ b/roles/btrfs/defaults/main.yml
@@ -0,0 +1,2 @@
+---
+btrfs_balance_period: weekly
diff --git a/roles/btrfs/handlers/main.yml b/roles/btrfs/handlers/main.yml
new file mode 100644
index 0000000000..c55eee4c16
--- /dev/null
+++ b/roles/btrfs/handlers/main.yml
@@ -0,0 +1,5 @@
+---
+- name: restart btrfs-balance
+  service:
+    name: btrfs-balance.timer
+    state: restarted
diff --git a/roles/btrfs/tasks/main.yml b/roles/btrfs/tasks/main.yml
new file mode 100644
index 0000000000..41a59bafb4
--- /dev/null
+++ b/roles/btrfs/tasks/main.yml
@@ -0,0 +1,34 @@
+---
+- name: Install btrfsmaintenance package
+  package:
+    state: present
+    name:
+    - btrfsmaintenance
+
+
+- name: Create /etc/systemd/system/btrfs-balance.timer.d
+  file:
+    state: directory
+    path: /etc/systemd/system/btrfs-balance.timer.d
+    owner: root
+    group: root
+    mode: 0755
+
+
+- name: Configure btrfs-balance timer
+  template:
+    src: schedule.conf.j2
+    dest: /etc/systemd/system/btrfs-balance.timer.d/schedule.conf
+    owner: root
+    group: root
+    mode: 0644
+  notify:
+  - reload systemd
+  - restart btrfs-balance
+
+
+- name: Ensure btrfs-balance is enabled and started
+  service:
+    name: btrfs-balance.timer
+    enabled: yes
+    state: started
diff --git a/roles/btrfs/templates/schedule.conf.j2 b/roles/btrfs/templates/schedule.conf.j2
new file mode 100644
index 0000000000..d39f65ea21
--- /dev/null
+++ b/roles/btrfs/templates/schedule.conf.j2
@@ -0,0 +1,2 @@
+[Timer]
+OnCalendar={{ btrfs_balance_period }}