@capillarytech/blaze-ui 0.1.6-alpha.6 → 0.1.6-alpha.61

package/CapUnifiedSelect/CapUnifiedSelect.js

@@ -1,121 +1,462 @@
-// CapUnifiedSelect component using Ant Design v5 Select and TreeSelect directly
-import React from 'react';
+import React, { useState, useEffect, useMemo, useCallback, memo } from 'react';
 import PropTypes from 'prop-types';
-import { Select, TreeSelect } from 'antd';
-import { SelectWrapper, HeaderWrapper, StyledInfoIcon } from './styles';
-import CapLabel from '../CapLabel';
-import CapTooltip from '../CapTooltip';
+import classnames from 'classnames';
+import { TreeSelect, Input, Button, Checkbox } from 'antd-v5';
+import styled from 'styled-components';
+import * as styledVars from '../styled/variables';
+import { CapLabel, CapTooltipWithInfo, CapRow, CapIcon } from '../';
+import withStyles from '../utils/withStyles';
+import withMemo from '../utils/withMemo';
+import { HeaderWrapper, selectStyles } from './styles';
 
-function CapUnifiedSelect({
+const StyledTreeSelect = styled(TreeSelect)`
+  ${selectStyles}
+`;
+
+const NoResult = memo(({ noResultCustomText, className, showUpload, options, noResultCustomIcon }) => (
+  <CapRow
+    className={classnames(className, 'cap-unified-select-no-result')}
+    align="middle"
+    gap={8}
+  >
+    <CapIcon type={noResultCustomIcon} size="m" />
+    <CapLabel className="cap-unified-select-no-result-text">
+      {showUpload && options?.length === 0
+        ? noResultCustomText
+        : 'No results found'}
+    </CapLabel>
+  </CapRow>
+));
+
+const SelectAllCheckbox = memo(({ currentItems, tempValue, setTempValue, processTreeData }) => {
+  const { leafValues } = processTreeData(currentItems);
+  const totalAvailable = leafValues.length;
+  const leafSet = new Set(leafValues);
+  const selectedInScope = Array.isArray(tempValue)
+    ? tempValue.filter((v) => leafSet.has(v)).length
+    : 0;
+
+  const allChecked = totalAvailable > 0 && selectedInScope === totalAvailable;
+  const indeterminate = selectedInScope > 0 && selectedInScope < totalAvailable;
+
+  const handleChange = (e) => {
+    if (e.target.checked) {
+      const merged = new Set(Array.isArray(tempValue) ? tempValue : []);
+      leafValues.forEach((v) => merged.add(v));
+      setTempValue(Array.from(merged));
+    } else {
+      const toRemove = new Set(leafValues);
+      const next = (Array.isArray(tempValue) ? tempValue : []).filter(
+        (v) => !toRemove.has(v)
+      );
+      setTempValue(next);
+    }
+  };
+
+  return (
+    <CapRow className="cap-unified-select-select-all-container" align="middle">
+      <Checkbox
+        className="cap-unified-select-select-all-checkbox"
+        checked={allChecked}
+        indeterminate={indeterminate}
+        onChange={handleChange}
+      >
+        <CapLabel type="label14" className="cap-unified-select-select-all-label">Select all</CapLabel>
+      </Checkbox>
+    </CapRow>
+  );
+});
+
+const buildTreeMaps = (nodes) => {
+  const result = {
+    leafValues: [],
+    parentChildMap: {},
+    nodeMap: {},
+  };
+  if (!nodes) return result;
+
+  const traverse = (items) => {
+    items.forEach((item) => {
+      result.nodeMap[item.value] = item;
+      if (item.children && item.children.length > 0) {
+        result.parentChildMap[item.value] = item.children.map(child => child.value);
+        traverse(item.children);
+      } else {
+        result.leafValues.push(item.value);
+      }
+    });
+  };
+  traverse(nodes);
+  return result;
+};
+
+const countSelectedLeaves = (treeMaps, selectedValues) => {
+  if (!Array.isArray(selectedValues) || !selectedValues.length) return 0;
+  const expandedSet = new Set(selectedValues);
+  const processNode = (value) => {
+    const children = treeMaps.parentChildMap[value];
+    if (!children) return;
+    children.forEach(childValue => {
+      expandedSet.add(childValue);
+      processNode(childValue);
+    });
+  };
+  selectedValues.forEach(processNode);
+  return treeMaps.leafValues.reduce((count, leaf) => expandedSet.has(leaf) ? count + 1 : count, 0);
+};
+
+const filterTreeData = (data, search, searchBasedOn) => {
+  if (!data?.length || !search) return data;
+  const searchLower = search.toLowerCase();
+  const nodeMatchesSearch = (node) => {
+    const target = searchBasedOn === 'value'
+      ? String(node.value ?? '')
+      : searchBasedOn === 'key'
+        ? String(node.key ?? '')
+        : String(node.label ?? node.title ?? '');
+    return target.toLowerCase().includes(searchLower);
+  };
+  const loop = (items) =>
+    items.reduce((acc, item) => {
+      if (!item) return acc;
+      const children = item.children?.length ? loop(item.children) : [];
+      if (nodeMatchesSearch(item) || children.length) {
+        acc.push({ ...item, children });
+      }
+      return acc;
+    }, []);
+  return loop(data);
+};
+
+const CapUnifiedSelect = ({
   type,
   options = [],
-  treeData,
   value,
   onChange,
   placeholder = 'Select an option',
   className,
   style,
+  isError = false,
+  errorMessage,
+  containerClassName,
+  popoverClassName,
   allowClear = false,
-  showSearch = false,
-  label,
+  headerLabel,
+  onUpload,
   tooltip,
+  bylineText,
   disabled = false,
-}) {
-  const selectVirtualizationProps = {
-    listHeight: 256,
-  };
+  showUpload = false,
+  customPopupRender = true,
+  showSearch = true,
+  searchBasedOn = 'label',
+  onConfirm,
+  onCancel,
+  noResultCustomText = 'No results found',
+  noResultCustomIcon = 'warning',
+  ...rest
+}) => {
+  const [searchText, setSearchText] = useState('');
+  const [tempValue, setTempValue] = useState(value);
+  const [dropdownOpen, setDropdownOpen] = useState(false);
 
-  const treeSelectVirtualizationProps = {
-    listHeight: 256,
-    listItemHeight: 32,
-  };
+  useEffect(() => {
+    const isEqual = Array.isArray(value) && Array.isArray(tempValue)
+      ? value.length === tempValue.length && value.every((v) => tempValue.includes(v))
+      : value === tempValue;
+    if (!isEqual) setTempValue(value);
+  }, [value]);
 
-  const renderHeader = () => {
-    if (!label && !tooltip) return null;
-    
+  const isMulti = useMemo(() => type === 'multiSelect' || type === 'multiTreeSelect', [type]);
+  const isTree = useMemo(() => type === 'treeSelect' || type === 'multiTreeSelect', [type]);
+
+  const dataSource = useMemo(() => {
+    if (!options?.length) return [];
+    return isTree
+      ? options
+      : options.map((opt) => ({
+          title: opt?.label,
+          value: opt?.value,
+          key: opt?.key || opt?.value,
+        }));
+  }, [isTree, options]);
+
+  const filteredTree = useMemo(
+    () => filterTreeData(dataSource, searchText, searchBasedOn),
+    [dataSource, searchText, searchBasedOn]
+  );
+
+  const treeMaps = useMemo(() => buildTreeMaps(options), [options]);
+  const selectedLeafCount = useMemo(() => countSelectedLeaves(treeMaps, tempValue), [treeMaps, tempValue]);
+
+  const displayValue = dropdownOpen ? tempValue : value;
+
+  const suffix = useMemo(() => {
+    const count = Array.isArray(displayValue) ? displayValue.length : (displayValue ? 1 : 0);
     return (
-      <HeaderWrapper className={disabled ? 'disabled' : ''}>
-        {label && (
-          <CapLabel type="label16" className={disabled ? 'disabled' : ''}>
-            {label}
-          </CapLabel>
-        )}
-        {tooltip && (
-          <CapTooltip title={tooltip}>
-            <StyledInfoIcon className={disabled ? 'disabled' : ''} />
-          </CapTooltip>
+      <>
+        {isMulti && count > 1 && <span>+{count - 1} more </span>}
+        <CapIcon
+          className="cap-unified-select-suffix-icon"
+          type={dropdownOpen ? 'up' : 'down'}
+          size="s"
+        />
+      </>
+    );
+  }, [isMulti, displayValue, dropdownOpen]);
+
+  const prefix = useMemo(() => {
+    if (isMulti && Array.isArray(displayValue) && displayValue.length > 0) {
+      const firstLeafValue = displayValue.find(val => treeMaps.leafValues.includes(val));
+      return treeMaps.nodeMap[firstLeafValue]?.label || treeMaps.nodeMap[firstLeafValue]?.title || null;
+    }
+    return null;
+  }, [isMulti, displayValue, treeMaps]);
+
+  const handleConfirm = useCallback(() => {
+    onChange?.(tempValue);
+    setDropdownOpen(false);
+    setSearchText('');
+    onConfirm?.(tempValue);
+  }, [onChange, onConfirm, tempValue]);
+
+  const handleCancel = useCallback(() => {
+    setTempValue(value);
+    setDropdownOpen(false);
+    onCancel?.();
+  }, [value, onCancel]);
+
+  const handleDropdownVisibilityChange = useCallback((open) => {
+    if (!open && !customPopupRender) {
+      onChange?.(tempValue);
+    } else if (!open) {
+      setTempValue(value);
+    }
+    setDropdownOpen(open);
+  }, [customPopupRender, value, onChange, tempValue]);
+
+  const renderHeader = useMemo(() => {
+    if (!headerLabel && !tooltip) return null;
+    return (
+      <>
+        <HeaderWrapper className={classnames(disabled && 'disabled', 'cap-unified-select-header')}>
+          {headerLabel && (
+            <CapLabel
+              type="label16"
+              className={classnames(disabled && 'disabled', 'cap-unified-select-header-label')}
+            >
+              {headerLabel}
+            </CapLabel>
+          )}
+          {tooltip && (
+            <CapTooltipWithInfo
+              title={tooltip}
+              className={classnames(disabled && 'disabled', 'cap-unified-select-header-tooltip')}
+              iconProps={{ disabled }}
+            />
+          )}
+        </HeaderWrapper>
+        {bylineText && (
+          <div className="cap-unified-select-header-byline-text">
+            <CapLabel
+              className={classnames(disabled && 'disabled', 'cap-unified-select-header-byline-text')}
+            >
+              {bylineText}
+            </CapLabel>
+          </div>
         )}
-      </HeaderWrapper>
+      </>
     );
-  };
+  }, [headerLabel, tooltip, bylineText, disabled]);
+
+  const renderCustomDropdown = useCallback(
+    (menu) => {
+      if (!customPopupRender) return menu;
+      const currentItems = filteredTree;
 
-  const renderDropdown = () => {
-    if (type === 'treeSelect' || type === 'multiTreeSelect') {
       return (
-        <TreeSelect
-          treeData={treeData || options}
-          value={value}
-          onChange={onChange}
-          placeholder={placeholder}
-          className={className}
-          style={style}
-          allowClear={allowClear}
-          showSearch={showSearch}
-          multiple={type === 'multiTreeSelect' ? true : false}
-          virtual
-          treeDefaultExpandAll
-          disabled={disabled}
-          {...treeSelectVirtualizationProps}
-        />
+        <div className={classnames(popoverClassName, `${type}-popup-container`)}>
+          {showSearch && (
+            <CapRow className="cap-unified-select-search-container" align="middle">
+              <Input
+                prefix={<CapIcon type="search" size="s" style={{ color: styledVars.CAP_G06 }} />}
+                placeholder="Search"
+                variant="borderless"
+                value={searchText}
+                onChange={(e) => setSearchText(e.target.value)}
+                allowClear
+              />
+            </CapRow>
+          )}
+
+          {isMulti && showUpload && (
+            <CapRow className="cap-unified-select-upload-container" align="middle" onClick={onUpload}>
+              <CapIcon type="upload" size="s" style={{ color: styledVars.CAP_SECONDARY.base }} />
+              <CapLabel type="label14" className="cap-unified-select-upload-label">Upload</CapLabel>
+            </CapRow>
+          )}
+
+          {isMulti && currentItems.length > 0 && (
+            <SelectAllCheckbox
+              currentItems={currentItems}
+              tempValue={tempValue}
+              setTempValue={setTempValue}
+              processTreeData={buildTreeMaps}
+            />
+          )}
+
+          {currentItems.length === 0 ? (
+            <NoResult
+              noResultCustomText={noResultCustomText}
+              className={className}
+              showUpload={showUpload}
+              options={options}
+              noResultCustomIcon={noResultCustomIcon}
+            />
+          ) : (
+            menu
+          )}
+
+          {currentItems.length > 0 && isMulti && (
+            <div className="cap-unified-select-confirm-container">
+              <div className="cap-unified-select-confirm-button-group">
+                <Button
+                  type="primary"
+                  size="small"
+                  className="cap-unified-select-confirm-button"
+                  onClick={handleConfirm}
+                >
+                  Confirm
+                </Button>
+                <Button
+                  type="text"
+                  className="cap-unified-select-cancel-button"
+                  size="small"
+                  onClick={handleCancel}
+                >
+                  Cancel
+                </Button>
+                <CapLabel className="cap-unified-select-selected-count">
+                  {selectedLeafCount} selected
+                </CapLabel>
+              </div>
+            </div>
+          )}
+        </div>
       );
-    }
+    },
+    [
+      customPopupRender,
+      filteredTree,
+      searchText,
+      isMulti,
+      showUpload,
+      onUpload,
+      noResultCustomText,
+      noResultCustomIcon,
+      options,
+      type,
+      tempValue,
+      handleConfirm,
+      handleCancel,
+      popoverClassName,
+      className,
+      selectedLeafCount,
+    ]
+  );
 
-    return (
-      <Select
-        value={value}
-        onChange={onChange}
+  const combinedClassName = useMemo(
+    () => classnames(containerClassName, 'cap-unified-tree-select', className),
+    [containerClassName, className]
+  );
+
+  return (
+    <CapRow className={classnames(className, 'cap-unified-select-container')}>
+      {renderHeader}
+      <StyledTreeSelect
+        type={type}
+        treeData={filteredTree}
+        value={customPopupRender ? tempValue : value}
+        onChange={isMulti ? setTempValue : onChange}
         placeholder={placeholder}
-        className={className}
+        showSearch={false}
+        maxTagCount={0}
+        maxTagPlaceholder={() => null}
+        prefix={prefix || undefined}
+        suffixIcon={suffix}
+        className={combinedClassName}
+        classNames={{
+          popup: { root: classnames('custom-popup-container', className) },
+        }}
         style={style}
+        status={isError ? 'error' : ''}
         allowClear={allowClear}
-        showSearch={showSearch}
-        options={options}
-        mode={type === 'multiSelect' ? 'multiple' : undefined}
+        multiple={isMulti}
+        treeCheckable={isMulti}
+        treeCheckStrictly={false}
+        showCheckedStrategy={TreeSelect.SHOW_CHILD}
+        open={dropdownOpen}
+        onOpenChange={handleDropdownVisibilityChange}
         virtual
         disabled={disabled}
-        {...selectVirtualizationProps}
+        filterTreeNode={false}
+        listHeight={256}
+        listItemHeight={32}
+        popupRender={renderCustomDropdown}
+        {...rest}
       />
-    );
-  };
-
-  return (
-    <SelectWrapper>
-      {renderHeader()}
-      {renderDropdown()}
-    </SelectWrapper>
+      {isError && (
+        <CapLabel className="cap-unified-select-status" style={{ color: 'red' }}>
+          {errorMessage}
+        </CapLabel>
+      )}
+    </CapRow>
   );
-}
+};
 
 CapUnifiedSelect.propTypes = {
   type: PropTypes.oneOf(['select', 'multiSelect', 'treeSelect', 'multiTreeSelect']),
   options: PropTypes.array,
-  treeData: PropTypes.array,
   value: PropTypes.any,
   onChange: PropTypes.func,
   placeholder: PropTypes.oneOfType([PropTypes.string, PropTypes.node]),
   className: PropTypes.string,
   style: PropTypes.object,
   allowClear: PropTypes.bool,
-  showSearch: PropTypes.bool,
-  label: PropTypes.string,
+  headerLabel: PropTypes.string,
   tooltip: PropTypes.oneOfType([PropTypes.string, PropTypes.node]),
   disabled: PropTypes.bool,
+  customPopupRender: PropTypes.bool,
+  showSearch: PropTypes.bool,
+  searchBasedOn: PropTypes.oneOf(['label', 'value', 'key']),
+  onConfirm: PropTypes.func,
+  onCancel: PropTypes.func,
+  isError: PropTypes.bool,
+  errorMessage: PropTypes.string,
+  popupClassName: PropTypes.string,
+  showUpload: PropTypes.bool,
+  onUpload: PropTypes.func,
 };
 
 CapUnifiedSelect.defaultProps = {
   type: 'select',
+  placeholder: 'Select an option',
+  searchBasedOn: 'label',
+  noResultCustomText: 'No results found',
+  noResultCustomIcon: 'warning',
+  options: [],
   allowClear: false,
-  showSearch: false,
+  customPopupRender: true,
+  showSearch: true,
+  className: '',
+  popupClassName: '',
+  disabled: false,
+  showUpload: false,
+  isError: false,
+  onUpload: () => {},
+  onChange: () => {},
+  onConfirm: () => {},
+  onCancel: () => {},
 };
 
-export default CapUnifiedSelect;
+export default withMemo(withStyles(CapUnifiedSelect, selectStyles));

package/CapUnifiedSelect/index.js

@@ -1,4 +1 @@
-import CapUnifiedSelect from './CapUnifiedSelect';
-import CapUnifiedSelectLoadable from './loadable';
-
-export default CapUnifiedSelectLoadable; 
+export { default } from './CapUnifiedSelect';