Rename repl to playground

v_repl_bot/cogs/playground.py

@@ -0,0 +1,141 @@
+from __future__ import annotations
+
+from io import BytesIO
+from typing import Literal, TYPE_CHECKING
+
+from discord import File
+from discord.ext.commands import Cog, command, param
+from jishaku.codeblocks import Codeblock, codeblock_converter
+
+if TYPE_CHECKING:
+    from discord import MessageReference, TextChannel
+    from discord.ext.commands import Context
+
+    from .. import ReplBot
+
+
+def sanitize_str_for_codeblock(string: str) -> str:
+    return string.replace("`", "\u200B`\u200B")  # Zero-width space.
+
+
+async def get_message_content(channel: TextChannel, ref: MessageReference) -> str:
+    if ref.resolved:
+        return ref.resolved.content
+    else:
+        message = await channel.fetch_message(ref.message_id)
+        return message.content
+
+
+class Playground(
+    Cog,
+    name = "Playground",
+    description = "V Playground commands.",
+):
+    def __init__(self, bot: ReplBot) -> None:
+        self.bot = bot
+
+    async def run_test_common(
+        self,
+        ctx: Context,
+        code: Codeblock | None,
+        *,
+        type: Literal["run"] | Literal["run_test"]
+    ) -> None:
+        if not code:
+            if not (reply := ctx.message.reference):
+                await ctx.reply("No code provided.")
+                return
+
+            content = await get_message_content(ctx.channel, reply)
+            code = codeblock_converter(content)
+
+        async with await self.bot.session.post(
+            f"https://play.vlang.io/{type}",
+            data = { "code": code.content },
+        ) as response:
+            text = sanitize_str_for_codeblock(await response.text())
+
+            if len(text) + 6 > 2000:
+                await ctx.reply(
+                    "The output was too long to be sent as a message. Here is a file instead:",
+                    file = File(BytesIO(text.encode()), filename = "output.txt")
+                )
+                return
+
+            await ctx.reply(
+                "```" + text + "```"
+            )
+
+    @command(
+        aliases = ("eval", "repl"),
+        brief = "Runs V code.",
+        help = "Runs V code."
+    )
+    async def run(
+        self,
+        ctx: Context,
+        *,
+        code: Codeblock | None = param(converter = codeblock_converter, default = None)
+    ) -> None:
+        await self.run_test_common(ctx, code, type = "run")
+
+    @command(
+        brief = "Runs tests of V code.",
+        help = "Runs tests of V code."
+    )
+    async def test(
+        self,
+        ctx: Context,
+        *,
+        code: Codeblock | None = param(converter = codeblock_converter, default = None)
+    ) -> None:
+        await self.run_test_common(ctx, code, type = "run_test")
+
+    @command(
+        aliases = ("download",),
+        brief = "Shows the code in a V playground link.",
+        help = "Shows the code in a V playground link."
+    )
+    async def show(self, ctx: Context, query: str | None = None) -> None:
+        if not query:
+            if not (reply := ctx.message.reference):
+                await ctx.reply("No query provided.")
+                return
+
+            content = await get_message_content(ctx.channel, reply)
+
+            if "play.vlang.io/?query=" in content:
+                query = content.split("play.vlang.io/?query=", 1)[1].split(" ", 1)[0]
+            else:
+                query = content.split(" ", 1)[0]
+
+        query = query.lstrip("https://").lstrip("play.vlang.io/?query=")
+
+        if not query:
+            await ctx.reply("No query provided.")
+            return
+
+        async with await self.bot.session.post(
+            f"https://play.vlang.io/query",
+            data = { "hash": query }
+        ) as response:
+            text = sanitize_str_for_codeblock(await response.text())
+
+            if text == "Not found.":
+                await ctx.reply("Invalid link.")
+                return
+
+            if len(text) + 8 > 2000:
+                await ctx.reply(
+                    "The code was too long to be sent as a message. Here is a file instead:",
+                    file = File(BytesIO(text.encode()), filename = "code.v")
+                )
+                return
+
+            await ctx.reply(
+                "```v\n" + text + "```"
+            )
+
+
+async def setup(bot: ReplBot) -> None:
+    await bot.add_cog(Playground(bot))