102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
import sys
|
|
import yaboli
|
|
from yaboli.utils import *
|
|
|
|
|
|
|
|
class InfoBot(yaboli.Bot):
|
|
"""
|
|
Display information about the clients connected to a room in its nick.
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__("()")
|
|
|
|
self.add_help("count", (
|
|
"This bot counts the number of clients connected to a room. If you"
|
|
" open a room in two different tabs, the bot counts you twice.\n"
|
|
"The euphoria client, on the other hand, usually displays all"
|
|
" connections of an account as one nick in the nick list. Because of"
|
|
" that, this bot's count is always as high as, or higher than, the"
|
|
" number of nicks on the nick list, similar to the number on the"
|
|
" button to toggle the nick list.\n\n"
|
|
"If the bot's count is off, try a !recount."
|
|
))
|
|
|
|
self.add_help("lurkers", (
|
|
"People or bots who are connected to the room but haven't chosen a"
|
|
" nick are lurkers. The euphoria client doesn't display them in the"
|
|
" nick list.\n"
|
|
"This bot differentiates between people (L) and bots (N) who are"
|
|
" lurking."
|
|
))
|
|
|
|
self.add_help("changelog", (
|
|
"- add !recount command\n"
|
|
"- fix bot counting incorrectly\n"
|
|
))
|
|
|
|
self.help_specific = (
|
|
"Displays information about the clients in a room in its nick:\n"
|
|
"(<people>P <bots>B <lurkers>L <bot-lurkers>N)\n\n"
|
|
"!recount @{nick} - Recount people in the room\n\n"
|
|
"Created by @Garmy using yaboli.\n"
|
|
"For additional info, try \"!help @{nick} <topic>\". Topics:\n"
|
|
)
|
|
self.help_specific += self.list_help_topics()
|
|
|
|
self.register_command("recount", self.command_recount, specific=False)
|
|
|
|
async def update_nick(self):
|
|
p = len(self.room.listing.get(types=["account", "agent"], lurker=False))
|
|
b = 1 + len(self.room.listing.get(types=["bot"], lurker=False))
|
|
l = len(self.room.listing.get(types=["account", "agent"], lurker=True))
|
|
n = len(self.room.listing.get(types=["bot"], lurker=True))
|
|
|
|
name = []
|
|
if p > 0: name.append(f"{p}P")
|
|
if b > 0: name.append(f"{b}B")
|
|
if l > 0: name.append(f"{l}L")
|
|
if n > 0: name.append(f"{n}N")
|
|
name = "\u0001(" + " ".join(name) + ")"
|
|
|
|
await self.set_nick(name)
|
|
|
|
async def on_join(self, session):
|
|
await self.update_nick()
|
|
await self.room.who()
|
|
await self.update_nick()
|
|
|
|
async def on_part(self, session):
|
|
await self.update_nick()
|
|
await self.room.who()
|
|
await self.update_nick()
|
|
|
|
async def on_nick(self, session_id, user_id, from_nick, to_nick):
|
|
await self.update_nick()
|
|
await self.room.who()
|
|
await self.update_nick()
|
|
|
|
async def on_snapshot(self, user_id, session_id, version, sessions,
|
|
messages, nick=None, pm_with_nick=None,
|
|
pm_with_user_id=None):
|
|
# Not needed because we're updating the nick anyways.
|
|
#super().on_snapshot(user_id, session_id, version, sessions, messages,
|
|
# nick, pm_with_nick, pm_with_user_id)
|
|
await self.update_nick()
|
|
|
|
async def command_recount(self, message, argstr):
|
|
await self.room.who()
|
|
await self.update_nick()
|
|
await self.room.send("Recalibrated.", message.mid)
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("USAGE:")
|
|
print(f" {sys.argv[0]} <room>")
|
|
return
|
|
|
|
run_bot(InfoBot, sys.argv[1])
|
|
|
|
if __name__ == "__main__":
|
|
main()
|