bitbot-3.11-fork/modules/admin.py

136 lines
4.5 KiB
Python

#--depends-on commands
#--depends-on permissions
from src import ModuleManager, utils
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.changenickname", min_args=1)
def change_nickname(self, event):
"""
:help: Change my nickname
:usage: <nickname>
:permission: changenickname
"""
nickname = event["args_split"][0]
event["server"].send_nick(nickname)
@utils.hook("received.command.raw", min_args=1)
def raw(self, event):
"""
:help: Send a line of raw IRC data
:usage: <raw line>
:permission: raw
"""
event["server"].send_raw(event["args"])
@utils.hook("received.command.part")
def part(self, event):
"""
:help: Part from the current or given channel
:usage: [channel]
:permission: part
"""
if event["args"]:
target = event["args_split"][0]
elif event["is_channel"]:
target = event["target"].name
else:
event["stderr"].write("No channel provided")
event["server"].send_part(target)
@utils.hook("received.command.reconnect")
def reconnect(self, event):
"""
:help: Reconnect to the current network
:permission: reconnect
"""
line = event["server"].send_quit("Reconnecting")
line.on_send(lambda: self.bot.reconnect(
event["server"].id, event["server"].connection_params))
@utils.hook("received.command.connect", min_args=1)
def connect(self, event):
"""
:help: Connect to a network
:usage: <server id>
:permission: connect
"""
alias = event["args"]
id = self.bot.database.servers.get_by_alias(alias)
if id == None:
raise utils.EventError("Unknown server alias")
existing_server = self.bot.get_server_by_id(id)
if existing_server:
raise utils.EventError("Already connected to %s" % str(
existing_server))
server = self.bot.add_server(id)
event["stdout"].write("Connecting to %s" % str(server))
@utils.hook("received.command.disconnect")
def disconnect(self, event):
"""
:help: Disconnect from a server
:usage: [server id]
:permission: disconnect
"""
id = event["server"].id
if event["args"]:
alias = event["args"]
id = self.bot.database.servers.get_by_alias(alias)
if id == None:
raise utils.EventError("Unknown server alias")
server = self.bot.get_server_by_id(id)
server.disconnect()
self.bot.disconnect(server)
event["stdout"].write("Disconnected from %s" % alias)
@utils.hook("received.command.shutdown")
def shutdown(self, event):
"""
:help: Shutdown bot
:usage: [reason]
:permission: shutdown
"""
reason = event["args"] or ""
for server in self.bot.servers.values():
line = server.send_quit(reason)
line.on_send(self._shutdown_hook(server))
def _shutdown_hook(self, server):
def shutdown():
server.disconnect()
self.bot.disconnect(server)
return shutdown
@utils.hook("received.command.addserver", min_args=3)
def add_server(self, event):
"""
:help: Add a new server
:usage: <alias> <hostname>:[+]<port> <nickname>!<username>[@<bindhost>]
:permission: addserver
"""
alias = event["args_split"][0]
hostname, sep, port = event["args_split"][1].partition(":")
tls = port.startswith("+")
port = port.lstrip("+")
if not hostname or not port or not port.isdigit():
raise utils.EventError("Please provide <hostname>:[+]<port>")
port = int(port)
hostmask = utils.irc.seperate_hostmask(event["args_split"][2])
nickname = hostmask.nickname
username = hostmask.username or nickname
realname = nickname
bindhost = hostmask.hostname or None
try:
server_id = self.bot.database.servers.add(alias, hostname, port, "",
tls, bindhost, nickname, username, realname)
except Exception as e:
event["stderr"].write("Failed to add server")
self.log.error("failed to add server \"%s\"", [alias],
exc_info=True)
return
event["stdout"].write("Added server '%s'" % alias)