1 files changed, 30 insertions, 0 deletions
diff --git a/.local/bin/krokify b/.local/bin/krokify
new file mode 100755
index 0000000..a01c1f6
--- /dev/null
+++ b/.local/bin/krokify
@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+
+import argparse
+import base64
+import sys
+import zlib
+
+def main():
+	arg_parser = argparse.ArgumentParser(
+		prog='krokify',
+		description='Turn a given file into a kroki link',
+	)
+	arg_parser.add_argument("file", default="-", help="Path to the file to krokify, or - to read from stdin")
+	arg_parser.add_argument("-f", "--format", default="svg", help="Which output format to use")
+	arg_parser.add_argument("-s", "--service", required=True, help="Which back-end service to use")
+	arg_parser.add_argument("-u", "--base-url", default="https://kroki.tyil.nl", help="The base URL of the Kroki server")
+
+	args = arg_parser.parse_args()
+
+	if args.file == "-":
+		stream = sys.stdin
+	else:
+		stream = open(args.file, "r")
+
+	url = base64.urlsafe_b64encode(zlib.compress(stream.read().encode('utf-8'), 9)).decode('ascii')
+
+	print(f"{args.base_url}/{args.service}/{args.format}/{url}")
+
+if __name__ == "__main__":
+    main()