"use client";

import * as React from "react";
import Link from "next/link";
import { usePathname } from "next/navigation";
import { Menu, ChevronDown } from "lucide-react";
import Image from 'next/image';

import { cn } from "@/lib/utils";
import { navLinks, allProducts } from "@/app/lib/site-data";
import { Button } from "@/components/ui/button";
import { Sheet, SheetContent, SheetTrigger, SheetHeader, SheetTitle } from "@/components/ui/sheet";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuTrigger,
  DropdownMenuLabel,
} from "@/components/ui/dropdown-menu";
import {
  Accordion,
  AccordionContent,
  AccordionItem,
  AccordionTrigger,
} from "@/components/ui/accordion";
import { ThemeToggle } from "./shared/ThemeToggle";

function Logo() {
  return (
      <Image
        src="https://demo.voitekk.com/voitekk/assets/images/voitekkLogo.png"
        alt="Voitekk Logo"
        width={52}
        height={13}
        priority
        className="object-contain"
      />
  );
}

const WhatsAppIcon = ({ className }: { className?: string }) => (
  <svg
    viewBox="0 0 24 24"
    className={cn("fill-current", className)}
    xmlns="http://www.w3.org/2000/svg"
  >
    <path d="M17.472 14.382c-.297-.149-1.758-.867-2.03-.967-.273-.099-.471-.148-.67.15-.197.297-.767.966-.94 1.164-.173.199-.347.223-.644.075-.297-.15-1.255-.463-2.39-1.353-.883-.788-1.48-1.76-1.653-2.059-.173-.297-.018-.458.13-.606.134-.133.298-.347.446-.52.149-.174.198-.298.298-.497.099-.198.05-.371-.025-.52-.05-.149-.669-1.612-.916-2.207-.242-.579-.487-.5-.669-.51-.173-.008-.371-.01-.57-.01-.198 0-.52.074-.792.372-.272.297-1.04 1.016-1.04 2.479 0 1.462 1.065 2.875 1.213 3.074.149.198 2.096 3.2 5.077 4.487.709.306 1.262.489 1.694.625.712.227 1.36.195 1.871.118.571-.085 1.758-.719 2.006-1.413.248-.694.248-1.289.173-1.413-.074-.124-.272-.198-.57-.347m-5.421 7.403h-.004a9.87 9.87 0 01-5.031-1.378l-.361-.214-3.741.982.998-3.648-.235-.374a9.86 9.86 0 01-1.51-5.26c.001-5.45 4.436-9.884 9.888-9.884 2.64 0 5.122 1.03 6.988 2.898a9.825 9.825 0 012.893 6.994c-.003 5.45-4.437 9.884-9.885 9.884m8.413-18.297A11.815 11.815 0 0012.05 0C5.495 0 .16 5.335.157 11.892c0 2.096.547 4.142 1.588 5.945L.057 24l6.305-1.654a11.882 11.882 0 005.683 1.448h.005c6.554 0 11.89-5.335 11.893-11.893a11.821 11.821 0 00-3.48-8.413z" />
  </svg>
);

export function Header() {
  const [isMenuOpen, setIsMenuOpen] = React.useState(false);
  const [isMounted, setIsMounted] = React.useState(false);
  const pathname = usePathname();

  React.useEffect(() => {
    setIsMounted(true);
  }, []);

  const renderNavLink = (link: typeof navLinks[0]) => {
    const isActive = link.href === '/' ? pathname === link.href : pathname.startsWith(link.href);
    const commonClasses = "text-sm font-medium transition-colors";

    if (link.label === "Capabilities") {
      const isProductActive = pathname.startsWith('/products') || pathname.startsWith('/solutions');
      return (
        <DropdownMenu>
          <DropdownMenuTrigger
            className={cn(
              commonClasses,
              "flex items-center gap-1 outline-none",
              isProductActive && isMounted
                ? "text-primary"
                : "text-foreground/60 hover:text-foreground/80"
            )}
          >
            {link.label}
            <ChevronDown className="h-4 w-4" />
          </DropdownMenuTrigger>
          <DropdownMenuContent>
            <DropdownMenuLabel>All Capabilities</DropdownMenuLabel>
            {allProducts.map((product) => (
              <DropdownMenuItem key={product.href} asChild>
                <Link href={product.href}>{product.title}</Link>
              </DropdownMenuItem>
            ))}
          </DropdownMenuContent>
        </DropdownMenu>
      );
    }
    
    return (
      <Link
        href={link.href}
        className={cn(
          commonClasses,
          "flex items-center gap-1",
          isActive && isMounted
            ? "text-primary"
            : "text-foreground/60 hover:text-foreground/80"
        )}
      >
        {link.label}
      </Link>
    );
  };
  
  const renderMobileNavLink = (link: typeof navLinks[0]) => {
      const commonClasses = "text-lg font-medium";
      const isActive = link.href === '/' ? pathname === link.href : pathname.startsWith(link.href);
      
      if (link.label === 'Capabilities') {
          return (
              <Accordion type="single" collapsible className="w-full">
                  <AccordionItem value="products" className="border-b-0">
                      <AccordionTrigger className={cn(
                          commonClasses,
                           "hover:no-underline",
                           (pathname.startsWith('/products') || pathname.startsWith('/solutions')) && isMounted ? "text-primary" : "text-foreground/70 hover:text-foreground"
                      )}>
                          {link.label}
                      </AccordionTrigger>
                      <AccordionContent className="pl-4">
                          <div className="flex flex-col gap-2">
                             <h4 className="font-semibold text-sm text-muted-foreground mt-4">All Capabilities</h4>
                            {allProducts.map((product) => (
                                <Link
                                    key={product.href}
                                    href={product.href}
                                    onClick={() => setIsMenuOpen(false)}
                                    className="text-base text-foreground/60 hover:text-foreground"
                                >
                                    {product.title}
                                </Link>
                            ))}
                          </div>
                      </AccordionContent>
                  </AccordionItem>
              </Accordion>
          )
      }

      return (
          <Link
            href={link.href}
            onClick={() => setIsMenuOpen(false)}
            className={cn(
              commonClasses,
              isActive && isMounted
                ? "text-primary"
                : "text-foreground/70 hover:text-foreground"
            )}
          >
            {link.label}
          </Link>
      )
  }

  return (
    <header className="sticky top-0 z-50 w-full border-b bg-background/50 backdrop-blur-lg supports-[backdrop-filter]:bg-background/50">
      <div className="container flex h-20 max-w-7xl items-center justify-between px-4 sm:px-6 lg:px-8">
        <Link href="/" className="flex items-center gap-2">
          <Logo />
          <div>
            <span className="font-bold text-3xl text-foreground">Voitekk</span>
            <span className="block text-xs text-muted-foreground -mt-1">Transforming Communication</span>
          </div>
        </Link>
        
        {/* Mobile Nav */}
        <div className="lg:hidden flex items-center gap-2">
           <ThemeToggle />
           <a 
              href="https://wa.me/919773122250" 
              target="_blank" 
              rel="noopener noreferrer" 
              className="text-[#25D366] hover:text-[#128C7E] transition-colors"
              aria-label="Contact us on WhatsApp"
            >
              <WhatsAppIcon className="h-8 w-8" />
            </a>
          <Sheet open={isMenuOpen} onOpenChange={setIsMenuOpen}>
            <SheetTrigger asChild>
              <Button variant="ghost" size="icon">
                <Menu className="h-6 w-6" />
                <span className="sr-only">Open Menu</span>
              </Button>
            </SheetTrigger>
            <SheetContent side="right" className="w-[240px] pt-10">
              <SheetHeader>
                  <SheetTitle className="sr-only">Mobile Navigation Menu</SheetTitle>
              </SheetHeader>
              <div className="flex flex-col gap-4">
                {navLinks.map((link) => (
                    <div key={`${link.href}-${link.label}`}>
                        {renderMobileNavLink(link)}
                    </div>
                ))}
                 <div className="border-t pt-4 mt-4 space-y-2">
                    <Button asChild className="w-full">
                        <Link href="/contact">Contact Us</Link>
                    </Button>
                </div>
              </div>
            </SheetContent>
          </Sheet>
        </div>

        {/* Desktop Nav */}
        <div className="hidden lg:flex items-center gap-x-8">
          <nav className="flex items-center gap-x-6">
            {navLinks.map((link) => (
                <React.Fragment key={`${link.href}-${link.label}`}>
                    {renderNavLink(link)}
                </React.Fragment>
            ))}
          </nav>
        
          <div className="flex items-center gap-4">
            <Button asChild size="sm" variant="outline" className="border-primary/20 hover:bg-primary/5 text-primary">
                <Link href="/contact">Contact Us</Link>
            </Button>
            <ThemeToggle />
            <a 
              href="https://wa.me/919773122250" 
              target="_blank" 
              rel="noopener noreferrer" 
              className="text-[#25D366] hover:text-[#128C7E] transition-colors"
              aria-label="Contact us on WhatsApp"
            >
              <WhatsAppIcon className="h-8 w-8" />
            </a>
          </div>
        </div>
      </div>
    </header>
  );
}