import twitchio.ext.commands
import collections
import zmq

from .filters import FFmpegChain, FFmpegFilter, filter_escape


def make_twitch_chat(name, nickname, token, channel, x=200, y=25, addr='tcp://localhost:5555', nick_width=160, nick_args={}, chat_width=300, chat_args={}):
    nick_id = 'drawtext@twitch-{}-nicks'.format(name)
    chat_id = 'drawtext@twitch-{}-messages'.format(name)
    runner = TwitchChatUpdater(nickname=nickname, token=token, channel=channel, nick_target=nick_id, chat_target=chat_id)
    return [runner.run], [FFmpegChain(
        FFmpegFilter(nick_id,
            text='Loading',
            expansion='none',
            x='{} + max(0, {} - text_w)'.format(x, nick_width),
            y=y,
            **nick_args
        ),
        FFmpegFilter(chat_id,
            text='chat...',
            expansion='none',
            x='{} + {}'.format(x, nick_width),
            y=y,
            **chat_args
        ),
        FFmpegFilter('zmq')
    )]

def zmq_escape(v):
    return (filter_escape(v)
        .replace('\\', '\\\\')
        .replace('\n', '\\\n')
        .replace(' ',  '\\ ')
    )

class TwitchChatUpdater(twitchio.ext.commands.Bot):
    def __init__(self, nickname, token, channel, nick_target, chat_target, addr='tcp://localhost:5555', maxlen=15):
        super().__init__(irc_token=token, nick=nickname, initial_channels=[channel], prefix='kwdakwdwajd')
        self.zmq_messages = collections.deque(maxlen=maxlen)
        self.zmq = zmq.Context()
        self.zmq_sock = self.zmq.socket(zmq.REQ)
        self.zmq_sock.connect(addr)
        self.zmq_nick_target = nick_target
        self.zmq_chat_target = chat_target

    def zmq_send_command(self, target, command, **args):
        cmd = '{} {} {}'.format(
            target, command,
            ':'.join(
                '{}={}'.format(k, zmq_escape(v)) for k, v in args.items()
            )
        ).encode('utf-8')
        self.zmq_sock.send(cmd)
        resp = self.zmq_sock.recv()

    def zmq_update(self):
        nick_text = '\n'.join('<{}>'.format(m.author.name) for m in self.zmq_messages if m)
        self.zmq_send_command(self.zmq_nick_target, 'reinit', text=nick_text)
        chat_text = '\n'.join(m.content for m in self.zmq_messages if m)
        self.zmq_send_command(self.zmq_chat_target, 'reinit', text=chat_text)

    async def event_ready(self):
        self.zmq_update()

    async def event_message(self, message):
        print(message.tags)
        self.zmq_messages.append(message)
        self.zmq_update()