import numpy as np
import pyaudio
import discord
import time


CHANNELS = 1
RATE = 48000

TOKEN = 'NDY5NTE2MDQ0OTgyMjg4Mzg0.DjJIhg.NAkdkg2IdKSuPs12qC4OEg3PyfA'


class StreamReader(object):
    def __init__(self, stream: pyaudio.Stream):
        self.stream = stream

    def read(self, frame_size: int):
        return self.stream.read(frame_size // 2, False)


def finalize(stream, p):
    stream.stop_stream()
    print("Stream is stopped")
    stream.close()
    p.terminate()


def main():
    client = discord.Client()
    p = pyaudio.PyAudio()
    stream = p.open(format=p.get_format_from_width(2), channels=CHANNELS, rate=RATE, input=True, start=False)

    @client.event
    async def on_message(message):
        # we do not want the bot to reply to itself
        if message.author == client.user:
            return

        if message.content.startswith('!test'):
            voice = await client.join_voice_channel(message.author.voice_channel)
            voice.encoder_options(sample_rate=RATE, channels=CHANNELS)

            stream.start_stream()
            player = voice.create_stream_player(StreamReader(stream), after=lambda: finalize(stream, p))
            player.start()
        else:
            print(message.content)

    @client.event
    async def on_ready():
        print('Logged in as')
        print(client.user.name)
        print(client.user.id)
        print('------')

    client.run(TOKEN)


if __name__ == '__main__':
    main()