from discord.ext import commands
import core_utils
import discord
import random
import requests
import os


class Canvass(commands.Cog):
    def __init__(self, bot):
        self.bot: commands.Bot = bot

    @commands.command(name="simp")
    async def simp(self, ctx: commands.Context, image_link=None):
        if not image_link:
            return await ctx.send("You must provide a link to an image")
        response = requests.get(
            "https://some-random-api.ml/canvas/simpcard/?avatar=%s" % image_link
        )
        with open("/app/simp.jpg", "wb") as f:
            f.write(response.content)
        await ctx.message.delete()
        await ctx.send(file=discord.File("/app/simp.jpg"))
        os.remove("/app/simp.jpg")

    @commands.command(name="horny")
    async def horny(self, ctx: commands.Context, image_link=None):
        if not image_link:
            return await ctx.send("You must provide a link to an image")
        response = requests.get(
            "https://some-random-api.ml/canvas/horny/?avatar=%s" % image_link
        )
        with open("/app/horny.jpg", "wb") as f:
            f.write(response.content)
        await ctx.message.delete()
        await ctx.send(file=discord.File("/app/horny.jpg"))
        os.remove("/app/horny.jpg")

    @commands.command(name="wasted")
    async def wasted(self, ctx: commands.Context, image_link=None):
        if not image_link:
            return await ctx.send("You must provide a link to an image")
        response = requests.get(
            "https://some-random-api.ml/canvas/wasted/?avatar=%s" % image_link
        )
        with open("/app/wasted.jpg", "wb") as f:
            f.write(response.content)
        await ctx.message.delete()
        await ctx.send(file=discord.File("/app/wasted.jpg"))
        os.remove("/app/wasted.jpg")

    @commands.command(name="pixelate")
    async def pixelate(self, ctx: commands.Context, image_link=None):
        if not image_link:
            return await ctx.send("You must provide a link to an image")
        response = requests.get(
            "https://some-random-api.ml/canvas/pixelate/?avatar=%s" % image_link
        )
        with open("/app/pixelate.jpg", "wb") as f:
            f.write(response.content)
        await ctx.message.delete()
        await ctx.send(file=discord.File("/app/pixelate.jpg"))
        os.remove("/app/pixelate.jpg")

    @commands.command(name="blur")
    async def blur(self, ctx: commands.Context, image_link=None):
        if not image_link:
            return await ctx.send("You must provide a link to an image")
        response = requests.get(
            "https://some-random-api.ml/canvas/blur/?avatar=%s" % image_link
        )
        with open("/app/blur.jpg", "wb") as f:
            f.write(response.content)
        await ctx.message.delete()
        await ctx.send(file=discord.File("/app/blur.jpg"))
        os.remove("/app/blur.jpg")


def setup(bot):
    bot.add_cog(Canvass(bot))