'use client';

import { useState, useEffect } from 'react';
import Link from 'next/link';
import { Button } from '@/components/ui/button';
import { ThemeToggle } from '@/components/shared/ThemeToggle';
import { PageTransition } from '@/components/shared/PageTransition';
import ProtectedRoute from '@/components/shared/ProtectedRoute';
import { useAuth } from '@/components/shared/AuthContext';
import { 
  ChevronLeftIcon, 
  ChevronRightIcon, 
  LayoutDashboardIcon, 
  CoinsIcon, 
  TargetIcon, 
  SettingsIcon,
  LogOutIcon,
  MenuIcon
} from 'lucide-react';
import { usePathname } from 'next/navigation';


export default function MainLayout({
  children,
}: Readonly<{
  children: React.ReactNode;
}>) {
  const pathname = usePathname();
  const { user, logout } = useAuth();
  const [isSidebarCollapsed, setIsSidebarCollapsed] = useState(false);
  const [isMobileMenuOpen, setIsMobileMenuOpen] = useState(false);

  useEffect(() => {
    // Load sidebar state from localStorage
    const savedSidebarState = localStorage.getItem('sidebarCollapsed');
    if (savedSidebarState !== null) {
      setIsSidebarCollapsed(savedSidebarState === 'true');
    }
  }, []);

  // Close mobile menu when changing routes
  useEffect(() => {
    setIsMobileMenuOpen(false);
  }, [pathname]);

  const handleLogout = () => {
    logout();
  };

  const toggleSidebar = () => {
    const newState = !isSidebarCollapsed;
    setIsSidebarCollapsed(newState);
    localStorage.setItem('sidebarCollapsed', newState.toString());
  };

  const toggleMobileMenu = () => {
    setIsMobileMenuOpen(!isMobileMenuOpen);
  };

  return (
    <ProtectedRoute>
      <div className="min-h-screen bg-background flex flex-col">
        {/* Top Navigation */}
        <header className="border-b shadow-sm backdrop-blur-sm bg-background/90 sticky top-0 z-50 w-full">
          <div className="w-full flex h-16 items-center px-4 md:px-6">
            <div className="flex items-center gap-4">
              {/* Mobile menu button - only visible on small screens */}
              <button 
                className="md:hidden rounded-full p-2 hover:bg-muted transition-colors"
                onClick={toggleMobileMenu}
                aria-label="Toggle mobile menu"
              >
                <MenuIcon size={20} />
              </button>
              <div className="font-semibold text-xl">
                <span className="bg-clip-text text-transparent bg-gradient-to-r from-primary to-primary/70">
                  Finance Management
                </span>
              </div>
            </div>
            
            <div className="flex items-center ms-auto gap-4">
              <div className="hidden md:flex items-center gap-1 bg-muted/50 px-3 py-1.5 rounded-full">
                <span className="relative flex h-2 w-2">
                  <span className="animate-ping absolute inline-flex h-full w-full rounded-full bg-primary/50 opacity-75"></span>
                  <span className="relative inline-flex rounded-full h-2 w-2 bg-primary"></span>
                </span>
                <span className='text-sm font-medium ml-1.5 transition-all duration-300'>
                  {user?.Name}
                </span>
              </div>
              <ThemeToggle />
              <Button 
                variant="ghost" 
                size="icon" 
                onClick={handleLogout}
                className="hover:bg-destructive/10 transition-colors duration-300"
                aria-label="Logout"
              >
                <LogOutIcon size={18} className="text-destructive/90 transition-transform hover:scale-110 duration-300" />
              </Button>
            </div>
          </div>
        </header>

        {/* Mobile Menu - outside the normal flow and only visible when toggled */}
        <div 
          className={`
            md:hidden fixed inset-0 z-40 bg-background/95 backdrop-blur-sm transition-all duration-300 
            ${isMobileMenuOpen ? 'opacity-100 pointer-events-auto' : 'opacity-0 pointer-events-none'}
          `}
        >
          <div className="pt-20 px-4">
            <nav className="space-y-4">
              <Link 
                href="/dashboard" 
                className={`
                  flex items-center p-3 rounded-lg transition-colors
                  ${pathname === '/dashboard' ? 'bg-primary/10 text-primary' : 'hover:bg-muted'}
                `}
              >
                <LayoutDashboardIcon size={20} className="mr-3" />
                Dashboard
              </Link>
              <Link 
                href="/loans" 
                className={`
                  flex items-center p-3 rounded-lg transition-colors
                  ${pathname === '/loans' ? 'bg-primary/10 text-primary' : 'hover:bg-muted'}
                `}
              >
                <CoinsIcon size={20} className="mr-3" />
                Loans
              </Link>
              <Link 
                href="/goals" 
                className={`
                  flex items-center p-3 rounded-lg transition-colors
                  ${pathname === '/goals' ? 'bg-primary/10 text-primary' : 'hover:bg-muted'}
                `}
              >
                <TargetIcon size={20} className="mr-3" />
                Goals
              </Link>
              <Link 
                href="/settings" 
                className={`
                  flex items-center p-3 rounded-lg transition-colors
                  ${pathname === '/settings' ? 'bg-primary/10 text-primary' : 'hover:bg-muted'}
                `}
              >
                <SettingsIcon size={20} className="mr-3" />
                Settings
              </Link>
            </nav>
            <div className="mt-8 border-t pt-4">
              <Button 
                variant="outline" 
                className="w-full justify-start text-destructive hover:text-destructive hover:bg-destructive/10"
                onClick={handleLogout}
              >
                <LogOutIcon size={18} className="mr-2" />
                Logout
              </Button>
            </div>
          </div>
        </div>

        {/* Main Content */}
        <div className="flex flex-1 overflow-hidden">
          {/* Sidebar - hidden on mobile */}
          <aside 
            className={`
              hidden md:flex flex-col h-[calc(100vh-4rem)] border-r bg-background flex-shrink-0 relative
              transition-all duration-300 ease-in-out
              ${isSidebarCollapsed ? 'w-16' : 'w-64'}
            `}
          >
            {/* Sidebar Toggle Button */}
            <button 
              onClick={toggleSidebar}
              className="absolute -right-3 top-10 bg-primary text-primary-foreground rounded-full p-1 shadow-md hover:bg-primary/90 transition-colors duration-300 hover:scale-110 group z-10"
              aria-label={isSidebarCollapsed ? "Expand sidebar" : "Collapse sidebar"}
            >
              <div className="transition-transform duration-300 group-hover:rotate-[360deg]">
                {isSidebarCollapsed ? <ChevronRightIcon size={16} /> : <ChevronLeftIcon size={16} />}
              </div>
            </button>

            <nav className="space-y-2 px-2 mt-4 flex-1">
              <Link 
                href="/dashboard" 
                className={`
                  flex items-center p-2 rounded-md transition-all duration-200 overflow-hidden
                  ${pathname === '/dashboard' 
                    ? 'bg-primary/10 text-primary font-medium' 
                    : 'hover:bg-muted text-foreground/80 hover:text-foreground'
                  }
                  ${isSidebarCollapsed ? 'justify-center' : 'justify-start'}
                `}
                title="Dashboard"
              >
                <LayoutDashboardIcon size={18} className={`transition-transform duration-300 ${pathname === '/dashboard' ? 'scale-110' : ''}`} />
                <span className={`ml-2 transition-all duration-300 overflow-hidden whitespace-nowrap ${isSidebarCollapsed ? 'w-0 opacity-0' : 'w-auto opacity-100'}`}>
                  Dashboard
                </span>
              </Link>
              <Link 
                href="/loans" 
                className={`
                  flex items-center p-2 rounded-md transition-all duration-200 overflow-hidden
                  ${pathname === '/loans' 
                    ? 'bg-primary/10 text-primary font-medium' 
                    : 'hover:bg-muted text-foreground/80 hover:text-foreground'
                  }
                  ${isSidebarCollapsed ? 'justify-center' : 'justify-start'}
                `}
                title="Loans"
              >
                <CoinsIcon size={18} className={`transition-transform duration-300 ${pathname === '/loans' ? 'scale-110' : ''}`} />
                <span className={`ml-2 transition-all duration-300 overflow-hidden whitespace-nowrap ${isSidebarCollapsed ? 'w-0 opacity-0' : 'w-auto opacity-100'}`}>
                  Loans
                </span>
              </Link>
              <Link 
                href="/goals" 
                className={`
                  flex items-center p-2 rounded-md transition-all duration-200 overflow-hidden
                  ${pathname === '/goals' 
                    ? 'bg-primary/10 text-primary font-medium' 
                    : 'hover:bg-muted text-foreground/80 hover:text-foreground'
                  }
                  ${isSidebarCollapsed ? 'justify-center' : 'justify-start'}
                `}
                title="Goals"
              >
                <TargetIcon size={18} className={`transition-transform duration-300 ${pathname === '/goals' ? 'scale-110' : ''}`} />
                <span className={`ml-2 transition-all duration-300 overflow-hidden whitespace-nowrap ${isSidebarCollapsed ? 'w-0 opacity-0' : 'w-auto opacity-100'}`}>
                  Goals
                </span>
              </Link>
              <Link 
                href="/settings" 
                className={`
                  flex items-center p-2 rounded-md transition-all duration-200 overflow-hidden
                  ${pathname === '/settings' 
                    ? 'bg-primary/10 text-primary font-medium' 
                    : 'hover:bg-muted text-foreground/80 hover:text-foreground'
                  }
                  ${isSidebarCollapsed ? 'justify-center' : 'justify-start'}
                `}
                title="Settings"
              >
                <SettingsIcon size={18} className={`transition-transform duration-300 ${pathname === '/settings' ? 'scale-110' : ''}`} />
                <span className={`ml-2 transition-all duration-300 overflow-hidden whitespace-nowrap ${isSidebarCollapsed ? 'w-0 opacity-0' : 'w-auto opacity-100'}`}>
                  Settings
                </span>
              </Link>
            </nav>
          </aside>

          {/* Page Content */}
          <main className="flex-1 p-4 md:p-8 overflow-auto h-[calc(100vh-4rem)] transition-all duration-300">
            <PageTransition>
              {children}
            </PageTransition>
          </main>
        </div>
      </div>
    </ProtectedRoute>
  );
}