[ui] Add auto-mount on first launch functionality #52

support/include/utils/windows.hpp

@@ -45,6 +45,7 @@ void free_console();
 struct shortcut_cfg final {
   std::wstring arguments;
   std::wstring exe_path;
+  std::wstring icon_path;
   std::wstring location{get_startup_folder()};
   std::wstring shortcut_name;
   std::wstring working_directory;

support/src/utils/windows.cpp

@@ -242,6 +242,17 @@ auto create_shortcut(const shortcut_cfg &cfg, bool overwrite_existing) -> bool {
     return false;
   }
 
+  if (not cfg.icon_path.empty()) {
+    result = psl->SetIconLocation(cfg.icon_path.c_str(), 0);
+    if (FAILED(result)) {
+      utils::error::handle_error(
+          function_name,
+          std::string("IShellLink::SetIconLocation failed: ") + hr_hex(result));
+      psl->Release();
+      return false;
+    }
+  }
+
   if (not utils::file::file{lnk_path}.remove()) {
     utils::error::handle_error(function_name,
                                "failed to remove existing shortcut|path|" +