From 5a8c43276e5466c40159c8a204dfa5a40f7a11ed Mon Sep 17 00:00:00 2001
From: Jay Kumar <jaykumar20march@gmail.com>
Date: Wed, 20 Nov 2024 13:27:23 +0000
Subject: [PATCH] feat: manual enrich field integration

---
 keep-ui/app/alerts/EnrichAlertModal.tsx | 181 +++++++++++++++++++++---
 keep-ui/app/alerts/alert-menu.tsx       |   3 +-
 keep-ui/app/alerts/alerts.tsx           |   3 +-
 3 files changed, 169 insertions(+), 18 deletions(-)

diff --git a/keep-ui/app/alerts/EnrichAlertModal.tsx b/keep-ui/app/alerts/EnrichAlertModal.tsx
index 247116818..534beb369 100644
--- a/keep-ui/app/alerts/EnrichAlertModal.tsx
+++ b/keep-ui/app/alerts/EnrichAlertModal.tsx
@@ -1,9 +1,10 @@
 import { AlertDto } from "./models";
 import Modal from "@/components/ui/Modal";
-import { Button, TextInput } from "@tremor/react";
+import { Button, TextInput, Divider } from "@tremor/react";
 import { useSession } from "next-auth/react";
 import { useApiUrl } from "utils/hooks/useConfig";
-import React from "react";
+import React, { useEffect, useState } from "react";
+import { toast } from "react-toastify";
 
 interface EnrichAlertModalProps {
   alert: AlertDto | null | undefined;
@@ -11,6 +12,38 @@ interface EnrichAlertModalProps {
   mutate: () => void;
 }
 
+const EXCLUDED_FIELDS = [
+  "id",
+  "status",
+  "severity",
+  "lastReceived",
+  "fingerprint",
+  "isPartialDuplicate",
+  "isFullDuplicate",
+  "apiKeyRef",
+  "firingStartTime",
+  "enriched_fields",
+  "isNoisy",
+  "deleted",
+  "startedAt",
+  "incident",
+  "providerId",
+  "providerType",
+  "dismissUntil",
+  "dismissed",
+  "assignee",
+  "source",
+  "pushed",
+  "environment"
+];
+
+const transformAlertToEditableFields = (alert: AlertDto | null | undefined) => {
+  if (!alert) return [];
+  return Object.entries(alert)
+    .filter(([key]) => !EXCLUDED_FIELDS.includes(key))
+    .map(([key, value]) => ({ key, value }));
+};
+
 const EnrichAlertModal: React.FC<EnrichAlertModalProps> = ({
   alert,
   handleClose,
@@ -20,41 +53,157 @@ const EnrichAlertModal: React.FC<EnrichAlertModalProps> = ({
   const { data: session } = useSession();
   const apiUrl = useApiUrl();
 
-  if (!alert) return null;
+  const [customFields, setCustomFields] = useState<{ key: string; value: string }[]>([]);
+  const [editedFields, setEditedFields] = useState<{ key: string; value: string }[]>([]);
+  const [finalData, setFinalData] = useState<Record<string, any>>({});
+
+  const addCustomField = () => {
+    setCustomFields((prev) => [...prev, { key: "", value: "" }]);
+  };
+
+  const updateCustomField = (index: number, field: "key" | "value", value: string) => {
+    setCustomFields((prev) =>
+      prev.map((item, i) => (i === index ? { ...item, [field]: value } : item))
+    );
+  };
+
+  const removeCustomField = (index: number) => {
+    setCustomFields((prev) => prev.filter((_, i) => i !== index));
+  };
+
+  const handleFieldChange = (key: string, value: string) => {
+    setEditedFields((prev) =>
+      prev.map((item) => (item.key === key ? { ...item, value } : item))
+    );
+  };
+
+  useEffect(() => {
+    setCustomFields([]);
+    setEditedFields(transformAlertToEditableFields(alert));
+  }, [alert]);
+
+  useEffect(() => {
+    const calculateFinalData = () => {
+      const changedFields = editedFields.reduce((acc, field) => {
+        const key = field.key as keyof AlertDto;
+        const originalValue = alert ? alert[key] : undefined;
+
+        if (String(originalValue) !== String(field.value)) {
+          acc[key] = field.value as any ?? null;
+        }
+        return acc;
+      }, {} as Partial<AlertDto>);
+
+      const customFieldData = customFields.reduce((acc, field) => {
+        if (field.key) {
+          acc[field.key] = field.value;
+        }
+        return acc;
+      }, {} as Record<string, string>);
+
+      return { ...changedFields, ...customFieldData };
+    };
+
+    setFinalData(calculateFinalData());
+  }, [customFields, editedFields, alert]);
 
-  const renderFormFields = () => {
-    return Object.entries(alert).map(([key, value]) => (
-      <div key={key} className="mb-4">
-        <label htmlFor={key} className="mb-1">
-          {key}:
+  const handleSave = async () => {
+    const requestData = {
+      enrichments: finalData,
+      fingerprint: alert?.fingerprint,
+    };
+
+    const response = await fetch(`${apiUrl}/alerts/enrich`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${session?.accessToken}`,
+      },
+      body: JSON.stringify(requestData),
+    });
+
+    if (response.ok) {
+      toast.success("Alert enriched successfully");
+      mutate();
+      handleClose();
+    } else {
+      toast.error("Failed to enrich alert");
+    }
+  };
+
+  const renderFormFields = () =>
+    editedFields.map((field) => (
+      <div key={field.key} className="mb-4 flex items-center gap-2">
+        <label htmlFor={field.key} className="mb-1 w-[20%] truncate">
+          {field.key}:
         </label>
         <TextInput
-          id={key}
-          name={key}
-          value={String(value || "")}
-          disabled
-          className="mt-1"
+          id={field.key}
+          name={field.key}
+          value={String(field.value || "")}
+          className="mt-1 w-full"
+          onChange={(e) => handleFieldChange(e.target.name, e.target.value)}
         />
       </div>
     ));
-  };
+
+  const renderCustomFields = () =>
+    customFields.map((field, index) => (
+      <div key={index} className="mb-4 flex items-center gap-2">
+        <TextInput
+          placeholder="Field Name"
+          value={field.key}
+          onChange={(e) => updateCustomField(index, "key", e.target.value)}
+          required
+          className="w-1/3"
+        />
+        <TextInput
+          placeholder="Field Value"
+          value={field.value}
+          onChange={(e) => updateCustomField(index, "value", e.target.value)}
+          className="w-1/3"
+        />
+        <Button color="red" onClick={() => removeCustomField(index)}>
+          ✕
+        </Button>
+      </div>
+    ));
 
   return (
     <Modal
       onClose={handleClose}
       isOpen={isOpen}
-      className="overflow-visible max-w-fit"
+      className="overflow-auto !max-w-full w-[50%]"
     >
       <div className="flex justify-between items-center mb-4 min-w-full">
         <h2 className="text-lg font-semibold">Enrich Alert</h2>
         <div className="flex gap-x-2">
+          <Button
+            onClick={handleSave}
+            color="orange"
+            variant="primary"
+            disabled={Object.keys(finalData).length === 0} // Disable button if finalData is empty
+          >
+            Save
+          </Button>
           <Button onClick={handleClose} color="orange" variant="secondary">
             Close
           </Button>
         </div>
       </div>
 
-      <div className="form-container">
+      <div>
+        <h3 className="text-md font-semibold mb-2">Custom Fields</h3>
+        {renderCustomFields()}
+        <Button onClick={addCustomField} className="mt-2 bg-orange-500">
+          + Add Field
+        </Button>
+      </div>
+
+      <Divider />
+
+      <div>
+        <h3 className="text-md font-semibold mb-2">Alert Data</h3>
         {alert ? renderFormFields() : <p>No data available.</p>}
       </div>
     </Modal>
diff --git a/keep-ui/app/alerts/alert-menu.tsx b/keep-ui/app/alerts/alert-menu.tsx
index 86d198ba7..d85d4bfdd 100644
--- a/keep-ui/app/alerts/alert-menu.tsx
+++ b/keep-ui/app/alerts/alert-menu.tsx
@@ -9,6 +9,7 @@ import {
   UserPlusIcon,
   PlayIcon,
   EyeIcon,
+  SparklesIcon,
 } from "@heroicons/react/24/outline";
 import { IoNotificationsOffOutline } from "react-icons/io5";
 
@@ -229,7 +230,7 @@ export default function AlertMenu({
               active ? "bg-slate-200" : "text-gray-900"
             } group flex w-full items-center rounded-md px-2 py-2 text-xs`}
           >
-            <ArchiveBoxIcon className="mr-2 h-4 w-4" aria-hidden="true" />
+            <SparklesIcon className="mr-2 h-4 w-4" aria-hidden="true" />
             Enrich
           </button>
         )}
diff --git a/keep-ui/app/alerts/alerts.tsx b/keep-ui/app/alerts/alerts.tsx
index 6389ca436..c4171eae2 100644
--- a/keep-ui/app/alerts/alerts.tsx
+++ b/keep-ui/app/alerts/alerts.tsx
@@ -112,6 +112,7 @@ export default function Alerts({ presetName }: AlertsProps) {
   useEffect(() => {
     const fingerprint = searchParams?.get("alertPayloadFingerprint");
     const enrich = searchParams?.get("enrich");
+    console.log(enrich, fingerprint)
     if (fingerprint && enrich) {
       const alert = alerts?.find((alert) => alert.fingerprint === fingerprint);
       setEnrichAlertModal(alert)
@@ -190,7 +191,7 @@ export default function Alerts({ presetName }: AlertsProps) {
         mutate={mutateAlerts}
       />
       <EnrichAlertModal
-        alert={viewAlertModal}
+        alert={viewEnrichAlertModal}
         handleClose={() => router.replace(`/alerts/${presetName}`)}
         mutate={mutateAlerts}
       />