aboutsummaryrefslogtreecommitdiffstats
path: root/bot/cogs/hacktober/spookysound.py
blob: e159851726a404fbe45a0b18c5988e120df8a8c2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import random
from pathlib import Path

import discord
from discord.ext import commands

from bot.constants import HACKTOBER_VOICE_CHANNEL_ID


class SpookySound:
    """
    A cog that plays a spooky sound in a voice channel on command.
    """

    def __init__(self, bot):
        self.bot = bot
        self.sound_files = list(Path("./bot/resources/spookysounds").glob("*.mp3"))
        self.channel = None

    @commands.cooldown(rate=1, per=1)
    @commands.command(brief="Play a spooky sound, restricted to once per 2 mins")
    async def spookysound(self, ctx):
        """
        Connect to the Hacktoberbot voice channel, play a random spooky sound, then disconnect. Cannot be used more than
        once in 2 minutes.
        """
        if not self.channel:
            await self.bot.wait_until_ready()
            self.channel = self.bot.get_channel(HACKTOBER_VOICE_CHANNEL_ID)

        await ctx.send("Initiating spooky sound...")
        file_path = random.choice(self.sound_files)
        src = discord.FFmpegPCMAudio(str(file_path.resolve()))
        voice = await self.channel.connect()
        voice.play(src, after=lambda e: self.bot.loop.create_task(self.disconnect(voice)))

    @staticmethod
    async def disconnect(voice):
        """
        Helper method to disconnect a given voice client.
        """
        await voice.disconnect()


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