Implement adding/editing hooks

core/views/hooks.py

@@ -35,27 +35,62 @@ class Hooks(LoginRequiredMixin, View):
         return render(request, template_name, context)
 
 
-class AddHook(LoginRequiredMixin, APIView):
+class HookAction(LoginRequiredMixin, APIView):
     template_name = "modals/add-hook.html"
     parser_classes = [FormParser]
 
-    def get(self, request):
-        form = HookForm()
-        context = {"form": form}
+    def get(self, request, hook_id=None):
+        """
+        Get the form for adding or editing a hook.
+        :param hook_id: The id of the hook to edit. Optional.
+        """
+        if hook_id:
+            try:
+                hook = Hook.objects.get(id=hook_id, user=request.user)
+                form = HookForm(instance=hook)
+            except Hook.DoesNotExist:
+                message = "Hook does not exist"
+                message_class = "danger"
+                context = {
+                    "message": message,
+                    "message_class": message_class,
+                }
+                return render(request, self.template_name, context)
+        else:
+            form = HookForm()
+        context = {"form": form, "hook_id": hook_id}
 
         return render(request, self.template_name, context)
 
-    def put(self, request):
+    def put(self, request, hook_id=None):
+        """
+        Add or edit a hook.
+        :param hook_id: The id of the hook to edit. Optional.
+        """
         message = None
         message_class = "success"
 
-        # data = request.data
-        form = HookForm(request.data)
+        if hook_id:
+            try:
+                form = HookForm(request.data, instance=Hook.objects.get(id=hook_id))
+            except Hook.DoesNotExist:
+                message = "Hook does not exist"
+                message_class = "danger"
+                context = {
+                    "message": message,
+                    "class": message_class,
+                }
+                return render(request, self.template_name, context)
+        else:
+            form = HookForm(request.data)
         if form.is_valid():
             hook = form.save(commit=False)
             hook.user = request.user
             hook.save()
-            message = "Hook added successfully"
+            if hook_id:
+                message = f"Hook {hook_id} edited successfully"
+            else:
+                message = f"Hook {hook.id} added successfully"
         else:
             message = "Error adding hook"
             message_class = "danger"
@@ -70,3 +105,30 @@ class AddHook(LoginRequiredMixin, APIView):
             context["class"] = message_class
         template_name = "partials/hook-list.html"
         return render(request, template_name, context)
+
+    def delete(self, request, hook_id):
+        """
+        Delete a hook.
+        :param hook_id: The id of the hook to delete.
+        """
+        message = None
+        message_class = "success"
+        try:
+            hook = Hook.objects.get(id=hook_id, user=request.user)
+            hook.delete()
+            message = "Hook deleted successfully"
+        except Hook.DoesNotExist:
+            message = "Error deleting hook"
+            message_class = "danger"
+
+        hooks = get_hooks(request.user)
+
+        context = {
+            "hooks": hooks,
+        }
+        if message:
+            context["message"] = message
+            context["class"] = message_class
+
+        template_name = "partials/hook-list.html"
+        return render(request, template_name, context)