import asyncio
import io

from pydub import AudioSegment # pip install pydub

import discord
from discord.ext import commands
from discord.sinks import MP3Sink


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

    async def finished_callback(
        self,
        sink: MP3Sink,
        channel: discord.TextChannel,
    ):
        mention_strs = []
        audio_segs: list[AudioSegment] = []
        files: list[discord.File] = []

        longest = AudioSegment.empty()

        for user_id, audio in sink.audio_data.items():

            mention_strs.append(f"<@{user_id}>")

            seg = AudioSegment.from_file(audio.file, format="mp3")

            # Determine the longest audio segment
            if len(seg) > len(longest):
                audio_segs.append(longest)
                longest = seg
            else:
                audio_segs.append(seg)

            audio.file.seek(0)
            files.append(discord.File(audio.file, filename=f"{user_id}.mp3"))

        for seg in audio_segs:
            longest = longest.overlay(seg)

        with io.BytesIO() as f:
            longest.export(f, format="mp3")
            await channel.send(
                f"Finished! Recorded audio for {', '.join(mention_strs)}.",
                files=files + [discord.File(f, filename="recording.mp3")],
            )

    @commands.command()
    async def join(self, ctx: commands.Context):
        await ctx.author.voice.channel.connect()

    @commands.command()
    async def record(self, ctx: commands.Context):
        vc: discord.VoiceClient = ctx.voice_client

        await ctx.send("Recording...")

        vc.start_recording(
            discord.sinks.MP3Sink(),
            self.finished_callback,
            ctx.channel,
            sync_start=True,
        )

        await asyncio.sleep(15) # edit to change recording duration
        vc.stop_recording()
        await ctx.send("Stopped recording.")


def setup(bot: commands.Bot):
    bot.add_cog(MyCog(bot))