Merge d4c0121c3e into 9f5eee1420

stdplugins/nicks.py

@@ -0,0 +1,87 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+import regex
+
+from telethon import events, utils
+
+nicks = storage.nicks or {}
+
+
+@borg.on(events.NewMessage(pattern=r'.nick( (?P<target>\S+)|$)', outgoing=True))
+@borg.on(events.NewMessage(pattern=r'.who( (?P<target>\S+)|$)', outgoing=True))
+async def on_check_nick(event):
+    # Space weirdness in regex required because argument is optional and other
+    # commands start with ".nick"
+    try:
+        target = await get_target_id(event)
+    except ValueError:
+        target = None
+    if target not in nicks:
+        await event.respond("no nick")
+    else:
+        await event.respond(f"nick: {nicks[target]}")
+
+    await event.delete()
+
+
+async def get_target_id(event):
+    """
+    Get the string target id, looking in the 'target' match group, the forward
+    from user, and the sender of the message.
+    """
+    target = event.pattern_match.group("target")
+    r_msg = await event.reply_message
+    # If they provided an explicit target, use that over the reply user
+    if target:
+        if regex.match(r'[^+]-?\d+', target):
+            target = int(target)
+
+        # Might raise ValueError if no such entity
+        target = utils.get_peer_id(await borg.get_entity(target))
+    elif r_msg:
+        # If it's a forward do the forwarded from, otherwise do the sender
+        if r_msg.fwd_from:
+            target = r_msg.fwd_from.from_id
+        else:
+            target = r_msg.from_id
+    else:
+        # Didn't specify a target and didn't reply to anything
+        raise ValueError
+
+    # JSON only does string keys lololol
+    return str(target)
+
+
+@borg.on(events.NewMessage(pattern=r'.nicks (-u (?P<target>\S+) )?(?P<nick>.+)', outgoing=True))
+async def on_nick_save(event):
+    nick = event.pattern_match.group("nick")
+    try:
+        target = await get_target_id(event)
+    except ValueError:
+        await event.reply("who?")
+        return
+    nicks[target] = nick
+    storage.nicks = nicks
+    await event.respond(f'set nick "{nick}" for user {target}')
+    await event.delete()
+
+
+@borg.on(events.NewMessage(pattern=r'.nickl', outgoing=True))
+async def on_nick_list(event):
+    await event.respond('set nicks: \n' + '\n'.join(
+        (f'{k}: "{v}"' for k, v in nicks.items())))
+    await event.delete()
+
+
+@borg.on(events.NewMessage(pattern=r'.nickd( (?P<target>\S+))?', outgoing=True))
+async def on_nick_delete(event):
+    # Space weirdness in regex is because argument is optional
+    try:
+        nick = nicks.pop(await get_target_id(event), None)
+    except ValueError:
+        await event.reply("who?")
+        return
+    storage.nicks = nicks
+    await event.respond(f'deleted nick "{nick}"')
+    await event.delete()