add filter for variant list

Author: namchuai

engine/cli/commands/engine_install_cmd.cc

@@ -4,19 +4,12 @@
 #include "utils/cli_selection_utils.h"
 #include "utils/download_progress.h"
 #include "utils/logging_utils.h"
+#include "utils/string_utils.h"
 
 namespace commands {
 bool EngineInstallCmd::Exec(const std::string& engine,
                             const std::string& version,
                             const std::string& src) {
-  // Start server if server is not started yet
-  if (!commands::IsServerAlive(host_, port_)) {
-    CLI_LOG("Starting server ...");
-    commands::ServerStartCmd ssc;
-    if (!ssc.Exec(host_, port_)) {
-      return false;
-    }
-  }
   // Handle local install, if fails, fallback to remote install
   if (!src.empty()) {
     auto res = engine_service_.UnzipEngine(engine, version, src);
@@ -30,6 +23,15 @@ bool EngineInstallCmd::Exec(const std::string& engine,
     }
   }
 
+  // Start server if server is not started yet
+  if (!commands::IsServerAlive(host_, port_)) {
+    CLI_LOG("Starting server ...");
+    commands::ServerStartCmd ssc;
+    if (!ssc.Exec(host_, port_)) {
+      return false;
+    }
+  }
+
   if (show_menu_) {
     DownloadProgress dp;
     dp.Connect(host_, port_);
@@ -81,7 +83,10 @@ bool EngineInstallCmd::Exec(const std::string& engine,
 
     std::vector<std::string> variant_selections;
     for (const auto& variant : variant_result.value()) {
-      variant_selections.push_back(variant["name"].asString());
+      auto v_name = variant["name"].asString();
+      if (string_utils::StringContainsIgnoreCase(v_name, hw_inf_.sys_inf->os)) {
+        variant_selections.push_back(variant["name"].asString());
+      }
     }
     auto selected_variant =
         cli_selection_utils::PrintSelection(variant_selections);

engine/cli/commands/engine_install_cmd.h

@@ -12,7 +12,9 @@ class EngineInstallCmd {
       : engine_service_{EngineService(download_service)},
         host_(host),
         port_(port),
-        show_menu_(show_menu) {};
+        show_menu_(show_menu),
+        hw_inf_{.sys_inf = system_info_utils::GetSystemInfo(),
+                .cuda_driver_version = system_info_utils::GetCudaVersion()} {};
 
   bool Exec(const std::string& engine, const std::string& version = "latest",
             const std::string& src = "");
@@ -22,5 +24,12 @@ class EngineInstallCmd {
   std::string host_;
   int port_;
   bool show_menu_;
+
+  struct HardwareInfo {
+    std::unique_ptr<system_info_utils::SystemInfo> sys_inf;
+    cortex::cpuid::CpuInfo cpu_inf;
+    std::string cuda_driver_version;
+  };
+  HardwareInfo hw_inf_;
 };
 }  // namespace commands