add thing

stdplugins/gifs_to_stickers.py

@@ -0,0 +1,87 @@
+from io import BytesIO
+
+from uniborg import util
+from PIL import Image
+from telethon import types, utils, events
+from telethon.tl.functions.messages import SaveGifRequest, UploadMediaRequest
+
+sticker_to_gif = storage.sticker_to_gif or {}
+access_hashes = storage.access_hashes or {}
+gif_to_sticker = {str(gif): int(sticker) for sticker, gif in sticker_to_gif.items()}
+
+
+async def convert_sticker_to_gif(sticker):
+    gif_id = sticker_to_gif.get(str(sticker.id), None)
+    if gif_id:
+        access_hash = access_hashes[str(gif_id)]
+        return types.InputDocument(gif_id, access_hash, b'')
+    file = BytesIO()
+    await borg.download_media(sticker, file=file)
+    file.seek(0)
+
+    # remove alpha
+    im = Image.open(file)
+    alpha = im.convert('RGBA').getchannel('A')
+    new_im = Image.new('RGBA', im.size, (255, 255, 255, 255))
+    new_im.paste(im, mask=alpha)
+    file = BytesIO()
+    new_im.save(file, format='gif')
+    file.seek(0)
+
+    # upload file
+    file = await borg.upload_file(file, part_size_kb=512)
+    file = types.InputMediaUploadedDocument(file, 'video/mp4', [])
+    media = await borg(UploadMediaRequest('me', file))
+    media = utils.get_input_document(media)
+
+    # save (that's right, this is relational json)
+    sticker_to_gif[str(sticker.id)] = media.id
+    gif_to_sticker[str(media.id)] = sticker.id
+    access_hashes[str(sticker.id)] = sticker.access_hash
+    access_hashes[str(media.id)] = media.access_hash
+    storage.sticker_to_gif = sticker_to_gif
+    storage.access_hashes = access_hashes
+
+    return media
+
+
+@borg.on(util.admin_cmd(r'^\.ss$'))
+async def on_save(event):
+    await event.delete()
+    target = await event.get_reply_message()
+    media = target.gif or target.sticker
+    if not media:
+        return
+    if target.sticker:
+        media = await convert_sticker_to_gif(media)
+    await borg(
+        SaveGifRequest(id=media, unsave=False)
+    )
+
+
+@borg.on(events.NewMessage(outgoing=True))
+async def on_sticker(event):
+    if not event.sticker:
+        return
+    media = await convert_sticker_to_gif(event.sticker)
+    await borg(
+        SaveGifRequest(id=media, unsave=False)
+    )
+
+
+@borg.on(events.NewMessage(outgoing=True))
+async def on_gif(event):
+    if not event.gif:
+        return
+    sticker_id = gif_to_sticker.get(str(event.gif.id), None)
+    if not sticker_id:
+        return
+    access_hash = access_hashes[str(sticker_id)]
+    sticker = types.InputDocument(sticker_id, access_hash, b'')
+
+    await event.delete()
+    await borg.send_message(
+        await event.get_input_chat(),
+        file=sticker,
+        reply_to=event.message.reply_to_msg_id
+    )