uniborg/stdplugins/snip.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/.
import asyncio
from telethon import events, utils
from telethon.tl import types

loop = asyncio.get_event_loop()

TYPE_TEXT = 0
TYPE_PHOTO = 1
TYPE_DOCUMENT = 2


# {name: {'text': text, 'id': id, 'hash': access_hash, 'type': type}}
snips = storage.snips or {}


@borg.on(events.NewMessage(pattern=r'\.snip (\S+)', outgoing=True))
async def on_snip(event):
    loop.create_task(event.delete())
    name = event.pattern_match.group(1)
    if name not in snips:
        return

    snip = snips[name]
    if snip['type'] == TYPE_PHOTO:
        media = types.InputPhoto(snip['id'], snip['hash'], file_reference=b'')
    elif snip['type'] == TYPE_DOCUMENT:
        media = types.InputDocument(snip['id'], snip['hash'], file_reference=b'')
    else:
        media = None

    await borg.send_message(await event.get_input_chat(), snip['text'],
                            file=media,
                            reply_to=event.message.reply_to_msg_id)


@borg.on(events.NewMessage(pattern=r'\.snips (\S+)', outgoing=True))
async def on_snip_save(event):
    loop.create_task(event.delete())
    name = event.pattern_match.group(1)
    msg = await event.get_reply_message()
    if not msg:
        return

    snips.pop(name, None)
    snip = {'type': TYPE_TEXT, 'text': msg.message or ''}
    if msg.media:
        media = None
        if isinstance(msg.media, types.MessageMediaPhoto):
            media = utils.get_input_photo(msg.media.photo)
            snip['type'] = TYPE_PHOTO
        elif isinstance(msg.media, types.MessageMediaDocument):
            media = utils.get_input_document(msg.media.document)
            snip['type'] = TYPE_DOCUMENT
        if media:
            snip['id'] = media.id
            snip['hash'] = media.access_hash

    snips[name] = snip
    storage.snips = snips


@borg.on(events.NewMessage(pattern=r'\.snipl', outgoing=True))
async def on_snip_list(event):
    loop.create_task(event.delete())
    await event.respond('available snips: ' + ', '.join(snips.keys()))


@borg.on(events.NewMessage(pattern=r'\.snipd (\S+)', outgoing=True))
async def on_snip_delete(event):
    loop.create_task(event.delete())
    snips.pop(event.pattern_match.group(1), None)
    storage.snips = snips


@borg.on(events.NewMessage(pattern=r'\.snipr (\S+)\s+(\S+)', outgoing=True))
async def on_snip_rename(event):
    loop.create_task(event.delete())
    snip = snips.pop(event.pattern_match.group(1), None)
    if snip:
        snips[event.pattern_match.group(2)] = snip
        storage.snips = snips