radiobot/main.py

import os
import sys
import re
import struct
import asyncio
from subprocess import Popen

import discord
from discord.ext import commands
import urllib.request as urllib2

login_token = os.environ['token']
voice_channel_id = os.environ['channel_voice']
text_channel_id = os.environ['channel_text']
source = os.environ['url']

# Configure the bot
description = '''Radiobot'''
bot = commands.Bot(command_prefix='!', description=description)

# Initialize some global variables
voice_client = None
text_channel = None
isConnected = False
encoding = 'latin1'
bot_version = '0.2'

print('[INFO] radiobot ' + bot_version + ' starting')

@bot.event
async def on_ready():
    print('[INFO] Logged in as ' + bot.user.name + ' #' + str(bot.user.id))

    # get channels
    voice_channel = await bot.fetch_channel(voice_channel_id)
    debug_channel = await bot.fetch_channel(text_channel_id)
    if not voice_channel:
        print("[WARN] No voice channel " + voice_channel_id + " found!")
    if not debug_channel:
        print("[WARN] No text channel " + text_channel_id + " found!")
    await debug_channel.send('] ready! :satellite_orbital:')

@bot.event
async def on_message(message):
    # don't respond to ourselves
    if message.author == bot.user:
        return

    if not str(message.channel.id) == text_channel_id:
        return

    print('<' + message.author.nick + '> ' + message.content)

    if message.content == '!help':
        await message.channel.send('] radiobot commands: help, version, song :star:')

    if message.content == '!version':
        await message.channel.send('] radiobot ' + bot_version + ' - python ' + os.environ['PYTHON_VERSION'] + ' - github.com/deflax/radiobot :purple_heart:')

    if message.content == '!song':
        request = urllib2.Request(source, headers={'Icy-MetaData': 1})  # request metadata
        response = urllib2.urlopen(request)
        metaint = int(response.headers['icy-metaint'])
        for _ in range(10): # # title may be empty initially, try several times
            response.read(metaint)  # skip to metadata
            metadata_length = struct.unpack('B', response.read(1))[0] * 16  # length byte
            metadata = response.read(metadata_length).rstrip(b'\0')
            m = re.search(br"StreamTitle='([^']*)';", metadata)
            if m:
                title = m.group(1)
                if title:
                    break
        else:
            print('no title found')
            return
        meta = title.decode(encoding, errors='replace')
        metasplit = meta.split('*', 1)[0]
        await message.channel.send('] ' + metasplit + ' :musical_note:')

@bot.event
async def on_voice_state_update(member, before, after):
    """
    Starts events when a user changes their voice state.
    Such as connecting, disconnecting and moving between channels.
    :type member: discord.Member
    :type before: discord.VoiceState
    :type after: discord.VoiceState
    :param member: The member that changed their voice state.
    :param before: The member as they were before the change.
    :param after: The member as they are after the change.
    :return:
    """
    global isConnected
    FFMPEG_OPTS = {'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5', 'options': '-vn'}
    member_msg = None

    if member.bot:
        #print("[INFO] self event detection")
        return
    
    debug_channel = await bot.fetch_channel(text_channel_id)
    voice_channel = await bot.fetch_channel(voice_channel_id)
    member_ids = len(voice_channel.voice_states.keys())

    if before.channel is None:
        prev_chan = "not_found"
    else:
        prev_chan = str(before.channel.id)
    #if prev_chan == str(voice_channel_id):
    #    member_msg = str(member.nick) + ' is back in the void :cyclone:'

    if after.channel is None:
        next_chan = "not_found"
    else:
        next_chan = str(after.channel.id)
    if next_chan == str(voice_channel_id):
        member_msg = str(member.nick) + ' enjoys! :satellite:'

    if prev_chan == next_chan:
        #print('[INFO] ' + str(member.nick) + ' activity')
        pass
    else:
        if member_msg is not None:
            print('[INFO] ' + member_msg)
            await debug_channel.send('] ' + member_msg)

    if member_ids > 0 and isConnected == False:
        isConnected = True
        voice_client = await voice_channel.connect()
        voice_client.play(discord.FFmpegPCMAudio(source, **FFMPEG_OPTS))
        voice_client.source = discord.PCMVolumeTransformer(voice_client.source)
        voice_client.source.volume = 0.3
        return

    if member_ids == 1 and isConnected == True:
        isConnected = False
        #await debug_channel.send('] sleeping. :satellite_orbital:')
        for voice_client in bot.voice_clients:
            await voice_client.disconnect()
        return

# Start the bot with multiprocess compatiblity
if __name__ == "__main__":
    try:
        bot.loop.run_until_complete(bot.start(login_token))
    finally:
        bot.loop.close()