bitbot-3.11-fork/modules/mumble.py

72 lines
2.3 KiB
Python
Raw Normal View History

2019-09-02 11:47:13 +00:00
import datetime, socket, struct
from src import ModuleManager, utils
DEFAULT_PORT = 64738
def _parse(s):
host, _, port = s.partition(":")
if port:
if not port.isdigit():
return None
else:
2019-09-17 11:23:05 +00:00
port = str(DEFAULT_PORT)
return "%s:%s" % (host, port)
SETTING = utils.FunctionSetting(_parse, "mumble-server",
"Set the mumble server for this channel",
example="example.com:%s" % DEFAULT_PORT)
@utils.export("channelset", SETTING)
@utils.export("serverset", SETTING)
2019-09-02 11:47:13 +00:00
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.mumble")
@utils.kwarg("help", "Get user and bandwidth stats for a mumble server")
@utils.kwarg("usage", "[server[:<port>]]")
2019-09-02 11:47:13 +00:00
def mumble(self, event):
server = None
if not event["args"]:
server = event["target"].get_setting("mumble-server",
event["server"].get_setting("mumble-server", None))
elif event["args"]:
server = event["args_split"][0]
if not server:
raise utils.EventError("Please provide a server")
server, _, port = server.partition(":")
2019-09-02 11:47:13 +00:00
if port:
if not port.isdigit():
raise utils.EventError("Port must be numeric")
port = int(port)
else:
port = DEFAULT_PORT
timestamp = datetime.datetime.utcnow().microsecond
ping_packet = struct.pack(">iQ", 0, timestamp)
2019-09-02 11:47:13 +00:00
s = socket.socket(type=socket.SOCK_DGRAM)
s.settimeout(5)
with utils.deadline():
try:
s.sendto(ping_packet, (server, port))
except socket.gaierror as e:
raise utils.EventError(str(e))
try:
pong_packet = s.recv(24)
except socket.timeout:
raise utils.EventError(
"Timed out waiting for response from %s:%d"
% (server, port))
2019-09-02 11:47:13 +00:00
pong = struct.unpack(">bbbbQiii", pong_packet)
2019-09-02 12:12:24 +00:00
version = ".".join(str(v) for v in pong[1:4])
ping = (datetime.datetime.utcnow().microsecond-timestamp)/1000
2019-09-02 11:47:13 +00:00
users = pong[5]
max_users = pong[6]
bandwidth = pong[7]/1000 # kbit/s
2019-09-02 12:12:24 +00:00
event["stdout"].write(
"%s:%d (v%s): %d/%d users, %.1fms ping, %dkbit/s bandwidth"
% (server, port, version, users, max_users, ping, bandwidth))