conditional-sidebar.tsx

"use client";

import { usePathname } from "next/navigation";
import { DocsLayoutProps } from "fumadocs-ui/layouts/docs";
import Sidebar from "./sidebar";
import IntegrationsSidebar from "./integrations-sidebar";
import { INTEGRATION_ORDER } from "@/lib/integrations";

const DEPLOY_ROUTES = ["agentcore"];
import { normalizeUrl } from "@/lib/analytics-utils";
import { useMemo } from "react";
import VersionSelector, {
  getVersionFromPathname,
} from "@/components/ui/reference-sidebar/version-selector";
import LearnHeader from "@/components/ui/learn-sidebar/learn-header";

interface ConditionalSidebarProps {
  pageTree: DocsLayoutProps["tree"];
}

type Node = DocsLayoutProps["tree"]["children"][number];

export default function ConditionalSidebar({
  pageTree,
}: ConditionalSidebarProps) {
  const pathname = usePathname();

  // Normalize the pathname to handle /integrations/... paths
  const normalizedPathname = normalizeUrl(pathname);
  // Check if this is an integration landing page (e.g., /langgraph)
  // Use the first segment of the normalized pathname to ensure correct matching
  const firstSegment = normalizedPathname.replace(/^\//, "").split("/")[0];
  const isIntegrationRoute =
    INTEGRATION_ORDER.includes(
      firstSegment as (typeof INTEGRATION_ORDER)[number],
    ) || DEPLOY_ROUTES.includes(firstSegment);

  // Check if this is a reference route (e.g., /reference)
  const isReferenceRoute = firstSegment === "reference";
  // Check if this is a learn route (e.g., /learn)
  const isLearnRoute = firstSegment === "learn";
  const currentVersion = getVersionFromPathname(pathname);

  // Find the learn folder and use its children
  const learnPageTree = useMemo(() => {
    if (!isLearnRoute) return null;

    const learnFolder = pageTree.children.find((node) => {
      if (node.type !== "folder") return false;
      const folderNode = node as any;
      const url = folderNode.index?.url || folderNode.url;
      const name =
        typeof folderNode.name === "string" ? folderNode.name : undefined;
      return url === "/learn" || name?.toLowerCase() === "learn";
    }) as Node | undefined;

    if (learnFolder && "children" in learnFolder) {
      return {
        ...pageTree,
        children: (learnFolder as any).children || [],
      };
    }

    return null;
  }, [isLearnRoute, pageTree]);

  // Find the reference folder and drill into the active version
  const referencePageTree = useMemo(() => {
    if (!isReferenceRoute) return null;

    // Find the reference folder
    const referenceFolder = pageTree.children.find((node) => {
      if (node.type !== "folder") return false;
      const folderNode = node as any;
      const url = folderNode.index?.url || folderNode.url;
      const name =
        typeof folderNode.name === "string" ? folderNode.name : undefined;
      return url === "/reference" || name?.toLowerCase() === "reference";
    }) as Node | undefined;

    if (referenceFolder && "children" in referenceFolder) {
      const referenceChildren = (referenceFolder as any).children || [];

      // Find the version folder (v1 or v2) within the reference folder
      const versionFolder = referenceChildren.find((node: any) => {
        if (node.type !== "folder") return false;
        const url = node.index?.url || node.url;
        const name = typeof node.name === "string" ? node.name : undefined;
        return (
          url === `/reference/${currentVersion}` ||
          name?.toLowerCase() === currentVersion
        );
      });

      if (versionFolder && "children" in versionFolder) {
        // Return a pageTree with only the version folder's children
        return {
          ...pageTree,
          children: (versionFolder as any).children || [],
        };
      }

      // Fallback: return the reference folder's children directly
      return {
        ...pageTree,
        children: referenceChildren,
      };
    }

    return null;
  }, [isReferenceRoute, pageTree, currentVersion]);

  if (isIntegrationRoute) {
    return <IntegrationsSidebar pageTree={pageTree} />;
  }

  if (isLearnRoute && learnPageTree) {
    return (
      <Sidebar
        pageTree={learnPageTree}
        showIntegrationSelector={false}
        headerSlot={<LearnHeader />}
      />
    );
  }

  if (isReferenceRoute && referencePageTree) {
    return (
      <Sidebar
        pageTree={referencePageTree}
        showIntegrationSelector={false}
        headerSlot={<VersionSelector />}
      />
    );
  }

  return <Sidebar pageTree={pageTree} />;
}