apps/rebuild/default.nix

@@ -7,6 +7,8 @@
   hostname,
   nix,
   nixos-rebuild-ng,
+  nix-output-monitor,
+  dix,
   ...
 }:
 
@@ -26,5 +28,7 @@ writeShellApplication {
     hostname
     nix
     nixos-rebuild-ng
+    nix-output-monitor
+    dix
   ];
 }

apps/rebuild/rebuild.sh

@@ -11,6 +11,11 @@ UPDATE_INPUTS=""                 # Default list of inputs to update. Empty means
 ROLLBACK=0                       # Default to not rollback
 SHOW_TRACE=0                     # Default to not show detailed error messages
 
+# Output functions
+_status() { echo -e "\033[0;34m> $1\033[0m"; }
+success() { echo -e "\033[0;32m$1\033[0m"; }
+error()   { echo -e "\033[0;31mError: $1\033[0m" >&2; exit 1; }
+
 # Function to display the help message
 Help() {
   echo "Wrapper script for 'nixos-rebuild switch' and 'home-manager switch' commands."
@@ -37,12 +42,6 @@ Help() {
   echo "  -u, --user <user>      Specify the username (as in 'homeConfigurations.<user>@<host>'). Default: $HOME_USER"
 }
 
-# Function to handle errors
-error() {
-  echo "Error: $1"
-  exit 1
-}
-
 # Function to rebuild NixOS configuration
 Rebuild_nixos() {
   local FLAKE="$FLAKE_PATH#$NIXOS_HOST"
@@ -56,20 +55,31 @@ Rebuild_nixos() {
   [ -n "$BUILD_HOST" ] && CMD+=("--build-host" "$BUILD_HOST")
   if [ "$NIXOS_HOST" != "$(hostname)" ] && [ -z "$TARGET_HOST" ]; then
     TARGET_HOST="$NIXOS_HOST"
-    echo "Using '$TARGET_HOST' as target host."
+    _status "Using '$TARGET_HOST' as target host."
   fi
   [ -n "$TARGET_HOST" ] && CMD+=("--target-host" "$TARGET_HOST")
 
-  # Rebuild NixOS configuration
-  if [ "$ROLLBACK" = 0 ]; then 
-    echo "Rebuilding NixOS configuration '$FLAKE'..." 
+  # Build config first so we can diff it
+  local BUILD_CMD=("nixos-rebuild" "build" "--flake" "$FLAKE")
+  [ "$SHOW_TRACE" = 1 ] && BUILD_CMD+=("--show-trace")
+  [ -n "$BUILD_HOST" ] && BUILD_CMD+=("--build-host" "$BUILD_HOST")
+
+  if [ "$ROLLBACK" = 0 ]; then
+    _status "Building NixOS configuration '$FLAKE'..."
+    _status "Executing command: ${BUILD_CMD[*]}"
+    "${BUILD_CMD[@]}" |&nom || error "NixOS build failed"
+    _status "Package diff:"
+    dix /run/current-system result
+    rm result
   else
-    echo "Rolling back to last NixOS generation..."
+    _status "Rolling back to last NixOS generation..."
   fi
 
-  echo "Executing command: ${CMD[*]}"
-  "${CMD[@]}" || error "NixOS rebuild failed"
-  echo "NixOS rebuild completed successfully."
+  _status "Switching to new NixOS configuration"
+  sudo -v
+  _status "Executing command: ${CMD[*]}"
+  "${CMD[@]}" |& nom || error "NixOS rebuild failed"
+  success "NixOS rebuild completed successfully."
 }
 
 # Function to rebuild Home Manager configuration
@@ -91,21 +101,29 @@ Rebuild_home() {
     [ "$SHOW_TRACE" = 1 ] && CMD+=("--show-trace")
   fi
 
-  # Rebuild Home Manager configuration
-  if [ "$ROLLBACK" = 0 ]; then 
-    echo "Rebuilding Home Manager configuration '$FLAKE'..." 
+  # Build config first so we can diff it
+  if [ "$ROLLBACK" = 0 ]; then
+    local BUILD_CMD=("home-manager" "build" "--flake" "$FLAKE")
+    [ "$SHOW_TRACE" = 1 ] && BUILD_CMD+=("--show-trace")
+    _status "Building Home Manager configuration '$FLAKE'..."
+    _status "Executing command: ${BUILD_CMD[*]}"
+    "${BUILD_CMD[@]}" |& nom || error "Home Manager build failed"
+    _status "Package diff:"
+    dix /run/current-system result
+    rm result
   else
-    echo "Rolling back to last Home Manager generation..."
+    _status "Rolling back to last Home Manager generation..."
   fi
 
-  echo "Executing command: ${CMD[*]}"
-  "${CMD[@]}" || error "Home Manager rebuild failed"
-  echo "Home Manager rebuild completed successfully."
+  _status "Switching to new Home Manager configuration"
+  _status "Executing command: ${CMD[*]}"
+  "${CMD[@]}" |& nom || error "Home Manager rebuild failed"
+  success "Home Manager rebuild completed successfully."
 }
 
-# Function to Update flake repositories
+# Function to update flake repositories
 Update() {
-  echo "Updating flake inputs..."
+  _status "Updating flake inputs..."
 
   # Construct update command as an array
   local CMD=("nix" "flake" "update" "--flake" "$FLAKE_PATH")
@@ -117,17 +135,18 @@ Update() {
     done
   fi
 
-  echo "Executing command: ${CMD[*]}"
+  _status "Executing command: ${CMD[*]}"
   "${CMD[@]}" || error "Failed to update flake repositories"
-  echo "Flake repositories updated successfully."
+  success "Flake repositories updated successfully."
 }
 
 # Parse command-line options
 if [[ -z "${1:-}" ]]; then
-  echo "Error: No command specified. Printing help page."
+  echo -e "\033[0;31mError: No command specified. Printing help page.\033[0m" >&2
   Help
   exit 1
 fi
+
 COMMAND=$1
 shift
 
@@ -198,9 +217,7 @@ while [ $# -gt 0 ]; do
       fi
       ;;
     *)
-      echo "Error: Unknown option '$1'"
-      Help
-      exit 1
+      error "Unknown option '$1'"
       ;;
   esac
 done
@@ -227,10 +244,10 @@ fi
 [ "$UPDATE" = 1 ] && Update
 
 case "$COMMAND" in
-  nixos)
+  nixos|os)
     Rebuild_nixos
     ;;
-  home)
+  home|hm)
     Rebuild_home
     ;;
   all)
@@ -238,9 +255,6 @@ case "$COMMAND" in
     Rebuild_home
     ;;
   *)
-    echo "Error: Unknown command '$COMMAND'"
-    echo "Printing help page:"
-    Help
-    exit 1
+    error "Unknown command '$COMMAND'"
     ;;
 esac

modules/shared/common/nix.nix

@@ -32,6 +32,8 @@ in
     };
   };
 
+  nix.settings.warn-dirty = mkDefault false;
+
   # fallback quickly if substituters are not available.
   nix.settings.connect-timeout = mkDefault 5;
   nix.settings.fallback = true;