Gamedata Search

cogs/gamedata.py

@@ -0,0 +1,98 @@
+import json
+import os
+import nextcord
+from nextcord.ext import commands
+
+class GamedataCog(commands.Cog):
+ def __init__(self, bot):
+ self.bot = bot
+ self.load_pals()
+ self.load_items()
+
+ def load_pals(self):
+ pals_path = os.path.join("gamedata", "pals.json")
+ with open(pals_path, "r", encoding="utf-8") as pals_file:
+ self.pals = json.load(pals_file)["creatures"]
+
+ def load_items(self):
+ items_path = os.path.join("gamedata", "items.json")
+ with open(items_path, "r", encoding="utf-8") as items_file:
+ self.items = json.load(items_file)["items"]
+
+ async def autocomplete_pal(self, interaction: nextcord.Interaction, current: str):
+ choices = [
+ pal["name"] for pal in self.pals if current.lower() in pal["name"].lower()
+ ][:25]
+ await interaction.response.send_autocomplete(choices)
+
+ async def autocomplete_item(self, interaction: nextcord.Interaction, current: str):
+ choices = [
+ item["name"] for item in self.items if current.lower() in item["name"].lower()
+ ][:25]
+ await interaction.response.send_autocomplete(choices)
+
+ @nextcord.slash_command(
+ default_member_permissions=nextcord.Permissions(administrator=True)
+ )
+ async def search(self, _interaction: nextcord.Interaction):
+ pass
+
+ @search.subcommand(description="Search for a pal by name.")
+ async def pal(
+ self,
+ interaction: nextcord.Interaction,
+ name: str = nextcord.SlashOption(
+ description="Name of the pal to search for", autocomplete=True),
+ ):
+ await interaction.response.defer(ephemeral=True)
+ pal = next((pal for pal in self.pals if pal["name"] == name), None)
+ if pal:
+ embed = nextcord.Embed(
+ title=pal['name'], color=nextcord.Color.blue())
+ embed.description = f"Spawn Code: `{pal['id']}`"
+ await interaction.followup.send(embed=embed)
+ else:
+ await interaction.followup.send("Pal not found.", ephemeral=True)
+
+ @pal.on_autocomplete("name")
+ async def autocomplete_pal_name(
+ self, interaction: nextcord.Interaction, current: str
+ ):
+ await self.autocomplete_pal(interaction, current)
+
+ @search.subcommand(description="Search for an item by name.")
+ async def item(
+ self,
+ interaction: nextcord.Interaction,
+ name: str = nextcord.SlashOption(
+ description="Name of the item to search for", autocomplete=True),
+ ):
+ await interaction.response.defer(ephemeral=True)
+ item = next(
+ (item for item in self.items if item["name"] == name), None)
+ if item:
+ embed = nextcord.Embed(
+ title=item['name'], color=nextcord.Color.green())
+ embed.description = f"Spawn Code: `{item['id']}`"
+ await interaction.followup.send(embed=embed)
+ else:
+ await interaction.followup.send("Item not found.", ephemeral=True)
+
+ @item.on_autocomplete("name")
+ async def autocomplete_item_name(
+ self, interaction: nextcord.Interaction, current: str
+ ):
+ await self.autocomplete_item(interaction, current)
+
+def setup(bot):
+ cog = GamedataCog(bot)
+ bot.add_cog(cog)
+ if not hasattr(bot, "all_slash_commands"):
+ bot.all_slash_commands = []
+ bot.all_slash_commands.extend(
+ [
+ cog.search,
+ cog.pal,
+ cog.item,
+ ]
+ )