bitbot-3.11-fork/modules/hostmask_tracking.py
2020-01-11 22:30:43 +00:00

40 lines
1.6 KiB
Python

from src import ModuleManager, utils
class Module(ModuleManager.BaseModule):
_name = "Hostmasks"
@utils.hook("new.user")
def new_user(self, event):
userhost = event["user"].userhost()
if not userhost == None:
known_hostmasks = event["user"].get_setting("known-hostmasks", [])
if not userhost in known_hostmasks:
known_hostmasks.append(userhost)
event["user"].set_setting("known-hostmasks", known_hostmasks)
@utils.hook("received.command.maskfind")
@utils.kwarg("min_args", 1)
@utils.kwarg("help", "Find all nicknames that used a given hostmask")
@utils.kwarg("usage", "<hostmask>")
@utils.kwarg("permission", "maskfind")
def maskfind(self, event):
all_userhosts = event["server"].get_all_user_settings("known-hostmasks")
nicknames = set([])
hostmask_str = event["args_split"][0]
hostmask = utils.irc.hostmask_parse(hostmask_str)
searched = 0
for nickname, userhosts in all_userhosts:
searched += len(userhosts)
for userhost in userhosts:
if hostmask.match(userhost):
nicknames.add((nickname, userhost))
if nicknames:
outs = []
for nickname, userhost in sorted(nicknames):
outs.append("%s (%s)" % (utils.irc.bold(nickname), userhost))
event["stdout"].write("%s (%d/%d): %s" %
(hostmask_str, len(nicknames), searched, ", ".join(outs)))
else:
event["stderr"].write("Hostmask not found")