dyolink/frontend/src/app/(dashboard)/layout.tsx

'use client';

import { memo, useEffect } from 'react';
import { useRouter } from 'next/navigation';
import { useAuth } from '@/lib/hooks/useAuth';
import Sidebar from '@/components/ui/Sidebar';
import { ThemeToggle } from '@/components/ui/ThemeToggle';
import { DashboardAccountMenu } from '@/components/ui/DashboardAccountMenu';

export default function DashboardLayout({ children }: { children: React.ReactNode }) {
  const { user, currentOrganization, isAuthReady } = useAuth();
  const router = useRouter();

  // ✅ AUTH GUARD (runs once per navigation group)
  useEffect(() => {
    if (!isAuthReady) return;

    if (!user) {
      router.replace('/login');
      return;
    }

    if (!currentOrganization) {
      router.replace('/select-organization');
      return;
    }
  }, [isAuthReady, user, currentOrganization, router]);

  // ✅ LOADING ONLY FOR INITIAL LOAD
  if (!isAuthReady) {
    return (
      <div className="h-screen flex items-center justify-center app-web-bg">
        Loading app...
      </div>
    );
  }

  if (!user || !currentOrganization) {
    return (
      <div className="h-screen flex items-center justify-center app-web-bg">
        Loading workspace...
      </div>
    );
  }

  return (
    <div className="flex h-screen app-web-bg text-text-primary">
      <Sidebar />

      <div className="flex-1 flex flex-col">
        <DashboardHeader organizationName={currentOrganization.name} />

        <main className="p-6 flex-1 overflow-y-auto">
          <div className="surface-panel p-6 min-h-full">
            {children}
          </div>
        </main>
      </div>
    </div>
  );
}

const DashboardHeader = memo(function DashboardHeader({
  organizationName,
}: {
  organizationName: string;
}) {
  return (
    <header className="h-[71px] flex justify-between items-center gap-4 px-6 border-b border-border/70 backdrop-blur-sm">
      <h2 className="text-lg font-medium truncate min-w-0">{organizationName}</h2>

      <div className="flex items-center gap-3 shrink-0">
        <ThemeToggle />
        <DashboardAccountMenu />
      </div>
    </header>
  );
});