From 0ffe0731e6aca624695dd528761d2e459256ee67 Mon Sep 17 00:00:00 2001
From: Jan200101 <sentrycraft123@gmail.com>
Date: Tue, 29 Mar 2022 11:35:51 +0200
Subject: Initial commit

---
 labbot/addons/merge-label.py | 73 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 labbot/addons/merge-label.py

(limited to 'labbot/addons')

diff --git a/labbot/addons/merge-label.py b/labbot/addons/merge-label.py
new file mode 100644
index 0000000..e980e6c
--- /dev/null
+++ b/labbot/addons/merge-label.py
@@ -0,0 +1,73 @@
+import os
+import re
+
+title_regex = r"^(?:#|)(\d+)\s*"
+word_regex = r"^#(\d+)$"
+relation_keywords = [
+    "related"
+]
+relation_distance = 2
+
+state_label = {
+    "opened": "Code-Review",
+    "merged": "Testing",
+    "closed": "In Progress",
+}
+async def merge_request_hook(event, gl, *args, **kwargs):
+    state = event.object_attributes["state"]
+    related_issues = []
+
+    match = re.search(title_regex, event.object_attributes["title"])
+    if match:
+        related_issues.append(match.group(1))
+
+    for line in event.object_attributes["description"].split("\\n"):
+        line = line.lower()
+        line_list = line.split(" ")
+
+        for keyword in relation_keywords:
+            try:
+                keyword_index = line_list.index(keyword)
+
+                min_pos = keyword_index - relation_distance
+                if min_pos < 0:
+                    min_pos = 0
+
+                max_pos = keyword_index + relation_distance
+                if max_pos >= len(line_list):
+                    max_pos = len(line_list)-1
+
+                for word in line_list[min_pos:max_pos+1]:
+                    match = re.search(word_regex, word)
+                    if match:
+                        related_issues.append(match.group(1))
+
+            except ValueError:
+                pass
+
+    for issue in related_issues:
+        if not issue: continue
+
+        base_url = f"/projects/{event.project_id}/issues/{issue}"
+
+        delete_labels = list(state_label.values())
+
+        try:
+            label = state_label[state]
+            if label in delete_labels:
+                delete_labels.remove(label)
+
+            remove_labels = ",".join(delete_labels)
+
+            await gl.put(base_url, data={
+                "add_labels": label,
+                "remove_labels": remove_labels,
+            })
+
+        except KeyError:
+            # unknown state
+            pass
+
+def setup(bot):
+    bot.register(merge_request_hook, "Merge Request Hook")
+    pass
\ No newline at end of file
-- 
cgit v1.2.3