Add script to fetch impact-db impacts data

Author: thvnx

Makefile

@@ -13,6 +13,7 @@ PYTHON=python
 MAVEN=mvn
 BUILD_DIR=/undefined
 LKQL_DIR=$(BUILD_DIR)/lkql
+IMPACTDB_DIR=/undefined
 GPRBUILD=gprbuild -j$(PROCS) -p -XBUILD_MODE=$(BUILD_MODE)
 GPRINSTALL=gprinstall --prefix=$(PREFIX) -p -XBUILD_MODE=$(BUILD_MODE)
 BUILD_FOR_JIT=false
@@ -36,6 +37,9 @@ doc: build_lkql_native_jit
 	cd user_manual && make clean html
 	cd lkql_checker/doc && make generate html-all
 
+impacts:
+	PYTHONPATH=$(IMPACTDB_DIR) ./utils/impact-db_impacts_gen.py $(IMPACTDB_DIR)
+
 format:
 	$(MAVEN) -f lkql_jit spotless:apply
 

utils/impact-db_impacts_gen.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+
+# This script should be run at the top level of the langkit-query-language with
+# PYTHONPATH set to the impact-db repository path. The impact-db repository path
+# should also be passed as first argument:
+#
+# $ PYTHONPATH=/path/to/impact-db ./impact-db_impacts_gen.py /path/to/impact-db
+
+import glob
+import json
+import sys
+from pathlib import Path
+from impactdb import db
+
+impactdb_dir = sys.argv[1]
+res = {}
+
+
+def format_impacts(impacts: [str]) -> str:
+    gnat = list(db.config["gnat"].keys())
+    return ",".join(list(map(lambda i: i + ".*" if i in gnat else i, impacts)))
+
+
+def format_name(name: str) -> str:
+    return "kp_" + name.lower().replace("-", "_")
+
+
+def list_impacts(issues: [str]) -> None:
+    entries = [e for e in db.load(impactdb_dir + "/entries") if e.type == "kp"]
+    for e in entries:
+        if e.name in issues:
+            res.update({format_name(e.name): format_impacts(e.impacts)})
+        elif e.origin in issues:
+            res.update({format_name(e.origin): format_impacts(e.impacts)})
+
+
+kps = [kp for kp in glob.glob("./lkql_checker/share/lkql/kp/KP-*")]
+ids = [Path(id).stem[3:] for id in kps]
+list_impacts(ids)
+
+impacts = {"impacts": res}
+impacts.update({"gnat": list(db.config["gnat"].keys())})
+
+with open("./lkql_checker/share/lkql/kp/kp.json", "w", encoding="utf-8") as f:
+    json.dump(impacts, f, ensure_ascii=False, indent=4)
+
+with open("./lkql_checker/share/lkql/kp/kp.json", "a", encoding="utf-8") as f:
+    f.write("\n")