#!/usr/bin/env bash

if [ -d "result" ];
then
  preserve_result=true
else
  preserve_result=false
fi

while [ $# -gt 0 ]; do
  case "$1" in
    --target*|-t*)
      if [[ "$1" != *=* ]]; then shift; fi # Value is next arg if no `=`
      target="${1#*=}"
      ;;
    --flake*|-h*)
      if [[ "$1" != *=* ]]; then shift; fi
      flake="${1#*=}"
      ;;
    --mode*|-m*)
      if [[ "$1" != *=* ]]; then shift; fi
      mode="${1#*=}"
      ;;
    --user*|-u*)
      if [[ "$1" != *=* ]]; then shift; fi
      user="${1#*=}"
      ;;
    --preserve-result)
      preserve_result=true
      ;;
    --no-preserve-result)
      preserve_result=false
      ;;
    --help|-h)
      echo "--help -h: print this message"
      echo "--target -t: set the target system to rebuild on"
      echo "--flake -f: set the flake to rebuild on the target system"
      echo "--mode -m: set the mode to rebuild flake as on the target system"
      echo "--user -u: set the user to rebuild flake as on the target system"
      echo "--preserve-result: do not remove the generated result folder after building"
      echo "--no-preserve-result: remove any result folder after building"
      exit 0
      ;;
    *)
      echo "Error: Invalid argument $1"
      exit 1
      ;;
  esac
  shift
done

target=${target:-$(hostname)}
flake=${flake:-$target}
mode=${mode:-switch}
user=${user:-$USER}

# path: prefixes on rebuilds here make nix not treat this flake like it has a git repo so we can
# access secret files in the submodule this is kinda bad and we should find a way to not need it 
if [[ "$target" == "$(hostname)" ]];
then
	nixos-rebuild $mode --use-remote-sudo --flake .#$flake
else
	nixos-rebuild $mode --use-remote-sudo --target-host $user@$target --flake .#$flake
fi

if [ -d "result" ];
then
  if [[ "$preserve_result" == "false" ]];
  then
    rm -r result
  fi
fi