Function useOptions

src/hooks/useOptions.ts:9–61

(
  options: OptionType[],
  children: React.ReactNode,
  fieldNames: FieldNames,
  optionFilterProp: string,
  optionLabelProp: string,
)

 * Then flatten the `options`.
 */
const useOptions = <OptionType>(
  options: OptionType[],
  children: React.ReactNode,
  fieldNames: FieldNames,
  optionFilterProp: string,
  optionLabelProp: string,
) => {
  return React.useMemo(() => {
    let mergedOptions = options;
    const childrenAsData = !options;

    if (childrenAsData) {
      mergedOptions = convertChildrenToData(children);
    }

    const valueOptions = new Map<RawValueType, OptionType>();
    const labelOptions = new Map<React.ReactNode, OptionType>();

    const setLabelOptions = (
      labelOptionsMap: Map<React.ReactNode, OptionType>,
      option: OptionType,
      key: string | number,
    ) => {
      if (key && typeof key === 'string') {
        labelOptionsMap.set(option[key], option);
      }
    };

    const dig = (optionList: OptionType[], isChildren = false) => {
      // for loop to speed up collection speed
      for (let i = 0; i < optionList.length; i += 1) {
        const option = optionList[i];
        if (!option[fieldNames.options] || isChildren) {
          valueOptions.set(option[fieldNames.value], option);
          setLabelOptions(labelOptions, option, fieldNames.label);
          // https://github.com/ant-design/ant-design/issues/35304
          setLabelOptions(labelOptions, option, optionFilterProp);
          setLabelOptions(labelOptions, option, optionLabelProp);
        } else {
          dig(option[fieldNames.options], true);
        }
      }
    };

    dig(mergedOptions);

    return {
      options: mergedOptions,
      valueOptions,
      labelOptions,
    };
  }, [options, children, fieldNames, optionFilterProp, optionLabelProp]);
};

export default useOptions;