vercel__ai-elements/apps/docs/components/geistdocs/provider.tsx

"use client";

import type { SharedProps } from "fumadocs-ui/contexts/search";
import type { ComponentProps } from "react";

import { Toaster } from "@repo/shadcn-ui/components/ui/sonner";
import { TooltipProvider } from "@repo/shadcn-ui/components/ui/tooltip";
import { useIsMobile } from "@repo/shadcn-ui/hooks/use-mobile";
import { cn } from "@repo/shadcn-ui/lib/utils";
import { Analytics } from "@vercel/analytics/next";
import { SpeedInsights } from "@vercel/speed-insights/next";
import { RootProvider } from "fumadocs-ui/provider/next";
import { useCallback } from "react";

import { useChatContext } from "@/hooks/geistdocs/use-chat";
import { i18n, i18nProvider } from "@/lib/geistdocs/i18n";

import { SearchDialog } from "./search";

type GeistdocsProviderProps = ComponentProps<typeof RootProvider> & {
  basePath: string | undefined;
  className?: string;
  lang?: string;
};

export const GeistdocsProvider = ({
  basePath,
  search,
  className,
  lang = i18n.defaultLanguage,
  ...props
}: GeistdocsProviderProps) => {
  const { isOpen } = useChatContext();
  const isMobile = useIsMobile();
  const isSidebarVisible = isOpen && !isMobile;

  const SearchDialogComponent = useCallback(
    (sdProps: SharedProps) => <SearchDialog basePath={basePath} {...sdProps} />,
    [basePath]
  );

  return (
    <div
      className={cn(
        "transition-all",
        isSidebarVisible ? "pr-96!" : null,
        className
      )}
    >
      <TooltipProvider>
        <RootProvider
          i18n={i18nProvider(lang)}
          search={{
            SearchDialog: SearchDialogComponent,
            ...search,
          }}
          {...props}
        />
      </TooltipProvider>
      <Analytics />
      <Toaster />
      <SpeedInsights />
    </div>
  );
};