|
|
# maunium-stickerpicker - A fast and simple Matrix sticker picker widget. # Copyright (C) 2020 Tulir Asokan # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <https://www.gnu.org/licenses/>. from functools import partial from io import BytesIO import os.path import json import tempfile import mimetypes
try: import magic except ImportError: print("[Warning] Magic is not installed, using file extensions to guess mime types") magic = None from PIL import Image, ImageSequence
from . import matrix
open_utf8 = partial(open, encoding='UTF-8')
def guess_mime(data: bytes) -> str: mime = None if magic: try: return magic.Magic(mime=True).from_buffer(data) except Exception: pass else: with tempfile.NamedTemporaryFile() as temp: temp.write(data) temp.close() mime, _ = mimetypes.guess_type(temp.name) return mime or "image/png"
def video_to_gif(data: bytes, mime: str) -> bytes: ext = mimetypes.guess_extension(mime) if mime.startswith("video/"): # run ffmpeg to fix duration with tempfile.NamedTemporaryFile(suffix=ext) as temp: temp.write(data) temp.flush() with tempfile.NamedTemporaryFile(suffix=ext) as temp_fixed: import subprocess print(".", end="", flush=True) result = subprocess.run(["ffmpeg", "-y", "-i", temp.name, "-codec", "copy", temp_fixed.name], capture_output=True) if result.returncode != 0: raise RuntimeError(f"Run ffmpeg failed with code {result.returncode}, Error occurred:\n{result.stderr}") temp_fixed.seek(0) data = temp_fixed.read() with tempfile.NamedTemporaryFile(suffix=ext) as temp: temp.write(data) temp.flush() with tempfile.NamedTemporaryFile(suffix=".gif") as gif: from moviepy.editor import VideoFileClip clip = VideoFileClip(temp.name) clip.write_gif(gif.name, logger=None) gif.seek(0) return gif.read()
def _convert_image(data: bytes, mimetype: str) -> (bytes, int, int): image: Image.Image = Image.open(BytesIO(data)) new_file = BytesIO() suffix = mimetypes.guess_extension(mimetype) if suffix: suffix = suffix[1:] # Determine if the image is a GIF is_animated = getattr(image, "is_animated", False) if is_animated: frames = [frame.convert("RGBA") for frame in ImageSequence.Iterator(image)] # Save the new GIF frames[0].save( new_file, format='GIF', save_all=True, append_images=frames[1:], loop=image.info.get('loop', 0), # Default loop to 0 if not present duration=image.info.get('duration', 100), # Set a default duration if not present transparency=image.info.get('transparency', 255), # Default to 255 if transparency is not present disposal=image.info.get('disposal', 2) # Default to disposal method 2 (restore to background) ) # Get the size of the first frame to determine resizing w, h = frames[0].size else: image = image.convert("RGBA") image.save(new_file, format=suffix) w, h = image.size if w > 256 or h > 256: # Set the width and height to lower values so clients wouldn't show them as huge images if w > h: h = int(h / (w / 256)) w = 256 else: w = int(w / (h / 256)) h = 256 return new_file.getvalue(), w, h
def _convert_sticker(data: bytes) -> (bytes, str, int, int): mimetype = guess_mime(data) if mimetype.startswith("video/"): data = video_to_gif(data, mimetype) print(".", end="", flush=True) mimetype = "image/gif" elif mimetype.startswith("application/gzip"): print(".", end="", flush=True) # unzip file import gzip with gzip.open(BytesIO(data), "rb") as f: data = f.read() mimetype = guess_mime(data) suffix = mimetypes.guess_extension(mimetype) with tempfile.NamedTemporaryFile(suffix=suffix) as temp: temp.write(data) with tempfile.NamedTemporaryFile(suffix=".gif") as gif: # run lottie_convert.py input output print(".", end="", flush=True) import subprocess cmd = ["lottie_convert.py", temp.name, gif.name] result = subprocess.run(cmd, capture_output=True, text=True) if result.returncode != 0: raise RuntimeError(f"Run {cmd} failed with code {retcode}, Error occurred:\n{result.stderr}") gif.seek(0) data = gif.read() mimetype = "image/gif" rlt = _convert_image(data, mimetype) suffix = mimetypes.guess_extension(mimetype) return rlt[0], mimetype, rlt[1], rlt[2]
def convert_sticker(data: bytes) -> (bytes, str, int, int): try: return _convert_sticker(data) except Exception as e: mimetype = guess_mime(data) print(f"Error converting image, mimetype: {mimetype}") ext = mimetypes.guess_extension(mimetype) with tempfile.NamedTemporaryFile(delete=False, suffix=ext) as temp: temp.write(data) print(f"Saved to {temp.name}") raise e
def add_to_index(name: str, output_dir: str) -> None: index_path = os.path.join(output_dir, "index.json") try: with open_utf8(index_path) as index_file: index_data = json.load(index_file) except (FileNotFoundError, json.JSONDecodeError): index_data = {"packs": []} if "homeserver_url" not in index_data and matrix.homeserver_url: index_data["homeserver_url"] = matrix.homeserver_url if name not in index_data["packs"]: index_data["packs"].append(name) with open_utf8(index_path, "w") as index_file: json.dump(index_data, index_file, indent=" ") print(f"Added {name} to {index_path}")
def make_sticker(mxc: str, width: int, height: int, size: int, mimetype: str, body: str = "") -> matrix.StickerInfo: return { "body": body, "url": mxc, "info": { "w": width, "h": height, "size": size, "mimetype": mimetype,
# Element iOS compatibility hack "thumbnail_url": mxc, "thumbnail_info": { "w": width, "h": height, "size": size, "mimetype": mimetype, }, }, "msgtype": "m.sticker", }
|