bitbot-3.11-fork/modules/admin.py

176 lines
5.9 KiB
Python
Raw Normal View History

#--depends-on commands
#--depends-on permissions
from src import ModuleManager, utils
2016-05-17 15:14:23 +00:00
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.changenickname", min_args=1)
2016-05-17 15:14:23 +00:00
def change_nickname(self, event):
"""
:help: Change my nickname
:usage: <nickname>
:permission: changenickname
"""
2016-05-17 15:14:23 +00:00
nickname = event["args_split"][0]
event["server"].send_nick(nickname)
@utils.hook("received.command.raw", min_args=1)
2016-05-17 15:14:23 +00:00
def raw(self, event):
"""
:help: Send a line of raw IRC data
:usage: <raw line>
:permission: raw
"""
event["server"].send_raw(event["args"])
2016-06-24 08:17:27 +00:00
@utils.hook("received.command.part")
2016-06-24 08:17:27 +00:00
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)
2018-09-03 17:21:13 +00:00
@utils.hook("received.command.reconnect")
2018-09-03 17:21:13 +00:00
def reconnect(self, event):
"""
:help: Reconnect to the current network
:permission: reconnect
"""
line = event["server"].send_quit("Reconnecting")
line.events.on("send").hook(lambda e: self.bot.reconnect(
event["server"].id, event["server"].connection_params))
2018-10-03 14:59:36 +00:00
@utils.hook("received.command.connect", min_args=1)
def connect(self, event):
"""
:help: Connect to a network
:usage: <server id>
2018-11-20 14:27:24 +00:00
:permission: connect
2018-10-03 14:59:36 +00:00
"""
alias = event["args"]
id = self.bot.database.servers.get_by_alias(alias)
if id == None:
raise utils.EventError("Unknown server alias")
2018-10-03 14:59:36 +00:00
existing_server = self.bot.get_server_by_id(id)
2018-10-03 14:59:36 +00:00
if existing_server:
raise utils.EventError("Already connected to %s" % str(
2018-10-03 14:59:36 +00:00
existing_server))
2018-10-03 14:59:36 +00:00
server = self.bot.add_server(id)
event["stdout"].write("Connecting to %s" % str(server))
2018-11-20 14:27:24 +00:00
@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)
2018-11-20 14:27:24 +00:00
server.disconnect()
self.bot.disconnect(server)
event["stdout"].write("Disconnected from %s" % str(server))
2019-02-10 23:53:33 +00:00
@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():
2019-02-10 23:53:33 +00:00
line = server.send_quit(reason)
line.events.on("send").hook(self._shutdown_hook(server))
2019-02-10 23:53:33 +00:00
def _shutdown_hook(self, server):
def shutdown(e):
server.disconnect()
self.bot.disconnect(server)
return shutdown
2019-05-21 09:16:20 +00:00
@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("+")
2019-05-21 09:16:20 +00:00
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
2019-05-21 09:45:15 +00:00
username = hostmask.username or nickname
2019-05-21 09:16:20 +00:00
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)
2019-07-04 12:15:51 +00:00
@utils.hook("received.command.editserver")
@utils.kwarg("min_args", 3)
@utils.kwarg("help", "Edit server details")
@utils.kwarg("usage", "<alias> <option> <value>")
@utils.kwarg("permission", "editserver")
def edit_server(self, event):
alias = event["args_split"][0]
server = self.bot.get_server_by_alias(alias)
if server == None:
raise utils.EventError("Unknown server '%s'" % alias)
option = event["args_split"][1].lower()
value = " ".join(event["args_split"][2:])
value_parsed = None
if option == "hostname":
value_parsed = value
elif option == "port":
if not value.isdigit():
raise utils.EventError("Invalid port")
value_parsed = int(value.lstrip("0"))
elif option == "tls":
value_lower = value.lower()
if not value_lower in ["yes", "no"]:
raise utils.EventError("TLS should be either 'yes' or 'no'")
value_parsed = value_lower == "yes"
elif option == "password":
value_parsed = value
elif option == "bindhost":
value_parsed = value
elif option in ["nickname", "username"]:
value_parsed = value
else:
raise utils.EventError("Unknown option '%s'" % option)
self.bot.database.servers.edit(server.id, option, value_parsed)
event["stdout"].write("Set %s for %s" % (option, alias))