From 5c19efbff4f1d2e74f7f8ddca2bb4c80fdd2ca55 Mon Sep 17 00:00:00 2001
From: Bauke <me@bauke.xyz>
Date: Sun, 25 Sep 2022 15:14:53 +0200
Subject: [PATCH] Add the project-avatar script.

---
 .local/bin/project-avatar | 125 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 125 insertions(+)
 create mode 100755 .local/bin/project-avatar

diff --git a/.local/bin/project-avatar b/.local/bin/project-avatar
new file mode 100755
index 0000000..1d6c0cf
--- /dev/null
+++ b/.local/bin/project-avatar
@@ -0,0 +1,125 @@
+#!/usr/bin/env python3
+
+import argparse
+import os
+import re
+import subprocess
+import sys
+
+
+def main() -> None:
+    parser = create_parser()
+    args = parser.parse_args()
+
+    if os.path.isfile(args.filename):
+        if args.overwrite:
+            os.remove(args.filename)
+        else:
+            print("Target file already exists, use --overwrite to overwrite.")
+            sys.exit(1)
+
+    subprocess.run(
+        [
+            "gegl",
+            "-o",
+            args.filename,
+            "--",
+            *gegl_graph(args.height, args.width, args.text),
+        ],
+        check=True,
+    )
+
+    if not os.path.isfile(args.filename):
+        print("Something went wrong with GEGL")
+        sys.exit(1)
+
+    subprocess.run(
+        [
+            "convert",
+            args.filename,
+            "-background",
+            "transparent",
+            "-gravity",
+            "center",
+            "-extent",
+            f"{args.width}x{args.height}",
+            args.filename,
+        ],
+        check=True,
+    )
+
+    if args.clean:
+        subprocess.run(
+            [
+                "mat2",
+                "--inplace",
+                args.filename,
+            ],
+            check=True,
+        )
+
+
+def create_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(description="Project Avatar Generator")
+    parser.add_argument(
+        "filename",
+        help="The image filename to write to.",
+    )
+    parser.add_argument(
+        "text",
+        help="The text for the project avatar.",
+    )
+    parser.add_argument(
+        "--width",
+        help="The width of the image.",
+        type=int,
+        default=256,
+    )
+    parser.add_argument(
+        "--height",
+        help="The height of the image.",
+        type=int,
+        default=256,
+    )
+    parser.add_argument(
+        "--overwrite",
+        help="Overwrite an existing image.",
+        action="store_true",
+    )
+    parser.add_argument(
+        "--clean",
+        help="Use MAT2 to clean the image.",
+        action="store_true",
+    )
+    return parser
+
+
+def gegl_graph(height: int, width: int, text: str) -> typing.List[str]:
+    graph = f"""
+gegl:text
+    string={text}
+    width={width}
+    height={height}
+    color=white
+    font=Heavitas
+    size=150
+    alignment=1
+    vertical-alignment=1
+gegl:dropshadow
+    x=0
+    y=0
+    color=black
+    opacity=1
+    grow-radius=4
+    radius=0
+gegl:long-shadow
+    angle=90
+    color=black
+    length=20
+"""
+
+    return re.sub("\s\s+", "\n", graph).strip().splitlines()
+
+
+if __name__ == "__main__":
+    main()