disko-install: check if it runs as root

nix-community · Mar 26, 2024 · a6717b1 · a6717b1
1 parent 5d2d3e4
commit a6717b1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/disko-install b/disko-install
@@ -90,7 +90,6 @@ parseArgs() {
         exit 1
         ;;
       esac
-      mode=$2
       shift
       ;;
     --extra-files)
@@ -155,6 +154,12 @@ main() {
     exit 1
   fi
 
+  # check if we are root
+  if [[ "$EUID" -ne 0 ]]; then
+    echo "This script must be run as root" >&2
+    exit 1
+  fi
+
   if [[ $flake =~ ^(.*)\#([^\#\"]*)$ ]]; then
     flake="${BASH_REMATCH[1]}"
     flakeAttr="${BASH_REMATCH[2]}"