From 0c35ecf54e64f072d26b78e41b8f167de8bc6ada Mon Sep 17 00:00:00 2001
From: funk-on-code <113871227+funk-on-code@users.noreply.github.com>
Date: Mon, 19 Sep 2022 06:10:51 +0000
Subject: [PATCH] Adding some basic sanity checking to the script

- Allows Control-C to terminate the script at any time.
- Confirms you have a UEFI/GPT capable system before operating.

Signed-off-by: funk-on-code <113871227+funk-on-code@users.noreply.github.com>
---
 install.sh | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/install.sh b/install.sh
index d686e63..f506900 100755
--- a/install.sh
+++ b/install.sh
@@ -3,6 +3,25 @@
 # Cleaning the TTY.
 clear
 
+# Add some basic sanity checking, just in case.
+if [[ ! "$(id -u)" == "0" ]]; then
+echo "[!] You MUST this script as root. Aborting."
+exit 255
+fi
+
+if [[ ! "$(ls /sys/firmware/efi/efivars)" ]]; then
+echo "[!] Please enable booting via UEFI. Aborting."
+exit 255
+fi
+
+# ... and trap Control-C correctly so we can bail out, when required.
+trap '_confirm_sigint' SIGINT
+
+_confirm_sigint() {
+    printf "\n"; read -rp "SIGINT caught: Are you sure you want to stop running this script? [y/N] " response
+    { [ "$response" == "y" ] || [ "$response" == "Y" ]; } && exit 1 || return
+}
+
 # Updating the live environment usually causes more problems than its worth, and quite often can't be done without remounting cowspace with more capacity, especially at the end of any given month.
 pacman -Sy