From 8a1af741dd358ce1c7542294717253a54c47af32 Mon Sep 17 00:00:00 2001
From: William <baduhai@proton.me>
Date: Sun, 13 Jul 2025 18:39:01 -0300
Subject: [PATCH] add nixos-deploy wrapper

---
 flake.nix                  |   1 +
 hosts/modules/programs.nix |   1 +
 packages/nixos-deploy.nix  | 111 +++++++++++++++++++++++++++++++++++++
 3 files changed, 113 insertions(+)
 create mode 100644 packages/nixos-deploy.nix

diff --git a/flake.nix b/flake.nix
index 9dfb40c..f465c91 100644
--- a/flake.nix
+++ b/flake.nix
@@ -143,6 +143,7 @@
         overlay = final: prev: {
         };
         workstationOverlay = final: prev: {
+          nixos-deploy = nixpkgs.legacyPackages."x86_64-linux".callPackage ./packages/nixos-deploy.nix { };
           plasticity = nixpkgs.legacyPackages."x86_64-linux".callPackage ./packages/plasticity.nix { };
           toggleaudiosink =
             nixpkgs.legacyPackages."x86_64-linux".callPackage ./packages/toggleaudiosink.nix
diff --git a/hosts/modules/programs.nix b/hosts/modules/programs.nix
index 8715d07..e3e9091 100644
--- a/hosts/modules/programs.nix
+++ b/hosts/modules/programs.nix
@@ -111,6 +111,7 @@
               kde-rounded-corners
               libfido2
               mission-center
+              nixos-deploy
               p7zip
               qbittorrent
               quickemu
diff --git a/packages/nixos-deploy.nix b/packages/nixos-deploy.nix
new file mode 100644
index 0000000..160c628
--- /dev/null
+++ b/packages/nixos-deploy.nix
@@ -0,0 +1,111 @@
+{
+  lib,
+  writeShellScript,
+  nixos-rebuild,
+  openssh,
+  coreutils,
+  gnugrep,
+  gawk,
+}:
+
+writeShellScript "nixos-deploy" ''
+  set -euo pipefail
+
+  LOCAL_BUILD=false
+  ACTION="switch"
+  FLAKE_URI=""
+  TARGET_HOST=""
+  SSH_USER=""
+  SSH_HOST=""
+
+  show_usage() {
+      echo -e "Usage: nixos-deploy [--local-build] [--boot] <flake-uri> [user@]host"
+      echo ""
+      echo -e "Arguments:"
+      echo "  flake-uri    Flake URI (e.g., .#hostname)"
+      echo "  [user@]host  Target host, optionally with user"
+      echo ""
+      echo -e "Options:"
+      echo "  --local-build  Build locally instead of on remote"
+      echo "  --boot         Use 'boot' instead of 'switch' action"
+      echo ""
+      echo -e "Examples:"
+      echo "  nixos-deploy .#hostname user@192.168.1.10"
+      echo "  nixos-deploy --local-build .#hostname 192.168.1.10"
+      echo "  nixos-deploy --boot .#hostname 192.168.1.10"
+      echo "  nixos-deploy .#hostname 192.168.1.10  # uses current user"
+      exit 1
+  }
+
+  while [[ $# -gt 0 ]]; do
+      case $1 in
+          --local-build)
+              LOCAL_BUILD=true
+              shift
+              ;;
+          --boot)
+              ACTION="boot"
+              shift
+              ;;
+          --help|-h)
+              show_usage
+              ;;
+          -*)
+              echo -e "Unknown option: $1"
+              show_usage
+              ;;
+          *)
+              if [[ -z "$FLAKE_URI" ]]; then
+                  FLAKE_URI="$1"
+              elif [[ -z "$TARGET_HOST" ]]; then
+                  TARGET_HOST="$1"
+              else
+                  echo -e "Too many arguments"
+                  show_usage
+              fi
+              shift
+              ;;
+      esac
+  done
+
+  if [[ -z "$FLAKE_URI" ]]; then
+      echo -e "flake-uri is required"
+      show_usage
+  fi
+
+  if [[ -z "$TARGET_HOST" ]]; then
+      echo -e "target host is required"
+      show_usage
+  fi
+
+  if [[ "$TARGET_HOST" == *"@"* ]]; then
+      SSH_USER="''${TARGET_HOST%@*}"
+      SSH_HOST="''${TARGET_HOST#*@}"
+  else
+      SSH_USER="$(${coreutils}/bin/whoami)"
+      SSH_HOST="$TARGET_HOST"
+  fi
+
+  echo "Deploying $FLAKE_URI to $SSH_HOST as user $SSH_USER (action: $ACTION)"
+
+  if [[ "$LOCAL_BUILD" != "true" ]]; then
+      GC_ROOT_PATH="/tmp/nixos-deploy-$SSH_HOST-$"
+  fi
+
+  REBUILD_CMD="${nixos-rebuild}/bin/nixos-rebuild $ACTION --flake $FLAKE_URI --target-host $TARGET_HOST"
+
+  if [[ "$LOCAL_BUILD" == "true" ]]; then
+      echo -e "Building locally and deploying to remote host"
+  else
+      REBUILD_CMD="$REBUILD_CMD --build-host $SSH_HOST"
+      echo -e "Building on remote host"
+  fi
+
+  if [[ "$SSH_USER" != "root" ]]; then
+      REBUILD_CMD="$REBUILD_CMD --use-remote-sudo"
+      echo -e "Using remote sudo for non-root user"
+  fi
+
+  echo -e "Running: $REBUILD_CMD"
+  exec $REBUILD_CMD
+''