#!/usr/bin/env nix-shell
#!nix-shell ../devshells/bootstrap/shell.nix
#!nix-shell -i bash

# Check if at least one argument is provided
if [ "$#" != "1" ] && [ "$#" != "2" ]; then
    echo "Usage: $0 $1 <disko-mode> [action]"
    exit 1
fi

# Ensure correct first argument type
if [ "$2" != "disko" ] && [ "$2" != "mount" ]; then
    echo "Disko action must be either 'disko' or 'mount'"
    exit 1
fi

# Ensure correct second argument type
if [ "$#" != "2" ] && [ "$3" != "install" ] && [ "$3" != "enter" ]; then
    echo "Action must either be empty, 'install' or 'enter'"
    exit 1
fi

if mountpoint -q /hermes; then
  echo "Keys already mounted"
else
  echo "Mounting keys"
  mkdir -p /hermes
  mount /dev/disk/by-uuid/7FE7-CA68 /hermes
fi

echo "Running disko"

if [ "$2" = "mount" ] && [ "$1" = "lapetus" ]; then
  zpool import -lfR /mnt zroot
fi

nix run disko -- --mode $1 ./hosts/nixos/lapetus/filesystems/partitions.nix

if [ "$3" = "install" ]; then
  echo "Installing nixos"
  nixos-install --flake ".#$1"
fi

if [ "$3" = "enter" ]; then
  echo "Entering nixos"
  nixos-enter --root /mnt
fi

echo "All done!"