From 6ec94335649e951c6c9cd84c00d3db9d775344cb Mon Sep 17 00:00:00 2001
From: Hari Krishna <hari.100.hks@outlook.com>
Date: Tue, 25 Jun 2024 12:35:49 +0000
Subject: [PATCH] making the select rules option work for pipelines

---
 backend/go-app/main.go                 |  4 ++
 frontend/src/views/AngularWorkflow.jsx | 93 ++++++++++++++++++--------
 2 files changed, 70 insertions(+), 27 deletions(-)

diff --git a/backend/go-app/main.go b/backend/go-app/main.go
index 202aa891..274f1145 100755
--- a/backend/go-app/main.go
+++ b/backend/go-app/main.go
@@ -5199,7 +5199,11 @@ func initHandlers() {
 	r.HandleFunc("/api/v1/files/detection/sigma_rules", shuffle.HandleGetSigmaRules).Methods("GET", "OPTIONS")
 	r.HandleFunc("/api/v1/files/detection/{fileId}/{action}", shuffle.HandleToggleRule).Methods("PUT", "OPTIONS")
 	r.HandleFunc("/api/v1/files/detection/{action}", shuffle.HandleFolderToggle).Methods("PUT", "OPTIONS")
+
 	r.HandleFunc("/api/v1/detection/siem/node_health", handleTenzirHealthUpdate).Methods("POST","OPTIONS")
+	r.HandleFunc("/api/v1/detection/{triggerId}/selected_rules", shuffle.HandleGetSelectedRules).Methods("GET","OPTIONS")
+	r.HandleFunc("/api/v1/detection/{triggerId}/selected_rules/save", shuffle.HandleSaveSelectedRules).Methods("POST","OPTIONS")
+
 
 	// Introduced in 0.9.21 to handle notifications for e.g. failed Workflow
 	r.HandleFunc("/api/v1/notifications", shuffle.HandleCreateNotification).Methods("POST", "OPTIONS")
diff --git a/frontend/src/views/AngularWorkflow.jsx b/frontend/src/views/AngularWorkflow.jsx
index cc4d0b40..f3acbf27 100755
--- a/frontend/src/views/AngularWorkflow.jsx
+++ b/frontend/src/views/AngularWorkflow.jsx
@@ -21194,9 +21194,37 @@ const releaseToConnectLabel = "Release to Connect"
   ) : null;
 
   const TenzirConfigModal = () => {
-    const [loading, setLoading] = useState(false);
+    const [loading, setLoading] = useState(true);
     const [selectedRules, setSelectedRules] = useState([]);
   
+    useEffect(() => {
+      if (tenzirConfigModalOpen) {
+        try {
+          const url = globalUrl + "/api/v1/detection/" + selectedTrigger.id + "/selected_rules"
+          fetch(url, {
+            method: "GET",
+            credentials: "include",
+            headers: {
+              "Content-Type": "application/json",
+            },
+          })
+            .then(response => response.json())
+            .then(data => {
+              const savedRules = data.selected_rules.map(rule => rule.file_id);
+              setSelectedRules(savedRules);
+              setLoading(false);
+            })
+            .catch(error => {
+              console.error('Error fetching selected rules:', error);
+              setLoading(false);
+            });
+        } catch (error) {
+          console.error('Error:', error);
+          setLoading(false);
+        }
+      }
+    }, [tenzirConfigModalOpen]);
+  
     const handleRuleChange = (event) => {
       setSelectedRules(event.target.value);
     };
@@ -21211,31 +21239,43 @@ const releaseToConnectLabel = "Release to Connect"
     };
   
     const handleSubmit = () => {
-      const selectedRuleFiles = rules
-        .filter(rule => selectedRules.includes(rule.file_id));
-
-        if (selectedTrigger.trigger_type !== "PIPELINE") {
-          toast("Unable to save the configuration");
-          return;
-        }
-      
-        selectedTrigger.parameters = selectedRuleFiles;
-    
-    
-      console.log('Selected Rule Files:', selectedRuleFiles);
-      console.log('Selected Rule File Names:', selectedRuleFiles.map(rule => rule.file_id));
-      
-      setTenzirConfigModalOpen(false);
-    };
-
-    useEffect(()=>{
-      if (selectedTrigger.trigger_type !== "PIPELINE") {
-        //toast("Unable to save the configuration");
-        return;
+      const selectedRuleFiles = rules.filter(rule => selectedRules.includes(rule.file_id));
+  
+      const payload = {
+        selected_rules: selectedRuleFiles.map(rule => ({
+          file_name: rule.file_name,
+          title: rule.title,
+          description: rule.description,
+          file_id: rule.file_id,
+          is_enabled: rule.is_enabled,
+        }))
+      };
+  
+      try {
+        const url = globalUrl + "/api/v1/detection/" + selectedTrigger.id + "/selected_rules/save"
+        fetch(url, {
+          method: 'POST',
+          credentials: "include",
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify(payload),
+        })
+          .then(response => response.json())
+          .then(data => {
+            console.log('Rules saved successfully:', data);
+            setTenzirConfigModalOpen(false);
+          })
+          .catch(error => {
+            console.error('Error saving selected rules:', error);
+            toast("Unable to save the configuration");
+          });
+      } catch (error) {
+        console.error('Error:', error);
+        toast("Unable to save the configuration");
       }
-      setSelectedRules(selectedTrigger.parameters);
-    },[])
-    
+    };
+  
     const enabledSigmaInfo = rules.filter(rule => rule.is_enabled);
   
     if (!tenzirConfigModalOpen) return null;
@@ -21319,8 +21359,7 @@ const releaseToConnectLabel = "Release to Connect"
         </IconButton>
       </Dialog>
     );
-  }
-  
+  };