Implement account-notify, extended-join and WHOX

Also integrate it with NickAuth.

Closes #778.
This commit is contained in:
nyuszika7h 2014-08-02 12:30:24 +02:00
parent 0810bcadc7
commit 871607614a
3 changed files with 111 additions and 32 deletions

View File

@ -167,6 +167,66 @@ class NickAuth(callbacks.Plugin):
else: else:
irc.error(_('No user has this nick on this network.')) irc.error(_('No user has this nick on this network.'))
def doAccount(self, irc, msg):
account = msg.args[0]
user = ircdb.users.getUserFromNick(irc.network, account)
if not user:
try:
user = ircdb.users.getUser(msg.prefix)
except KeyError:
user = None
if user:
if account == '*':
user.clearAuth()
else:
user.addAuth(msg.prefix)
ircdb.users.setUser(user, flush=False)
def doJoin(self, irc, msg):
# extended-join is not supported
if len(msg.args) < 2:
return
account = msg.args[1]
user = ircdb.users.getUserFromNick(irc.network, account)
if not user:
try:
user = ircdb.users.getUser(msg.prefix)
except KeyError:
user = None
if user:
if account == '*':
user.clearAuth()
else:
user.addAuth(msg.prefix)
ircdb.users.setUser(user, flush=False)
def do354(self, irc, msg):
(_, ident, host, nick, account) = msg.args
prefix = '%s!%s@%s' % (nick, ident, host)
user = ircdb.users.getUserFromNick(irc.network, account)
if not user:
try:
user = ircdb.users.getUser(prefix)
except KeyError:
user = None
if user:
#if account == '0':
# user.clearAuth()
#else:
# user.addAuth(prefix)
# ircdb.users.setUser(user, flush=False)
if account != '0':
user.addAuth(prefix)
ircdb.users.setUser(user, flush=False)
Class = NickAuth Class = NickAuth

View File

@ -481,9 +481,10 @@ class IrcState(IrcCommandDispatcher):
else: else:
self.supported[arg] = None self.supported[arg] = None
def do352(self, irc, msg): def do354(self, irc, msg):
# WHO reply. # WHO reply.
(nick, user, host) = (msg.args[5], msg.args[2], msg.args[3])
(_, user, host, nick, __) = msg.args
hostmask = '%s!%s@%s' % (nick, user, host) hostmask = '%s!%s@%s' % (nick, user, host)
self.nicksToHostmasks[nick] = hostmask self.nicksToHostmasks[nick] = hostmask
@ -927,24 +928,38 @@ class Irc(IrcCommandDispatcher):
if self.zombie: if self.zombie:
self.driver.die() self.driver.die()
self._reallyDie() self._reallyDie()
else:
return
caps = ['account-notify', 'extended-join']
if self.sasl_password: if self.sasl_password:
if not self.sasl_username: if not self.sasl_username:
log.warning('%s: SASL username is not set, unable to ' log.warning('%s: SASL username is not set, unable to '
'identify.', self.network) 'identify.', self.network)
else: else:
log.debug("%s: Requesting capability 'sasl'.", caps.append('sasl')
self.network)
self.queueMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', 'sasl'))) log.debug('%s: Requesting capabilities: %s',
self.network, ' '.join(caps))
for cap in caps:
self.queueMsg(ircmsgs.IrcMsg(command='CAP', args=('REQ', cap)))
if self.password: if self.password:
log.info('%s: Queuing PASS command, not logging the password.', log.info('%s: Queuing PASS command, not logging the password.',
self.network) self.network)
self.queueMsg(ircmsgs.password(self.password)) self.queueMsg(ircmsgs.password(self.password))
log.debug('%s: Queuing NICK command, nick is %s.', log.debug('%s: Queuing NICK command, nick is %s.',
self.network, self.nick) self.network, self.nick)
self.queueMsg(ircmsgs.nick(self.nick)) self.queueMsg(ircmsgs.nick(self.nick))
log.debug('%s: Queuing USER command, ident is %s, user is %s.', log.debug('%s: Queuing USER command, ident is %s, user is %s.',
self.network, self.ident, self.user) self.network, self.ident, self.user)
self.queueMsg(ircmsgs.user(self.ident, self.user)) self.queueMsg(ircmsgs.user(self.ident, self.user))
def doAuthenticate(self, msg): def doAuthenticate(self, msg):
@ -960,15 +975,19 @@ class Irc(IrcCommandDispatcher):
self.queueMsg(ircmsgs.IrcMsg(command='AUTHENTICATE', args=(authstring,))) self.queueMsg(ircmsgs.IrcMsg(command='AUTHENTICATE', args=(authstring,)))
def doCap(self, msg): def doCap(self, msg):
if msg.args[2] == 'sasl': for cap in msg.args[2].split(' '):
if msg.args[1] == 'ACK': if msg.args[1] == 'ACK':
log.debug("%s: Server acknowledged 'sasl' capability", log.info('%s: Server acknowledged %r capability',
self.network) self.network, cap)
if cap == 'sasl':
self.queueMsg(ircmsgs.IrcMsg(command='AUTHENTICATE', self.queueMsg(ircmsgs.IrcMsg(command='AUTHENTICATE',
args=('PLAIN',))) args=('PLAIN',)))
elif msg.args[1] == 'NAK': elif msg.args[1] == 'NAK':
log.warning("%s: Server refused 'sasl' capability", log.warning('%s: Server refused %r capability',
self.network) self.network, cap)
if cap == 'sasl':
self.queueMsg(ircmsgs.IrcMsg(command='CAP', args=('END',))) self.queueMsg(ircmsgs.IrcMsg(command='CAP', args=('END',)))
def do903(self, msg): def do903(self, msg):
@ -1068,7 +1087,7 @@ class Irc(IrcCommandDispatcher):
def doJoin(self, msg): def doJoin(self, msg):
if msg.nick == self.nick: if msg.nick == self.nick:
channel = msg.args[0] channel = msg.args[0]
self.queueMsg(ircmsgs.who(channel)) # Ends with 315. self.queueMsg(ircmsgs.who(channel, args=('%uhna',))) # Ends with 315.
self.queueMsg(ircmsgs.mode(channel)) # Ends with 329. self.queueMsg(ircmsgs.mode(channel)) # Ends with 329.
for channel in msg.args[0].split(','): for channel in msg.args[0].split(','):
self.queueMsg(ircmsgs.mode(channel, '+b')) self.queueMsg(ircmsgs.mode(channel, '+b'))

View File

@ -736,7 +736,7 @@ def user(ident, user, prefix='', msg=None):
return IrcMsg(prefix=prefix, command='USER', return IrcMsg(prefix=prefix, command='USER',
args=(ident, '0', '*', user), msg=msg) args=(ident, '0', '*', user), msg=msg)
def who(hostmaskOrChannel, prefix='', msg=None): def who(hostmaskOrChannel, prefix='', msg=None, args=()):
"""Returns a WHO for the hostmask or channel hostmaskOrChannel.""" """Returns a WHO for the hostmask or channel hostmaskOrChannel."""
if conf.supybot.protocols.irc.strictRfc(): if conf.supybot.protocols.irc.strictRfc():
assert isChannel(hostmaskOrChannel) or \ assert isChannel(hostmaskOrChannel) or \
@ -744,7 +744,7 @@ def who(hostmaskOrChannel, prefix='', msg=None):
if msg and not prefix: if msg and not prefix:
prefix = msg.prefix prefix = msg.prefix
return IrcMsg(prefix=prefix, command='WHO', return IrcMsg(prefix=prefix, command='WHO',
args=(hostmaskOrChannel,), msg=msg) args=(hostmaskOrChannel,) + args, msg=msg)
def _whois(COMMAND, nick, mask='', prefix='', msg=None): def _whois(COMMAND, nick, mask='', prefix='', msg=None):
"""Returns a WHOIS for nick.""" """Returns a WHOIS for nick."""