Free yamls for those who want them, take 2 (ArchipelagoMW#10)

Author: silasary

worlds/tracker/TrackerClient.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+import tempfile
 import traceback
 import typing
 from collections.abc import Callable
@@ -481,9 +482,20 @@ def on_package(self, cmd: str, args: dict):
                     logger.error(tb)
                     return
             else:
-                # TODO consider allowing worlds that self-attest to not need an options file for UT
-                self.log_to_tab(f"Player's Yaml not in tracker's list. Known players: {list(self.launch_multiworld.world_name_lookup.keys())}", False)
-                return
+                if getattr(AutoWorld.AutoWorldRegister.world_types[self.game], "ut_can_gen_without_yaml", False):
+                    with tempfile.TemporaryDirectory() as tempdir:
+                        self.write_empty_yaml(self.game, slot_name, tempdir)
+                        self.run_generator(None, tempdir)
+                        self.player_id = 1
+                        if callable(getattr(self.multiworld.worlds[self.player_id], "interpret_slot_data", None)):
+                            temp = self.multiworld.worlds[self.player_id].interpret_slot_data(args["slot_data"])
+                            if temp:
+                                self.re_gen_passthrough = {self.game: temp}
+
+                        self.run_generator(args["slot_data"], tempdir)
+                else:
+                    self.log_to_tab(f"Player's Yaml not in tracker's list. Known players: {list(self.launch_multiworld.world_name_lookup.keys())}", False)
+                    return
 
             if self.ui is not None and getattr(self.multiworld.worlds[self.player_id], "tracker_world", None):
                 self.tracker_world = UTMapTabData(**self.multiworld.worlds[self.player_id].tracker_world)
@@ -499,6 +511,13 @@ def on_package(self, cmd: str, args: dict):
         elif cmd == 'RoomUpdate':
             updateTracker(self)
 
+    def write_empty_yaml(self, game, player_name, tempdir):
+        path = os.path.join(tempdir, f'{game}_{player_name}.yaml')
+        with open(path, 'w') as f:
+            f.write('name: ' + player_name + '\n')
+            f.write('game: ' + game + '\n')
+            f.write(game + ': {}\n')
+
     async def disconnect(self, allow_autoreconnect: bool = False):
         if "Tracker" in self.tags:
             self.game = ""
@@ -528,7 +547,7 @@ def _set_host_settings(self):
         return tracker_settings['player_files_path'], report_type, tracker_settings[
             'hide_excluded_locations']
 
-    def run_generator(self, slot_data: Optional[Dict] = None):
+    def run_generator(self, slot_data: Optional[Dict] = None, override_yaml_path: Optional[str] = None):
         def move_slots(args: "Namespace", slot_name: str):
             """
             helper function to copy all the proper option values into slot 1,
@@ -547,7 +566,9 @@ def move_slots(args: "Namespace", slot_name: str):
             # strip command line args, they won't be useful from the client anyway
             sys.argv = sys.argv[:1]
             args = mystery_argparse()
-            if yaml_path:
+            if override_yaml_path:
+                args.player_files_path = override_yaml_path
+            elif yaml_path:
                 args.player_files_path = yaml_path
             args.skip_output = True