From 76d86764c088363abd119480f84cca0534c545a8 Mon Sep 17 00:00:00 2001
From: aniket-engg <aniketengg.225@gmail.com>
Date: Thu, 24 Jun 2021 14:08:35 +0530
Subject: [PATCH] show only slither analysis

---
 .../src/lib/remix-ui-static-analyser.tsx      | 23 ++++++++++++++-----
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/libs/remix-ui/static-analyser/src/lib/remix-ui-static-analyser.tsx b/libs/remix-ui/static-analyser/src/lib/remix-ui-static-analyser.tsx
index e2dcd06777..3993d23005 100644
--- a/libs/remix-ui/static-analyser/src/lib/remix-ui-static-analyser.tsx
+++ b/libs/remix-ui/static-analyser/src/lib/remix-ui-static-analyser.tsx
@@ -57,8 +57,9 @@ export const RemixUiStaticAnalyser = (props: RemixUiStaticAnalyserProps) => {
   const [autoRun, setAutoRun] = useState(true)
   const [slitherEnabled, setSlitherEnabled] = useState(false)
   const [showSlither, setShowSlither] = useState('hidden')
+  // Show checkbox to select to display only Slither Analysis
   const [showSlitherResult, setShowSlitherResult] = useState('hidden')
-  const [showSlitherResultEnabled, setShowSlitherResultEnabled] = useState(false)
+  const [slitherResultEnabled, setSlitherResultEnabled] = useState(false)
   const [categoryIndex, setCategoryIndex] = useState(groupedModuleIndex(groupedModules))
 
   const warningContainer = React.useRef(null)
@@ -128,6 +129,14 @@ export const RemixUiStaticAnalyser = (props: RemixUiStaticAnalyserProps) => {
     setWarningState(groupedCategory)
   }
 
+  const showWarningsByModule = (showOnlyModule: string) => {
+    if(showOnlyModule && warningState[showOnlyModule]) {
+      const newWarningState = {}
+      newWarningState[showOnlyModule] = warningState[showOnlyModule]
+      setWarningState({[showOnlyModule]: warningState[showOnlyModule]})
+    }
+  }
+
   const run = (lastCompilationResult, lastCompilationSource, currentFile) => {
     if (state.data !== null) {
       if (lastCompilationResult && (categoryIndex.length > 0 || slitherEnabled)) {
@@ -243,6 +252,7 @@ export const RemixUiStaticAnalyser = (props: RemixUiStaticAnalyserProps) => {
                     warningMessage.push({ msg, options, hasWarning: true, warningModuleName: 'Slither Analysis' })
                   })
                   showWarnings(warningMessage, 'warningModuleName')
+                  setShowSlitherResult('visible')
                   props.event.trigger('staticAnaysisWarning', [warningCount])
                 }
               }).catch((error) => {
@@ -291,7 +301,7 @@ export const RemixUiStaticAnalyser = (props: RemixUiStaticAnalyserProps) => {
     }
   }
 
-  const handleShowSlitherResultEnabled = () => {
+  const handleSlitherEnabled = () => {
     if (slitherEnabled) {
       setSlitherEnabled(false)
     } else {
@@ -299,11 +309,12 @@ export const RemixUiStaticAnalyser = (props: RemixUiStaticAnalyserProps) => {
     }
   }
 
-  const handleSlitherEnabled = () => {
-    if (showSlitherResultEnabled) {
-      setShowSlitherResultEnabled(false)
+  const handleShowSlitherResultEnabled = () => {
+    if (slitherResultEnabled) {
+      setSlitherResultEnabled(false)
     } else {
-      setShowSlitherResultEnabled(true)
+      setSlitherResultEnabled(true)
+      showWarningsByModule('Slither Analysis')
     }
   }