"use client";

import React from "react";
import Link from "next/link";
import { useParams, usePathname } from "next/navigation";
import { cn } from "@app/lib/cn";
import { useUserContext } from "@app/hooks/useUserContext";
import { Badge } from "@app/components/ui/badge";
import { useLicenseStatusContext } from "@app/hooks/useLicenseStatusContext";
import { useTranslations } from "next-intl";
import {
    Tooltip,
    TooltipContent,
    TooltipProvider,
    TooltipTrigger
} from "@app/components/ui/tooltip";
import {
    Collapsible,
    CollapsibleContent,
    CollapsibleTrigger
} from "@app/components/ui/collapsible";
import {
    Popover,
    PopoverContent,
    PopoverTrigger
} from "@app/components/ui/popover";
import { ChevronRight } from "lucide-react";
import { build } from "@server/build";

export type SidebarNavItem = {
    href?: string;
    title: string;
    icon?: React.ReactNode;
    showEE?: boolean;
    isBeta?: boolean;
    items?: SidebarNavItem[];
};

export type SidebarNavSection = {
    heading: string;
    items: SidebarNavItem[];
};

export interface SidebarNavProps extends React.HTMLAttributes<HTMLElement> {
    sections: SidebarNavSection[];
    disabled?: boolean;
    onItemClick?: () => void;
    isCollapsed?: boolean;
    notificationCounts?: Record<string, number | undefined>;
}

type CollapsibleNavItemProps = {
    item: SidebarNavItem;
    level: number;
    isActive: boolean;
    isChildActive: boolean;
    isDisabled: boolean;
    isCollapsed: boolean;
    renderNavItem: (item: SidebarNavItem, level: number) => React.ReactNode;
    t: (key: string) => string;
    build: string;
    isUnlocked: () => boolean;
    getNotificationCount: (item: SidebarNavItem) => number | undefined;
};

function CollapsibleNavItem({
    item,
    level,
    isActive,
    isChildActive,
    isDisabled,
    isCollapsed,
    renderNavItem,
    t,
    build,
    isUnlocked,
    getNotificationCount
}: CollapsibleNavItemProps) {
    const notificationCount = getNotificationCount(item);
    const storageKey = `pangolin-sidebar-expanded-${item.title}`;

    // Get initial state from localStorage or use isChildActive
    const getInitialState = (): boolean => {
        if (typeof window === "undefined") {
            return isChildActive;
        }
        const saved = localStorage.getItem(storageKey);
        if (saved !== null) {
            return saved === "true";
        }
        return isChildActive;
    };

    const [isOpen, setIsOpen] = React.useState(getInitialState);

    // Update open state when child active state changes (but don't override user preference)
    React.useEffect(() => {
        if (isChildActive) {
            setIsOpen(true);
        }
    }, [isChildActive]);

    // Save state to localStorage when it changes
    const handleOpenChange = (open: boolean) => {
        setIsOpen(open);
        if (typeof window !== "undefined") {
            localStorage.setItem(storageKey, String(open));
        }
    };

    return (
        <Collapsible
            key={item.title}
            open={isOpen}
            onOpenChange={handleOpenChange}
            className="group/collapsible"
        >
            <CollapsibleTrigger asChild>
                <button
                    className={cn(
                        "flex items-center w-full rounded-md transition-colors",
                        level === 0 ? "px-3 py-1.5" : "px-3 py-1",
                        isActive
                            ? "bg-secondary font-medium"
                            : "text-muted-foreground hover:bg-secondary/80 dark:hover:bg-secondary/50 hover:text-foreground",
                        isDisabled && "cursor-not-allowed opacity-60"
                    )}
                    disabled={isDisabled}
                >
                    {item.icon && (
                        <span className="flex-shrink-0 mr-3 w-5 h-5 flex items-center justify-center">
                            {item.icon}
                        </span>
                    )}
                    <div className="flex items-center gap-2 flex-1 min-w-0">
                        <span className="text-left truncate">
                            {t(item.title)}
                        </span>
                        {item.isBeta && (
                            <span className="uppercase font-mono text-yellow-600 dark:text-yellow-800 font-black text-xs">
                                {t("beta")}
                            </span>
                        )}
                    </div>
                    <div className="flex items-center gap-1.5 flex-shrink-0 ml-2">
                        {notificationCount !== undefined &&
                            notificationCount > 0 && (
                                <Badge variant="secondary">
                                    {notificationCount > 99
                                        ? "99+"
                                        : notificationCount}
                                </Badge>
                            )}
                        {build === "enterprise" &&
                            item.showEE &&
                            !isUnlocked() && (
                                <Badge variant="outlinePrimary">
                                    {t("licenseBadge")}
                                </Badge>
                            )}
                        <ChevronRight
                            className={cn(
                                "h-4 w-4 transition-transform duration-300 ease-in-out text-muted-foreground",
                                "group-data-[state=open]/collapsible:rotate-90"
                            )}
                        />
                    </div>
                </button>
            </CollapsibleTrigger>
            <CollapsibleContent>
                <div
                    className={cn(
                        "border-l ml-3 pl-3 mt-0 space-y-0",
                        "border-border"
                    )}
                >
                    {item.items!.map((childItem) =>
                        renderNavItem(childItem, level + 1)
                    )}
                </div>
            </CollapsibleContent>
        </Collapsible>
    );
}

export function SidebarNav({
    className,
    sections,
    disabled = false,
    onItemClick,
    isCollapsed = false,
    notificationCounts,
    ...props
}: SidebarNavProps) {
    const pathname = usePathname();
    const params = useParams();
    const orgId = params.orgId as string;
    const niceId = params.niceId as string;
    const resourceId = params.resourceId as string;
    const userId = params.userId as string;
    const apiKeyId = params.apiKeyId as string;
    const clientId = params.clientId as string;
    const { licenseStatus, isUnlocked } = useLicenseStatusContext();
    const { user } = useUserContext();
    const t = useTranslations();

    function getNotificationCount(item: SidebarNavItem): number | undefined {
        if (!notificationCounts) return undefined;
        return notificationCounts[item.title];
    }

    function hydrateHref(val?: string): string | undefined {
        if (!val) return undefined;
        return val
            .replace("{orgId}", orgId)
            .replace("{niceId}", niceId)
            .replace("{resourceId}", resourceId)
            .replace("{userId}", userId)
            .replace("{apiKeyId}", apiKeyId)
            .replace("{clientId}", clientId);
    }

    function isItemOrChildActive(item: SidebarNavItem): boolean {
        const hydratedHref = hydrateHref(item.href);
        if (hydratedHref && pathname.startsWith(hydratedHref)) {
            return true;
        }
        if (item.items) {
            return item.items.some((child) => isItemOrChildActive(child));
        }
        return false;
    }

    const renderNavItem = (
        item: SidebarNavItem,
        level: number = 0
    ): React.ReactNode => {
        const hydratedHref = hydrateHref(item.href);
        const hasNestedItems = item.items && item.items.length > 0;
        const isActive = hydratedHref
            ? pathname.startsWith(hydratedHref)
            : false;
        const isChildActive = hasNestedItems
            ? isItemOrChildActive(item)
            : false;
        const isEE = build === "enterprise" && item.showEE && !isUnlocked();
        const isDisabled = disabled || isEE;
        const tooltipText =
            item.showEE && !isUnlocked()
                ? `${t(item.title)} (${t("licenseBadge")})`
                : t(item.title);

        // If item has nested items, render as collapsible
        if (hasNestedItems && !isCollapsed) {
            return (
                <CollapsibleNavItem
                    key={item.title}
                    item={item}
                    level={level}
                    isActive={isActive}
                    isChildActive={isChildActive}
                    isDisabled={isDisabled || false}
                    isCollapsed={isCollapsed}
                    renderNavItem={renderNavItem}
                    t={t}
                    build={build}
                    isUnlocked={isUnlocked}
                    getNotificationCount={getNotificationCount}
                />
            );
        }

        const notificationCount = getNotificationCount(item);

        // Regular item without nested items
        const itemContent = hydratedHref ? (
            <Link
                href={isDisabled ? "#" : hydratedHref}
                className={cn(
                    "flex items-center rounded-md transition-colors relative",
                    isCollapsed
                        ? "px-2 py-2 justify-center"
                        : level === 0
                          ? "px-3 py-1.5"
                          : "px-3 py-1",
                    isActive
                        ? "bg-secondary font-medium"
                        : "text-muted-foreground hover:bg-secondary/80 dark:hover:bg-secondary/50 hover:text-foreground",
                    isDisabled && "cursor-not-allowed opacity-60"
                )}
                onClick={(e) => {
                    if (isDisabled) {
                        e.preventDefault();
                    } else if (onItemClick) {
                        onItemClick();
                    }
                }}
                tabIndex={isDisabled ? -1 : undefined}
                aria-disabled={isDisabled}
            >
                {item.icon && (
                    <span
                        className={cn(
                            "flex-shrink-0 w-5 h-5 flex items-center justify-center",
                            !isCollapsed && "mr-3"
                        )}
                    >
                        {item.icon}
                    </span>
                )}
                {!isCollapsed && (
                    <>
                        <div className="flex items-center gap-2 flex-1 min-w-0">
                            <span className="truncate">{t(item.title)}</span>
                            {item.isBeta && (
                                <span className="uppercase font-mono text-yellow-600 dark:text-yellow-800 font-black text-xs">
                                    {t("beta")}
                                </span>
                            )}
                        </div>
                        <div className="flex items-center gap-1.5 flex-shrink-0">
                            {notificationCount !== undefined &&
                                notificationCount > 0 && (
                                    <Badge variant="secondary">
                                        {notificationCount > 99
                                            ? "99+"
                                            : notificationCount}
                                    </Badge>
                                )}
                            {build === "enterprise" &&
                                item.showEE &&
                                !isUnlocked() && (
                                    <Badge
                                        variant="outlinePrimary"
                                        className="flex-shrink-0"
                                    >
                                        {t("licenseBadge")}
                                    </Badge>
                                )}
                        </div>
                    </>
                )}
                {isCollapsed &&
                    notificationCount !== undefined &&
                    notificationCount > 0 && (
                        <Badge
                            variant="secondary"
                            className="absolute -top-1 -right-1 h-5 min-w-5 px-1.5 flex items-center justify-center text-xs"
                        >
                            {notificationCount > 99 ? "99+" : notificationCount}
                        </Badge>
                    )}
            </Link>
        ) : (
            <div
                className={cn(
                    "flex items-center rounded-md transition-colors",
                    level === 0 ? "px-3 py-1.5" : "px-3 py-1",
                    "text-muted-foreground",
                    isDisabled && "cursor-not-allowed opacity-60"
                )}
            >
                {item.icon && (
                    <span className="flex-shrink-0 mr-3 w-5 h-5 flex items-center justify-center">
                        {item.icon}
                    </span>
                )}
                <div className="flex items-center gap-2 flex-1 min-w-0">
                    <span className="truncate">{t(item.title)}</span>
                    {item.isBeta && (
                        <span className="uppercase font-mono text-yellow-600 dark:text-yellow-800 font-black text-xs">
                            {t("beta")}
                        </span>
                    )}
                </div>
                <div className="flex items-center gap-1.5 flex-shrink-0 ml-2">
                    {notificationCount !== undefined &&
                        notificationCount > 0 && (
                            <Badge
                                variant="secondary"
                                className="flex-shrink-0 bg-primary text-primary-foreground"
                            >
                                {notificationCount > 99
                                    ? "99+"
                                    : notificationCount}
                            </Badge>
                        )}
                    {build === "enterprise" && item.showEE && !isUnlocked() && (
                        <Badge
                            variant="outlinePrimary"
                            className="flex-shrink-0"
                        >
                            {t("licenseBadge")}
                        </Badge>
                    )}
                </div>
            </div>
        );

        if (isCollapsed) {
            // If item has nested items, show both tooltip and popover
            if (hasNestedItems) {
                return (
                    <TooltipProvider key={item.title}>
                        <Tooltip>
                            <Popover>
                                <PopoverTrigger asChild>
                                    <TooltipTrigger asChild>
                                        <button
                                            className={cn(
                                                "flex items-center rounded-md transition-colors px-2 py-2 justify-center w-full",
                                                isActive || isChildActive
                                                    ? "bg-secondary font-medium"
                                                    : "text-muted-foreground hover:bg-secondary/80 dark:hover:bg-secondary/50 hover:text-foreground",
                                                isDisabled &&
                                                    "cursor-not-allowed opacity-60"
                                            )}
                                            disabled={isDisabled}
                                        >
                                            {item.icon && (
                                                <span className="flex-shrink-0 w-5 h-5 flex items-center justify-center">
                                                    {item.icon}
                                                </span>
                                            )}
                                        </button>
                                    </TooltipTrigger>
                                </PopoverTrigger>
                                <TooltipContent side="right" sideOffset={8}>
                                    <p>{tooltipText}</p>
                                </TooltipContent>
                                <PopoverContent
                                    side="right"
                                    align="start"
                                    className="w-56 p-1"
                                >
                                    <div className="space-y-1">
                                        {item.items!.map((childItem) => {
                                            const childHydratedHref =
                                                hydrateHref(childItem.href);
                                            const childIsActive =
                                                childHydratedHref
                                                    ? pathname.startsWith(
                                                          childHydratedHref
                                                      )
                                                    : false;
                                            const childIsEE =
                                                build === "enterprise" &&
                                                childItem.showEE &&
                                                !isUnlocked();
                                            const childIsDisabled =
                                                disabled || childIsEE;

                                            if (!childHydratedHref) {
                                                return null;
                                            }

                                            return (
                                                <Link
                                                    key={childItem.title}
                                                    href={
                                                        childIsDisabled
                                                            ? "#"
                                                            : childHydratedHref
                                                    }
                                                    className={cn(
                                                        "flex items-center rounded-md transition-colors px-3 py-1.5 text-sm",
                                                        childIsActive
                                                            ? "bg-secondary font-medium"
                                                            : "text-muted-foreground hover:bg-secondary/50 hover:text-foreground",
                                                        childIsDisabled &&
                                                            "cursor-not-allowed opacity-60"
                                                    )}
                                                    onClick={(e) => {
                                                        if (childIsDisabled) {
                                                            e.preventDefault();
                                                        } else if (
                                                            onItemClick
                                                        ) {
                                                            onItemClick();
                                                        }
                                                    }}
                                                >
                                                    {childItem.icon && (
                                                        <span className="flex-shrink-0 mr-3 w-5 h-5 flex items-center justify-center">
                                                            {childItem.icon}
                                                        </span>
                                                    )}
                                                    <div className="flex items-center gap-2 flex-1 min-w-0">
                                                        <span className="truncate">
                                                            {t(childItem.title)}
                                                        </span>
                                                        {childItem.isBeta && (
                                                            <span className="uppercase font-mono text-yellow-600 dark:text-yellow-800 font-black text-xs">
                                                                {t("beta")}
                                                            </span>
                                                        )}
                                                    </div>
                                                    {build === "enterprise" &&
                                                        childItem.showEE &&
                                                        !isUnlocked() && (
                                                            <Badge
                                                                variant="outlinePrimary"
                                                                className="flex-shrink-0 ml-2"
                                                            >
                                                                {t(
                                                                    "licenseBadge"
                                                                )}
                                                            </Badge>
                                                        )}
                                                </Link>
                                            );
                                        })}
                                    </div>
                                </PopoverContent>
                            </Popover>
                        </Tooltip>
                    </TooltipProvider>
                );
            }

            // Regular item without nested items - show tooltip
            return (
                <TooltipProvider key={item.title}>
                    <Tooltip>
                        <TooltipTrigger asChild>{itemContent}</TooltipTrigger>
                        <TooltipContent side="right" sideOffset={8}>
                            <p>{tooltipText}</p>
                        </TooltipContent>
                    </Tooltip>
                </TooltipProvider>
            );
        }

        return <React.Fragment key={item.title}>{itemContent}</React.Fragment>;
    };

    return (
        <nav
            className={cn(
                "flex flex-col text-sm",
                disabled && "pointer-events-none opacity-60",
                className
            )}
            {...props}
        >
            {sections.map((section, sectionIndex) => (
                <div
                    key={section.heading}
                    className={cn(sectionIndex > 0 && "mt-4")}
                >
                    {!isCollapsed && (
                        <div className="px-3 py-2 text-xs font-medium text-muted-foreground/80 uppercase tracking-wider">
                            {t(`${section.heading}`)}
                        </div>
                    )}
                    <div className="flex flex-col gap-0">
                        {section.items.map((item) => renderNavItem(item, 0))}
                    </div>
                </div>
            ))}
        </nav>
    );
}
