irc: add drop command

This commit is contained in:
William Pitcock 2018-08-11 10:27:04 -05:00
parent f048e2efff
commit 9c92a6bc0e
2 changed files with 10 additions and 2 deletions

View file

@ -57,3 +57,7 @@ def fetch_auth(account):
return DATABASE["auths"][account] return DATABASE["auths"][account]
return None return None
def drop_auth(account):
DATABASE["auths"].pop(account, None)

View file

@ -7,7 +7,7 @@ from blinker import signal
from . import CONFIG from . import CONFIG
from .irc_envelope import RFC1459Message from .irc_envelope import RFC1459Message
from .authreqs import new_auth_req, set_irc_bot, check_auth, fetch_auth from .authreqs import new_auth_req, set_irc_bot, check_auth, fetch_auth, drop_auth
IRC_CONFIG = CONFIG.get('irc', {}) IRC_CONFIG = CONFIG.get('irc', {})
# SASL_PAYLOAD = base64.b64encode(b'\x00'.join([IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_password']])) # SASL_PAYLOAD = base64.b64encode(b'\x00'.join([IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_username'], IRC_CONFIG['sasl_password']]))
@ -111,6 +111,10 @@ class IRCProtocol(asyncio.Protocol):
self.voice(nickname) self.voice(nickname)
elif action == 'invite': elif action == 'invite':
self.invite(nickname) self.invite(nickname)
elif action == 'drop':
data = fetch_auth(account)
drop_auth(account)
self.say(nickname, "The association of \x02{0}\x02 with \x02{1}\x02 has been dropped.".format(account, data))
elif 'whois' in action: elif 'whois' in action:
self.whois(nickname, action['whois'], account) self.whois(nickname, action['whois'], account)
@ -148,7 +152,7 @@ class IRCProtocol(asyncio.Protocol):
source_nick = message.source.split('!')[0] source_nick = message.source.split('!')[0]
if message.params[1] == 'auth': if message.params[1] == 'auth':
self.fetch_account_whox(message) self.fetch_account_whox(message)
elif message.params[1] in ('voice', 'invite'): elif message.params[1] in ('voice', 'invite', 'drop'):
self.set_pending_action(source_nick, message.params[1]) self.set_pending_action(source_nick, message.params[1])
self.fetch_account_whox(message) self.fetch_account_whox(message)