feat(commands:ralgo|Crypto): add ralgo api

.idea/vcs.xml

@@ -7,5 +7,6 @@
   </component>
   <component name="VcsDirectoryMappings">
     <mapping directory="$PROJECT_DIR$" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/tuxbot/cogs/Crypto/deps/ralgo" vcs="Git" />
   </component>
 </project>

setup.cfg

@@ -24,6 +24,7 @@ install_requires =
     jishaku @ git+https://github.com/Gorialis/jishaku
     psutil==5.7.2
     pydig==0.3.0
+    ralgo @ git+https://github.com/Rom1-J/ralgo
     rich==9.10.0
     sentry_sdk>=0.20.2
     structured_config==4.12

tuxbot/cogs/Crypto/__init__.py

@@ -0,0 +1,19 @@
+from collections import namedtuple
+
+from tuxbot.core.bot import Tux
+from .crypto import Crypto
+from .config import CryptoConfig, HAS_MODELS
+
+VersionInfo = namedtuple("VersionInfo", "major minor micro release_level")
+version_info = VersionInfo(major=1, minor=0, micro=0, release_level="alpha")
+
+__version__ = "v{}.{}.{}-{}".format(
+    version_info.major,
+    version_info.minor,
+    version_info.micro,
+    version_info.release_level,
+).replace("\n", "")
+
+
+def setup(bot: Tux):
+    bot.add_cog(Crypto(bot))

tuxbot/cogs/Crypto/config.py

@@ -0,0 +1,10 @@
+from structured_config import Structure
+
+HAS_MODELS = False
+
+
+class CryptoConfig(Structure):
+    pass
+
+
+extra = {}

tuxbot/cogs/Crypto/crypto.py

@@ -0,0 +1,79 @@
+import logging
+from io import BytesIO
+
+import discord
+from discord.ext import commands
+from ralgo.ralgo import Ralgo
+
+from tuxbot.cogs.Crypto.functions.extractor import extract
+from tuxbot.cogs.Crypto.functions.file import find_ext
+from tuxbot.core.bot import Tux
+from tuxbot.core.i18n import (
+    Translator,
+)
+from tuxbot.core.utils.functions.extra import group_extra, ContextPlus
+
+
+log = logging.getLogger("tuxbot.cogs.Crypto")
+_ = Translator("Crypto", __file__)
+
+
+class Crypto(commands.Cog, name="Crypto"):
+    def __init__(self, bot: Tux):
+        self.bot = bot
+
+    @group_extra(name="ralgo")
+    async def _ralgo(self, ctx: commands.Context):
+        if ctx.invoked_subcommand is None:
+            await ctx.send_help("ralgo")
+
+    @_ralgo.command(name="encode")
+    async def _ralgo_encode(self, ctx: ContextPlus, *, data: str = None):
+        try:
+            params = await extract(ctx.message.attachments, data, 100000)
+        except ValueError:
+            return await ctx.send("Invalid data provided")
+
+        statement = Ralgo(params["message"])
+        params = params["params"]
+        encoded = statement.encode(chars=params["chars"])
+
+        if params["compressed"]:
+            return await ctx.send(str(encoded.compress()))
+        if params["graphical"]:
+            output = encoded.graphical().encode()
+            file = discord.File(BytesIO(output.to_bytes()), "output.png")
+
+            return await ctx.send(file=file)
+
+        await ctx.send(str(encoded))
+
+    @_ralgo.command(name="decode")
+    async def _ralgo_decode(self, ctx: ContextPlus, *, data: str = None):
+        try:
+            params = await extract(ctx.message.attachments, data, 5000000)
+        except ValueError:
+            return await ctx.send("Invalid data provided")
+
+        statement = Ralgo(params["message"])
+        params = params["params"]
+
+        if params["graphical"]:
+            output = Ralgo(statement.graphical().decode()).decode()
+
+            output = discord.utils.escape_markdown(str(output))
+            output = discord.utils.escape_mentions(output)
+        elif params["compressed"]:
+            output = Ralgo(statement.decompress()).decode()
+        else:
+            output = statement.decode(chars=params["chars"])
+
+        if isinstance(output, bytes):
+            file = discord.File(BytesIO(output), f"output.{find_ext(output)}")
+
+            return await ctx.send(file=file)
+
+        output = discord.utils.escape_markdown(str(output))
+        output = discord.utils.escape_mentions(output)
+
+        await ctx.send(output if len(output) > 0 else "no content...")

tuxbot/cogs/Crypto/functions/extractor.py

@@ -0,0 +1,28 @@
+from typing import Optional, NoReturn, Union
+
+from discord import Attachment
+from tuxbot.cogs.Crypto.functions.parser import data_parser
+
+
+async def extract(
+    attachments: list[Optional[Attachment]], data: str, max_size: int
+) -> Union[dict, NoReturn]:
+    if not data and len(attachments) == 0:
+        raise ValueError
+
+    kwargs = data_parser(data)
+
+    if len(attachments) > 0:
+        file = attachments[0]
+        if file.size > max_size:
+            raise ValueError
+
+        kwargs["message"] = await file.read()
+
+    params = {
+        "compressed": "compressed" in kwargs.keys(),
+        "graphical": "graphical" in kwargs.keys(),
+        "chars": kwargs["chars"] if "chars" in kwargs.keys() else (".", ","),
+    }
+
+    return {"message": kwargs["message"], "params": params}

tuxbot/cogs/Crypto/functions/file.py

@@ -0,0 +1,10 @@
+# https://python-forum.io/Thread-read-a-binary-file-to-find-its-type
+def find_ext(content: bytes) -> str:
+    magic_numbers = {
+        'png': bytes([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A])
+    }
+
+    if content.startswith(magic_numbers["png"]):
+        return "png"
+
+    return "txt"

tuxbot/cogs/Crypto/functions/parser.py

@@ -0,0 +1,32 @@
+import re
+
+
+def data_parser(data: str) -> dict:
+    output = {
+        "message": None,
+    }
+
+    if not data:
+        return output
+
+    if "--compressed" in data:
+        output["compressed"] = True
+        data = "".join(data.rsplit("--compressed", 1))
+
+    if "--graphical" in data:
+        output["graphical"] = True
+        data = "".join(data.rsplit("--graphical", 1))
+
+        if "compressed" in output.keys():
+            del output["compressed"]
+
+    if "--chars" in data:
+        regex = r"--chars=(\S\S)"
+        match = re.search(regex, data)
+
+        output["chars"] = tuple(match.group()[-2:])
+        data = "".join(data.rsplit(match.group(), 1))
+
+    output["message"] = data.strip()
+
+    return output

tuxbot/cogs/Crypto/locales/en-US.po

@@ -0,0 +1,57 @@
+# French translations for Tuxbot-bot package
+# Traductions françaises du paquet Tuxbot-bot.
+# Copyright (C) 2020 THE Tuxbot-bot'S COPYRIGHT HOLDER
+# This file is distributed under the same license as the Tuxbot-bot package.
+# Automatically generated, 2020.
+#
+msgid ""
+msgstr ""
+"Project-Id-Version: Tuxbot-bot\n"
+"Report-Msgid-Bugs-To: rick@gnous.eu\n"
+"POT-Creation-Date: 2020-10-21 01:15+0200\n"
+"PO-Revision-Date: 2020-10-21 01:15+0200\n"
+"Last-Translator: Automatically generated\n"
+"Language-Team: none\n"
+"Language: en_US\n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=UTF-8\n"
+"Content-Transfer-Encoding: 8bit\n"
+"Plural-Forms: nplurals=2; plural=(n != 1);\n"
+
+#: tuxbot/cogs/Polls/polls.py:55 tuxbot/cogs/Polls/polls.py:176
+msgid "**Preparation**"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:204
+#, python-brace-format
+msgid "Proposed addition for poll #{id}"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:209
+msgid "Poll"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:211
+msgid "here"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:215
+#, python-brace-format
+msgid "Requested by {author}"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:21
+#: tuxbot/cogs/Polls/functions/converters.py:49
+msgid "Please provide a message in this channel"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:30
+#: tuxbot/cogs/Polls/functions/converters.py:36
+#: tuxbot/cogs/Polls/functions/converters.py:41
+#: tuxbot/cogs/Polls/functions/converters.py:46
+msgid "Unable to find this poll"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:58
+msgid "Your proposal must be smaller than 30"
+msgstr ""

tuxbot/cogs/Crypto/locales/fr-FR.po

@@ -0,0 +1,58 @@
+# French translations for Tuxbot-bot package
+# Traductions françaises du paquet Tuxbot-bot.
+# Copyright (C) 2020 THE Tuxbot-bot'S COPYRIGHT HOLDER
+# This file is distributed under the same license as the Tuxbot-bot package.
+# Automatically generated, 2020.
+#
+msgid ""
+msgstr ""
+"Project-Id-Version: Tuxbot-bot\n"
+"Report-Msgid-Bugs-To: rick@gnous.eu\n"
+"POT-Creation-Date: 2020-10-21 01:15+0200\n"
+"PO-Revision-Date: 2020-10-21 01:15+0200\n"
+"Last-Translator: Automatically generated\n"
+"Language-Team: none\n"
+"Language: fr\n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=UTF-8\n"
+"Content-Transfer-Encoding: 8bit\n"
+"Plural-Forms: nplurals=2; plural=(n > 1);\n"
+
+#: tuxbot/cogs/Polls/polls.py:55 tuxbot/cogs/Polls/polls.py:176
+msgid "**Preparation**"
+msgstr "**Préparation**"
+
+#: tuxbot/cogs/Polls/polls.py:204
+#, python-brace-format
+msgid "Proposed addition for poll #{id}"
+msgstr "Proposition d'ajout pour le sondage #{id}"
+
+#: tuxbot/cogs/Polls/polls.py:209
+msgid "Poll"
+msgstr "Sondage"
+
+#: tuxbot/cogs/Polls/polls.py:211
+msgid "here"
+msgstr "ici"
+
+#: tuxbot/cogs/Polls/polls.py:215
+#, python-brace-format
+msgid "Requested by {author}"
+msgstr "Demandée par {author}"
+
+#: tuxbot/cogs/Polls/functions/converters.py:21
+#: tuxbot/cogs/Polls/functions/converters.py:49
+msgid "Please provide a message in this channel"
+msgstr "Veuillez fournir un message dans ce salon"
+
+#: tuxbot/cogs/Polls/functions/converters.py:30
+#: tuxbot/cogs/Polls/functions/converters.py:36
+#: tuxbot/cogs/Polls/functions/converters.py:41
+#: tuxbot/cogs/Polls/functions/converters.py:46
+msgid "Unable to find this poll"
+msgstr "Impossible de trouver ce sondage"
+
+#: tuxbot/cogs/Polls/functions/converters.py:58
+msgid "Your proposal must be smaller than 30"
+msgstr "Votre proposition doit être inférieure à 30"
+

tuxbot/cogs/Crypto/locales/messages.pot

@@ -0,0 +1,56 @@
+# SOME DESCRIPTIVE TITLE.
+# Copyright (C) YEAR THE PACKAGE'S COPYRIGHT HOLDER
+# This file is distributed under the same license as the Tuxbot-bot package.
+# FIRST AUTHOR <EMAIL@ADDRESS>, YEAR.
+#
+#, fuzzy
+msgid ""
+msgstr ""
+"Project-Id-Version: Tuxbot-bot\n"
+"Report-Msgid-Bugs-To: rick@gnous.eu\n"
+"POT-Creation-Date: 2021-01-26 16:12+0100\n"
+"PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
+"Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
+"Language-Team: LANGUAGE <LL@li.org>\n"
+"Language: \n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=CHARSET\n"
+"Content-Transfer-Encoding: 8bit\n"
+
+#: tuxbot/cogs/Polls/polls.py:55 tuxbot/cogs/Polls/polls.py:176
+msgid "**Preparation**"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:204
+#, python-brace-format
+msgid "Proposed addition for poll #{id}"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:209
+msgid "Poll"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:211
+msgid "here"
+msgstr ""
+
+#: tuxbot/cogs/Polls/polls.py:215
+#, python-brace-format
+msgid "Requested by {author}"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:21
+#: tuxbot/cogs/Polls/functions/converters.py:49
+msgid "Please provide a message in this channel"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:30
+#: tuxbot/cogs/Polls/functions/converters.py:36
+#: tuxbot/cogs/Polls/functions/converters.py:41
+#: tuxbot/cogs/Polls/functions/converters.py:46
+msgid "Unable to find this poll"
+msgstr ""
+
+#: tuxbot/cogs/Polls/functions/converters.py:58
+msgid "Your proposal must be smaller than 30"
+msgstr ""

tuxbot/cogs/Logs/logs.py

@@ -66,6 +66,7 @@ class Logs(commands.Cog, name="Logs"):
         self.old_on_error = bot.on_error
         bot.on_error = self.on_error
 
+        if self.bot.instance_name != "dev":
             sentry_sdk.init(
                 dsn=self.__config.sentryKey,
                 traces_sample_rate=1.0,
@@ -232,6 +233,7 @@ class Logs(commands.Cog, name="Logs"):
         if isinstance(error, (discord.Forbidden, discord.NotFound)):
             return
 
+        if self.bot.instance_name != "dev":
             sentry_sdk.capture_exception(error)
         self.bot.console.log(
             "Command Error, check sentry or discord error channel"
@@ -273,6 +275,7 @@ class Logs(commands.Cog, name="Logs"):
         e.remove_field(1)
         e.remove_field(1)
 
+        if self.bot.instance_name != "dev":
             e.set_footer(text=sentry_sdk.last_event_id())
 
         await ctx.send(embed=e)

tuxbot/core/bot.py

@@ -42,6 +42,7 @@ packages: List[str] = [
     "tuxbot.cogs.Polls",
     "tuxbot.cogs.Custom",
     "tuxbot.cogs.Network",
+    "tuxbot.cogs.Crypto",
 ]
 
 

tuxbot/core/utils/functions/extra.py

@@ -1,5 +1,6 @@
 import asyncio
 import random
+from io import BytesIO
 
 import aiohttp
 import discord
@@ -23,7 +24,7 @@ class ContextPlus(commands.Context):
         delete_after=None,
         nonce=None,
         allowed_mentions=None,
-        deletable=True
+        deletable=True,
     ):  # i know *args and **kwargs but, i prefer work with same values
         from tuxbot.core.utils.functions.utils import (
             replace_in_dict,
@@ -39,6 +40,10 @@ class ContextPlus(commands.Context):
                 )
                 .replace(self.bot.config.Core.ip, IP_REPLACEMENT)
             )
+
+            if len(content) > 1800:
+                file = discord.File(BytesIO(content.encode()), "output.txt")
+                content = "output too long..."
         if embed:
             e = embed.to_dict()
             for key, value in e.items():