99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
"""Channel management: kick, ban, unban, topic, mode, invite-join."""
|
|
|
|
import logging
|
|
|
|
from derp.plugin import command, event
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def _require_channel(message):
|
|
"""Return True if the message originated in a channel."""
|
|
return message.is_channel
|
|
|
|
|
|
@command("kick", help="Kick a user: !kick <nick> [reason]", admin=True)
|
|
async def cmd_kick(bot, message):
|
|
"""Kick a user from the current channel."""
|
|
if not _require_channel(message):
|
|
await bot.reply(message, "Must be used in a channel")
|
|
return
|
|
parts = message.text.split(None, 2)
|
|
if len(parts) < 2:
|
|
await bot.reply(message, "Usage: !kick <nick> [reason]")
|
|
return
|
|
nick = parts[1]
|
|
reason = parts[2] if len(parts) > 2 else ""
|
|
await bot.kick(message.target, nick, reason)
|
|
|
|
|
|
@command("ban", help="Ban a hostmask: !ban <mask>", admin=True)
|
|
async def cmd_ban(bot, message):
|
|
"""Set +b on a hostmask in the current channel."""
|
|
if not _require_channel(message):
|
|
await bot.reply(message, "Must be used in a channel")
|
|
return
|
|
parts = message.text.split(None, 1)
|
|
if len(parts) < 2:
|
|
await bot.reply(message, "Usage: !ban <mask>")
|
|
return
|
|
mask = parts[1]
|
|
await bot.mode(message.target, "+b", mask)
|
|
|
|
|
|
@command("unban", help="Remove a ban: !unban <mask>", admin=True)
|
|
async def cmd_unban(bot, message):
|
|
"""Remove +b from a hostmask in the current channel."""
|
|
if not _require_channel(message):
|
|
await bot.reply(message, "Must be used in a channel")
|
|
return
|
|
parts = message.text.split(None, 1)
|
|
if len(parts) < 2:
|
|
await bot.reply(message, "Usage: !unban <mask>")
|
|
return
|
|
mask = parts[1]
|
|
await bot.mode(message.target, "-b", mask)
|
|
|
|
|
|
@command("topic", help="Set or query channel topic: !topic [text]", admin=True)
|
|
async def cmd_topic(bot, message):
|
|
"""Set or query the channel topic."""
|
|
if not _require_channel(message):
|
|
await bot.reply(message, "Must be used in a channel")
|
|
return
|
|
parts = message.text.split(None, 1)
|
|
if len(parts) < 2:
|
|
# Query current topic
|
|
from derp.irc import format_msg
|
|
await bot.conn.send(format_msg("TOPIC", message.target))
|
|
else:
|
|
await bot.set_topic(message.target, parts[1])
|
|
|
|
|
|
@command("mode", help="Set channel mode: !mode <mode> [args]", admin=True)
|
|
async def cmd_mode(bot, message):
|
|
"""Set a mode on the current channel."""
|
|
if not _require_channel(message):
|
|
await bot.reply(message, "Must be used in a channel")
|
|
return
|
|
parts = message.text.split(None)
|
|
if len(parts) < 2:
|
|
await bot.reply(message, "Usage: !mode <mode> [args]")
|
|
return
|
|
mode_str = parts[1]
|
|
args = parts[2:]
|
|
await bot.mode(message.target, mode_str, *args)
|
|
|
|
|
|
@event("INVITE")
|
|
async def on_invite(bot, message):
|
|
"""Join a channel when invited by an admin or IRC operator."""
|
|
if not bot._is_admin(message):
|
|
log.info("ignoring invite from non-admin %s", message.nick)
|
|
return
|
|
channel = message.params[1] if len(message.params) > 1 else None
|
|
if not channel:
|
|
return
|
|
log.info("accepting invite to %s from %s", channel, message.nick)
|
|
await bot.join(channel)
|