From 8b3c8e506d12f3cefc44e34eae414f198f07c959 Mon Sep 17 00:00:00 2001
From: Ariejan de Vroom <ariejan@devroom.io>
Date: Thu, 6 Jun 2024 22:22:51 +0200
Subject: [PATCH] Enable Wayland and GPU drivers

---
 main.yml                                      |  5 +-
 roles/03_wayland/tasks/main.yml               | 71 +++++++++++++++++++
 .../tasks/main.yml                            |  0
 roles/{04_user => 07_user}/tasks/main.yml     |  0
 4 files changed, 74 insertions(+), 2 deletions(-)
 create mode 100644 roles/03_wayland/tasks/main.yml
 rename roles/{03_packages => 05_packages}/tasks/main.yml (100%)
 rename roles/{04_user => 07_user}/tasks/main.yml (100%)

diff --git a/main.yml b/main.yml
index a591706..2202cf5 100644
--- a/main.yml
+++ b/main.yml
@@ -4,5 +4,6 @@
   roles:
     - { role: 01_host_info } 
     - { role: 02_basics } 
-    - { role: 03_packages }
-    - { role: 04_user }
\ No newline at end of file
+    - { role: 03_wayland }
+    - { role: 05_packages }
+    - { role: 07_user }
\ No newline at end of file
diff --git a/roles/03_wayland/tasks/main.yml b/roles/03_wayland/tasks/main.yml
new file mode 100644
index 0000000..2f2692f
--- /dev/null
+++ b/roles/03_wayland/tasks/main.yml
@@ -0,0 +1,71 @@
+---
+- name: Install Xorg
+  community.general.pacman:
+    name:
+      - xorg-xwayland
+    state: present
+  become: true
+
+- name: Detect GPU
+  # ansible.builtin.shell: set -o pipefail && lspci -v | grep -A1 -e VGA -e 3D  # noqa command-instead-of-shell
+  ansible.builtin.shell: set -o pipefail && lspci | grep -i 'VGA\|3D\|display'  # noqa command-instead-of-shell
+  register: gpu_result
+  changed_when: no
+
+- name: Install GPU drivers (AMD)
+  community.general.pacman:
+    name:
+      - mesa
+      - mesa-utils
+      # https://wiki.archlinux.org/title/Xorg
+      - xf86-video-amdgpu
+      # https://wiki.archlinux.org/title/Hardware_video_acceleration
+      - mesa-vdpau
+      - libva-mesa-driver
+      # https://wiki.archlinux.org/title/Vulkan
+      - vulkan-radeon
+      # https://wiki.archlinux.org/title/GPGPU
+      - rocm-hip-runtime
+      - rocm-hip-sdk
+      - hip-runtime-amd
+      - rocm-core
+      - rocm-opencl-runtime
+      - rocm-opencl-sdk
+    state: present
+  become: true
+  when: ("'radeon' in gpu_result.stdout.lower()") or ("'amd/ati' in gpu_result.stdout.lower()")
+
+- name: Install GPU drivers (Intel)
+  community.general.pacman:
+    name:
+      - mesa
+      - mesa-utils
+      # https://wiki.archlinux.org/title/Xorg
+      - xf86-video-intel
+      # https://wiki.archlinux.org/title/Hardware_video_acceleration
+      - intel-media-driver
+      - libva-intel-driver
+      # https://wiki.archlinux.org/title/Vulkan
+      - vulkan-intel
+      # https://wiki.archlinux.org/title/GPGPU
+      - intel-compute-runtime
+      - opencl-clover-mesa
+    state: present
+  become: true
+  when: "'intel' in gpu_result.stdout.lower()"
+
+- name: Install GPU drivers (Nvidia)
+  community.general.pacman:
+    name:
+      - mesa
+      - mesa-utils
+      # https://wiki.archlinux.org/title/Xorg
+      - nvidia
+      # https://wiki.archlinux.org/title/Hardware_video_acceleration
+      - nvidia-utils
+      # https://wiki.archlinux.org/title/GPGPU
+      - cuda
+      - opencl-nvidia
+    state: present
+  become: true
+  when: "'nvidia' in gpu_result.stdout.lower()"
\ No newline at end of file
diff --git a/roles/03_packages/tasks/main.yml b/roles/05_packages/tasks/main.yml
similarity index 100%
rename from roles/03_packages/tasks/main.yml
rename to roles/05_packages/tasks/main.yml
diff --git a/roles/04_user/tasks/main.yml b/roles/07_user/tasks/main.yml
similarity index 100%
rename from roles/04_user/tasks/main.yml
rename to roles/07_user/tasks/main.yml