tld_wms/packages/app/src/pages/role/table/tree-modal/info/hooks.ts

import {getTreeMenu} from '@apis';
import {useContext} from '@hooks';
import {useQuery} from '@tanstack/react-query';
import {context} from '../context';

export function useMenuTree() {
  const {data} = useQuery({
    queryKey: [getTreeMenu.name],
    async queryFn() {
      const data = await getTreeMenu();

      if (data.msg === '200') return data.data;

      throw new Error(data.errMsg);
    },
    suspense: true,
  });

  return data;
}

export function useCheck() {
  const [list, setList] = useContext(context);

  function onCheck(keys: string[]) {
    setList(keys);
  }

  return [list, {onCheck}] as const;
}