41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
from io import BytesIO
|
||
|
|
from typing import Union
|
||
|
|
|
||
|
|
import discord
|
||
|
|
from ralgo.ralgo import Ralgo
|
||
|
|
from tuxbot.cogs.Crypto.functions.file import find_ext
|
||
|
|
|
||
|
|
|
||
|
|
def encode(params: dict) -> Union[str, discord.File]:
|
||
|
|
statement = Ralgo(params["message"])
|
||
|
|
params = params["params"]
|
||
|
|
encoded = statement.encode(chars=params["chars"])
|
||
|
|
|
||
|
|
if params["compressed"]:
|
||
|
|
return str(encoded.compress())
|
||
|
|
if params["graphical"]:
|
||
|
|
output = encoded.graphical().encode()
|
||
|
|
return discord.File(BytesIO(output.to_bytes()), "output.png")
|
||
|
|
|
||
|
|
return str(encoded)
|
||
|
|
|
||
|
|
|
||
|
|
def decode(params: dict) -> Union[str, discord.File]:
|
||
|
|
statement = Ralgo(params["message"])
|
||
|
|
params = params["params"]
|
||
|
|
|
||
|
|
if params["graphical"]:
|
||
|
|
output = Ralgo(statement.graphical().decode()).decode()
|
||
|
|
elif params["compressed"]:
|
||
|
|
output = Ralgo(statement.decompress()).decode()
|
||
|
|
else:
|
||
|
|
output = statement.decode(chars=params["chars"])
|
||
|
|
|
||
|
|
if isinstance(output, bytes):
|
||
|
|
return discord.File(BytesIO(output), f"output.{find_ext(output)}")
|
||
|
|
|
||
|
|
output = discord.utils.escape_markdown(str(output))
|
||
|
|
output = discord.utils.escape_mentions(output)
|
||
|
|
|
||
|
|
return output if len(output) > 0 else "no content..."
|