precommit: Introduce new server-based framework

tools/precommit/precommit_server.py

@@ -0,0 +1,83 @@
+#!/usr/bin/python3
+
+# author: Ole Schuett
+
+import os
+import logging
+from os import path
+from time import time
+import tempfile
+import subprocess
+from subprocess import PIPE, STDOUT
+from flask import Flask, request, abort
+
+app = Flask(__name__)
+app.config["MAX_CONTENT_LENGTH"] = 1024 * 1024  # 1MB
+app.logger.setLevel(logging.INFO)
+app.logger.info("CP2K Precommit Server is up and running :-)")
+
+# ======================================================================================
+@app.route("/")
+def hello():
+    return "cp2k precommit server revision: " + os.environ["REVISION"]
+
+
+# ======================================================================================
+@app.route("/black", methods=["POST"])
+def black():
+    return run_tool(["black"])
+
+
+# ======================================================================================
+@app.route("/shellcheck", methods=["POST"])
+def shellcheck():
+    return run_tool(["shellcheck"])
+
+
+# ======================================================================================
+@app.route("/markdownlint", methods=["POST"])
+def markdownlint():
+    return run_tool(["mdl"])
+
+
+# ======================================================================================
+@app.route("/clangformat", methods=["POST"])
+def clangformat():
+    return run_tool(["clang-format", "--style=llvm", "-i"])
+
+
+# ======================================================================================
+def run_tool(cmd, timeout=30):
+    assert len(request.files) == 1
+    orig_fn = list(request.files.keys())[0]
+    data_before = request.files[orig_fn].read()
+    data_kb = len(data_before) / 1024.0
+    fn = path.basename(orig_fn)
+    workdir = tempfile.TemporaryDirectory()
+    abs_fn = path.join(workdir.name, fn)
+    open(abs_fn, "wb").write(data_before)
+
+    t1 = time()
+    try:
+        p = subprocess.run(
+            cmd + [fn], cwd=workdir.name, timeout=timeout, stdout=PIPE, stderr=STDOUT
+        )
+    except subprocess.TimeoutExpired:
+        app.logger.info(f"Timeout of {cmd[0]} on {data_kb:.1f}KB after {timeout}s.")
+        return f"Timeout while running {cmd[0]} - please try again.", 504
+    t2 = time()
+    app.logger.info(f"Ran {cmd[0]} on {data_kb:.1f}KB in {t2-t1:.1f}s.")
+
+    if p.returncode != 0:
+        return p.stdout, 422  # Unprocessable Entity
+    data_after = open(abs_fn, "rb").read()
+    if data_after == data_before:
+        return "Not Modified", 304
+    return data_after, 200
+
+
+# ======================================================================================
+if __name__ == "__main__":
+    app.run()
+
+# EOF