134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
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"]:
|
|
print("alias")
|
|
alias = event["args"]
|
|
id = self.bot.database.servers.get_by_alias(alias)
|
|
if id == None:
|
|
raise utils.EventError("Unknown server alias")
|
|
print(id)
|
|
server = self.bot.get_server_by_id(id)
|
|
server.disconnect()
|
|
self.bot.disconnect(server)
|
|
|
|
@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)
|