Enso-Bot/cogs/interactive.py

# Ensō~Chan - A Multi Purpose Discord Bot That Has Everything Your Server Needs!
# Copyright (C) 2020  Goudham Suresh

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

import datetime
import random

import aiohttp
from decouple import config
from discord import Embed, Member
from discord.ext.commands import cooldown, command, BucketType, bot_has_permissions, Cog


# Gets the member and user avatar
def getMember(ctx):
    # Set member as the author
    member = ctx.message.author
    # Get the member avatar
    userAvatar = member.avatar_url

    return member, userAvatar


# Set up the Cog
class Interactive(Cog):
    """Interactive Commands! (E.G Kiss/Hug/Cuddle)"""

    def __init__(self, bot):
        self.bot = bot

    @Cog.listener()
    async def on_ready(self):
        """Printing out that Cog is ready on startup"""
        print(f"{self.__class__.__name__} Cog has been loaded\n-----")

    @command(name="kiss")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def kiss(self, ctx, member: Member):
        """Kiss your partner"""

        # Get the guild
        guild = ctx.guild

        # Error handling to make sure that the user can kiss themselves
        if member.id == ctx.author.id:
            kiss = False
            title = f":kissing_heart: :kissing_heart: | **{ctx.author.display_name}** kissed **themselves**"
        else:
            kiss = True
            title = f":kissing_heart: :kissing_heart: | **{ctx.author.display_name}** kissed **{member.display_name}**"

        # Get author record from cache/database
        result = await self.bot.check_cache(ctx.author.id, guild.id)

        married_user = result["married"]
        if married_user is None and kiss:
            await self.bot.generate_embed(ctx,
                                          desc="Σ(‘◉⌓◉’) You need to be married in order to use this command! Baka!")
            return
        elif not member.id == married_user and kiss:
            await self.bot.generate_embed(ctx, desc="Σ(‘◉⌓◉’) You can only kiss your partner! Baka!")
            return

        # Set details for search
        apikey = config("TENOR_AUTH")  # test value
        search_term = "anime-kiss"
        url = f"https://api.tenor.com/v1/random?q={search_term}&key={apikey}&limit=1"

        # get random results using default locale of EN_US
        # Searching API for the current airing shows
        async with aiohttp.ClientSession() as session:
            async with session.get(url) as resp:
                if resp.status == 200:
                    gifs = await resp.json()
                    url = gifs["results"][0]["media"][0]["gif"]["url"]
                else:
                    self.bot.generate_embed(ctx, desc="**Something Went Wrong With Tenor!**")

        # Get the member and the userAvatar
        member, userAvatar = getMember(ctx)

        # Set up the embed to display a random kissing gif
        embed = Embed(
            title=title,
            colour=self.bot.random_colour(),
            timestamp=datetime.datetime.utcnow())
        embed.set_image(url=url)
        embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

        # Send the embedded message to the user
        await ctx.send(embed=embed)

    @command(name="cuddle")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def cuddle(self, ctx, member: Member):
        """Cuddle your partner"""

        # Get the guild
        guild = ctx.guild

        # Error handling to make sure that the user can cuddle themselves
        if member.id == ctx.author.id:
            cuddle = False
            title = f":blush: :blush: | **{ctx.author.display_name}** cuddled **themselves**"
        else:
            cuddle = True
            title = f":blush: :blush: | **{ctx.author.display_name}** cuddled **{member.display_name}**"

        # Get author record from cache/database
        result = await self.bot.check_cache(ctx.author.id, guild.id)

        married_user = result["married"]
        if married_user is None and cuddle:
            await self.bot.generate_embed(ctx,
                                          desc="Σ(‘◉⌓◉’) You need to be married in order to use this command! Baka!")
            return
        elif not member.id == married_user and cuddle:
            await self.bot.generate_embed(ctx, desc="Σ(‘◉⌓◉’) You can only cuddle your partner! Baka!")
            return

        try:

            # Open the file containing the cuddling gifs
            with open('images/FunCommands/cuddling.txt') as file:
                # Store content of the file in cuddling_array
                cuddling_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random cuddling gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(cuddling_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="kill")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def kill(self, ctx, member: Member):
        """Kill a member"""

        if member is ctx.author:
            title = f":scream: :scream: | **{ctx.author.display_name}** killed **themselves**"
        else:
            title = f":scream: :scream: | **{ctx.author.display_name}** killed **{member.display_name}**"

        try:

            # Open the file containing the killing gifs
            with open('images/FunCommands/killing.txt') as file:
                # Store content of the file in killing_array
                killing_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random killing gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(killing_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="slap")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def slap(self, ctx, member: Member):
        """Slap a member"""

        if member is ctx.author:
            title = f":cold_sweat: :cold_sweat: | **{ctx.author.display_name}** slapped **themselves**"
        else:
            title = f":cold_sweat: :cold_sweat: | **{ctx.author.display_name}** slapped **{member.display_name}**"

        try:

            # Open the file containing the cuddling gifs
            with open('images/FunCommands/slapping.txt') as file:
                # Store content of the file in cuddling_array
                slapping_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random slapping gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(slapping_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="pat")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def pat(self, ctx, member: Member):
        """Pat a member"""

        if member is ctx.author:
            title = f"👉 👈 | **{ctx.author.display_name}** patted **themselves**"
        else:
            title = f"👉 👈 | **{ctx.author.display_name}** patted **{member.display_name}**"

        try:

            # Open the file containing the patting gifs
            with open('images/FunCommands/patting.txt') as file:
                # Store content of the file in patting_array
                patting_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random patting gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(patting_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="lemon")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def lemon(self, ctx, member: Member):
        """Give Lemon to member"""

        if member is ctx.author:
            title = f":relaxed: :relaxed: | **{ctx.author.display_name}** gave a lemon to **themselves**"
        else:
            title = f":relaxed: :relaxed: | **{ctx.author.display_name}** gave a lemon to **{member.display_name}**"

        lemon_array = ["https://media.discordapp.net/attachments/669812887564320769/720093589056520202/lemon.gif",
                       "https://media.discordapp.net/attachments/669812887564320769/720093575492272208/lemon2.gif",
                       "https://media.discordapp.net/attachments/718484280925224981/719629805263257630/lemon.gif"]

        try:

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random lemon gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(lemon_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="choke")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def choke(self, ctx, member: Member):
        """Choke a member"""

        if member is ctx.author:
            title = f":confounded: :confounded: | **{ctx.author.display_name}** choked **themselves**"
        else:
            title = f":confounded: :confounded: | **{ctx.author.display_name}** choked **{member.display_name}**"

        try:
            # Open the file containing the choking gifs
            with open('images/FunCommands/choking.txt') as file:
                # Store content of the file in choking_array
                choking_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random choking gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(choking_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)

    @command(name="hug")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 3, BucketType.user)
    async def hug(self, ctx, member: Member):
        """Hug a member"""

        if member is ctx.author:
            title = f":smiling_face_with_3_hearts: :smiling_face_with_3_hearts: | **{ctx.author.display_name}** hugged **themselves**"
        else:
            title = f":smiling_face_with_3_hearts: :smiling_face_with_3_hearts: | **{ctx.author.display_name}** hugged **{member.display_name}**"

        try:

            # Open the file containing the hug gifs
            with open('images/FunCommands/hugging.txt') as file:
                # Store content of the file in hugging_array
                hugging_array = file.readlines()

            # Get the member and the userAvatar
            member, userAvatar = getMember(ctx)

            # Set up the embed to display a random hugging gif
            embed = Embed(
                title=title,
                colour=self.bot.random_colour(),
                timestamp=datetime.datetime.utcnow())
            embed.set_image(url=random.choice(hugging_array))
            embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

            # Send the embedded message to the user
            await ctx.send(embed=embed)

        except FileNotFoundError as e:
            print(e)


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