Add nvidia driver role w/testing

Fix nvidia role for clean installs and make hyprland nvidia env vars conditional
- Add nouveau blacklist to prevent driver conflicts
- Remove redundant DKMS install task (AUR package handles this automatically)
- Fix mkinitcpio MODULES to append nvidia modules instead of overwriting
- Add missing initramfs rebuild notification to modprobe.d template
- Make hyprland NVIDIA environment variables conditional via hyprland_nvidia flag

Co-authored-by: Claude.ai

6 files changed, 80 insertions, 0 deletions

diff --git a/roles/nvidia/defaults/main.yml b/roles/nvidia/defaults/main.yml
new file mode 100644
index 00000000..8fc2a510
--- /dev/null
+++ b/roles/nvidia/defaults/main.yml
@@ -0,0 +1,15 @@
+---
+nvidia_aur_packages:
+  - "nvidia-580xx-dkms"
+  - "nvidia-580xx-utils"
+
+nvidia_packages:
+  - "linux-headers"
+  - "nvidia-settings"
+  - "libva-nvidia-driver"
+
+nvidia_modules:
+  - "nvidia"
+  - "nvidia_modeset"
+  - "nvidia_uvm"
+  - "nvidia_drm"
diff --git a/roles/nvidia/handlers/main.yml b/roles/nvidia/handlers/main.yml
new file mode 100644
index 00000000..67408c79
--- /dev/null
+++ b/roles/nvidia/handlers/main.yml
@@ -0,0 +1,6 @@
+---
+- name: "Rebuild initramfs"
+  become: true
+  ansible.builtin.command:
+    cmd: "mkinitcpio -P"
+  changed_when: true
diff --git a/roles/nvidia/meta/main.yml b/roles/nvidia/meta/main.yml
new file mode 100644
index 00000000..cc0ec526
--- /dev/null
+++ b/roles/nvidia/meta/main.yml
@@ -0,0 +1,12 @@
+---
+dependencies:
+  - role: "yay"
+galaxy_info:
+  author: "a14m"
+  description: "Install and configure NVIDIA proprietary drivers"
+  license: "MIT"
+  min_ansible_version: "2.18"
+  platforms:
+    - name: "ArchLinux"
+      versions:
+        - "all"
diff --git a/roles/nvidia/tasks/install-archlinux.yml b/roles/nvidia/tasks/install-archlinux.yml
new file mode 100644
index 00000000..cbcc2221
--- /dev/null
+++ b/roles/nvidia/tasks/install-archlinux.yml
@@ -0,0 +1,40 @@
+---
+- name: "Install NVIDIA packages"
+  become: true
+  ansible.builtin.package:
+    name: "{{ nvidia_packages }}"
+    state: "present"
+
+- name: "Install NVIDIA driver from AUR"
+  become: true
+  become_user: "aur_builder"
+  kewlfft.aur.aur:
+    name: "{{ nvidia_aur_packages }}"
+    state: "present"
+    use: "yay"
+
+- name: "Configure NVIDIA DRM kernel mode setting"
+  become: true
+  ansible.builtin.template:
+    src: "nvidia.conf.j2"
+    dest: "/etc/modprobe.d/nvidia.conf"
+    mode: "0644"
+  notify: "Rebuild initramfs"
+
+- name: "Get current mkinitcpio MODULES"
+  become: true
+  ansible.builtin.shell:
+    cmd: "grep '^MODULES=' /etc/mkinitcpio.conf | sed 's/MODULES=(\\(.*\\))/\\1/'"
+  register: "current_modules"
+  changed_when: false
+
+- name: "Add NVIDIA modules to mkinitcpio"
+  become: true
+  vars:
+    existing_modules: "{{ current_modules.stdout.split() | reject('in', nvidia_modules) | list }}"
+    all_modules: "{{ existing_modules + nvidia_modules }}"
+  ansible.builtin.lineinfile:
+    path: "/etc/mkinitcpio.conf"
+    regexp: "^MODULES=\\(.*\\)"
+    line: "MODULES=({{ all_modules | join(' ') }})"
+  notify: "Rebuild initramfs"
diff --git a/roles/nvidia/tasks/main.yml b/roles/nvidia/tasks/main.yml
new file mode 100644
index 00000000..71bcbd0d
--- /dev/null
+++ b/roles/nvidia/tasks/main.yml
@@ -0,0 +1,3 @@
+---
+- name: "Include OS-specific tasks"
+  ansible.builtin.include_tasks: "install-{{ ansible_facts['os_family'] | lower }}.yml"
diff --git a/roles/nvidia/templates/nvidia.conf.j2 b/roles/nvidia/templates/nvidia.conf.j2
new file mode 100644
index 00000000..9672977e
--- /dev/null
+++ b/roles/nvidia/templates/nvidia.conf.j2
@@ -0,0 +1,4 @@
+blacklist nouveau
+
+options nvidia_drm modeset=1
+options nvidia_drm fbdev=1