mirror of
https://github.com/wah-su/stickerbridge.git
synced 2025-04-05 15:54:41 +00:00
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import tempfile
|
|
|
|
from nio import MatrixRoom, AsyncClient
|
|
|
|
from chat_functions import has_permission, is_stickerpack_existing, send_text_to_room, upload_image, upload_stickerpack
|
|
from sticker_types import Sticker, MatrixStickerset
|
|
from telegram_exporter import TelegramExporter
|
|
|
|
|
|
class MatrixReuploader:
|
|
|
|
STATUS_OK = 0
|
|
STATUS_NO_PERMISSION = 1
|
|
STATUS_PACK_EXISTS = 2
|
|
STATUS_PACK_EMPTY = 3
|
|
|
|
STATUS_DOWNLOADING = 4
|
|
STATUS_UPLOADING = 5
|
|
STATUS_UPDATING_ROOM_STATE = 6
|
|
|
|
def __init__(self, client: AsyncClient, room: MatrixRoom, exporter: TelegramExporter = None,
|
|
pack: list[Sticker] = None):
|
|
|
|
if not exporter and not pack:
|
|
raise ValueError('Either exporter or the pack must be set')
|
|
|
|
self.client = client
|
|
self.room = room
|
|
self.exporter = exporter
|
|
self.pack = pack
|
|
|
|
async def _has_permission_to_upload(self) -> bool:
|
|
return await has_permission(self.client, self.room.room_id, 'state_default')
|
|
|
|
async def import_stickerset_to_room(self, pack_name: str):
|
|
if not await self._has_permission_to_upload():
|
|
yield self.STATUS_NO_PERMISSION
|
|
return
|
|
|
|
stickerset = MatrixStickerset(pack_name)
|
|
if await is_stickerpack_existing(self.client, self.room.room_id, stickerset.name()):
|
|
yield self.STATUS_PACK_EXISTS
|
|
return
|
|
|
|
yield self.STATUS_DOWNLOADING
|
|
converted_stickerset = await self.exporter.get_stickerset(stickerset.name())
|
|
yield self.STATUS_UPLOADING
|
|
for sticker in converted_stickerset:
|
|
with tempfile.NamedTemporaryFile('w+b') as file:
|
|
file.write(sticker.image_data)
|
|
sticker_mxc = await upload_image(self.client, file.name, sticker.mime_type)
|
|
stickerset.add_sticker(sticker_mxc, sticker.alt_text)
|
|
|
|
if not stickerset.count():
|
|
yield self.STATUS_PACK_EMPTY
|
|
return
|
|
|
|
yield self.STATUS_UPDATING_ROOM_STATE
|
|
await upload_stickerpack(self.client, self.room.room_id, stickerset)
|
|
|
|
yield self.STATUS_OK
|