From ce219f1b74ed6ab3df937bed6c2f23f225c4d1e7 Mon Sep 17 00:00:00 2001
From: Mateusz Kwasniewski <kwasniewski.mateusz@gmail.com>
Date: Fri, 26 Jan 2024 17:03:10 +0100
Subject: [PATCH] fix: hidden envs (#6050)

---
 .../FeatureOverviewEnvironment.tsx             |  8 +++++++-
 frontend/src/hooks/useHiddenEnvironments.ts    | 18 +++++++++++++++---
 2 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/frontend/src/component/feature/FeatureView/FeatureOverview/FeatureOverviewEnvironments/FeatureOverviewEnvironment/FeatureOverviewEnvironment.tsx b/frontend/src/component/feature/FeatureView/FeatureOverview/FeatureOverviewEnvironments/FeatureOverviewEnvironment/FeatureOverviewEnvironment.tsx
index 62d5287d56..31015ad43b 100644
--- a/frontend/src/component/feature/FeatureView/FeatureOverview/FeatureOverviewEnvironments/FeatureOverviewEnvironment/FeatureOverviewEnvironment.tsx
+++ b/frontend/src/component/feature/FeatureView/FeatureOverview/FeatureOverviewEnvironments/FeatureOverviewEnvironment/FeatureOverviewEnvironment.tsx
@@ -134,7 +134,13 @@ const FeatureOverviewEnvironment = ({
 
     return (
         <ConditionallyRender
-            condition={!new Set(globalStore.hiddenEnvironments).has(env.name)}
+            condition={
+                !new Set(
+                    Array.isArray(globalStore.hiddenEnvironments)
+                        ? globalStore.hiddenEnvironments
+                        : [],
+                ).has(env.name)
+            }
             show={
                 <StyledFeatureOverviewEnvironment enabled={env.enabled}>
                     <StyledAccordion
diff --git a/frontend/src/hooks/useHiddenEnvironments.ts b/frontend/src/hooks/useHiddenEnvironments.ts
index 7aba77e69a..28dc1a32a6 100644
--- a/frontend/src/hooks/useHiddenEnvironments.ts
+++ b/frontend/src/hooks/useHiddenEnvironments.ts
@@ -4,15 +4,26 @@ import { usePlausibleTracker } from 'hooks/usePlausibleTracker';
 
 export const useHiddenEnvironments = () => {
     const { trackEvent } = usePlausibleTracker();
+
     const { value: globalStore, setValue: setGlobalStore } =
         useGlobalLocalStorage();
     const [hiddenEnvironments, setStoredHiddenEnvironments] = useState<
         Set<string>
-    >(new Set(globalStore.hiddenEnvironments));
+    >(
+        new Set(
+            Array.isArray(globalStore.hiddenEnvironments)
+                ? globalStore.hiddenEnvironments
+                : [],
+        ),
+    );
 
     const setHiddenEnvironments = (environment: string) => {
         setGlobalStore((params) => {
-            const hiddenEnvironments = new Set(params.hiddenEnvironments);
+            const hiddenEnvironments = new Set(
+                Array.isArray(globalStore.hiddenEnvironments)
+                    ? globalStore.hiddenEnvironments
+                    : [],
+            );
             if (hiddenEnvironments.has(environment)) {
                 hiddenEnvironments.delete(environment);
                 trackEvent('hidden_environment', {
@@ -29,9 +40,10 @@ export const useHiddenEnvironments = () => {
                 });
             }
             setStoredHiddenEnvironments(hiddenEnvironments);
+
             return {
                 ...globalStore,
-                hiddenEnvironments: hiddenEnvironments,
+                hiddenEnvironments: [...hiddenEnvironments],
             };
         });
     };