import discord
import asyncio
import time
import json
client = discord.Client()

staff_channel_id = "224239361779040274"
allothers_channel_id = "225068884959559680"
allchannels_id = "224582137934905346"

udp_ip = "127.0.0.1"
udp_port = 5000
_debug = False

channel_list = {}

class MyDiscordClient(discord.Client):

    @asyncio.coroutine
    def on_ready(self):
        print('Logged in as ', self.user.name, self.user.id)

        # get registered gamechat channels...
        chans = self.get_all_channels()
        for chan in chans:
            if chan.name.startswith("srv_"):
                srv_code = chan.name[-3:]
                channel_list[srv_code] = chan.id
                if _debug: print(srv_code, chan.id)


    def write_message(self, cmd, extension):
        file = "command_" + str(int(time.time() * 1000)) + "." + extension
        with open ("/home/steam/sharedcontent/data/shared/ulxcmd/"+file, "w") as myfile:
            myfile.write(cmd )


    # We are receiving a message from Discord.
    @asyncio.coroutine
    def on_message(self, message):

        if (message.author.name != "THABBot"):  # Don't consider messages from THABBot itself

            # Helpful tips from THABBot
            msgLower = message.content.lower()
            if msgLower.startswith('why'):
                yield from self.send_message(message.channel, 'Ask someone who cares.')
            if msgLower.startswith('how do i'):
                if "points" in msgLower:
                    yield from self.send_message(message.channel, 'You earn points by spending time in-game on each THAB server. There are also opportunities to win points, including a regularly-scheduled lottery in every Pointshop-enabled gamemmode and gamemmode-specific perks like Slots in Elevator')
                else:
                    yield from self.send_message(message.channel, 'I have absolutely no idea.')



            for mention in message.mentions:
                # Did someone mention THABBot? If so, send a snarky response.
                if mention.name == "THABBot":
                    yield from self.send_message(message.channel, 'Here`s some bleach, ' + message.author.display_name + ". You know what to do." )

            if _debug: print(message.timestamp.strftime("%Y-%m-%d %H:%M:%S") + " - (#" + message.channel.name + ") " + message.author.display_name + ": " + message.content)


            # relay functionality here
            if message.channel.id == staff_channel_id:  # Did we say something on staff channel? Then gasay it to all servers
                cmd = "ulx asay [@Discord] " + message.author.display_name  + ": " + message.content
                self.write_message(cmd, "txt")
            elif message.channel.name.startswith("srv_"): # did someone say something on a server chat chan? Relay it.
                srv_code = message.channel.name[-3:]

                # was it a command?
                if (message.content.startswith("!")):
                    if message.author.id == "224309318697091074":
                        cmd = message.content[1:]
                        self.write_message(cmd, srv_code)
                else:
                    # Assume this is just chat
                    cmd = "ulx tsay [@Discord#" + srv_code + "] " + message.author.display_name  + ": " + message.content
                    self.write_message(cmd, srv_code)
                    cmd = "say AdminHide: [@Discord#" + srv_code + "] "  + message.author.display_name  + ": " + message.content
                    self.write_message(cmd, srv_code)



class MyUDPServer(asyncio.DatagramProtocol):

    def connection_made(self, transport):
        print('start', transport)
        self.transport = transport

    # We are receiving a transmission from globalcommand.py
    def datagram_received(self, data, addr):

        payload = data.decode('ISO-8859-1')
        if _debug: print('Data received:', addr, payload)
        data = json.loads(payload)
        msg = data["msg"]

        # completely ignore anything coming from globalcommand that looks like something originating from Discord.
        # ie, don't get stuck in a feedback loop.  This should probably be stopped in globalcommand itself.
        #if "[@Discord]" in msg:
        #   return

        # a global asay? Broadcast it to the staff channel
        if msg.startswith("ulx asay "):
            msg_send = msg[9:]                  # strip ulx asay
            asyncio.async(client.send_message(client.get_channel(staff_channel_id), msg_send ))

        # a global botasay?, post that to the staff channel too
        elif msg.startswith("ulx botasay "):
            msg_send = msg[12:]                     # strip ulx asay
            asyncio.async(client.send_message(client.get_channel(staff_channel_id), msg_send ))

        # a chat message on a specific, registered server? Post it to the appropriate server channel
        elif data["server"] in channel_list:
            msg_send = "[@" + data["serverName"] + "] " + data["from"] + " : " + msg
            asyncio.async(client.send_message(client.get_channel(channel_list[data["server"]]), msg_send ))
            asyncio.async(client.send_message(client.get_channel(allchannels_id), msg_send ))

        # a chat message on a channel we don't have registered yet? Post it to the catch-all channel.
        elif data["server"] !="":
            msg_send = "[@" + data["serverName"] + "] " + data["from"] + " : " + msg
            asyncio.async(client.send_message(client.get_channel(allothers_channel_id), msg_send ))
            asyncio.async(client.send_message(client.get_channel(allchannels_id), msg_send ))

        else:
            if _debug: print(" - Ignored! - ")

    def error_received(self, exc):
        print('error_received: ', exc)

    def connection_lost(self, exc):
        print('connection_lost: ', exc)



loop = asyncio.get_event_loop()
coro = loop.create_datagram_endpoint(MyUDPServer, local_addr=(udp_ip, udp_port))
server_udp = loop.run_until_complete(coro)

#loop.run_forever()


client.run('NjkzNjUxMzE4ODE2MTEyNzcw.XoAxGg.MeuL6JXCFwr8sIHrHA_Ce4Ho-JM')