From 766b530ace50ec7adb483cbb7c74e3851f5cb743 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Mon, 1 Aug 2022 19:34:36 +0100
Subject: [PATCH] Implement automatic provisioning

---
 app/urls.py                                   |  6 +++++
 core/lib/manage/threshold.py                  |  7 ++++++
 .../manage/threshold/irc/network/actions.html |  4 ++++
 core/templatetags/nsep.py                     |  3 ++-
 core/views/manage/threshold/irc.py            | 23 +++++++++++++++++++
 5 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/app/urls.py b/app/urls.py
index 3b303c9..2aebd43 100644
--- a/app/urls.py
+++ b/app/urls.py
@@ -27,6 +27,7 @@ from core.views.manage.threshold.irc import (
     ThresholdIRCAliases,
     ThresholdIRCAliasesEdit,
     ThresholdIRCNetworkActions,
+    ThresholdIRCNetworkActionsAuto,
     ThresholdIRCNetworkActionsRelay,
     ThresholdIRCNetworkChannels,
     ThresholdIRCNetworkInfo,
@@ -171,6 +172,11 @@ urlpatterns = [
         ThresholdIRCNetworkActionsRelay.as_view(),
         name="threshold_irc_network_actions_add_relay",
     ),
+    path(
+        "manage/threshold/irc/auto/<str:net>/",
+        ThresholdIRCNetworkActionsAuto.as_view(),
+        name="threshold_irc_network_actions_auto",
+    ),
     ##
     path("api/chans/", ThresholdChans.as_view(), name="chans"),
     path("api/users/", ThresholdUsers.as_view(), name="users"),
diff --git a/core/lib/manage/threshold.py b/core/lib/manage/threshold.py
index f4d2efa..064be53 100644
--- a/core/lib/manage/threshold.py
+++ b/core/lib/manage/threshold.py
@@ -111,3 +111,10 @@ def update_aliases(aliases):
     payload = aliases
     deleted = threshold_request(url, payload, method="POST")
     return deleted
+
+
+def run_auto(net):
+    url = f"irc/auto/{net}"
+    payload = {}
+    deleted = threshold_request(url, payload, method="POST")
+    return deleted
diff --git a/core/templates/manage/threshold/irc/network/actions.html b/core/templates/manage/threshold/irc/network/actions.html
index b0d725b..19ef3eb 100644
--- a/core/templates/manage/threshold/irc/network/actions.html
+++ b/core/templates/manage/threshold/irc/network/actions.html
@@ -21,6 +21,10 @@
     </button>
     <button
       hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
+      hx-post="{% url 'threshold_irc_network_actions_auto' net %}"
+      hx-trigger="click"
+      hx-target="#actions"
+      hx-swap="outerHTML"
       class="button is-success">
       <span class="icon-text">
         <span class="icon">
diff --git a/core/templatetags/nsep.py b/core/templatetags/nsep.py
index 515e807..742ae57 100644
--- a/core/templatetags/nsep.py
+++ b/core/templatetags/nsep.py
@@ -2,6 +2,7 @@ from django import template
 
 register = template.Library()
 
+
 @register.filter
 def nsep(lst):
-    return "\n".join(lst)
\ No newline at end of file
+    return "\n".join(lst)
diff --git a/core/views/manage/threshold/irc.py b/core/views/manage/threshold/irc.py
index 9909305..6a60c73 100644
--- a/core/views/manage/threshold/irc.py
+++ b/core/views/manage/threshold/irc.py
@@ -287,6 +287,29 @@ class ThresholdIRCNetworkActions(SuperUserRequiredMixin, View):
         return render(request, self.template_name, context)
 
 
+class ThresholdIRCNetworkActionsAuto(SuperUserRequiredMixin, View):
+    template_name = "manage/threshold/irc/network/actions.html"
+
+    def post(self, request, net):
+        """
+        Run network automation.
+        """
+        ran = threshold.run_auto(net)
+        if ran["success"]:
+            message = f"Ran auto on network {net}"
+            message_class = "success"
+        else:
+            message = ran["reason"]
+            message_class = "danger"
+
+        context = {
+            "net": net,
+            "message": message,
+            "class": message_class,
+        }
+        return render(request, self.template_name, context)
+
+
 class ThresholdIRCNetworkActionsRelay(SuperUserRequiredMixin, APIView):
     template_name = "manage/threshold/irc/network/actions.html"
     parser_classes = [FormParser]