From 8e12382f60bbabff5df62d38d37e558f41f0b641 Mon Sep 17 00:00:00 2001
From: Mohammad Rafiq <rafiq@rrv.sh>
Date: Thu, 29 May 2025 16:47:22 +0800
Subject: [PATCH] feat(packages/rebuild): add multi host rebuild to rebuild
 script

---
 packages/rebuild/default.nix | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/packages/rebuild/default.nix b/packages/rebuild/default.nix
index c21e47e..b7482dc 100644
--- a/packages/rebuild/default.nix
+++ b/packages/rebuild/default.nix
@@ -1,14 +1,22 @@
 { pkgs, ... }:
 pkgs.writeShellScriptBin "rebuild" # sh
   ''
-    CURRENT_GENERATION=$(readlink /nix/var/nix/profiles/system | cut -d- -f2)
-
     if [ ! -f "flake.nix" ]; then
       echo "Error: flake.nix not found in the current directory. Exiting."
       exit 1  # Indicate an error
     fi
 
     git add .
+
+    if [ $# -gt 0 ]; then
+      for arg in "$@"; do
+        nixos-rebuild switch --flake .#"$arg" --target-host "$arg" --use-remote-sudo
+      done
+      exit 0
+    fi
+
+    CURRENT_GENERATION=$(readlink /nix/var/nix/profiles/system | cut -d- -f2)
+
     nh os test . || {
       echo "Error: nixos-rebuild switch failed.  Check the output for details."
       exit 1