fix downloading engines with progress bar

Author: namchuai

engine/cli/commands/engine_install_cmd.cc

@@ -35,8 +35,9 @@ bool EngineInstallCmd::Exec(const std::string& engine,
     DownloadProgress dp;
     dp.Connect(host_, port_);
     // engine can be small, so need to start ws first
-    auto dp_res = std::async(std::launch::deferred,
-                             [&dp, &engine] { return dp.Handle(engine); });
+    auto dp_res = std::async(std::launch::deferred, [&dp, &engine] {
+      return dp.Handle(DownloadType::Engine);
+    });
     CLI_LOG("Validating download items, please wait..")
 
     auto versions_url = url_parser::Url{
@@ -118,7 +119,7 @@ bool EngineInstallCmd::Exec(const std::string& engine,
 
     bool check_cuda_download = !system_info_utils::GetCudaVersion().empty();
     if (check_cuda_download) {
-      if (!dp.Handle("cuda"))
+      if (!dp.Handle(DownloadType::CudaToolkit))
         return false;
     }
 
@@ -130,9 +131,8 @@ bool EngineInstallCmd::Exec(const std::string& engine,
   DownloadProgress dp;
   dp.Connect(host_, port_);
   // engine can be small, so need to start ws first
-  auto dp_res = std::async(std::launch::deferred, [&dp] {
-    return dp.Handle(DownloadType::Engine);
-  });
+  auto dp_res = std::async(std::launch::deferred,
+                           [&dp] { return dp.Handle(DownloadType::Engine); });
   CLI_LOG("Validating download items, please wait..")
 
   auto install_url = url_parser::Url{

engine/cli/commands/engine_update_cmd.cc

@@ -23,8 +23,9 @@ bool EngineUpdateCmd::Exec(const std::string& host, int port,
   DownloadProgress dp;
   dp.Connect(host, port);
   // engine can be small, so need to start ws first
-  auto dp_res = std::async(std::launch::deferred,
-                           [&dp, &engine] { return dp.Handle(engine); });
+  auto dp_res = std::async(std::launch::deferred, [&dp, &engine] {
+    return dp.Handle(DownloadType::Engine);
+  });
   CLI_LOG("Validating download items, please wait..")
 
   auto update_url = url_parser::Url{
@@ -43,7 +44,7 @@ bool EngineUpdateCmd::Exec(const std::string& host, int port,
 
   bool check_cuda_download = !system_info_utils::GetCudaVersion().empty();
   if (check_cuda_download) {
-    if (!dp.Handle("cuda"))
+    if (!dp.Handle(DownloadType::CudaToolkit))
       return false;
   }