natsuko/cogs/fun.py

import discord
import requests

import utils.XOR as XOR
import utils.FFC as FFC

from discord import app_commands
from discord.ext import commands
from os import remove
from os.path import splitext, join

tmp_dir = "tmp"

class Fun(commands.Cog):
    htip_gr = app_commands.Group(name="htip", description="Hide text in picture")

    @htip_gr.command()
    async def write(self, itr: discord.Interaction, text: str, pic: discord.Attachment, key: str = None):
        if key is not None:
            text = XOR.char_encode(text, key)

        filename = f"{itr.id}.pic.filename"
        fln, ext = splitext(filename)
        if ext not in ('.jpg', '.png'):
            await itr.response.send_message("Supported only JPEG and PNG")
            return

        url = pic.url
        img = requests.get(url)
        with open(join(tmp_dir, filename), 'wb') as img_file:
            img_file.write(img.content)

        FFC.write(join(tmp_dir, filename), join(tmp_dir, f'{fln}.secret{ext}'), text)
        with open(join(tmp_dir, f'{fln}.secret{ext}'), 'rb') as f:
            await itr.response.send_message(file=discord.File(f))
        remove(join(tmp_dir, filename))
        remove(join(tmp_dir, f'{fln}.secret{ext}'))


    @htip_gr.command()
    async def read(self, itr: discord.Interaction, pic: discord.Attachment, key: str = None):
        filename = f"{itr.id}.pic.filename"

        fln, ext = splitext(filename)
        if ext not in ('.jpg', '.png'):
            await itr.response.send_message('Supported only JPEG or PNG')
            return

        url = pic.url
        img = requests.get(url)
        with open(join(tmp_dir, filename), 'wb') as img_file:
            img_file.write(img.content)

        secret_text = FFC.read(join(tmp_dir, filename))
        if key is not None:
            secret_text = XOR.char_encode(secret_text, key)
        await itr.response.send_message(secret_text)

        remove(join(tmp_dir, filename))


    @app_commands.command()
    async def demotivator(self, inter: discord.Interaction, title: str, text: str, image: discord.Attachment):
        logger.debug((title, text))

        if not "image" in image.content_type:
            await inter.response.send_message("Это не изображение")
            return

        logger.debug("Meow")
        filename = join("tmp", f"{inter.id}_{image.filename}")
        logger.debug(filename)
        await image.save(filename)
        Dem(title, text).create(filename, font_name="FreeSans.ttf")
        with open('demresult.jpg', 'rb') as f:
            await inter.response.send_message(file=discord.File(f))
        remove(filename)
        remove('demresult.jpg')


async def setup(bot): 
    await bot.add_cog(Fun())