aboutsummaryrefslogtreecommitdiffstats
path: root/bot/cogs/spookysound.py
blob: dd607097875fc321d33972237b540029edb2b84d (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
import random
from pathlib import Path

import discord
from discord.ext import commands

HACKTOBERBOT_VOICE_CHANNEL_ID = 498804789287714816


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

    async def on_ready(self):
        self.channel = self.bot.get_channel(HACKTOBERBOT_VOICE_CHANNEL_ID)

    @commands.cooldown(rate=1, per=120)
    @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.
        """
        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))