Input Group

"use client"

import * as React from "react"
import { cva, type VariantProps } from "class-variance-authority"
import { cn } from "cn"

import { Button } from "@/registry/paragon/ui/button"
import { Input } from "@/registry/paragon/ui/input"
import { Textarea } from "@/registry/paragon/ui/textarea"

/**
 * One field surface with icons, text, keys or buttons attached. The group
 * owns border, focus and invalid states; the control inside stays bare.
 */
function InputGroup({ className, ...props }: React.ComponentProps<"div">) {
  return (
    <div
      data-slot="input-group"
      role="group"
      className={cn(
        "group/input-group relative flex h-[2.625rem] w-full min-w-0 items-center rounded-button border border-input bg-card/70 outline-none",
        "transition-[border-color,box-shadow,background-color] duration-(--duration-base) ease-paragon",
        "hover:border-foreground/35",
        "has-[[data-slot=input-group-control]:focus-visible]:border-ring has-[[data-slot=input-group-control]:focus-visible]:bg-card has-[[data-slot=input-group-control]:focus-visible]:ring-4 has-[[data-slot=input-group-control]:focus-visible]:ring-ring/15",
        "has-[[data-slot][aria-invalid=true]]:border-destructive has-[[data-slot][aria-invalid=true]:focus-visible]:ring-destructive/15",
        "has-disabled:pointer-events-none has-disabled:bg-secondary/60 has-disabled:opacity-55",
        "has-[>[data-align=block-end]]:h-auto has-[>[data-align=block-end]]:flex-col has-[>[data-align=block-start]]:h-auto has-[>[data-align=block-start]]:flex-col has-[>textarea]:h-auto",
        "has-[>[data-align=block-end]]:[&>input]:pt-3 has-[>[data-align=block-start]]:[&>input]:pb-3 has-[>[data-align=inline-end]]:[&>input]:pe-2 has-[>[data-align=inline-start]]:[&>input]:ps-2",
        "dark:bg-input/15 dark:has-[[data-slot=input-group-control]:focus-visible]:bg-input/25",
        className
      )}
      {...props}
    />
  )
}

const inputGroupAddonVariants = cva(
  "flex h-auto cursor-text items-center justify-center gap-2 py-1.5 text-sm text-muted-foreground select-none has-[>kbd]:[direction:ltr] [&>kbd]:inline-flex [&>kbd]:h-5 [&>kbd]:min-w-5 [&>kbd]:items-center [&>kbd]:justify-center [&>kbd]:rounded-md [&>kbd]:bg-secondary [&>kbd]:px-1.5 [&>kbd]:font-mono [&>kbd]:text-[0.6875rem] [&>kbd]:text-muted-foreground [&>svg:not([class*='size-'])]:size-4",
  {
    variants: {
      align: {
        "inline-start": "order-first ps-3.5 has-[>button]:ms-[-0.375rem] has-[>kbd]:ms-[-0.25rem]",
        "inline-end": "order-last pe-3.5 has-[>button]:me-[-0.375rem] has-[>kbd]:me-[-0.25rem]",
        "block-start":
          "order-first w-full justify-start px-3.5 pt-2.5 group-has-[>input]/input-group:pt-2.5 [.border-b]:pb-2.5",
        "block-end":
          "order-last w-full justify-start px-3.5 pb-2.5 group-has-[>input]/input-group:pb-2.5 [.border-t]:pt-2.5",
      },
    },
    defaultVariants: {
      align: "inline-start",
    },
  }
)

function InputGroupAddon({
  className,
  align = "inline-start",
  ...props
}: React.ComponentProps<"div"> & VariantProps<typeof inputGroupAddonVariants>) {
  return (
    <div
      role="group"
      data-slot="input-group-addon"
      data-align={align}
      className={cn(inputGroupAddonVariants({ align }), className)}
      onClick={(e) => {
        if ((e.target as HTMLElement).closest("button")) {
          return
        }
        e.currentTarget.parentElement?.querySelector<HTMLElement>("[data-slot=input-group-control]")?.focus()
      }}
      {...props}
    />
  )
}

function InputGroupButton({
  type = "button",
  variant = "ghost",
  size = "xs",
  ...props
}: Omit<React.ComponentProps<typeof Button>, "size" | "type"> & {
  size?: "xs" | "sm" | "icon-xs" | "icon-sm"
  type?: "button" | "submit" | "reset"
}) {
  return <Button type={type} data-size={size} variant={variant} size={size} {...props} />
}

function InputGroupText({ className, ...props }: React.ComponentProps<"span">) {
  return (
    <span
      className={cn(
        "flex items-center gap-2 text-sm text-muted-foreground [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4",
        className
      )}
      {...props}
    />
  )
}

function InputGroupInput({ className, ...props }: React.ComponentProps<typeof Input>) {
  return (
    <Input
      data-slot="input-group-control"
      className={cn(
        "h-full flex-1 rounded-none border-0 bg-transparent shadow-none ring-0 hover:border-0 focus-visible:bg-transparent focus-visible:ring-0 disabled:bg-transparent disabled:opacity-100 aria-invalid:ring-0 dark:bg-transparent dark:focus-visible:bg-transparent",
        className
      )}
      {...props}
    />
  )
}

function InputGroupTextarea({ className, ...props }: React.ComponentProps<"textarea">) {
  return (
    <Textarea
      data-slot="input-group-control"
      className={cn(
        "min-h-20 flex-1 resize-none rounded-none border-0 bg-transparent py-3 shadow-none ring-0 hover:border-0 focus-visible:bg-transparent focus-visible:ring-0 disabled:bg-transparent disabled:opacity-100 aria-invalid:ring-0 dark:bg-transparent dark:focus-visible:bg-transparent",
        className
      )}
      {...props}
    />
  )
}

export { InputGroup, InputGroupAddon, InputGroupButton, InputGroupText, InputGroupInput, InputGroupTextarea }