Feat UI tweaks

frontend/app/src/components/molecules/nav-bar/banner.tsx

@@ -0,0 +1,57 @@
+import React from 'react';
+import { Button } from '@/components/ui/button';
+
+export interface BannerProps {
+  message: React.ReactNode;
+  type?: 'info' | 'warning' | 'success' | 'error';
+  actionText?: string;
+  onAction?: () => void;
+  showBanner?: boolean;
+}
+
+export const Banner: React.FC<BannerProps> = ({
+  message,
+  type = 'info',
+  actionText,
+  onAction,
+  showBanner = true,
+}) => {
+  if (!showBanner) {
+    return null;
+  }
+
+  const getBgColor = () => {
+    switch (type) {
+      case 'warning':
+        return 'bg-amber-50 dark:bg-amber-900/20 text-amber-800 dark:text-amber-200';
+      case 'success':
+        return 'bg-green-50 dark:bg-green-900/20 text-green-800 dark:text-green-200';
+      case 'error':
+        return 'bg-red-50 dark:bg-red-900/20 text-red-800 dark:text-red-200';
+      default:
+        return 'bg-blue-50 dark:bg-blue-900/20 text-blue-800 dark:text-blue-200';
+    }
+  };
+
+  return (
+    <div className={`w-full py-2 px-4 h-12 ${getBgColor()}`}>
+      <div className="flex items-center justify-between">
+        <div className="flex flex-1 items-center">
+          <p className="text-sm font-medium">{message}</p>
+        </div>
+        <div className="flex items-center gap-2">
+          {actionText && onAction && (
+            <Button
+              variant="outline"
+              size="sm"
+              onClick={onAction}
+              className="text-xs font-medium hover:bg-opacity-20"
+            >
+              {actionText}
+            </Button>
+          )}
+        </div>
+      </div>
+    </div>
+  );
+};

frontend/app/src/components/molecules/nav-bar/nav-bar.tsx

@@ -17,7 +17,7 @@ import {
   useNavigate,
   useSearchParams,
 } from 'react-router-dom';
-import api, { TenantVersion, User } from '@/lib/api';
+import api, { Tenant, TenantVersion, User } from '@/lib/api';
 import { useApiError } from '@/lib/hooks';
 import { useMutation } from '@tanstack/react-query';
 import hatchet from '@/assets/hatchet_logo.png';
@@ -33,13 +33,14 @@ import {
   BiUserCircle,
 } from 'react-icons/bi';
 import { useTheme } from '@/components/theme-provider';
-import { useMemo } from 'react';
+import { useEffect, useMemo } from 'react';
 import useApiMeta from '@/pages/auth/hooks/use-api-meta';
 import { VersionInfo } from '@/pages/main/info/components/version-info';
 import { useTenant } from '@/lib/atoms';
 import { routes } from '@/router';
 import { TooltipContent, TooltipProvider } from '@/components/ui/tooltip';
 import { Tooltip, TooltipTrigger } from '@/components/ui/tooltip';
+import { Banner, BannerProps } from './banner'; // Import the new Banner component
 
 function HelpDropdown() {
   const meta = useApiMeta();
@@ -105,7 +106,7 @@ function HelpDropdown() {
   );
 }
 
-function AccountDropdown({ user }: MainNavProps) {
+function AccountDropdown({ user, tenant }: MainNavProps) {
   const navigate = useNavigate();
   const { handleApiError } = useApiError({});
 
@@ -148,6 +149,12 @@ function AccountDropdown({ user }: MainNavProps) {
         <DropdownMenuItem>
           <VersionInfo />
         </DropdownMenuItem>
+        {tenant?.version == TenantVersion.V1 &&
+          location.pathname.includes('v1') && (
+            <DropdownMenuItem onClick={() => navigate('/')}>
+              View Legacy V0 Data
+            </DropdownMenuItem>
+          )}
         <DropdownMenuSeparator />
         <DropdownMenuItem onClick={() => toggleTheme()}>
           Toggle Theme
@@ -215,29 +222,89 @@ function VersionUpgradeButton() {
 
 interface MainNavProps {
   user: User;
+  tenant?: Tenant;
+  bannerProps?: BannerProps;
+  setHasBanner?: (state: boolean) => void;
 }
 
-export default function MainNav({ user }: MainNavProps) {
+export default function MainNav({ user, setHasBanner }: MainNavProps) {
   const { toggleSidebarOpen } = useSidebar();
   const { theme } = useTheme();
+  const { tenant } = useTenant();
+  const { pathname } = useLocation();
+  const navigate = useNavigate();
+  const [params] = useSearchParams();
+
+  const versionedRoutes = useMemo(
+    () =>
+      routes
+        .at(0)
+        ?.children?.find((r) => r.path === '/v1/')
+        ?.children?.find((r) => r.path === '/v1/' && r.children?.length)
+        ?.children?.map((c) => c.path)
+        ?.map((p) => p?.replace('/v1', '')) || [],
+    [],
+  );
+
+  const tenantVersion = tenant?.version || TenantVersion.V0;
+
+  const banner: BannerProps | undefined = useMemo(() => {
+    const shouldShowVersionUpgradeButton =
+      versionedRoutes.includes(pathname) && // It is a versioned route
+      !pathname.includes('/v1') && // The user is not already on the v1 version
+      tenantVersion === TenantVersion.V1; // The tenant is on the v1 version
+
+    if (shouldShowVersionUpgradeButton) {
+      return {
+        message: (
+          <>
+            You are viewing legacy V0 data for a tenant that was upgraded to V1
+            runtime.
+          </>
+        ),
+        type: 'warning',
+        actionText: 'Upgrade',
+        onAction: () => {
+          navigate({
+            pathname: '/v1' + pathname,
+            search: params.toString(),
+          });
+        },
+      };
+    }
+
+    return;
+  }, [navigate, params, pathname, tenantVersion, versionedRoutes]);
+
+  useEffect(() => {
+    if (!setHasBanner) {
+      return;
+    }
+    setHasBanner(!!banner);
+  }, [setHasBanner, banner]);
 
   return (
-    <div className="fixed top-0 w-screen h-16 border-b">
-      <div className="flex h-16 items-center pr-4 pl-4">
-        <button
-          onClick={() => toggleSidebarOpen()}
-          className="flex flex-row gap-4 items-center"
-        >
-          <img
-            src={theme == 'dark' ? hatchet : hatchetDark}
-            alt="Hatchet"
-            className="h-9 rounded"
-          />
-        </button>
-        <div className="ml-auto flex items-center">
-          <HelpDropdown />
-          <AccountDropdown user={user} />
-          <VersionUpgradeButton />
+    <div className="fixed top-0 w-screen">
+      {banner && <Banner {...banner} />}
+
+      {/* Main Navigation Bar */}
+      <div className="h-16 border-b">
+        <div className="flex h-16 items-center pr-4 pl-4">
+          <button
+            onClick={() => toggleSidebarOpen()}
+            className="flex flex-row gap-4 items-center"
+          >
+            <img
+              src={theme == 'dark' ? hatchet : hatchetDark}
+              alt="Hatchet"
+              className="h-9 rounded"
+            />
+          </button>
+          <div className="ml-auto flex items-center">
+            <HelpDropdown />
+            <AccountDropdown user={user} tenant={tenant} />
+            <VersionUpgradeButton />
+          </div>
         </div>
       </div>
     </div>

frontend/app/src/lib/atoms.ts

@@ -1,7 +1,7 @@
 import { atom, useAtom } from 'jotai';
-import { Tenant, queries } from './api';
-import { useSearchParams } from 'react-router-dom';
-import { useCallback, useEffect, useMemo } from 'react';
+import { Tenant, TenantVersion, queries } from './api';
+import { useLocation, useNavigate, useSearchParams } from 'react-router-dom';
+import { useCallback, useEffect, useMemo, useState } from 'react';
 import { useQuery } from '@tanstack/react-query';
 
 const getInitialValue = <T>(key: string, defaultValue?: T): T | undefined => {
@@ -121,6 +121,40 @@ export function useTenant(): TenantContext {
     }
   }, [searchParams, tenant, setTenant]);
 
+  // Set the correct path for tenant version
+  // NOTE: this is hacky and not ideal
+
+  const { pathname } = useLocation();
+  const navigate = useNavigate();
+  const [params] = useSearchParams();
+
+  const [lastRedirected, setLastRedirected] = useState<string | undefined>();
+
+  useEffect(() => {
+    // Only redirect on initial tenant load
+    if (lastRedirected == tenant?.slug) {
+      return;
+    }
+
+    setLastRedirected(tenant?.slug);
+
+    if (tenant?.version == TenantVersion.V0 && pathname.startsWith('/v1')) {
+      return navigate({
+        pathname: pathname.replace('/v1', ''),
+        search: params.toString(),
+      });
+    }
+
+    if (tenant?.version == TenantVersion.V1 && !pathname.startsWith('/v1')) {
+      return navigate({
+        pathname: '/v1' + pathname,
+        search: params.toString(),
+      });
+    }
+
+    console.log(tenant?.version);
+  }, [lastRedirected, navigate, params, pathname, tenant]);
+
   if (!tenant) {
     return {
       tenant: undefined,

frontend/app/src/pages/authenticated.tsx

@@ -12,6 +12,7 @@ import { Loading } from '@/components/ui/loading.tsx';
 import { useQuery } from '@tanstack/react-query';
 import SupportChat from '@/components/molecules/support-chat';
 import AnalyticsProvider from '@/components/molecules/analytics-provider';
+import { useState } from 'react';
 
 const authMiddleware = async (currentUrl: string) => {
   try {
@@ -104,6 +105,8 @@ export default function Authenticated() {
     memberships: listMembershipsQuery.data?.rows || memberships,
   });
 
+  const [hasHasBanner, setHasBanner] = useState(false);
+
   if (!user || !memberships) {
     return <Loading />;
   }
@@ -112,8 +115,10 @@ export default function Authenticated() {
     <AnalyticsProvider user={user}>
       <SupportChat user={user}>
         <div className="flex flex-row flex-1 w-full h-full">
-          <MainNav user={user} />
-          <div className="pt-16 flex-grow overflow-y-auto overflow-x-hidden">
+          <MainNav user={user} setHasBanner={setHasBanner} />
+          <div
+            className={`pt-${hasHasBanner ? 28 : 16} flex-grow overflow-y-auto overflow-x-hidden`}
+          >
             <Outlet context={ctx} />
           </div>
         </div>

frontend/app/src/pages/onboarding/verify-email/index.tsx

@@ -43,7 +43,7 @@ export default function VerifyEmail() {
   }
 
   return (
-    <div className="flex flex-row flex-1 w-full h-full pt-16">
+    <div className="flex flex-row flex-1 w-full h-full">
       <MainNav user={res.user} />
       <div className="container relative hidden flex-col items-center justify-center md:grid lg:max-w-none lg:grid-cols-2 lg:px-0">
         <div className="lg:p-8 mx-auto w-screen">