From 3671d94e593597cff6f24e3595c403e11da5f7a6 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Sun, 14 Aug 2022 12:43:13 +0100
Subject: [PATCH] Implement re-checking and resetting authentication status

---
 app/urls.py                                   |  6 ++
 core/lib/manage/threshold.py                  |  7 ++
 .../irc/network/modals/registration.html      | 79 +++++++++++++------
 core/views/manage/threshold/irc.py            | 26 ++++++
 4 files changed, 96 insertions(+), 22 deletions(-)

diff --git a/app/urls.py b/app/urls.py
index 0cf05b9..2d3ebb3 100644
--- a/app/urls.py
+++ b/app/urls.py
@@ -30,6 +30,7 @@ from core.views.manage.threshold.irc import (
     ThresholdIRCActions,
     ThresholdIRCActionsAddNetwork,
     ThresholdIRCActionsRegistration,
+    ThresholdIRCActionsRegistrationAuth,
     ThresholdIRCAliases,
     ThresholdIRCAliasesEdit,
     ThresholdIRCNetworkActions,
@@ -135,6 +136,11 @@ urlpatterns = [
         ThresholdIRCActionsAddNetwork.as_view(),
         name="threshold_irc_actions_add-network",
     ),
+    path(
+        "manage/threshold/irc/actions/registration/auth/",
+        ThresholdIRCActionsRegistrationAuth.as_view(),
+        name="threshold_irc_actions_registration_auth",
+    ),
     path(
         "manage/threshold/irc/actions/registration/<str:net>/",
         ThresholdIRCActionsRegistration.as_view(),
diff --git a/core/lib/manage/threshold.py b/core/lib/manage/threshold.py
index 5671e95..5f4d2fc 100644
--- a/core/lib/manage/threshold.py
+++ b/core/lib/manage/threshold.py
@@ -246,3 +246,10 @@ def irc_enable_auth(net, num):
     payload = {}
     enabled = threshold_request(url, payload, method="POST")
     return enabled
+
+
+def irc_check_auth(data):
+    url = "irc/network/auth"
+    payload = data
+    updated = threshold_request(url, payload, method="POST")
+    return updated
diff --git a/core/templates/manage/threshold/irc/network/modals/registration.html b/core/templates/manage/threshold/irc/network/modals/registration.html
index d78ce14..ddeade3 100644
--- a/core/templates/manage/threshold/irc/network/modals/registration.html
+++ b/core/templates/manage/threshold/irc/network/modals/registration.html
@@ -6,31 +6,66 @@
   <div class="modal-background"></div>
   <div class="modal-content">
     <div class="box">
+      {% include 'manage/threshold/partials/notify.html' %}
       <h4 class="subtitle is-4">Registration</h4>
-      <form
-        hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
-        hx-put="{% url 'threshold_irc_actions_registration_net' net %}"
-        hx-target="#actions"
-        hx-swap="outerHTML">
-        {% for network, items in unreg.items %}
-          <h4 class="title is-4">{{ network }}</h4>
-          {% for nick, num in items %}
-            <div class="field">
-              <label class="label">{{ nick }}/{{ num }}</label>
-              <div class="control">
-                <input class="input" type="text" name="{{ network }}|{{ num }}" placeholder="Enter token">
-              </div>
-            </div>
-          {% endfor %}
-        {% endfor %}
+      <div class="buttons">
         <button
-          type="button"
-          class="button is-light modal-close-button">
-          Cancel
+          hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
+          hx-post="{% url 'threshold_irc_actions_registration_auth' %}"
+          hx-vals='{"net": "{{ net }}", "func": "recheckauth"}'
+          hx-trigger="click"
+          hx-target="#modals-here"
+          class="button is-info">
+          <span class="icon-text">
+            <span class="icon">
+              <i class="fa-solid fa-wrench"></i>
+            </span>
+            <span>Check auth</span>
+          </span>
         </button>
-        <button type="submit" class="button is-info modal-close-button">Submit</button>
-        {# <script>activateButtons();</script> #}
-      </form>
+        <button
+          hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
+          hx-post="{% url 'threshold_irc_actions_registration_auth' %}"
+          hx-vals='{"net": "{{ net }}", "func": "resetauth"}'
+          hx-trigger="click"
+          hx-target="#modals-here"
+          class="button is-info">
+          <span class="icon-text">
+            <span class="icon">
+              <i class="fa-solid fa-wrench"></i>
+            </span>
+            <span>Reset auth</span>
+          </span>
+        </button>
+      </div>
+      {% if unreg %}
+        <form
+          hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
+          hx-put="{% url 'threshold_irc_actions_registration_net' net %}"
+          hx-target="#actions"
+          hx-swap="outerHTML">
+          {% for network, items in unreg.items %}
+            <h4 class="title is-4">{{ network }}</h4>
+            {% for nick, num in items %}
+              <div class="field">
+                <label class="label">{{ nick }}/{{ num }}</label>
+                <div class="control">
+                  <input class="input" type="text" name="{{ network }}|{{ num }}" placeholder="Enter token">
+                </div>
+              </div>
+            {% endfor %}
+          {% endfor %}
+          <button
+            type="button"
+            class="button is-light modal-close-button">
+            Cancel
+          </button>
+          <button type="submit" class="button is-info modal-close-button">Submit</button>
+          {# <script>activateButtons();</script> #}
+        </form>
+      {% else %}
+        <p>No unregistered relays.</p>
+      {% endif %}
   
     </div>
     <button class="modal-close is-large" aria-label="close"></button>
diff --git a/core/views/manage/threshold/irc.py b/core/views/manage/threshold/irc.py
index b5fa024..3a348eb 100644
--- a/core/views/manage/threshold/irc.py
+++ b/core/views/manage/threshold/irc.py
@@ -447,6 +447,32 @@ class ThresholdIRCActionsRegistration(SuperUserRequiredMixin, APIView):
         return render(request, template_name, context)
 
 
+class ThresholdIRCActionsRegistrationAuth(SuperUserRequiredMixin, APIView):
+    template_name = "manage/threshold/irc/network/modals/registration.html"
+    parser_classes = [FormParser]
+
+    def post(self, request):
+        """
+        Confirm registration for networks.
+        """
+        updated = threshold.irc_check_auth(request.data)
+
+        message = "Re-checked authentication successfully."
+        message_class = "success"
+
+        if not updated["success"]:
+            message = updated["reason"]
+            message_class = "danger"
+
+        context = {
+            "message": message,
+            "class": message_class,
+        }
+        if "net" in request.data:
+            context["net"] = request.data["net"]
+        return render(request, self.template_name, context)
+
+
 class ThresholdIRCAliasesEdit(SuperUserRequiredMixin, APIView):
     template_name = "manage/threshold/irc/overview/modals/edit-aliases.html"
     parser_classes = [FormParser]