All files / src/ahuora-design-system/ui combobox.tsx

90% Statements 18/20
77.27% Branches 17/22
87.5% Functions 7/8
90% Lines 18/20

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150                                                                                          37x                     574x 574x   574x 31x     31x     574x 287452x                 574x                                 29x                 2296x   1722x 1722x       287452x     287452x       31x                                       574x 574x                        
"use client";
 
import * as React from "react";
import { Check } from "lucide-react";
import { cn } from "@/lib/utils";
import {
  Command,
  CommandEmpty,
  CommandGroup,
  CommandInput,
  CommandItem,
  CommandList,
  CommandSeparator,
} from "@/ahuora-design-system/ui/command";
import {
  Popover,
  PopoverContent,
  PopoverTrigger,
} from "@/ahuora-design-system/ui/popover";
import { ScrollArea } from "./scroll-area";
 
export interface ComboboxOption {
  value: string;
  label: string;
}
 
export interface ComboboxProps {
  options: {
    type: "group" | "item" | "separator";
    label?: string; // Only for groups
    options?: ComboboxOption[]; // Only for groups
    value?: string; // Only for items
  }[];
  selectedValues: string[];
  onChange: (values: string[]) => void;
  searchPlaceholder?: string;
  renderItem?: (option: ComboboxOption, isSelected: boolean) => React.ReactNode;
  emptyMessage?: string;
  noGroupItemsMessage?: React.ReactNode;
  trigger: React.ReactNode;
  renderSelectedValues?: (
    setOpen?: (isOpen: boolean) => void,
  ) => React.ReactNode;
}
 
export const Combobox: React.FC<ComboboxProps> = ({
  options,
  selectedValues,
  onChange,
  searchPlaceholder = "Search...",
  renderItem,
  emptyMessage = "No options found.",
  noGroupItemsMessage = "No items selected.",
  trigger,
  renderSelectedValues,
}) => {
  const [open, setOpen] = React.useState(false);
  const commandList = React.useRef<HTMLDivElement>(null);
 
  const handleToggle = (value: string) => {
    const updatedValues = selectedValues.includes(value)
      ? selectedValues.filter((v) => v !== value) // Remove if already selected
      : [...selectedValues, value]; // Add if not selected
    onChange(updatedValues);
  };
 
  const defaultRenderItem = (option: ComboboxOption, isSelected: boolean) => (
    <>
      {option.label}
      <Check
        size={16}
        className={cn("ml-auto", isSelected ? "opacity-100" : "opacity-0")}
      />
    </>
  );
 
  return (
    <Popover open={open} onOpenChange={setOpen} modal={true}>
      <PopoverTrigger asChild>{trigger}</PopoverTrigger>
      <PopoverContent
        className="w-[250px] p-2"
        side="right"
        sideOffset={26}
        align="start"
      >
        <Command position="popper" className="h-[300px]">
          {renderSelectedValues && (
            <div className="p-2">{renderSelectedValues(setOpen)}</div>
          )}
          <CommandInput
            placeholder={searchPlaceholder}
            onValueChange={() => {
              // Scroll to top when search input changes
              commandList.current
                ?.querySelector("[data-radix-scroll-area-viewport]")
                .scrollTo(0, 0);
            }}
          />
          <CommandList className="h-[300px]">
            <ScrollArea ref={commandList} className="h-[300px]">
              <CommandEmpty>{emptyMessage}</CommandEmpty>
              {options.map((option, idx) => {
                if (option.type === "group") {
                  const groupHasOptions =
                    option.options && option.options.length > 0;
                  return (
                    <CommandGroup key={idx} heading={option.label}>
                      {groupHasOptions
                        ? option.options?.map((item) => {
                            const isSelected = selectedValues?.includes(
                              item?.value,
                            );
                            return (
                              <CommandItem
                                key={item.value}
                                value={item.value}
                                onSelect={() => handleToggle(item.value)}
                              >
                                {renderItem
                                  ? renderItem(item, isSelected)
                                  : defaultRenderItem(item, isSelected)}
                              </CommandItem>
                            );
                          })
                        : (option.label === "Selected compounds" ||
                            option.label === "Recommended") && (
                            <small className="px-2 py-0 text-zinc-500 italic">
                              {option.label === "Selected compounds"
                                ? noGroupItemsMessage
                                : "No recommended compounds"}
                            </small>
                          )}
                    </CommandGroup>
                  );
                }
 
                if (option.type === "separator") {
                  return <CommandSeparator key={idx} />;
                }
 
                return null;
              })}
            </ScrollArea>
          </CommandList>
        </Command>
      </PopoverContent>
    </Popover>
  );
};