Add discord dynamic command and message

Dockerfile

@@ -14,16 +14,18 @@ RUN apt-get install ffmpeg -y
 COPY ./requirements.txt /discordbot/requirements.txt
 RUN pip install -U -r /discordbot/requirements.txt
 
-COPY ./discord_bot.py /discordbot/discord_bot.py
-COPY ./pingpong.py /discordbot/pingpong.py
-COPY ./server.py /discordbot/server.py
+
+COPY ./client.js /discordbot/static/client.js
+COPY ./style.css /discordbot/static/style.css
 COPY ./index.html /discordbot/templates/index.html
 COPY ./disclaimer.html /discordbot/templates/disclaimer.html
 COPY ./eula.html /discordbot/templates/eula.html
 COPY ./privacy_policy.html /discordbot/templates/privacy_policy.html
 COPY ./terms_of_service.html /discordbot/templates/terms_of_service.html
-COPY ./client.js /discordbot/static/client.js
-COPY ./style.css /discordbot/static/style.css
+COPY ./server.py /discordbot/server.py
+COPY ./pingpong.py /discordbot/pingpong.py
+COPY ./discord_bot.py /discordbot/discord_bot.py
 COPY ./horde.py /discordbot/horde.py
+COPY ./discord.json /discordbot/discord.json
 
 CMD ["python", "/discordbot/server.py"]

discord.json

@@ -0,0 +1,57 @@
+{
+    "message": [
+        {
+            "type": "equals",
+            "content": "ping",
+            "response": "pong"
+        },
+        {
+            "type": "startswith",
+            "content": "$hello",
+            "response": "Hello!"
+        },
+        {
+            "type": "equals",
+            "get kudos",
+            "function": "getKudos"
+        }
+    ],
+    "command": [
+        {
+            "name": "hello",
+            "description": "Sends a greeting!",
+            "function": "hello",
+            "parameters": []
+        },
+        {
+            "name": "greet",
+            "description": "Greets the specified user",
+            "function": "greet",
+            "parameters": [
+                {
+                    "name": "name",
+                    "type": "str",
+                    "description": "The user to greet"
+                }
+            ]
+        },
+        {
+            "name": "get-kudos",
+            "description": "The amount of Kudos this user has.",
+            "function": "getKudos",
+            "parameters": []
+        },
+        {
+            "name": "generate-status",
+            "description" "Retrieve the status of an Asynchronous generation request.",
+            "function": "generateStatus",
+            "parameters": [
+                {
+                    "name": "id",
+                    "type": "str",
+                    "description": "The ID of asyncronous generation request."
+                }
+            ]
+        }
+    ]
+}

discord_bot.py

@@ -10,12 +10,33 @@ import json
 from horde import HordeAPI
 
 
+TYPE_MAPPING = {
+    "str": str,
+    "int": int,
+    "float": float
+}
+
+
+# 创建一个字典，将规则类型映射到相应的条件检查函数
+check_functions = {
+    "equals": lambda content, message: message.content == content,
+    "contains": lambda content, message: content in message.content,
+    "startswith": lambda content, message: message.content.startswith(content),
+    "endswith": lambda content, message: message.content.endswith(content)
+}
+
 
 intents = discord.Intents.default()
 intents.message_content = True
 bot = commands.Bot(command_prefix='>', intents=intents)
 tree = bot.tree
 
+
+with open("discord.json", "r") as f:
+    json_data = json.load(f)
+
+
+"""
 @tree.command(name="hello", description="Sends a greeting!")
 async def hello(interaction: discord.Interaction):
     await interaction.response.send_message(f"Hello, {interaction.user.mention}!")
@@ -58,11 +79,70 @@ async def generateStatus(interaction: discord.Interaction, id: str):
             await interaction.followup.send(f'Processing image: {details["processing"]}/{total}')
             return
         await interaction.followup.send(f'Position in queue: {details["queue_position"]}, wait time: {details["wait_time"]}s')
+"""
 
 @bot.command()
 async def ping(ctx):
     await ctx.send('pong')
 
+async def hello():
+    return f"Hello, {interaction.user.mention}!"
+
+async def greet(name: str):
+    return f"Hello, {name}!"
+
+async def getKudos():
+    async with HordeAPI.getUserDetails() as details:
+        if "kudos" not in details:
+            return f'Error: {details["code"]} {details["reason"]}'
+        return f'The amount of Kudos this user has is {details["kudos"]}'
+
+async def generateStatus(id: str):
+    async with HordeAPI.generateCheck(id) as details:
+        if "kudos" not in details:
+            return f'Check Error: {details["code"]} {details["reason"]}'
+        if bool(details["is_possible"]) == False:
+            return "This generation is impossible."
+        if bool(details["faulted"]) == True:
+            return "This generation is faulted."
+        if bool(details["done"]) == True:
+            async with HordeAPI.generateStatus(id) as generationDetail:
+                if "generations" not in generationDetail:
+                    return f'Status Error: {generationDetail["code"]} {generationDetail["reason"]}'
+                for i in range(len(generationDetail["generations"])):
+                    return generationDetail["generations"][i]["img"]
+        if int(details["processing"]) > 0:
+            total = int(details["finished"]) + int(details["processing"]) + int(details["queue_position"]) + int(details["restarted"]) + int(details["waiting"])
+            return f'Processing image: {details["processing"]}/{total}'
+        return f'Position in queue: {details["queue_position"]}, wait time: {details["wait_time"]}s'
+
+
+# 根据 json 数据动态创建命令
+for command in json_data["command"]:
+    async def dynamic_command(interaction: discord.Interaction, **kwargs):
+        await interaction.response.defer()
+        # 动态调用命令对应的函数
+        function_name = command["function"]
+        function = globals()[function_name]
+        result = await function(**kwargs)
+        await interaction.followup.send(result)
+
+    # 动态创建参数
+    params = [app_commands.Parameter(name=param["name"], description=param["description"], type=TYPE_MAPPING[param["type"]])
+              for param in command["parameters"]]
+
+    # 动态创建命令并注册
+    tree_command = app_commands.Command(
+        name=command["name"],
+        description=command["description"],
+        callback=dynamic_command,
+        params=params
+    )
+
+    # 将命令添加到 bot 的 command tree
+    bot.tree.add_command(tree_command)
+
+
 @bot.event
 async def on_ready():
     await tree.sync()
@@ -74,11 +154,34 @@ async def on_message(message):
     if message.author == bot.user:
         return
 
+    """
     if message.content == 'ping':
         await message.channel.send('pong')
 
     if message.content.startswith('$hello'):
         await message.channel.send('Hello!')
+    """
+    for rule in json_data["message"]:
+        rule_type = rule["type"]
+        content = rule["content"]
+        response = rule["response"]
+        
+        # 根据规则类型动态调用对应的判断函数
+        if check_functions.get(rule_type, lambda c, m: False)(content, message):
+            # 如果规则指定了函数，则调用对应的函数
+            if "function" in rule:
+                function_name = rule["function"]
+                function = globals()[function_name]
+                await function(message)
+                break
+            # 否则发送预定义的响应消息
+            elif "response" in rule:
+                await message.channel.send(rule["response"])
+                break
+
+    # 确保命令系统正常工作
+    await bot.process_commands(message)
+
 
 async def sendMessageToChannelHelper(data):
     channel = await bot.fetch_channel(os.environ.get("CHANNEL_ID"))
@@ -106,9 +209,11 @@ async def sendMessageToChannelHelper(data):
     # 发送 embed 消息
     await channel.send(embed=embed)
 
+
 def sendMessageToChannel(data):
     bot.loop.create_task(sendMessageToChannelHelper(data))
 
+
 def run():
     try:
       token = os.environ.get("TOKEN") or ""
@@ -126,6 +231,7 @@ def run():
         else:
             raise e
 
+
 def discord_bot():
     print("Running discord_bot")
     run()