diff --git a/sticker/lib/util.py b/sticker/lib/util.py index 2b3fe2a..b48181e 100644 --- a/sticker/lib/util.py +++ b/sticker/lib/util.py @@ -57,7 +57,7 @@ def add_to_index(name: str, output_dir: str) -> None: def make_sticker(mxc: str, width: int, height: int, size: int, - body: str = "") -> matrix.StickerInfo: + body: str = "", mimetype: str = "image/png") -> matrix.StickerInfo: return { "body": body, "url": mxc, @@ -65,7 +65,7 @@ def make_sticker(mxc: str, width: int, height: int, size: int, "w": width, "h": height, "size": size, - "mimetype": "image/png", + "mimetype": mimetype, # Element iOS compatibility hack "thumbnail_url": mxc, @@ -73,7 +73,7 @@ def make_sticker(mxc: str, width: int, height: int, size: int, "w": width, "h": height, "size": size, - "mimetype": "image/png", + "mimetype": mimetype, }, }, "msgtype": "m.sticker", diff --git a/sticker/stickerimport.py b/sticker/stickerimport.py index db176e4..2f4fc10 100644 --- a/sticker/stickerimport.py +++ b/sticker/stickerimport.py @@ -19,13 +19,13 @@ import asyncio import os.path import json import re - +import os from telethon import TelegramClient from telethon.tl.functions.messages import GetAllStickersRequest, GetStickerSetRequest from telethon.tl.types.messages import AllStickers from telethon.tl.types import InputStickerSetShortName, Document, DocumentAttributeSticker from telethon.tl.types.messages import StickerSet as StickerSetFull - +from subprocess import run from .lib import matrix, util @@ -39,6 +39,19 @@ async def reupload_document(client: TelegramClient, document: Document) -> matri print(".", flush=True) return util.make_sticker(mxc, width, height, len(data)) +async def reupload_document_gif(client: TelegramClient, document: Document) -> matrix.StickerInfo: + print(f"Reuploading {document.id}", end="", flush=True) + data = await client.download_media(document, file=bytes) + print(".", end="", flush=True) + # run LottieConverter + convert_data = run(['lottieconverter', '-', '-', 'gif', '512x512', '60'],capture_output = True, input=data).stdout + print(".", end="", flush=True) + mxc = await matrix.upload(convert_data, "image/gif", f"{document.id}.gif") + print(".", flush=True) + # 512x512 is mandatory for all stickers + return util.make_sticker(mxc, 512, 512, len(data), mimetype="image/gif") + + def add_meta(document: Document, info: matrix.StickerInfo, pack: StickerSetFull) -> None: for attr in document.attributes: @@ -57,8 +70,7 @@ def add_meta(document: Document, info: matrix.StickerInfo, pack: StickerSetFull) async def reupload_pack(client: TelegramClient, pack: StickerSetFull, output_dir: str) -> None: if pack.set.animated: - print("Animated stickerpacks are currently not supported") - return + print("Warning, to convert animated stickers, you need lottieconverter installed") pack_path = os.path.join(output_dir, f"{pack.set.short_name}.json") try: @@ -85,7 +97,10 @@ async def reupload_pack(client: TelegramClient, pack: StickerSetFull, output_dir reuploaded_documents[document.id] = already_uploaded[document.id] print(f"Skipped reuploading {document.id}") except KeyError: - reuploaded_documents[document.id] = await reupload_document(client, document) + if pack.set.animated: + reuploaded_documents[document.id] = await reupload_document_gif(client,document) + else: + reuploaded_documents[document.id] = await reupload_document(client, document) # Always ensure the body and telegram metadata is correct add_meta(document, reuploaded_documents[document.id], pack) diff --git a/sticker/version.py b/sticker/version.py index 47f7fff..a6c0452 100644 --- a/sticker/version.py +++ b/sticker/version.py @@ -1 +1,6 @@ -from .get_version import git_tag, git_revision, version, linkified_version +# Generated in setup.py + +git_tag = None +git_revision = 'f59406a4' +version = '0.1.0+dev.f59406a4' +linkified_version = '0.1.0+dev.[f59406a4](https://github.com/maunium/stickerpicker/commit/f59406a47a6778cd402e656ffb64f667335f665a)'