bitbot-3.11-fork/modules/admin.py
2018-11-20 14:27:24 +00:00

89 lines
2.8 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(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
"""
event["server"].send_quit("Reconnecting")
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
"""
id = event["args_split"][0]
if not id.isdigit():
raise utils.EventError("Please provide a numeric server ID")
id = int(id)
if not self.bot.database.servers.get(id):
raise utils.EventError("Unknown server ID")
existing_server = self.bot.get_server(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"]:
id = event["args_split"][0]
if not id.isdigit():
raise utils.EventError("Please provide a numeric server ID")
id = int(id)
if not self.bot.database.servers.get(id):
raise utils.EventError("Unknown server ID")
server = self.bot.get_server(id)
server.disconnect()
self.bot.disconnect(server)