Multi-DJ livestream toolkit
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 

70 lines
2.5 KiB

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 [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'),
runners=[runner.run]
)]
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()