RGBCube/VReplBot
1
Fork 0
mirror of https://github.com/RGBCube/VReplBot synced 2025-07-27 16:37:47 +00:00
Code Issues Activity Actions

Stuff

Browse source
This commit is contained in:
RGBCube 2023-01-31 19:43:56 +03:00
parent 9a416f5000
commit 75b6c2e8c1
1 changed files with 19 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

25
v_repl_bot/cogs/repl.py
View file

@ -8,11 +8,20 @@ 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
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 REPL(
Cog,
name = "REPL",
@ -32,9 +41,13 @@ class REPL(
*,
code: Codeblock | None = param(converter = codeblock_converter, default = None)
) -> None:
if code is None:
await ctx.reply("No code provided.")
return
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(
"https://play.vlang.io/run",
@ -60,13 +73,12 @@ class REPL(
help = "Shows the code in a V playground query."
)
async def show(self, ctx: Context, query: str | None = None) -> None:
if query is None:
if not query:
if not (reply := ctx.message.reference):
await ctx.reply("No query provided.")
return
replied_message = await ctx.channel.fetch_message(reply.message_id)
content = replied_message.content
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]
@ -84,6 +96,7 @@ class REPL(
data = { "hash": query }
) as response:
text = await response.text()
text = text.replace("`", "\u200B`\u200B") # Zero-width space.
if text == "Not found.":
await ctx.reply("Invalid link.")