Use CAP LS.

This commit is contained in:
Valentin Lorentz 2015-05-24 12:25:42 +02:00
parent f85395d8b1
commit e2b3b917e5
2 changed files with 79 additions and 34 deletions

View File

@ -350,7 +350,8 @@ class IrcState(IrcCommandDispatcher):
__firewalled__ = {'addMsg': None}
def __init__(self, history=None, supported=None,
nicksToHostmasks=None, channels=None,
capabilities_ack=None, capabilities_nak=None):
capabilities_ack=None, capabilities_nak=None,
capabilities_ls=None):
if history is None:
history = RingBuffer(conf.supybot.protocols.irc.maxHistoryLength())
if supported is None:
@ -361,6 +362,7 @@ class IrcState(IrcCommandDispatcher):
channels = ircutils.IrcDict()
self.capabilities_ack = capabilities_ack or set()
self.capabilities_nak = capabilities_nak or set()
self.capabilities_ls = capabilities_ls or {}
self.ircd = None
self.supported = supported
self.history = history
@ -949,6 +951,9 @@ class Irc(IrcCommandDispatcher):
self.outstandingPing = False
REQUEST_CAPABILITIES = {'account-notify', 'extended-join', 'multi-prefix',
'metadata-notify', 'account-tag'}
def _queueConnectMessages(self):
if self.zombie:
self.driver.die()
@ -987,14 +992,7 @@ class Irc(IrcCommandDispatcher):
# in telling between ACK and NAK
# * using sendMsg instead of queueMsg because these messages cannot
# be throttled.
for cap in ('account-notify', 'extended-join', 'multi-prefix',
'metadata-notify', 'account-tag'):
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', cap)))
if self.sasl:
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', 'sasl')))
else:
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('END',)))
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('LS', '302')))
def doAuthenticate(self, msg):
if len(msg.args) == 1 and msg.args[0] == '+':
@ -1026,10 +1024,17 @@ class Irc(IrcCommandDispatcher):
self.sendMsg(ircmsgs.IrcMsg(command='AUTHENTICATE', args=(authstring,)))
def doCap(self, msg):
if msg.args[1] == 'ACK':
self.doCapAck(msg)
elif msg.args[1] == 'NAK':
self.doCapNak(msg)
elif msg.args[1] == 'LS':
self.doCapLs(msg)
def doCapAck(self, msg):
if len(msg.args) != 3:
log.warning('Bad CAP ACK from server: %r', msg)
return
for cap in msg.args[2].split(' '):
if msg.args[1] == 'ACK':
log.info('%s: Server acknowledged capability %r',
self.network, cap)
self.state.capabilities_ack.add(cap)
@ -1038,7 +1043,11 @@ class Irc(IrcCommandDispatcher):
self.sendMsg(ircmsgs.IrcMsg(
command='AUTHENTICATE',
args=(self.sasl.upper(),)))
elif msg.args[1] == 'NAK':
def doCapNak(self, msg):
if len(msg.args) != 3:
log.warning('Bad CAP NAK from server: %r', msg)
return
for cap in msg.args[2].split(' '):
self.state.capabilities_nak.add(cap)
log.warning('%s: Server refused capability %r',
self.network, cap)
@ -1047,6 +1056,37 @@ class Irc(IrcCommandDispatcher):
self.sendMsg(ircmsgs.IrcMsg(
command='CAP',
args=('END',)))
def _addCapabilities(self, capstring):
for item in capstring.split():
while item.startswith(('=', '~')):
item = item[1:]
if '=' in item:
(cap, value) = item.split('=', 1)
self.state.capabilities_ls[cap] = value
else:
self.state.capabilities_ls[item] = None
def doCapLs(self, msg):
if len(msg.args) == 4:
# Multi-line LS
if msg.args[2] != '*':
log.warning('Bad CAP LS from server: %r', msg)
return
self._addCapabilities(msg.args[3])
elif len(msg.args) == 3: # End of LS
self._addCapabilities(msg.args[2])
common_supported_capabilities = set(self.state.capabilities_ls) & \
self.REQUEST_CAPABILITIES
for cap in common_supported_capabilities:
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', cap)))
if 'sasl' in self.state.capabilities_ls and self.sasl:
# TODO: use the value of self.state.capabilities_ls['sasl']
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', 'sasl')))
else:
self.sendMsg(ircmsgs.IrcMsg(command='CAP', args=('END',)))
else:
log.warning('Bad CAP LS from server: %r', msg)
return
def monitor(self, targets):
"""Increment a counter of how many callbacks monitor each target;

View File

@ -386,16 +386,26 @@ class IrcTestCase(SupyTestCase):
self.failUnless(m.command == 'USER', 'Expected USER, got %r.' % m)
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.failUnless(m.args == ('LS', '302'), 'Expected CAP LS 302, got %r.' % m)
# TODO
self.irc.feedMsg(ircmsgs.IrcMsg(command='CAP',
args=('*', 'LS', '*', 'account-tag multi-prefix')))
self.irc.feedMsg(ircmsgs.IrcMsg(command='CAP',
args=('*', 'LS', 'extended-join')))
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.assertEqual(m.args, ('REQ', 'account-tag'), m)
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.assertEqual(m.args, ('REQ', 'multi-prefix'), m)
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.assertEqual(m.args, ('REQ', 'extended-join'), m)
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.assertEqual(m.args, ('END',), m)
m = self.irc.takeMsg()
self.failUnless(m.command == 'CAP', 'Expected CAP, got %r.' % m)
self.failUnless(m is None, m)
def testPingResponse(self):
self.irc.feedMsg(ircmsgs.ping('123'))
@ -493,16 +503,11 @@ class IrcCallbackTestCase(SupyTestCase):
expected = [
ircmsgs.nick(nick),
ircmsgs.user('limnoria', user),
ircmsgs.IrcMsg(command='CAP', args=('REQ', 'account-notify')),
ircmsgs.IrcMsg(command='CAP', args=('REQ', 'extended-join')),
ircmsgs.IrcMsg(command='CAP', args=('REQ', 'multi-prefix')),
ircmsgs.IrcMsg(command='CAP', args=('REQ', 'metadata-notify')),
ircmsgs.IrcMsg(command='CAP', args=('REQ', 'account-tag')),
ircmsgs.IrcMsg(command='CAP', args=('END',)),
ircmsgs.IrcMsg(command='CAP', args=('LS', '302')),
]
irc = irclib.Irc('test')
msgs = [irc.takeMsg()]
while msgs[-1] != None:
while msgs[-1] is not None:
msgs.append(irc.takeMsg())
msgs.pop()
self.assertEqual(msgs, expected)
@ -510,7 +515,7 @@ class IrcCallbackTestCase(SupyTestCase):
conf.supybot.networks.test.password.setValue(password)
irc = irclib.Irc('test')
msgs = [irc.takeMsg()]
while msgs[-1] != None:
while msgs[-1] is not None:
msgs.append(irc.takeMsg())
msgs.pop()
expected.insert(0, ircmsgs.password(password))