Launcher Icon

Author: silasary

src/__init__.py

@@ -4,9 +4,10 @@
 import json
 from typing import Callable, Optional
 
+import requests
 import Utils
 from worlds.generic.Rules import forbid_items_for_player
-from worlds.LauncherComponents import Component, SuffixIdentifier, components, Type, launch_subprocess
+from worlds.LauncherComponents import Component, SuffixIdentifier, components, Type, launch_subprocess, icon_paths
 
 from .Data import item_table, location_table, region_table, category_table, meta_table
 from .Game import game_name, filler_item_name, starting_items
@@ -435,21 +436,30 @@ def launch_client(*args):
         Main()
 
 class VersionedComponent(Component):
-    def __init__(self, display_name: str, script_name: Optional[str] = None, func: Optional[Callable] = None, version: int = 0, file_identifier: Optional[Callable[[str], bool]] = None):
-        super().__init__(display_name=display_name, script_name=script_name, func=func, component_type=Type.CLIENT, file_identifier=file_identifier)
+    def __init__(self, display_name: str, script_name: Optional[str] = None, func: Optional[Callable] = None, version: int = 0, file_identifier: Optional[Callable[[str], bool]] = None, icon: Optional[str] = None):
+        super().__init__(display_name=display_name, script_name=script_name, func=func, component_type=Type.CLIENT, file_identifier=file_identifier, icon=icon)
         self.version = version
 
 def add_client_to_launcher() -> None:
     version = 2024_09_29 # YYYYMMDD
     found = False
+
+    icon_url = "https://cdn.discordapp.com/icons/1097532591650910289/31019cd41a862df824f886c6cefbeb68.png?size=96"
+    if "manual" not in icon_paths:
+        icon_paths["manual"] = Utils.user_path('data', 'manual.png')
+        if not os.path.exists(icon_paths["manual"]):
+            with open(icon_paths["manual"], 'wb') as f:
+                f.write(requests.get(icon_url).content)
+
     for c in components:
         if c.display_name == "Manual Client":
             found = True
             if getattr(c, "version", 0) < version:  # We have a newer version of the Manual Client than the one the last apworld added
                 c.version = version
                 c.func = launch_client
+                c.icon = "manual"
                 return
     if not found:
-        components.append(VersionedComponent("Manual Client", "ManualClient", func=launch_client, version=version, file_identifier=SuffixIdentifier('.apmanual')))
+        components.append(VersionedComponent("Manual Client", "ManualClient", func=launch_client, version=version, file_identifier=SuffixIdentifier('.apmanual'), icon="manual"))
 
 add_client_to_launcher()