Enso-Bot/cogs/interactive.py

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

# 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

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


def random_line(file):
    """Return a random line from the chosen file"""

    lines = open(f'images/FunCommands/{file}.txt').read().splitlines()
    return random.choice(lines)


# 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, 1, 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

        # 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=random_line("kissing"))
        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, 1, 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

        # 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_line("cuddling"))
        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="kill")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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}**"

        # 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_line("killing"))
        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="slap")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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}**"

        # 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_line("slapping"))
        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="pat")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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}**"

        # 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_line("patting"))
        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="lemon")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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"]

        # 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)

    @command(name="choke")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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}**"

        # 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_line("choking"))
        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="hug")
    @bot_has_permissions(embed_links=True)
    @cooldown(1, 1, 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}**"

        # 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_line("hugging"))
        embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)

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


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