Fix installed osu detection (#4)

UncleGoogle · Jun 22, 2020 · 038f8ef · 038f8ef
1 parent c3c81ef
commit 038f8ef
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## Version 1.0.1
+[Fix]
+- Installed game detection #3
+
 ## Version 1.0.0
 [Added]
 - install

diff --git a/src/local.py b/src/local.py
@@ -40,9 +40,12 @@ def _find_exe(self) -> t.Optional[str]:
         if not WIN:
             raise NotImplementedError('Only Windows supported for now')
 
-        UNINSTALL_KEY = R'SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall\{86ce7517-ee79-4be9-a314-128183321391}'
+        UNINSTALL_REG = R'SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall'
         try:
-            with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, UNINSTALL_KEY) as key:
+            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, R'Software\osu!') as key:
+                uninstall_id = winreg.QueryValueEx(key, 'UninstallId')[0]
+                uninstall_key = UNINSTALL_REG + fR'\{{{uninstall_id}}}'
+            with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, uninstall_key) as key:
                 exe = winreg.QueryValueEx(key, 'DisplayIcon')[0]
                 return pathlib.Path(exe)
         except FileNotFoundError:

diff --git a/src/manifest.json b/src/manifest.json
@@ -1,8 +1,8 @@
 {
-    "name": "Integration for Osu!",
+    "name": "Integration for osu!",
     "platform": "egg",
     "guid": "79764115-b455-4ff0-be13-a316c7c53dec",
-    "version": "1.0.0",
+    "version": "1.0.1",
     "description": "Until GOG provide osu id this plugin is shown as Newegg",
     "author": "Mesco",
     "email": "[email protected]",