shadcn-ui/apps/v4/app/(create)/components/style-picker.tsx

"use client"

import * as React from "react"
import { useQueryStates } from "nuqs"

import { type Style, type StyleName } from "@/registry/config"
import { LockButton } from "@/app/(create)/components/lock-button"
import {
  Picker,
  PickerContent,
  PickerGroup,
  PickerRadioGroup,
  PickerRadioItem,
  PickerSeparator,
  PickerTrigger,
} from "@/app/(create)/components/picker"
import { designSystemSearchParams } from "@/app/(create)/lib/search-params"

export function StylePicker({
  styles,
  isMobile,
  anchorRef,
}: {
  styles: readonly Style[]
  isMobile: boolean
  anchorRef: React.RefObject<HTMLDivElement | null>
}) {
  const [params, setParams] = useQueryStates(designSystemSearchParams, {
    shallow: false,
    history: "push",
  })

  const currentStyle = styles.find((style) => style.name === params.style)

  return (
    <Picker>
      <div className="group/picker relative">
        <PickerTrigger>
          <div className="flex flex-col justify-start text-left">
            <div className="text-muted-foreground text-xs">Style</div>
            <div className="text-foreground text-sm font-medium">
              {currentStyle?.title}
            </div>
          </div>
          {currentStyle?.icon && (
            <div className="absolute top-1/2 right-4 flex size-4 -translate-y-1/2 items-center justify-center">
              {React.cloneElement(currentStyle.icon, {
                className: "size-4",
              })}
            </div>
          )}
        </PickerTrigger>
        <LockButton
          param="style"
          className="absolute top-1/2 right-10 -translate-y-1/2"
        />
      </div>
      <PickerContent
        anchor={isMobile ? anchorRef : undefined}
        side={isMobile ? "top" : "right"}
        align={isMobile ? "center" : "start"}
        className="md:w-64"
      >
        <PickerRadioGroup
          value={currentStyle?.name}
          onValueChange={(value) => {
            setParams({ style: value as StyleName })
          }}
        >
          <PickerGroup>
            {styles.map((style, index) => (
              <React.Fragment key={style.name}>
                <PickerRadioItem value={style.name}>
                  <div className="flex items-start gap-2">
                    {style.icon && (
                      <div className="flex size-4 translate-y-0.5 items-center justify-center">
                        {React.cloneElement(style.icon, {
                          className: "size-4",
                        })}
                      </div>
                    )}
                    <div className="flex flex-col justify-start pointer-coarse:gap-1">
                      <div>{style.title}</div>
                      <div className="text-muted-foreground text-xs pointer-coarse:text-sm">
                        {style.description}
                      </div>
                    </div>
                  </div>
                </PickerRadioItem>
                {index < styles.length - 1 && (
                  <PickerSeparator className="opacity-50" />
                )}
              </React.Fragment>
            ))}
          </PickerGroup>
        </PickerRadioGroup>
      </PickerContent>
    </Picker>
  )
}