info.py

# 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/.
"""
Show all .info about the replied message
"""
from uniborg.util import parse_pre

from telethon import events
from telethon.tl.functions.channels import GetParticipantRequest
from telethon.tl.types import InputPeerChannel, User, DocumentAttributeFilename
from telethon.tl.tlobject import TLObject
from telethon.errors.rpcerrorlist import UserNotParticipantError, MessageTooLongError
import datetime


def yaml_format(obj, indent=0, max_str_len=256, max_byte_len=64):
    """
    Pretty formats the given object as a YAML string which is returned.
    (based on TLObject.pretty_format)
    """
    result = []
    if isinstance(obj, TLObject):
        obj = obj.to_dict()

    if isinstance(obj, dict):
        if not obj:
            return 'dict:'
        items = obj.items()
        has_items = len(items) > 1
        has_multiple_items = len(items) > 2
        result.append(obj.get('_', 'dict') + (':' if has_items else ''))
        if has_multiple_items:
            result.append('\n')
            indent += 2
        for k, v in items:
            if k == '_' or v is None:
                continue
            formatted = yaml_format(v, indent, max_str_len, max_byte_len)
            if not formatted.strip():
                continue
            result.append(' ' * (indent if has_multiple_items else 1))
            result.append(f'{k}:')
            if not formatted[0].isspace():
                result.append(' ')
            result.append(f'{formatted}')
            result.append('\n')
        if has_items:
            result.pop()
        if has_multiple_items:
            indent -= 2
    elif isinstance(obj, str):
        # truncate long strings and display elipsis
        result = repr(obj[:max_str_len])
        if len(obj) > max_str_len:
            result += '…'
        return result
    elif isinstance(obj, bytes):
        # repr() bytes if it's printable, hex like "FF EE BB" otherwise
        if all(0x20 <= c < 0x7f for c in obj):
            return repr(obj)
        else:
            return ('<…>' if len(obj) > max_byte_len else
                    ' '.join(f'{b:02X}' for b in obj))
    elif isinstance(obj, datetime.datetime):
        # ISO-8601 without timezone offset (telethon dates are always UTC)
        return obj.strftime('%Y-%m-%d %H:%M:%S')
    elif hasattr(obj, '__iter__'):
        # display iterables one after another at the base indentation level
        result.append('\n')
        indent += 2
        for x in obj:
            result.append(f"{' ' * indent}- {yaml_format(x, indent + 2, max_str_len, max_byte_len)}")
            result.append('\n')
        result.pop()
        indent -= 2
    else:
        return repr(obj)

    return ''.join(result)


@borg.on(borg.cmd("info"))
async def _(event):
    if not event.message.is_reply:
        await who(event)
        return
    msg = await event.message.get_reply_message()
    yaml_text = yaml_format(msg)
    action = event.edit if not borg.me.bot else event.respond
    try:
        await action(yaml_text, parse_mode=parse_pre)
    except MessageTooLongError:
        if not borg.me.bot:
            await event.delete()
        yaml_text = yaml_format(
            msg,
            max_str_len=9999,
            max_byte_len=9999
        )
        await borg.send_file(
            await event.get_input_chat(),
            f'<pre>{yaml_text}</pre>'.encode('utf-8'),
            reply_to=msg,
            attributes=[
                DocumentAttributeFilename('info.html')
            ],
            allow_cache=False
        )


@borg.on(borg.cmd("who"))
async def who(event):
    participant = None
    if not event.message.is_reply:
        who = await event.get_chat()
    else:
        msg = await event.message.get_reply_message()
        if msg.forward:
            if msg.forward.from_name is not None:
                who = msg.forward.original_fwd
            else:
                who = await borg.get_entity(
                    msg.forward.sender_id or msg.forward.chat_id)
        else:
            who = await msg.get_sender()
            ic = await event.get_input_chat()
            if who is None:
                who = await event.get_chat()
            elif (isinstance(ic, InputPeerChannel) and
                    isinstance(who, User)):
                try:
                    participant = (await borg(GetParticipantRequest(
                        ic,
                        who
                    ))).participant
                except UserNotParticipantError:
                    pass
    yaml_text = yaml_format(who)
    if participant is not None:
        yaml_text += "\n"
        yaml_text += yaml_format(participant)
    action = event.edit if not borg.me.bot else event.respond
    await action(yaml_text, parse_mode=parse_pre)