Update emergency script

scripts/emergency.sh

@@ -3,24 +3,24 @@
 #!nix-shell -i bash
 
 # Check if at least one argument is provided
-if [ "$#" -eq 0 ]; then
+if [ "$#" != "1" ] && [ "$#" != "2" ]; then
-    echo "Usage: $0 <disko-mode> [action]"
+    echo "Usage: $0 $1 <disko-mode> [action]"
     exit 1
 fi
 
 # Ensure correct first argument type
-if [ "$1" != "disko" ] && [ "$1" != "mount" ]; then
+if [ "$2" != "disko" ] && [ "$2" != "mount" ]; then
     echo "Disko action must be either 'disko' or 'mount'"
     exit 1
 fi
 
 # Ensure correct second argument type
-if [ "$#" != "1" ] && [ "$2" != "install" ] && [ "$2" != "enter" ]; then
+if [ "$#" != "2" ] && [ "$3" != "install" ] && [ "$3" != "enter" ]; then
     echo "Action must either be empty, 'install' or 'enter'"
     exit 1
 fi
 
-if mountpoint -q /hermes; then 
+if mountpoint -q /hermes; then
   echo "Keys already mounted"
 else
   echo "Mounting keys"
@@ -30,18 +30,18 @@ fi
 
 echo "Running disko"
 
-if [ "$1" = "mount" ]; then
+if [ "$2" = "mount" ] && [ "$1" = "lapetus" ]; then
   zpool import -lfR /mnt zroot
 fi
 
 nix run disko -- --mode $1 ./hosts/nixos/lapetus/filesystems/partitions.nix
 
-if [ "$2" = "install" ]; then
+if [ "$3" = "install" ]; then
   echo "Installing nixos"
-  nixos-install --flake ".#lapetus"
+  nixos-install --flake ".#$1"
 fi
 
-if [ "$2" = "enter" ]; then
+if [ "$3" = "enter" ]; then
   echo "Entering nixos"
   nixos-enter --root /mnt
 fi