diff --git a/app/dashboard/company/layout.tsx b/app/dashboard/company/layout.tsx index a799fbe3..690a6e6a 100644 --- a/app/dashboard/company/layout.tsx +++ b/app/dashboard/company/layout.tsx @@ -6,7 +6,7 @@ import { useParams } from 'next/navigation' import { Button } from '@/components/ui/button' import { CompanySidebar } from '@/components/dashboard/CompanySidebar' import { CompanyHeader } from '@/components/dashboard/CompanyHeader' -import { CompanyProvider } from '@/contexts/CompanyContext' +import { CompanyProvider, useCompanyContext } from '@/contexts/CompanyContext' import { LayoutDashboard, Calendar, @@ -110,6 +110,7 @@ export default function CompanyDashboardLayout({ ) } +// Component that uses CompanyContext to conditionally render sidebar function CompanyDashboardContent({ avatar, name, @@ -123,6 +124,21 @@ function CompanyDashboardContent({ }) { const params = useParams() const companySlug = params?.slug as string + const { currentCompany, loading, error } = useCompanyContext() + + // Show loading while company context is loading + if (loading) { + return ( +