initTreeNode.tsx

import _ from 'lodash';
import style from './style.less'

const getTreeList = (data: any, dataList: any[])=> {
  data?.forEach((item: any) => {
    if (_.isUndefined(item?.selectable) || item?.selectable) dataList.push(item);
    if (item?.children) {
      getTreeList(item.children, dataList);
    }
  })
  return dataList
};

export const initTreeNode = (data: any[], searchValue?: string , searchCode?: boolean): any[] => {
  if (searchValue) {
    const allData = getTreeList(data, []);
    const finalData = allData.filter(item =>{
      if(searchCode){
        return item?.title?.toLowerCase()?.indexOf(searchValue?.toLowerCase() as string) > -1 ||
        item?.code?.indexOf(searchValue) > -1
      }
        return item?.title?.toLowerCase()?.indexOf(searchValue?.toLowerCase() as string) > -1
    }
    )
    return finalData?.map(item => {
      const index = item?.title?.toLowerCase()?.indexOf(searchValue?.toLowerCase() as string);
      const beforeStr = item?.title?.substr(0, index);
      const afterStr = item?.title?.substr(index + searchValue.length);
      const title = index > -1 ? (
        <span>
          {beforeStr}
          <span className={style.search_value_keyword}>{searchValue}</span>
          {afterStr}
        </span>
      ) : item.title;
      const extraItem = _.omit(item, ['children'])
      return {
        ...extraItem,
        title,
        extraTitle: item.title,
      };
    }) || [];
  }
  return data?.map(item => {
    const extraItem = _.omit(item, ['children'])
    return item?.children?.length > 0 ? ({
      ...extraItem,
      extraTitle: item.title,
      children: initTreeNode(item.children)
    }) : ({
      ...extraItem,
      extraTitle: item.title,
    })
  })
}