rhl-lieferscheine/components/app-shell/QuickNav.jsx

"use client";

import React from "react";
import Link from "next/link";
import { usePathname, useRouter } from "next/navigation";
import { FolderOpen, Search as SearchIcon } from "lucide-react";

import { useAuth } from "@/components/auth/authContext";
import { getBranches } from "@/lib/frontend/apiClient";
import { branchPath, searchPath } from "@/lib/frontend/routes";
import { isValidBranchParam } from "@/lib/frontend/params";
import {
	buildNextUrlForBranchSwitch,
	readRouteBranchFromPathname,
	safeReadLocalStorageBranch,
	safeWriteLocalStorageBranch,
} from "@/lib/frontend/quickNav/branchSwitch";
import {
	getPrimaryNavFromPathname,
	PRIMARY_NAV,
} from "@/lib/frontend/nav/activeRoute";

import { Button } from "@/components/ui/button";
import {
	DropdownMenu,
	DropdownMenuContent,
	DropdownMenuLabel,
	DropdownMenuRadioGroup,
	DropdownMenuRadioItem,
	DropdownMenuSeparator,
	DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";

const STORAGE_KEY_LAST_BRANCH = "rhl_last_branch";

const BRANCH_LIST_STATE = Object.freeze({
	IDLE: "idle",
	LOADING: "loading",
	READY: "ready",
	ERROR: "error",
});

export default function QuickNav() {
	const router = useRouter();
	const pathname = usePathname() || "/";

	const { status, user } = useAuth();

	const isAuthenticated = status === "authenticated" && user;
	const isAdminDev =
		isAuthenticated && (user.role === "admin" || user.role === "dev");
	const isBranchUser = isAuthenticated && user.role === "branch";

	const [selectedBranch, setSelectedBranch] = React.useState(null);

	const [branchList, setBranchList] = React.useState({
		status: BRANCH_LIST_STATE.IDLE,
		branches: null,
	});

	const activePrimaryNav = React.useMemo(() => {
		return getPrimaryNavFromPathname(pathname);
	}, [pathname]);

	const isExplorerActive = activePrimaryNav?.active === PRIMARY_NAV.EXPLORER;
	const isSearchActive = activePrimaryNav?.active === PRIMARY_NAV.SEARCH;

	React.useEffect(() => {
		if (!isAuthenticated) return;

		// Branch users: selection is fixed to their own branch.
		if (isBranchUser) {
			const own = user.branchId;
			setSelectedBranch(own && isValidBranchParam(own) ? own : null);
			return;
		}

		// Admin/dev: prefer current route branch, fallback to last-used localStorage.
		const fromRoute = readRouteBranchFromPathname(pathname);
		const fromStorage = safeReadLocalStorageBranch(STORAGE_KEY_LAST_BRANCH);

		const initial = fromRoute || fromStorage || null;

		// Avoid unnecessary state updates.
		if (initial && initial !== selectedBranch) {
			setSelectedBranch(initial);
			safeWriteLocalStorageBranch(STORAGE_KEY_LAST_BRANCH, initial);
		}
	}, [isAuthenticated, isBranchUser, user?.branchId, pathname, selectedBranch]);

	React.useEffect(() => {
		// Fetch the branch list once for admin/dev users (or when the user changes),
		// not on every selectedBranch change.
		if (!isAdminDev) return;

		let cancelled = false;

		setBranchList({ status: BRANCH_LIST_STATE.LOADING, branches: null });

		(async () => {
			try {
				const res = await getBranches();
				if (cancelled) return;

				const branches = Array.isArray(res?.branches) ? res.branches : [];
				setBranchList({ status: BRANCH_LIST_STATE.READY, branches });
			} catch (err) {
				if (cancelled) return;

				console.error("[QuickNav] getBranches failed:", err);
				setBranchList({ status: BRANCH_LIST_STATE.ERROR, branches: null });
			}
		})();

		return () => {
			cancelled = true;
		};
	}, [isAdminDev, user?.userId]);

	React.useEffect(() => {
		// After we have the branch list, ensure selectedBranch is valid and known.
		// This effect does NOT trigger any refetches (only local state).
		if (!isAdminDev) return;
		if (branchList.status !== BRANCH_LIST_STATE.READY) return;

		const branches = Array.isArray(branchList.branches)
			? branchList.branches
			: [];
		if (branches.length === 0) return;

		if (!selectedBranch || !branches.includes(selectedBranch)) {
			const next = branches[0];
			setSelectedBranch(next);
			safeWriteLocalStorageBranch(STORAGE_KEY_LAST_BRANCH, next);
		}
	}, [isAdminDev, branchList.status, branchList.branches, selectedBranch]);

	React.useEffect(() => {
		if (!isAdminDev) return;
		if (!selectedBranch) return;

		safeWriteLocalStorageBranch(STORAGE_KEY_LAST_BRANCH, selectedBranch);
	}, [isAdminDev, selectedBranch]);

	if (!isAuthenticated) return null;

	const effectiveBranch = isBranchUser ? user.branchId : selectedBranch;
	const canNavigate = Boolean(
		effectiveBranch && isValidBranchParam(effectiveBranch),
	);

	function navigateToBranchKeepingContext(nextBranch) {
		if (!isValidBranchParam(nextBranch)) return;

		// IMPORTANT:
		// Avoid useSearchParams() here to prevent build-time prerender failures on static routes.
		// We only need the current query string at click-time (client-only), so window is fine.
		const currentPathname =
			typeof window !== "undefined"
				? window.location.pathname || pathname
				: pathname;

		const currentSearch =
			typeof window !== "undefined" ? window.location.search || "" : "";

		const nextUrl = buildNextUrlForBranchSwitch({
			pathname: currentPathname,
			search: currentSearch,
			nextBranch,
		});

		if (!nextUrl) return;

		// Client navigation: keeps providers mounted and avoids hard reload flicker.
		router.push(nextUrl);
	}

	const explorerVariant = isExplorerActive ? "secondary" : "ghost";
	const searchVariant = isSearchActive ? "secondary" : "ghost";

	return (
		<div className="hidden items-center gap-2 md:flex">
			{isAdminDev ? (
				<DropdownMenu>
					<DropdownMenuTrigger asChild>
						<Button
							variant="outline"
							size="sm"
							type="button"
							title="Niederlassung auswählen"
						>
							{canNavigate ? effectiveBranch : "Niederlassung wählen"}
						</Button>
					</DropdownMenuTrigger>

					<DropdownMenuContent align="end" className="min-w-56">
						<DropdownMenuLabel>Niederlassung</DropdownMenuLabel>
						<DropdownMenuSeparator />

						{branchList.status === BRANCH_LIST_STATE.ERROR ? (
							<div className="px-2 py-2 text-xs text-muted-foreground">
								Konnte nicht geladen werden.
							</div>
						) : (
							<DropdownMenuRadioGroup
								value={canNavigate ? effectiveBranch : ""}
								onValueChange={(value) => {
									if (!value) return;
									if (!isValidBranchParam(value)) return;

									setSelectedBranch(value);
									safeWriteLocalStorageBranch(STORAGE_KEY_LAST_BRANCH, value);

									navigateToBranchKeepingContext(value);
								}}
							>
								{(Array.isArray(branchList.branches)
									? branchList.branches
									: []
								).map((b) => (
									<DropdownMenuRadioItem key={b} value={b}>
										{b}
									</DropdownMenuRadioItem>
								))}
							</DropdownMenuRadioGroup>
						)}
					</DropdownMenuContent>
				</DropdownMenu>
			) : null}

			<Button
				variant={explorerVariant}
				size="sm"
				asChild
				disabled={!canNavigate}
			>
				<Link
					href={canNavigate ? branchPath(effectiveBranch) : "#"}
					title="Explorer öffnen"
					aria-current={isExplorerActive ? "page" : undefined}
				>
					<FolderOpen className="h-4 w-4" />
					Explorer
				</Link>
			</Button>

			<Button variant={searchVariant} size="sm" asChild disabled={!canNavigate}>
				<Link
					href={canNavigate ? searchPath(effectiveBranch) : "#"}
					title="Suche öffnen"
					aria-current={isSearchActive ? "page" : undefined}
				>
					<SearchIcon className="h-4 w-4" />
					Suche
				</Link>
			</Button>
		</div>
	);
}