Autocomplete mission filenames in /zeus-set-mission (zeusops#9)

Author: Gehock

CHANGELOG.md

@@ -9,6 +9,7 @@ The project uses semantic versioning (see [semver](https://semver.org)).
 
 - `/zeus-upload` command now accepts an optional modlist (JSON snippet exported
   from Mod Presets menu in Reforger).
+- `/zeus-set-mission` now autocompletes based on known missions.
 - Command line flag `--debug` for enabling more verbose debug logging.
 
 ### Fixed

src/zeusops_bot/cogs/zeus_upload.py

@@ -1,32 +1,43 @@
 """zeus_upload extension"""
 
+import typing
+
 import discord
+from discord.commands import option
 from discord.ext import commands
+from discord.utils import basic_autocomplete
 from pydantic import TypeAdapter, ValidationError
 
+if typing.TYPE_CHECKING:
+    from zeusops_bot.discord import ZeusopsBot
 from zeusops_bot.errors import (
     ConfigFileInvalidJson,
     ConfigFileNotFound,
     ConfigPatchingError,
 )
 from zeusops_bot.models import ModDetail
 from zeusops_bot.reforger_config_gen import ReforgerConfigGenerator, extract_mods
-from zeusops_bot.settings import ZeusopsBotConfig
 
 modlist_typeadapter = TypeAdapter(list[ModDetail])
 
 
+def _autocomplete_missions(ctx: discord.AutocompleteContext) -> list[str]:
+    """List known missions
+
+    Used to populate the autocomplete list in /zeus-set-mission.
+
+    TODO: Return list[discord.OptionChoice] instead?
+    """
+    return ctx.bot.reforger_confgen.list_missions()
+
+
 class ZeusUpload(commands.Cog):
     """ZeusUpload cog for handling mission uploads"""
 
-    def __init__(self, bot: discord.Bot, config: ZeusopsBotConfig):
+    def __init__(self, bot: "ZeusopsBot", reforger_confgen: ReforgerConfigGenerator):
         """Initialise the cog"""
         self.bot = bot
-        self.config = config
-        self.reforger_confgen = ReforgerConfigGenerator(
-            base_config_file=config.reforger.reference_config,
-            target_folder=config.reforger.config_folder,
-        )
+        self.reforger_confgen = reforger_confgen
 
     @commands.slash_command(name="zeus-upload")
     @discord.option(
@@ -91,7 +102,16 @@ async def zeus_upload(
         await ctx.respond(f"Mission uploaded successfully under {path=}")
 
     @commands.slash_command(name="zeus-set-mission")
-    async def zeus_set_mission(self, ctx: discord.ApplicationContext, filename: str):
+    @option(
+        "filename",
+        description="Mission filename",
+        autocomplete=basic_autocomplete(_autocomplete_missions),
+    )
+    async def zeus_set_mission(
+        self,
+        ctx: discord.ApplicationContext,
+        filename: str,
+    ):
         """Activate the given a mission file as a Zeus"""
         try:
             self.reforger_confgen.zeus_set_mission(filename)

src/zeusops_bot/discord.py

@@ -5,6 +5,7 @@
 from discord import Bot
 
 from zeusops_bot.cogs import ZeusUpload
+from zeusops_bot.reforger_config_gen import ReforgerConfigGenerator
 from zeusops_bot.settings import ZeusopsBotConfig
 
 
@@ -16,8 +17,12 @@ def __init__(self, config: ZeusopsBotConfig, logger: Logger, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.config = config
         self.logger = logger
+        self.reforger_confgen = ReforgerConfigGenerator(
+            base_config_file=config.reforger.reference_config,
+            target_folder=config.reforger.config_folder,
+        )
 
-        cog = ZeusUpload(self, config)
+        cog = ZeusUpload(self, self.reforger_confgen)
         self.add_cog(cog)
 
     def run(self):