import { IAntdTreeNode, ICategory } from 'src/type';

export function generateAntdTreeData(data: ICategory[], pid: string = '0'): IAntdTreeNode[] | [] {
  if (!data || !data.length) return [];

  return data
    .filter((item) => item.pid === pid)
    .map((item) => {
      return {
        title: item.name,
        value: item.id,
        children: generateAntdTreeData(
          data.filter((item) => item.pid !== pid),
          item.id,
        ),
      };
    });
}

export const generateEllipses = (str: string) => {
  if (str.length < 60) return str;
  return str.slice(0, 60) + '...';
};

export function getQueryParams(name: string, url: string) {
  if (!url) url = location.href;
  name = name.replace(/[\[]/, '\\[').replace(/[\]]/, '\\]');
  var regexS = '[\\?&]' + name + '=([^&#]*)';
  var regex = new RegExp(regexS);
  var results = regex.exec(url);
  return results == null ? null : results[1];
}