Reorganize Python stuff and add command to create packs

Fixes #11

1 files changed, 99 insertions, 0 deletions

diff --git a/sticker/pack.py b/sticker/pack.py
new file mode 100644
index 0000000..9959d8e
--- /dev/null
+++ b/sticker/pack.py
@@ -0,0 +1,99 @@
+# Copyright (c) 2020 Tulir Asokan
+#
+# 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/.
+from hashlib import sha256
+import argparse
+import os.path
+import asyncio
+import string
+import json
+
+import magic
+
+from .lib import matrix, util
+
+
+def convert_name(name: str) -> str:
+    name_translate = {
+        ord(" "): ord("_"),
+    }
+    allowed_chars = string.ascii_letters + string.digits + "_-/.#"
+    return "".join(filter(lambda char: char in allowed_chars, name.translate(name_translate)))
+
+
+async def main(args: argparse.Namespace) -> None:
+    await matrix.load_config(args.config)
+
+    dirname = os.path.basename(os.path.abspath(args.path))
+    meta_path = os.path.join(args.path, "pack.json")
+    try:
+        with open(meta_path) as pack_file:
+            pack = json.load(pack_file)
+            print(f"Loaded existing pack meta from {meta_path}")
+    except FileNotFoundError:
+        pack = {
+            "title": args.title or dirname,
+            "id": args.id or convert_name(dirname),
+            "stickers": [],
+        }
+        old_stickers = {}
+    else:
+        old_stickers = {sticker["id"]: sticker for sticker in pack["stickers"]}
+        pack["stickers"] = []
+    for file in os.listdir(args.path):
+        if file.startswith("."):
+            continue
+        path = os.path.join(args.path, file)
+        if not os.path.isfile(path):
+            continue
+        mime = magic.from_file(path, mime=True)
+        if not mime.startswith("image/"):
+            continue
+
+        try:
+            with open(path, "rb") as image_file:
+                image_data = image_file.read()
+        except Exception as e:
+            print(f"Failed to read {file}: {e}")
+            continue
+        print(f"Processing {file}", end="", flush=True)
+        name = os.path.splitext(file)[0]
+        sticker_id = f"sha256:{sha256(image_data).hexdigest()}"
+        print(".", end="", flush=True)
+        if sticker_id in old_stickers:
+            pack["stickers"].append({
+                **old_stickers[sticker_id],
+                "body": name,
+            })
+            print(f".. using existing upload")
+        else:
+            image_data, width, height = util.convert_image(image_data)
+            print(".", end="", flush=True)
+            mxc = await matrix.upload(image_data, "image/png", file)
+            print(".", end="", flush=True)
+            sticker = util.make_sticker(mxc, width, height, len(image_data), name)
+            sticker["id"] = sticker_id
+            pack["stickers"].append(sticker)
+            print(" uploaded", flush=True)
+    with open(meta_path, "w") as pack_file:
+        json.dump(pack, pack_file)
+    print(f"Wrote pack to {meta_path}")
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--config",
+                    help="Path to JSON file with Matrix homeserver and access_token",
+                    type=str, default="config.json")
+parser.add_argument("--title", help="Override the sticker pack displayname", type=str)
+parser.add_argument("--id", help="Override the sticker pack ID", type=str)
+parser.add_argument("path", help="Path to the sticker pack directory", type=str)
+
+
+def cmd():
+    asyncio.get_event_loop().run_until_complete(main(parser.parse_args()))
+
+
+if __name__ == "__main__":
+    cmd()