From 8348ea63a0d217e1fd9a45013fae22943db32cd0 Mon Sep 17 00:00:00 2001
From: Your Name <j@842.be>
Date: Sat, 23 Aug 2025 18:19:43 +1200
Subject: [PATCH] feat: Update 2 files

---
 build.sh                        |  1 +
 source/src/commands/install.cpp | 20 ++++++++++----------
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/build.sh b/build.sh
index b59e07f..0a6b1bc 100755
--- a/build.sh
+++ b/build.sh
@@ -31,6 +31,7 @@ if [ "${INSTALL_LOCAL:-true}" = "true" ]; then
     INSTALL_DIR="${HOME}/.local/bin"
     mkdir -p "${INSTALL_DIR}"
     cp "${SCRIPT_DIR}/output/dropshell" "${INSTALL_DIR}/dropshell"
+    "${INSTALL_DIR}/dropshell" install
     echo "Dropshell installed to ${INSTALL_DIR}/dropshell"
 fi
 
diff --git a/source/src/commands/install.cpp b/source/src/commands/install.cpp
index 3348867..d32f1ee 100644
--- a/source/src/commands/install.cpp
+++ b/source/src/commands/install.cpp
@@ -449,15 +449,6 @@ complete -F _dropshell_completions ds
         if (rval != 0)
             return rval;
 
-        // install the dropshell agent on all servers.
-        std::vector<ServerConfig> servers = get_configured_servers();
-        for (const auto &server : servers)
-        {
-            rval = install_server(server);
-            if (rval != 0)
-                return rval;
-        }
-
         std::cout << "Installation complete." << std::endl;
         return 0;
     }
@@ -482,7 +473,16 @@ complete -F _dropshell_completions ds
 
         if (ctx.args.size() == 1)
         { // install server
-            return install_server(server);
+            if (server == "all")
+            { // install the dropshell agent on all servers.
+                int rval = 0;
+                std::vector<ServerConfig> servers = get_configured_servers();
+                for (const auto &server : servers)
+                    rval += abs(install_server(server)); // capture any non-zero return values.
+                return rval;
+            }
+            else
+                return install_server(server);
         }
 
         // install service(s)