93 lines
2.7 KiB
TypeScript
93 lines
2.7 KiB
TypeScript
|
import { useQueryClient } from "@tanstack/react-query";
|
||
|
|
import { getQueryKey } from "@trpc/react-query";
|
||
|
|
import { DataNode, DepartmentDto, ObjectType } from "@nicestack/common";
|
||
|
|
import { api } from "../trpc";
|
||
|
|
import { findQueryData, getCacheDataFromQuery } from "../utils";
|
||
|
|
import { CrudOperation, emitDataChange } from "../../event";
|
||
|
|
export function useDepartment() {
|
||
|
|
const queryClient = useQueryClient();
|
||
|
|
const queryKey = getQueryKey(api.department);
|
||
|
|
const create = api.department.create.useMutation({
|
||
|
|
onSuccess: (result) => {
|
||
|
|
queryClient.invalidateQueries({ queryKey });
|
||
|
|
emitDataChange(ObjectType.DEPARTMENT, result as any, CrudOperation.CREATED)
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
const update = api.department.update.useMutation({
|
||
|
|
onSuccess: (result) => {
|
||
|
|
|
||
|
|
queryClient.invalidateQueries({ queryKey });
|
||
|
|
emitDataChange(ObjectType.DEPARTMENT, result as any, CrudOperation.UPDATED)
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
const softDeleteByIds = api.department.softDeleteByIds.useMutation({
|
||
|
|
onSuccess: (result) => {
|
||
|
|
queryClient.invalidateQueries({ queryKey });
|
||
|
|
emitDataChange(ObjectType.DEPARTMENT, result as any, CrudOperation.DELETED)
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
const buildTree = (
|
||
|
|
data: DepartmentDto[],
|
||
|
|
parentId: string | null = null
|
||
|
|
): DataNode[] => {
|
||
|
|
return data
|
||
|
|
.filter((department) => department.parentId === parentId)
|
||
|
|
.sort((a, b) => a.order - b.order)
|
||
|
|
.map((department) => {
|
||
|
|
const node: DataNode = {
|
||
|
|
title: department.name,
|
||
|
|
key: department.id,
|
||
|
|
value: department.id,
|
||
|
|
hasChildren: department.hasChildren,
|
||
|
|
children: department.hasChildren
|
||
|
|
? buildTree(data, department.id)
|
||
|
|
: undefined,
|
||
|
|
data: department,
|
||
|
|
};
|
||
|
|
return node;
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// const getTreeData = () => {
|
||
|
|
// const uniqueData: DepartmentDto[] = getCacheDataFromQuery(
|
||
|
|
// queryClient,
|
||
|
|
// api.department
|
||
|
|
// );
|
||
|
|
// const treeData: DataNode[] = buildTree(uniqueData);
|
||
|
|
// return treeData;
|
||
|
|
// };
|
||
|
|
// const getTreeData = () => {
|
||
|
|
// const cacheArray = queryClient.getQueriesData({
|
||
|
|
// queryKey: getQueryKey(api.department.getChildren),
|
||
|
|
// });
|
||
|
|
// const data: DepartmentDto[] = cacheArray
|
||
|
|
// .flatMap((cache) => cache.slice(1))
|
||
|
|
// .flat()
|
||
|
|
// .filter((item) => item !== undefined) as any;
|
||
|
|
// const uniqueDataMap = new Map<string, DepartmentDto>();
|
||
|
|
|
||
|
|
// data?.forEach((item) => {
|
||
|
|
// if (item && item.id) {
|
||
|
|
// uniqueDataMap.set(item.id, item);
|
||
|
|
// }
|
||
|
|
// });
|
||
|
|
// // Convert the Map back to an array
|
||
|
|
// const uniqueData: DepartmentDto[] = Array.from(uniqueDataMap.values());
|
||
|
|
// const treeData: DataNode[] = buildTree(uniqueData);
|
||
|
|
// return treeData;
|
||
|
|
// };
|
||
|
|
const getDept = <T = DepartmentDto>(key: string) => {
|
||
|
|
return findQueryData<T>(queryClient, api.department, key);
|
||
|
|
};
|
||
|
|
return {
|
||
|
|
softDeleteByIds,
|
||
|
|
update,
|
||
|
|
create,
|
||
|
|
// getTreeData,
|
||
|
|
getDept
|
||
|
|
};
|
||
|
|
}
|