6 files changed, 82 insertions, 0 deletions
diff --git a/roles/nvidia_gtx1060/defaults/main.yml b/roles/nvidia_gtx1060/defaults/main.yml
new file mode 100644
index 00000000..6bdba8ed
--- /dev/null
+++ b/roles/nvidia_gtx1060/defaults/main.yml
@@ -0,0 +1,14 @@
+---
+nvidia_gtx1060_packages:
+  - "libva-nvidia-driver"
+
+nvidia_gtx1060_aur_packages:
+  - "nvidia-580xx-dkms"
+  - "nvidia-580xx-utils"
+  - "nvidia-580xx-settings"
+
+nvidia_gtx1060_modules:
+  - "nvidia"
+  - "nvidia_modeset"
+  - "nvidia_uvm"
+  - "nvidia_drm"
diff --git a/roles/nvidia_gtx1060/handlers/main.yml b/roles/nvidia_gtx1060/handlers/main.yml
new file mode 100644
index 00000000..67408c79
--- /dev/null
+++ b/roles/nvidia_gtx1060/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_gtx1060/meta/main.yml b/roles/nvidia_gtx1060/meta/main.yml
new file mode 100644
index 00000000..6bdd4b5c
--- /dev/null
+++ b/roles/nvidia_gtx1060/meta/main.yml
@@ -0,0 +1,13 @@
+---
+dependencies:
+  - role: "linux_headers"
+  - role: "yay"
+galaxy_info:
+  author: "a14m"
+  description: "Install and configure NVIDIA GTX 1060 proprietary drivers"
+  license: "MIT"
+  min_ansible_version: "2.18"
+  platforms:
+    - name: "ArchLinux"
+      versions:
+        - "all"
diff --git a/roles/nvidia_gtx1060/tasks/install-archlinux.yml b/roles/nvidia_gtx1060/tasks/install-archlinux.yml
new file mode 100644
index 00000000..488dd15d
--- /dev/null
+++ b/roles/nvidia_gtx1060/tasks/install-archlinux.yml
@@ -0,0 +1,42 @@
+---
+- name: "Install libva NVIDIA driver"
+  become: true
+  ansible.builtin.package:
+    name: "{{ nvidia_gtx_1060_packages }}"
+    state: "present"
+
+- name: "Install NVIDIA driver from AUR"
+  become: true
+  become_user: "aur_builder"
+  kewlfft.aur.aur:
+    name: "{{ nvidia_gtx_1060_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: |
+    set -euo pipefail
+    grep '^MODULES=' /etc/mkinitcpio.conf | sed 's/MODULES=(\(.*\))/\1/'
+  # noqa var-naming[no-role-prefix] as it describes the current kernel modules and not the nvidia modules
+  register: "current_kernel_modules"
+  changed_when: false
+
+- name: "Add NVIDIA modules to mkinitcpio"
+  become: true
+  vars:
+    existing_modules: "{{ current_kernel_modules.stdout.split() | reject('in', nvidia_gtx_1060_modules) | list }}"
+    all_modules: "{{ existing_modules + nvidia_gtx_1060_modules }}"
+  ansible.builtin.lineinfile:
+    path: "/etc/mkinitcpio.conf"
+    regexp: "^MODULES=\\(.*\\)"
+    line: "MODULES=({{ all_modules | join(' ') }})"
+  notify: "Rebuild initramfs"
diff --git a/roles/nvidia_gtx1060/tasks/main.yml b/roles/nvidia_gtx1060/tasks/main.yml
new file mode 100644
index 00000000..71bcbd0d
--- /dev/null
+++ b/roles/nvidia_gtx1060/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_gtx1060/templates/nvidia.conf.j2 b/roles/nvidia_gtx1060/templates/nvidia.conf.j2
new file mode 100644
index 00000000..9672977e
--- /dev/null
+++ b/roles/nvidia_gtx1060/templates/nvidia.conf.j2
@@ -0,0 +1,4 @@
+blacklist nouveau
+
+options nvidia_drm modeset=1
+options nvidia_drm fbdev=1