From 456a4cf8db7d1971308183cd91a40d5912af3523 Mon Sep 17 00:00:00 2001
From: evavirseda <evirseda@boxfish.studio>
Date: Wed, 30 Oct 2024 12:50:34 +0100
Subject: [PATCH] feat(wallet-dashboard): add feature flags for migration &
 supply increase vesting (#3670)

* feat: add features flags

* feat: add growthbook

* feat:add redirects if feature flag are disabled

* feat: call init outside function

* feat: rename anum and variabel

* fix

* feat: remove comment
---
 .../src/features/features.controller.ts       |  6 +++
 apps/core/src/constants/features.enum.ts      |  2 +
 .../components/sidebar/Sidebar.tsx            | 13 ++++-
 .../app/(protected)/migrations/page.tsx       | 15 ++++++
 .../app/(protected)/vesting/page.tsx          | 13 +++++
 apps/wallet-dashboard/app/layout.tsx          | 50 ++++++++++---------
 apps/wallet-dashboard/lib/utils/growthbook.ts | 11 ++++
 apps/wallet-dashboard/lib/utils/index.ts      |  1 +
 apps/wallet-dashboard/package.json            |  2 +
 pnpm-lock.yaml                                |  6 +++
 10 files changed, 94 insertions(+), 25 deletions(-)
 create mode 100644 apps/wallet-dashboard/lib/utils/growthbook.ts

diff --git a/apps/apps-backend/src/features/features.controller.ts b/apps/apps-backend/src/features/features.controller.ts
index e1400b1d57a..e71e4b326a4 100644
--- a/apps/apps-backend/src/features/features.controller.ts
+++ b/apps/apps-backend/src/features/features.controller.ts
@@ -64,6 +64,12 @@ export class FeaturesController {
                 [Feature.AccountFinder]: {
                     defaultValue: false,
                 },
+                [Feature.WalletDashboardMigration]: {
+                    defaultValue: false,
+                },
+                [Feature.WalletDashboardSupplyIncreaseVesting]: {
+                    defaultValue: false,
+                },
             },
             dateUpdated: new Date().toISOString(),
         };
diff --git a/apps/core/src/constants/features.enum.ts b/apps/core/src/constants/features.enum.ts
index 615c80386e3..3670db4edc1 100644
--- a/apps/core/src/constants/features.enum.ts
+++ b/apps/core/src/constants/features.enum.ts
@@ -18,4 +18,6 @@ export enum Feature {
     NetworkOutageOverride = 'network-outage-override',
     ModuleSourceVerification = 'module-source-verification',
     WalletEffectsOnlySharedTransaction = 'wallet-effects-only-shared-transaction',
+    StardustMigration = 'migration',
+    SupplyIncreaseVesting = 'supply-increase-vesting',
 }
diff --git a/apps/wallet-dashboard/app/(protected)/components/sidebar/Sidebar.tsx b/apps/wallet-dashboard/app/(protected)/components/sidebar/Sidebar.tsx
index 0f49225192d..03742f9e8e4 100644
--- a/apps/wallet-dashboard/app/(protected)/components/sidebar/Sidebar.tsx
+++ b/apps/wallet-dashboard/app/(protected)/components/sidebar/Sidebar.tsx
@@ -4,13 +4,24 @@
 import { PROTECTED_ROUTES } from '@/lib/constants/routes.constants';
 import { IotaLogoMark } from '@iota/ui-icons';
 import { SidebarItem } from './SidebarItem';
+import { Feature } from '@iota/core';
+import { useFeature } from '@growthbook/growthbook-react';
 
 export function Sidebar() {
+    const featureFlags = {
+        Migrations: useFeature<boolean>(Feature.StardustMigration).value,
+        Vesting: useFeature<boolean>(Feature.SupplyIncreaseVesting).value,
+    };
+
+    const filteredRoutes = PROTECTED_ROUTES.filter(({ title }) => {
+        return title in featureFlags ? featureFlags[title as keyof typeof featureFlags] : true;
+    });
+
     return (
         <nav className="flex h-screen flex-col items-center gap-y-2xl bg-neutral-100 py-xl dark:bg-neutral-6">
             <IotaLogoMark className="h-10 w-10 text-neutral-10 dark:text-neutral-92" />
             <div className="flex flex-col gap-y-xs">
-                {PROTECTED_ROUTES.map((route) => (
+                {filteredRoutes.map((route) => (
                     <SidebarItem key={route.path} {...route} />
                 ))}
             </div>
diff --git a/apps/wallet-dashboard/app/(protected)/migrations/page.tsx b/apps/wallet-dashboard/app/(protected)/migrations/page.tsx
index b57170a8ed6..01f8a8fbb4f 100644
--- a/apps/wallet-dashboard/app/(protected)/migrations/page.tsx
+++ b/apps/wallet-dashboard/app/(protected)/migrations/page.tsx
@@ -1,7 +1,22 @@
 // Copyright (c) 2024 IOTA Stiftung
 // SPDX-License-Identifier: Apache-2.0
+'use client';
+
+import { useFeature } from '@growthbook/growthbook-react';
+import { Feature } from '@iota/core';
+import { useRouter } from 'next/navigation';
+import { useEffect } from 'react';
 
 function MigrationDashboardPage(): JSX.Element {
+    const router = useRouter();
+    const stardustMigrationEnabled = useFeature<boolean>(Feature.StardustMigration).value;
+
+    useEffect(() => {
+        if (!stardustMigrationEnabled) {
+            router.push('/');
+        }
+    }, [stardustMigrationEnabled, router]);
+
     return (
         <div className="flex items-center justify-center pt-12">
             <h1>MIGRATIONS</h1>
diff --git a/apps/wallet-dashboard/app/(protected)/vesting/page.tsx b/apps/wallet-dashboard/app/(protected)/vesting/page.tsx
index c83b89b7962..e4c7b10badb 100644
--- a/apps/wallet-dashboard/app/(protected)/vesting/page.tsx
+++ b/apps/wallet-dashboard/app/(protected)/vesting/page.tsx
@@ -14,7 +14,9 @@ import {
     TimelockedStakedObjectsGrouped,
 } from '@/lib/utils';
 import { NotificationType } from '@/stores/notificationStore';
+import { useFeature } from '@growthbook/growthbook-react';
 import {
+    Feature,
     TIMELOCK_IOTA_TYPE,
     useGetActiveValidatorsInfo,
     useGetAllOwnedObjects,
@@ -24,11 +26,14 @@ import {
 import { useCurrentAccount, useIotaClient, useSignAndExecuteTransaction } from '@iota/dapp-kit';
 import { IotaValidatorSummary } from '@iota/iota-sdk/client';
 import { useQueryClient } from '@tanstack/react-query';
+import { useRouter } from 'next/navigation';
+import { useEffect } from 'react';
 
 function VestingDashboardPage(): JSX.Element {
     const account = useCurrentAccount();
     const queryClient = useQueryClient();
     const iotaClient = useIotaClient();
+    const router = useRouter();
     const { addNotification } = useNotifications();
     const { openPopup, closePopup } = usePopups();
     const { data: currentEpochMs } = useGetCurrentEpochStartTimestamp();
@@ -39,6 +44,8 @@ function VestingDashboardPage(): JSX.Element {
     const { data: timelockedStakedObjects } = useGetTimelockedStakedObjects(account?.address || '');
     const { mutateAsync: signAndExecuteTransaction } = useSignAndExecuteTransaction();
 
+    const supplyIncreaseVestingEnabled = useFeature<boolean>(Feature.SupplyIncreaseVesting).value;
+
     const timelockedMapped = mapTimelockObjects(timelockedObjects || []);
     const timelockedstakedMapped = formatDelegatedTimelockedStake(timelockedStakedObjects || []);
 
@@ -134,6 +141,12 @@ function VestingDashboardPage(): JSX.Element {
         );
     }
 
+    useEffect(() => {
+        if (!supplyIncreaseVestingEnabled) {
+            router.push('/');
+        }
+    }, [router, supplyIncreaseVestingEnabled]);
+
     return (
         <div className="flex flex-row">
             <div className="flex w-1/2 flex-col items-center justify-center space-y-4 pt-12">
diff --git a/apps/wallet-dashboard/app/layout.tsx b/apps/wallet-dashboard/app/layout.tsx
index 328f5fa085e..a4c71f018d6 100644
--- a/apps/wallet-dashboard/app/layout.tsx
+++ b/apps/wallet-dashboard/app/layout.tsx
@@ -5,17 +5,18 @@
 import { Inter } from 'next/font/google';
 
 import './globals.css';
-
+import { GrowthBookProvider } from '@growthbook/growthbook-react';
 import { IotaClientProvider, lightTheme, darkTheme, WalletProvider } from '@iota/dapp-kit';
 import { getAllNetworks, getDefaultNetwork } from '@iota/iota-sdk/client';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
 import React from 'react';
-
 import '@iota/dapp-kit/dist/index.css';
 import { Popup, PopupProvider } from '@/components/Popup';
+import { growthbook } from '@/lib/utils';
 import { Toaster } from 'react-hot-toast';
 
 const inter = Inter({ subsets: ['latin'] });
+growthbook.init();
 
 export default function RootLayout({
     children,
@@ -31,28 +32,29 @@ export default function RootLayout({
     return (
         <html lang="en">
             <body className={inter.className} ref={bodyRef}>
-                <QueryClientProvider client={queryClient}>
-                    <IotaClientProvider networks={allNetworks} defaultNetwork={defaultNetwork}>
-                        <WalletProvider
-                            autoConnect={true}
-                            theme={[
-                                {
-                                    variables: lightTheme,
-                                },
-                                {
-                                    selector: '.dark',
-                                    variables: darkTheme,
-                                },
-                            ]}
-                        >
-                            <PopupProvider>
-                                {children}
-                                <Toaster />
-                                <Popup />
-                            </PopupProvider>
-                        </WalletProvider>
-                    </IotaClientProvider>
-                </QueryClientProvider>
+                <GrowthBookProvider growthbook={growthbook}>
+                    <QueryClientProvider client={queryClient}>
+                        <IotaClientProvider networks={allNetworks} defaultNetwork={defaultNetwork}>
+                            <WalletProvider
+                                theme={[
+                                    {
+                                        variables: lightTheme,
+                                    },
+                                    {
+                                        selector: '.dark',
+                                        variables: darkTheme,
+                                    },
+                                ]}
+                            >
+                                <PopupProvider>
+                                    {children}
+                                    <Toaster />
+                                    <Popup />
+                                </PopupProvider>
+                            </WalletProvider>
+                        </IotaClientProvider>
+                    </QueryClientProvider>
+                </GrowthBookProvider>
             </body>
         </html>
     );
diff --git a/apps/wallet-dashboard/lib/utils/growthbook.ts b/apps/wallet-dashboard/lib/utils/growthbook.ts
new file mode 100644
index 00000000000..ee2eed0addf
--- /dev/null
+++ b/apps/wallet-dashboard/lib/utils/growthbook.ts
@@ -0,0 +1,11 @@
+// Copyright (c) 2024 IOTA Stiftung
+// SPDX-License-Identifier: Apache-2.0
+
+import { GrowthBook } from '@growthbook/growthbook';
+import { getAppsBackend } from '@iota/iota-sdk/client';
+
+export const growthbook = new GrowthBook({
+    apiHost: getAppsBackend(),
+    clientKey: process.env.NODE_ENV === 'production' ? 'production' : 'development',
+    enableDevMode: process.env.NODE_ENV === 'development',
+});
diff --git a/apps/wallet-dashboard/lib/utils/index.ts b/apps/wallet-dashboard/lib/utils/index.ts
index d52fd5f097b..78b0f2eb9c5 100644
--- a/apps/wallet-dashboard/lib/utils/index.ts
+++ b/apps/wallet-dashboard/lib/utils/index.ts
@@ -6,3 +6,4 @@ export * from './vesting';
 export * from './time';
 export * from './timelock';
 export * from './transaction';
+export * from './growthbook';
diff --git a/apps/wallet-dashboard/package.json b/apps/wallet-dashboard/package.json
index e8a4a107e2f..9093f6da53e 100644
--- a/apps/wallet-dashboard/package.json
+++ b/apps/wallet-dashboard/package.json
@@ -15,6 +15,8 @@
         "node": ">= v18.17.0"
     },
     "dependencies": {
+        "@growthbook/growthbook": "^1.0.0",
+        "@growthbook/growthbook-react": "^1.0.0",
         "@iota/apps-ui-kit": "workspace:*",
         "@iota/core": "workspace:*",
         "@iota/dapp-kit": "workspace:*",
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 92e2c8a9a2f..d987ac3fc3a 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -991,6 +991,12 @@ importers:
 
   apps/wallet-dashboard:
     dependencies:
+      '@growthbook/growthbook':
+        specifier: ^1.0.0
+        version: 1.2.0
+      '@growthbook/growthbook-react':
+        specifier: ^1.0.0
+        version: 1.2.0(react@18.3.1)
       '@iota/apps-ui-kit':
         specifier: workspace:*
         version: link:../ui-kit