fix(inject): restore button

close #49
std-microblock · Feb 4, 2025 · 2c94566 · 2c94566
1 parent db7ff4c
commit 2c94566
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 19 deletions.
diff --git a/src/inject/inject.cc b/src/inject/inject.cc
@@ -454,24 +454,10 @@ struct restore_btn : public button_widget {
   restore_btn() : button_widget("还原") {}
 
   void on_click() override {
-    // reg delete "HKCU\Software\Classes\CLSID\{86ca1aa0-34aa-4e8b-a509-50c905bae2a2}" /f
-    // reg delete "HKCU\Software\Classes\CLSID\{86ca1aa0-34aa-4e8b-a509-50c905bae2a2}\InprocServer32" /f
-
-    HKEY key;
-    RegOpenKeyExW(HKEY_CURRENT_USER,
-                  L"HKCU\\Software\\Classes\\CLSID\\{86ca1aa0-34aa-4e8b-a509-"
-                  L"50c905bae2a2}\\InprocServer32",
-                  0, KEY_SET_VALUE, &key);
-    RegDeleteValueW(key, L"");
-    RegCloseKey(key);
-
-    RegOpenKeyExW(HKEY_CURRENT_USER,
-                  L"HKCU\\Software\\Classes\\CLSID\\{86ca1aa0-34aa-4e8b-a509-"
-                  L"50c905bae2a2}",
-                  0, KEY_SET_VALUE, &key);
-
-    RegDeleteValueW(key, L"");
-    RegCloseKey(key);
+    RegDeleteKeyValueW(HKEY_CURRENT_USER,
+                       L"Software\\Classes\\CLSID\\{86ca1aa0-34aa-4e8b-a509-"
+                       L"50c905bae2a2}\\InprocServer32",
+                       nullptr);
   }
 };
 

diff --git a/src/shell/entry.cc b/src/shell/entry.cc
@@ -35,6 +35,7 @@
 #include <consoleapi.h>
 #include <debugapi.h>
 #include <type_traits>
+#include <winreg.h>
 #include <winuser.h>
 
 #define NOMINMAX
@@ -109,7 +110,6 @@ void main() {
                                                       lptpm);
     }
 
-
     has_active_menu = true;
     menu menu = menu::construct_with_hmenu(hMenu, hWnd);
     auto menu_render = menu_render::create(x, y, menu);