av-planner/src/components/common/CategorySelect.tsx

import { useMemo, useState } from 'react'
import { allDeviceCategories } from '../../domain/diagram'
import { useCatalogStore } from '../../state/catalogStore'

const NEW_CATEGORY_VALUE = '__new__'

/** A device-category <select> with a built-in "+ New category…" option that
 *  switches to a small inline text field, creates the category, and selects
 *  it — so adding a category never requires leaving the form you're in. */
export default function CategorySelect({
  value,
  onChange,
  className = 'mt-1 w-full rounded border border-slate-300 px-2 py-1 text-sm',
  publishPublic = false,
}: {
  value: string
  onChange: (categoryId: string) => void
  className?: string
  /** When true, a category created here goes straight into the public
   *  catalog (Admin-only) instead of your own private one — used when the
   *  containing form is itself publishing directly, so a brand-new category
   *  it references doesn't end up private and invisible to everyone else. */
  publishPublic?: boolean
}) {
  const catalog = useCatalogStore((s) => s.catalog)
  const addCustomDeviceCategory = useCatalogStore((s) => s.addCustomDeviceCategory)
  const adminAddDeviceCategory = useCatalogStore((s) => s.adminAddDeviceCategory)
  const [creating, setCreating] = useState(false)
  const [draftName, setDraftName] = useState('')

  const categories = useMemo(
    () => [...allDeviceCategories(catalog)].sort((a, b) => a.name.localeCompare(b.name)),
    [catalog],
  )

  const commitNewCategory = async () => {
    const trimmed = draftName.trim()
    if (trimmed) {
      onChange(await (publishPublic ? adminAddDeviceCategory(trimmed) : addCustomDeviceCategory(trimmed)))
    }
    setCreating(false)
    setDraftName('')
  }

  if (creating) {
    return (
      <div className="mt-1 flex gap-1.5">
        <input
          autoFocus
          value={draftName}
          onChange={(e) => setDraftName(e.target.value)}
          onKeyDown={(e) => {
            if (e.key === 'Enter') {
              e.preventDefault()
              commitNewCategory()
            } else if (e.key === 'Escape') {
              setCreating(false)
              setDraftName('')
            }
          }}
          placeholder="New category name"
          className="w-full min-w-0 flex-1 rounded border border-slate-300 px-2 py-1 text-sm"
        />
        <button
          onClick={commitNewCategory}
          className="shrink-0 rounded bg-indigo-600 px-2 text-xs font-medium text-white hover:bg-indigo-500"
        >
          Add
        </button>
        <button
          onClick={() => {
            setCreating(false)
            setDraftName('')
          }}
          className="shrink-0 rounded px-2 text-xs text-slate-500 hover:bg-slate-100"
        >
          Cancel
        </button>
      </div>
    )
  }

  return (
    <select
      value={value}
      onChange={(e) => {
        if (e.target.value === NEW_CATEGORY_VALUE) setCreating(true)
        else onChange(e.target.value)
      }}
      className={className}
    >
      {categories.map((c) => (
        <option key={c.id} value={c.id}>
          {c.name}
        </option>
      ))}
      <option value={NEW_CATEGORY_VALUE}>+ New category…</option>
    </select>
  )
}