218 lines
8.1 KiB
Python
218 lines
8.1 KiB
Python
import asyncio
|
|
import logging
|
|
import base64
|
|
|
|
from blinker import signal
|
|
|
|
from . import CONFIG
|
|
from .irc_envelope import RFC1459Message
|
|
|
|
from .authreqs import new_auth_req, set_irc_bot, check_auth, fetch_auth
|
|
IRC_CONFIG = CONFIG.get('irc', {})
|
|
|
|
# SASL_PAYLOAD = base64.b64encode(b'\x00'.join([IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_password']]))
|
|
|
|
|
|
class IRCProtocol(asyncio.Protocol):
|
|
def connection_made(self, transport):
|
|
self.pending_actions = {}
|
|
|
|
self.caps_available = []
|
|
self.caps_requested = []
|
|
self.caps_acknowledged = []
|
|
|
|
self.transport = transport
|
|
self.recv_buffer = b''
|
|
|
|
self.send_message(verb='CAP', params=['LS'])
|
|
self.send_message(verb='NICK', params=[IRC_CONFIG['nickname']])
|
|
self.send_message(verb='USER', params=[IRC_CONFIG['username'], '*', '*', IRC_CONFIG['realname']])
|
|
|
|
def data_received(self, data):
|
|
self.recv_buffer += data
|
|
recvd = self.recv_buffer.replace(b'\r', b'').split(b'\n')
|
|
self.recv_buffer = recvd.pop(-1)
|
|
|
|
[self.line_received(m) for m in recvd]
|
|
|
|
def line_received(self, data):
|
|
data = data.decode('UTF-8', 'replace').strip('\r\n')
|
|
|
|
m = RFC1459Message.from_message(data)
|
|
self.message_received(m)
|
|
|
|
def request_caps(self, caps):
|
|
caps = [cap for cap in caps if cap in self.caps_available]
|
|
caps = [cap for cap in caps if cap not in self.caps_requested]
|
|
|
|
logging.debug('IRC: requesting caps: %r', caps)
|
|
|
|
self.caps_requested += caps
|
|
self.send_message(verb='CAP', params=['REQ', ' '.join(caps)])
|
|
|
|
def end_caps(self):
|
|
self.send_message(verb='CAP', params=['END'])
|
|
|
|
def do_blind_authenticate(self, username, password):
|
|
username = username.encode('ascii')
|
|
password = password.encode('ascii')
|
|
|
|
payload = b'\x00'.join([username, username, password])
|
|
payload = base64.b64encode(payload).decode('ascii')
|
|
|
|
self.send_message(verb='AUTHENTICATE', params=['PLAIN'])
|
|
self.send_message(verb='AUTHENTICATE', params=[payload])
|
|
|
|
def handle_cap_message(self, message):
|
|
self.cap_requested = True
|
|
if message.params[1] == 'LS':
|
|
caps = message.params[2].split()
|
|
logging.debug('IRC: available caps: %r', caps)
|
|
|
|
self.caps_available += message.params[2].split()
|
|
self.request_caps(['sasl', 'extended-join', 'account-tag', 'account-notify'])
|
|
elif message.params[1] == 'ACK':
|
|
caps = message.params[2].split()
|
|
logging.debug('IRC: acknowledged caps: %r', caps)
|
|
|
|
self.caps_acknowledged += caps
|
|
if 'sasl' in self.caps_acknowledged:
|
|
self.do_blind_authenticate(IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_password'])
|
|
else:
|
|
self.end_caps()
|
|
|
|
def join_channels(self):
|
|
self.send_message(verb='JOIN', params=[','.join(IRC_CONFIG['channels'])])
|
|
|
|
def say(self, target, message, verb='NOTICE'):
|
|
self.send_message(verb=verb, params=[target, message])
|
|
|
|
def invite(self, nickname):
|
|
[self.send_message(verb="INVITE", params=[nickname, chan]) for chan in IRC_CONFIG['channels']]
|
|
|
|
def voice(self, nickname):
|
|
[self.send_message(verb="MODE", params=[chan, "+v", nickname]) for chan in IRC_CONFIG['channels']]
|
|
|
|
def whois(self, nickname, chan, account):
|
|
data = fetch_auth(account)
|
|
if not data:
|
|
return
|
|
self.say(chan, '\x02{0}\x02: \x02{1}\x02'.format(nickname, data), verb='PRIVMSG')
|
|
|
|
def set_pending_action(self, nickname, action):
|
|
if nickname not in self.pending_actions:
|
|
self.pending_actions[nickname] = action
|
|
|
|
def process_pending_action(self, nickname, account=None):
|
|
if nickname not in self.pending_actions:
|
|
return
|
|
action = self.pending_actions.pop(nickname)
|
|
if action == 'voice':
|
|
self.voice(nickname)
|
|
elif action == 'invite':
|
|
self.invite(nickname)
|
|
elif 'whois' in action:
|
|
self.whois(nickname, action['whois'], account)
|
|
|
|
def handle_auth_req(self, req):
|
|
self.say(req.irc_nickname, "The actor \x02{0}\x02 is now linked to the IRC account \x02{1}\x02.".format(req.actor, req.irc_account))
|
|
self.set_pending_action(req.irc_nickname, 'voice')
|
|
self.process_pending_action(req.irc_nickname)
|
|
|
|
def pending_whois(self, nickname, pop=False):
|
|
if nickname not in self.pending_actions:
|
|
return False
|
|
data = self.pending_actions[nickname]
|
|
if isinstance(data, dict) and 'whois' in data:
|
|
return True
|
|
if pop:
|
|
self.pending_actions.pop(nickname)
|
|
|
|
def handle_whox(self, message):
|
|
nickname = message.params[1]
|
|
account = message.params[2]
|
|
|
|
if not check_auth(account) and not self.pending_whois(nickname, True):
|
|
auth = new_auth_req(nickname, account)
|
|
self.say(nickname, "Authentication is required for this action. In order to prove your identity, you need to send me a token via the fediverse.")
|
|
self.say(nickname, "On most platforms, posting like this will work: \x02@viera@viera.dereferenced.org {}\x02".format(auth))
|
|
self.say(nickname, "This token is ephemeral, so you can send it to me publicly if your platform does not support direct messages.")
|
|
else:
|
|
self.process_pending_action(nickname, account)
|
|
|
|
def fetch_account_whox(self, message):
|
|
source_nick = message.source.split('!')[0]
|
|
self.send_message(verb='WHO', params=[source_nick, "%na"])
|
|
|
|
def handle_private_message(self, message):
|
|
source_nick = message.source.split('!')[0]
|
|
if message.params[1] == 'auth':
|
|
self.fetch_account_whox(message)
|
|
elif message.params[1] in ('voice', 'invite'):
|
|
self.set_pending_action(source_nick, message.params[1])
|
|
self.fetch_account_whox(message)
|
|
|
|
def handle_public_message(self, message):
|
|
if not message.params[1].startswith(IRC_CONFIG['nickname']):
|
|
return
|
|
|
|
chunks = message.params[1].split()
|
|
if chunks[1] == 'whois':
|
|
self.set_pending_action(chunks[2], {'whois': message.params[0]})
|
|
self.send_message(verb='WHO', params=[chunks[2], "%na"])
|
|
|
|
def handle_chat_message(self, message):
|
|
if message.params[0] == IRC_CONFIG['nickname']:
|
|
self.handle_private_message(message)
|
|
else:
|
|
self.handle_public_message(message)
|
|
|
|
def handle_join(self, message):
|
|
source_nick = message.source.split('!')[0]
|
|
if check_auth(message.params[1]):
|
|
self.set_pending_action(source_nick, 'voice')
|
|
self.process_pending_action(source_nick, message.params[1])
|
|
|
|
def message_received(self, message):
|
|
if message.verb in ('PRIVMSG', 'NOTICE'):
|
|
self.handle_chat_message(message)
|
|
elif message.verb == '001':
|
|
self.join_channels()
|
|
elif message.verb == 'JOIN':
|
|
self.handle_join(message)
|
|
elif message.verb == 'CAP':
|
|
self.handle_cap_message(message)
|
|
elif message.verb == '354':
|
|
self.handle_whox(message)
|
|
elif message.verb == '433':
|
|
self.send_message(verb='NICK', params=[message.params[0] + '_'])
|
|
elif message.verb in ('900', '901', '902', '903', '904', '905', '906', '907'):
|
|
self.end_caps()
|
|
elif message.verb == 'PING':
|
|
self.send_message(verb='PONG', params=message.params)
|
|
elif message.verb in ('AUTHENTICATE',):
|
|
pass
|
|
else:
|
|
logging.debug('IRC: unhandled inbound message: %r', message)
|
|
|
|
def send_message(self, **kwargs):
|
|
m = RFC1459Message.from_data(**kwargs)
|
|
logging.debug('> %r', m)
|
|
self.transport.write(m.to_message().encode('utf-8') + b'\r\n')
|
|
|
|
|
|
async def irc_bot():
|
|
loop = asyncio.get_event_loop()
|
|
|
|
if 'host' not in IRC_CONFIG:
|
|
return
|
|
|
|
server = IRC_CONFIG['host']
|
|
port = IRC_CONFIG['port']
|
|
ssl = IRC_CONFIG['ssl']
|
|
|
|
transport, protocol = await loop.create_connection(IRCProtocol, host=server, port=port, ssl=ssl)
|
|
set_irc_bot(protocol)
|
|
|
|
logging.info('IRC bot ready.')
|