Section Header 2

import { useState } from "react";
import clsx from "clsx";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { Input } from "@/components/ui/input";
import { KeyboardArrowDown, Search } from "relume-icons";

type SortMenu = {
  title: string;
  criteria: string[];
};

type Props = {
  heading: string;
  description: string;
  inputPlaceholder?: string;
  inputIcon: React.ReactNode;
  sortMenu: SortMenu;
};

export type SectionHeader2Props = React.ComponentPropsWithoutRef<"section"> & Partial<Props>;

export const SectionHeader2 = (props: SectionHeader2Props) => {
  const { heading, description, sortMenu, inputPlaceholder, inputIcon } = {
    ...SectionHeader2Defaults,
    ...props,
  };

  const [searchInput, setSearchInput] = useState<string>("");
  const [isDropdownOpen, setIsDropdownOpen] = useState(false);

  return (
    <section className="pb-5 md:pb-6">
      <div className="grid grid-cols-1 items-end justify-between gap-4 md:grid-cols-[1fr_max-content] md:gap-6">
        <div className="max-w-lg">
          <h2 className="text-h5 font-bold">{heading}</h2>
          <p className="mt-2">{description}</p>
        </div>
        <div className="flex items-center justify-between md:justify-normal">
          <div>
            <Input
              id="email"
              type="email"
              value={searchInput}
              placeholder={inputPlaceholder}
              icon={inputIcon}
              className="mr-4"
              onChange={(e) => setSearchInput(e.target.value)}
            />
          </div>
          <DropdownMenu onOpenChange={(isDropdownOpen) => setIsDropdownOpen(isDropdownOpen)}>
            <DropdownMenuTrigger className="flex items-center gap-2 border border-scheme-border px-4">
              <p className="whitespace-nowrap">{sortMenu.title}</p>
              <KeyboardArrowDown
                className={clsx("shrink-0 text-scheme-text transition-transform duration-300", {
                  "rotate-180": isDropdownOpen,
                  "rotate-0": !isDropdownOpen,
                })}
              />
            </DropdownMenuTrigger>
            <DropdownMenuContent>
              {sortMenu.criteria.map((criteria, index) => (
                <DropdownMenuItem key={index}>{criteria}</DropdownMenuItem>
              ))}
            </DropdownMenuContent>
          </DropdownMenu>
        </div>
      </div>
    </section>
  );
};

export const SectionHeader2Defaults: Props = {
  heading: "Heading goes here",
  description:
    "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Suspendisse varius enim in eros.",
  sortMenu: {
    title: "Sort by",
    criteria: ["Option One", "Option Two", "Option Three", "Option Four"],
  },
  inputPlaceholder: "Search",
  inputIcon: <Search className="size-6 text-scheme-text" />,
};