Fix autocomplete bugs.

src/commands/command_registry.cpp

@@ -1,4 +1,7 @@
 #include "command_registry.hpp"
+#include "utils/utils.hpp"
+
+namespace dropshell {
 
 CommandRegistry& CommandRegistry::instance() {
     static CommandRegistry reg;
@@ -41,18 +44,25 @@ std::vector<std::string> CommandRegistry::list_primary_commands(bool include_hid
 }
 
 
-void CommandRegistry::autocomplete(const std::vector<std::string>& args) const {
+void CommandRegistry::autocomplete(const CommandContext& ctx) const {
     // dropshell autocomplete <command> <arg> <arg> ...
-    if (args.size() < 3) {
+    if (ctx.args.size() < 1) {
         for (const auto& name : list_primary_commands(false)) {
             std::cout << name << std::endl;
         }
         return;
     }
-    std::string cmd = args[2];
-    auto* info = find_command(cmd);
+
+    // ctx command is autocomplete, so recreate ctx with the first arg removed
+    CommandContext childcontext = {
+        ctx.exename,
+        ctx.args[0],
+        std::vector<std::string>(ctx.args.begin() + 1, ctx.args.end())
+    };
+    auto* info = find_command(childcontext.command);
     if (info && info->autocomplete) {
-        CommandContext ctx{args[0], cmd, std::vector<std::string>(args.begin() + 2, args.end())};
-        info->autocomplete(ctx);
+        info->autocomplete(childcontext);
     }
-}
+}
+
+} // namespace dropshell

src/commands/command_registry.hpp

@@ -1,4 +1,6 @@
-#pragma once
+#ifndef COMMAND_REGISTRY_HPP
+#define COMMAND_REGISTRY_HPP
+
 #include <string>
 #include <vector>
 #include <functional>
@@ -7,6 +9,8 @@
 #include <memory>
 #include <iostream>
 
+namespace dropshell {
+
 struct CommandContext {
     std::string exename;
     std::string command;
@@ -42,10 +46,14 @@ public:
     std::vector<std::string> list_primary_commands(bool include_hidden = false) const;
 
     // For autocomplete
-    void autocomplete(const std::vector<std::string>& args) const;
+    void autocomplete(const CommandContext& ctx) const;
 
 private:
     CommandRegistry() = default;
     std::map<std::string, std::shared_ptr<CommandInfo>> command_map_;
     std::vector<std::shared_ptr<CommandInfo>> all_commands_;
 };
+
+} // namespace dropshell
+
+#endif // COMMAND_REGISTRY_HPP

src/commands/health.cpp

@@ -2,11 +2,11 @@
 #include "config.hpp"
 #include "utils/utils.hpp"
 #include "utils/directories.hpp"
-#include "templates.hpp"
 #include "shared_commands.hpp"
 #include "server_env_manager.hpp"
 #include "services.hpp"
 #include "servers.hpp"
+#include "transwarp.hpp"
 
 namespace dropshell
 {
@@ -118,13 +118,18 @@ namespace dropshell
         if (ctx.args.size() == 1) {
             // get all services on server
             std::vector<LocalServiceInfo> services = get_server_services_info(server);
-            for (const auto& service : services) {
-                std::cout << service.service_name << " " << healthtick(server, service.service_name) << std::endl;
-            }
+            transwarp::parallel exec{services.size()};
+            auto task = transwarp::for_each(exec, services.begin(), services.end(), [&](const LocalServiceInfo& service) {
+                std::string status = healthtick(server, service.service_name);
+                std::cout << status << " " << service.service_name << " (" << service.template_name << ")" << std::endl << std::flush;
+            });
+            task->wait();
+            return 0;
         } else {
             // get service status
             std::string service = safearg(ctx.args, 1);
-            std::cout << service << " " << healthtick(server, service) << std::endl;
+            LocalServiceInfo service_info = get_service_info(server, service);
+            std::cout << healthtick(server, service) << " " << service << " (" << service_info.template_name << ")" << std::endl << std::flush;
         }
         return 0;
     }

src/commands/standard_autocomplete.cpp

@@ -10,17 +10,16 @@ namespace dropshell {
 
 void std_autocomplete(const CommandContext &ctx)
 {
-    ASSERT(ctx.args.size() > 0);
-    if (ctx.args.size() == 1) {
+    if (ctx.args.size() == 0) { // just the command health.
         // list servers
         std::vector<ServerInfo> servers = get_configured_servers();
         for (const auto& server : servers) {
             std::cout << server.name << std::endl;
         }
-    }
-    else if (ctx.args.size() == 2) {
+    }    
+    else if (ctx.args.size() == 1) {
         // list services
-        std::vector<LocalServiceInfo> services = get_server_services_info(ctx.args[2]);
+        std::vector<LocalServiceInfo> services = get_server_services_info(ctx.args[0]);
         for (const auto& service : services) {
             std::cout << service.service_name << std::endl;
         }