making enable / disable button to work only when connected to the siem

frontend/src/views/DetectionDashboard.jsx

@@ -1,68 +1,45 @@
 import React, { useState, useEffect } from "react";
-import { Container} from "@mui/material";
+import { Container, CircularProgress, Typography } from "@mui/material";
 import { toast } from "react-toastify";
 import Detection from "./Detection";
-import EditComponent from "./EditRules";
-
-const getSigmaInfo = (globalUrl, setRuleInfo, setFolderDisabled, setIsTenzirActive) => {
-  const url = globalUrl + "/api/v1/files/detection/sigma_rules";
-
-  fetch(url, {
-    method: "GET",
-    credentials: "include",
-    headers: {
-      "Content-Type": "application/json",
-    },
-  })
-    .then((response) =>
-      response.json().then((responseJson) => {
-        if (responseJson["success"] === false) {
-          toast("Failed to get sigma rules");
-        } else {
-          setRuleInfo(responseJson.sigma_info);
-          setFolderDisabled(responseJson.folder_disabled);
-          setIsTenzirActive(responseJson.is_tenzir_active);
-
-        }
-      })
-    )
-    .catch((error) => {
-      console.log("Error in getting sigma files: ", error);
-      toast("An error occurred while fetching sigma rules");
-    });
-};
 
 const DetectionDashBoard = (props) => {
   const { globalUrl } = props;
-  const [ruleInfo, setRuleInfo] = useState([]);
-  const [selectedRule, setSelectedRule] = useState(null);
-  const [fileData, setFileData] = React.useState("");
-  const [isTenzirActive, setIsTenzirActive] = React.useState(false);
- 
+  const [ruleInfo, setRuleInfo] = useState(null);
+  const [, setSelectedRule] = useState(null);
+  const [, setFileData] = useState("");
+  const [isTenzirActive, setIsTenzirActive] = useState(false);
   const [folderDisabled, setFolderDisabled] = useState(false);
+  const [isLoading, setIsLoading] = useState(false);
+  const [importAttempts, setImportAttempts] = useState(0);
+  const maxImportAttempts = 2;
 
   useEffect(() => {
-    getSigmaInfo(globalUrl, setRuleInfo, setFolderDisabled, setIsTenzirActive);
-  }, [folderDisabled]);
+    const fetchTimeout = setTimeout(() => {
+      fetchSigmaInfo();
+    }, 1000); // Delay by 1 second
+
+    return () => clearTimeout(fetchTimeout);
+  }, [globalUrl]);
 
   useEffect(() => {
-    if (ruleInfo?.length > 0) {
-      openEditBar(ruleInfo[0]);
+    if (ruleInfo && ruleInfo.length === 0 && importAttempts < maxImportAttempts) {
+      importSigmaFromUrl();
     }
   }, [ruleInfo]);
 
   const openEditBar = (rule) => {
     setSelectedRule(rule);
-    getFileContent(rule.file_id)
+    fetchFileContent(rule.file_id);
   };
 
   const handleSave = (updatedContent) => {
-   toast("this will be saved");
+    toast("This will be saved");
   };
 
-  const getFileContent = (file_id) => {
+  const fetchFileContent = (file_id) => {
     setFileData("");
-    fetch(globalUrl + "/api/v1/files/" + file_id + "/content", {
+    fetch(`${globalUrl}/api/v1/files/${file_id}/content`, {
       method: "GET",
       headers: {
         "Content-Type": "application/json",
@@ -77,38 +54,109 @@ const DetectionDashBoard = (props) => {
         }
         return response.text();
       })
-      .then((respdata) => {    
+      .then((respdata) => {
         if (respdata.length === 0) {
           toast("Failed getting file. Is it deleted?");
           return;
         }
-        return respdata
-      })
-      .then((responseData) => {
-      
-      setFileData(responseData);
+        setFileData(respdata);
       })
       .catch((error) => {
         toast(error.toString());
       });
   };
 
+  const fetchSigmaInfo = () => {
+    const url = `${globalUrl}/api/v1/files/detection/sigma_rules`;
+    setIsLoading(true);
+
+    fetch(url, {
+      method: "GET",
+      credentials: "include",
+      headers: {
+        "Content-Type": "application/json",
+      },
+    })
+      .then((response) => response.json())
+      .then((responseJson) => {
+        if (responseJson["success"] === false) {
+          toast("Failed to get sigma rules");
+        } else {
+          setRuleInfo(responseJson.sigma_info || []);
+          setFolderDisabled(responseJson.folder_disabled);
+          setIsTenzirActive(responseJson.is_tenzir_active);
+        }
+        setIsLoading(false);
+      })
+      .catch((error) => {
+        setIsLoading(false);
+        console.log("Error in getting sigma files: ", error);
+        toast("An error occurred while fetching sigma rules");
+        setRuleInfo([]);
+      });
+  };
+
+  const importSigmaFromUrl = () => {
+    setIsLoading(true);
+    setImportAttempts((prevAttempts) => prevAttempts + 1);
+
+    const url = "https://github.com/satti-hari-krishna-reddy/shuffle_sigma";
+    const folder = "sigma";
+
+    const parsedData = {
+      url: url,
+      path: folder,
+      field_3: "main",
+    };
+
+    toast(`Getting files from url ${url}. This may take a while if the repository is large. Please wait...`);
+    fetch(`${globalUrl}/api/v2/files/download_remote`, {
+      method: "POST",
+      mode: "cors",
+      headers: {
+        Accept: "application/json",
+      },
+      body: JSON.stringify(parsedData),
+      credentials: "include",
+    })
+      .then((response) => response.json())
+      .then((responseJson) => {
+        if (responseJson.success) {
+          toast("Successfully loaded files from " + url);
+          fetchSigmaInfo(); // Fetch again after successful import
+        } else {
+          toast(responseJson.reason ? `Failed loading: ${responseJson.reason}` : "Failed loading");
+        }
+        setIsLoading(false);
+      })
+      .catch((error) => {
+        toast(error.toString());
+        setIsLoading(false);
+      });
+  };
+
+  if (isLoading && (!ruleInfo || ruleInfo.length === 0)) {
+    return (
+      <Container style={{ display: "flex", justifyContent: "center", alignItems: "center", height: "100vh" }}>
+        <div>
+          <CircularProgress />
+          <Typography variant="h6" style={{ marginTop: 20 }}>Downloading rules, please wait...</Typography>
+        </div>
+      </Container>
+    );
+  }
+
   return (
-    <Container style={{display: "flex"}}>
-      {/* {selectedRule ? (
-        <EditComponent
-          ruleName={selectedRule.title}
-          description={selectedRule.description}
-          content={fileData}
-          setContent={setFileData}
-          lastEdited={selectedRule.lastEdited}
-          editedBy={selectedRule.editedBy}
-          onSave={handleSave}
-        />
-      ) : null} */}
-      <Detection globalUrl={globalUrl} ruleInfo={ruleInfo} folderDisabled={folderDisabled} setFolderDisabled={setFolderDisabled} openEditBar={openEditBar} isTenzirActive={isTenzirActive} />
+    <Container style={{ display: "flex" }}>
+      <Detection
+        globalUrl={globalUrl}
+        ruleInfo={ruleInfo}
+        folderDisabled={folderDisabled}
+        setFolderDisabled={setFolderDisabled}
+        isTenzirActive={isTenzirActive}
+      />
     </Container>
-  );  
+  );
 };
 
 export default DetectionDashBoard;