From 96238f8fa614c0e2ae0ce127fe343fa085d62b3a Mon Sep 17 00:00:00 2001
From: Mark Nipper <nipsy@bitgnome.net>
Date: Tue, 8 Jul 2025 23:51:40 -0700
Subject: Fix short hostname flake names

---
 scripts/pretty-rebuild | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/scripts/pretty-rebuild b/scripts/pretty-rebuild
index 613ce2b..889a3d1 100755
--- a/scripts/pretty-rebuild
+++ b/scripts/pretty-rebuild
@@ -1,9 +1,15 @@
 #!/usr/bin/env nix-shell
 #!nix-shell -i zsh --packages nvd zsh
 
+if [[ ${@} =~ "--flake" ]]; then
+	args=(${=@})
+else
+	args=("--flake .#$(hostname -s)" ${=@})
+fi
+
 cd /etc/nixos && \
 	nix flake update && \
-	nixos-rebuild switch --upgrade --show-trace ${@} && \
+	nixos-rebuild switch --upgrade --show-trace ${args} && \
 	echo && \
 	nixos-rebuild list-generations | cat && \
 	echo && \
-- 
cgit v1.2.3