GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2025-02-02 23:54:08 +01:00
Code Issues Projects Releases Wiki Activity

bots: allow specifying channel prefixes (e.g. @+) in 'join'

Browse Source 
This functionality should really be merged with what ServiceBot does, but whatever...
This commit is contained in:
James Lu 2017-02-13 16:59:54 -08:00
parent 9fac7cb1f3
commit 902b246f96
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
plugins/bots.py
View File

@ -48,9 +48,14 @@ def quit(irc, source, args):
irc.callHooks([u, 'PYLINK_BOTSPLUGIN_QUIT', {'text': quitmsg, 'parse_as': 'QUIT'}]) irc.callHooks([u, 'PYLINK_BOTSPLUGIN_QUIT', {'text': quitmsg, 'parse_as': 'QUIT'}])
def joinclient(irc, source, args): def joinclient(irc, source, args):
"""[<target>] <channel1>,[<channel2>], etc. """[<target>] <channel1>[,<channel2>,<channel3>,...]
Admin-only. Joins <target>, the nick of a PyLink client, to a comma-separated list of channels. If <target> is not given, it defaults to the main PyLink client.""" Admin-only. Joins <target>, the nick of a PyLink client, to a comma-separated list of channels.
If <target> is not given, it defaults to the main PyLink client.
For the channel arguments, prefixes can also be specified to join the given client with
(e.g. @#channel will join the client with op, while ~@#channel will join it with +qo.
"""
permissions.checkPermissions(irc, source, ['bots.joinclient']) permissions.checkPermissions(irc, source, ['bots.joinclient'])
try: try:
@ -79,15 +84,26 @@ def joinclient(irc, source, args):
irc.error("Cannot force join a protected PyLink services client.") irc.error("Cannot force join a protected PyLink services client.")
return return
prefix_to_mode = {v: k for k, v in irc.prefixmodes.items()}
for channel in clist: for channel in clist:
if not utils.isChannel(channel): real_channel = channel.lstrip(''.join(prefix_to_mode))
irc.error("Invalid channel name %r." % channel) # XXX we need a better way to do this, but only the other option I can think of is regex...
prefixes = channel[:len(channel)-len(real_channel)]
joinmodes = ''.join(prefix_to_mode[prefix] for prefix in prefixes)
if not utils.isChannel(real_channel):
irc.error("Invalid channel name %r." % real_channel)
return return
irc.proto.join(u, channel)
# join() doesn't support prefixes.
if prefixes:
irc.proto.sjoin(irc.sid, real_channel, [(joinmodes, u)])
else:
irc.proto.join(u, real_channel)
# Call a join hook manually so other plugins like relay can understand it. # Call a join hook manually so other plugins like relay can understand it.
irc.callHooks([u, 'PYLINK_BOTSPLUGIN_JOIN', {'channel': channel, 'users': [u], irc.callHooks([u, 'PYLINK_BOTSPLUGIN_JOIN', {'channel': real_channel, 'users': [u],
'modes': irc.channels[channel].modes, 'modes': irc.channels[real_channel].modes,
'parse_as': 'JOIN'}]) 'parse_as': 'JOIN'}])
utils.add_cmd(joinclient, name='join') utils.add_cmd(joinclient, name='join')