bitbot-3.11-fork/modules/line_handler/user.py

102 lines
3.6 KiB
Python

from src import utils
def handle_311(event):
nickname = event["line"].args[1]
username = event["line"].args[2]
hostname = event["line"].args[3]
realname = event["line"].args[4]
if event["server"].is_own_nickname(nickname):
event["server"].username = username
event["server"].hostname = hostname
event["server"].realname = realname
target = event["server"].get_user(nickname)
target.username = username
target.hostname = hostname
target.realname = realname
def quit(events, event):
nickname = None
if event["direction"] == utils.Direction.Recv:
nickname = event["line"].source.nickname
reason = event["line"].args.get(0)
if event["direction"] == utils.Direction.Recv:
nickname = event["line"].source.nickname
if (not event["server"].is_own_nickname(nickname) and
not event["line"].source.hostmask == "*"):
user = event["server"].get_user(nickname)
events.on("received.quit").call(reason=reason, user=user,
server=event["server"])
event["server"].remove_user(user)
else:
event["server"].disconnect()
else:
events.on("send.quit").call(reason=reason, server=event["server"])
def nick(events, event):
new_nickname = event["line"].args.get(0)
user = event["server"].get_user(event["line"].source.nickname)
old_nickname = user.nickname
user.set_nickname(new_nickname)
event["server"].change_user_nickname(old_nickname, new_nickname)
if not event["server"].is_own_nickname(event["line"].source.nickname):
events.on("received.nick").call(new_nickname=new_nickname,
old_nickname=old_nickname, user=user, server=event["server"])
else:
events.on("self.nick").call(server=event["server"],
new_nickname=new_nickname, old_nickname=old_nickname)
event["server"].set_own_nickname(new_nickname)
def away(events, event):
user = event["server"].get_user(event["line"].source.nickname)
message = event["line"].args.get(0)
if message:
user.away = True
user.away_message = message
events.on("received.away.on").call(user=user, server=event["server"],
message=message)
else:
user.away = False
user.away_message = None
events.on("received.away.off").call(user=user, server=event["server"])
def chghost(events, event):
nickname = event["line"].source.nickname
username = event["line"].args[0]
hostname = event["line"].args[1]
if event["server"].is_own_nickname(nickname):
event["server"].username = username
event["server"].hostname = hostname
target = event["server"].get_user(nickname)
events.on("received.chghost").call(user=target, server=event["server"],
username=username, hostname=hostname)
target.username = username
target.hostname = hostname
def setname(event):
nickname = event["line"].source.nickname
realname = event["line"].args[0]
user = event["server"].get_user(nickname)
user.realname = realname
if event["server"].is_own_nickname(nickname):
event["server"].realname = realname
def account(events, event):
user = event["server"].get_user(event["line"].source.nickname)
if not event["line"].args[0] == "*":
user.account = event["line"].args[0]
events.on("received.account.login").call(user=user,
server=event["server"], account=event["line"].args[0])
else:
user.account = None
events.on("received.account.logout").call(user=user,
server=event["server"])