From 8e2850c3a1c7210c790edfd493f2d35d083b7e6c Mon Sep 17 00:00:00 2001
From: Adam Perkowski <adas1per@protonmail.com>
Date: Fri, 27 Sep 2024 16:41:37 +0200
Subject: [PATCH] Linux Neptune installation script

---
 core/tabs/system-setup/arch/linux-neptune.sh | 60 ++++++++++++++++++++
 core/tabs/system-setup/tab_data.toml         |  6 ++
 tui/src/state.rs                             |  1 +
 3 files changed, 67 insertions(+)
 create mode 100755 core/tabs/system-setup/arch/linux-neptune.sh

diff --git a/core/tabs/system-setup/arch/linux-neptune.sh b/core/tabs/system-setup/arch/linux-neptune.sh
new file mode 100755
index 00000000..d3f98d41
--- /dev/null
+++ b/core/tabs/system-setup/arch/linux-neptune.sh
@@ -0,0 +1,60 @@
+#! /bin/sh -e
+
+. ../../common-script.sh
+
+setUpRepos() {
+    if ! grep -q "^\s*$$jupiter-staging$$" /etc/pacman.conf; then
+	printf "%b\n" "${CYAN}Adding jupiter-staging to pacman repositories...${RC}"
+        echo "[jupiter-staging]" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+	echo "Server = https://steamdeck-packages.steamos.cloud/archlinux-mirror/\$repo/os/\$arch" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+	echo "SigLevel = Never" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+    fi
+    if ! grep -q "^\s*$$holo-staging$$" /etc/pacman.conf; then
+	printf "%b\n" "${CYAN}Adding holo-staging to pacman repositories...${RC}"
+	echo "[holo-staging]" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+	echo "Server = https://steamdeck-packages.steamos.cloud/archlinux-mirror/\$repo/os/\$arch" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+	echo "SigLevel = Never" | "$ESCALATION_TOOL" tee -a /etc/pacman.conf
+    fi
+}
+
+installKernel() {
+    if ! "$PACKAGER" -Q | grep -q "\blinux-neptune"; then
+	printf "%b\n" "${CYAN}Installing linux-neptune..."
+        "$ESCALATION_TOOL" "$PACKAGER" -Syyu --noconfirm
+        "$ESCALATION_TOOL" "$PACKAGER" -S --noconfirm linux-neptune linux-neptune-headers
+	"$ESCALATION_TOOL" mkinitcpio -P
+    else
+	printf "%b\n" "${GREEN}linux-neptune detected. Skipping installation.${RC}"
+    fi
+
+    if [ -f /etc/default/grub ]; then
+	printf "%b\n" "${CYAN}Updating GRUB...${RC}"
+	if ! grep -q '^UPDATEDEFAULT=' /etc/default/grub; then
+	    echo 'UPDATEDEFAULT=yes' | sudo tee -a /etc/default/grub
+	else
+   	    sudo sed -i 's/^UPDATEDEFAULT=.*/UPDATEDEFAULT=yes/' /etc/default/grub
+	fi
+	if [ -f /boot/grub/grub.cfg ]; then
+	    "$ESCALATION_TOOL" grub-mkconfig -o /boot/grub/grub.cfg
+        else
+	    printf "%b\n" "${RED}GRUB configuration file not found. Run grub-mkconfig manually.${RC}"
+	fi
+    else
+        printf "%b\n" "${RED}GRUB not detected. Manually set your bootloader to use linux-neptune.${RC}"
+    fi
+}
+
+copyFirmwareFiles() {
+    printf "%b\n" "${CYAN}Copying firmware files...${RC}"
+    "$ESCALATION_TOOL" mkdir -p /usr/lib/firmware/cirrus
+    "$ESCALATION_TOOL" cp /usr/lib/firmware/cs35l41-dsp1-spk-cali.bin /usr/lib/firmware/cirrus/
+    "$ESCALATION_TOOL" cp /usr/lib/firmware/cs35l41-dsp1-spk-cali.wmfw  /usr/lib/firmware/cirrus/
+    "$ESCALATION_TOOL" cp /usr/lib/firmware/cs35l41-dsp1-spk-prot.bin  /usr/lib/firmware/cirrus/
+    "$ESCALATION_TOOL" cp /usr/lib/firmware/cs35l41-dsp1-spk-prot.wmfw  /usr/lib/firmware/cirrus/
+}
+
+checkEnv
+checkEscalationTool
+setUpRepos
+installKernel
+copyFirmwareFiles
diff --git a/core/tabs/system-setup/tab_data.toml b/core/tabs/system-setup/tab_data.toml
index a68d2fdf..6a413902 100644
--- a/core/tabs/system-setup/tab_data.toml
+++ b/core/tabs/system-setup/tab_data.toml
@@ -15,6 +15,12 @@ description = "This command installs a minimal arch server setup under 5 minutes
 script = "arch/server-setup.sh"
 task_list = "SI D"
 
+[[data.entries]]
+name ="Linux Neptune for SteamDeck"
+description = "Valve's fork of Linux Kernel for the SteamDeck"
+script = "arch/linux-neptune.sh"
+task_list = "I PFM K"
+
 [[data.entries]]
 name = "Paru AUR Helper"
 description = "Paru is your standard pacman wrapping AUR helper with lots of features and minimal interaction.\nTo know more about AUR helpers visit: https://wiki.archlinux.org/title/AUR_helpers"
diff --git a/tui/src/state.rs b/tui/src/state.rs
index 4aee0aaa..623bd740 100644
--- a/tui/src/state.rs
+++ b/tui/src/state.rs
@@ -28,6 +28,7 @@ D  - disk modifications (ex. partitioning) (privileged)
 FI - flatpak installation
 FM - file modification
 I  - installation (privileged)
+K  - kernel modifications (privileged)
 MP - package manager actions
 SI - full system installation
 SS - systemd actions (privileged)