🔄 Merge pull request #8 from steeeee0223/feature/ui

🔧 `ui` Fix tree actions

packages/ui/src/components/custom/tree/tree-actions.tsx

@@ -24,13 +24,15 @@ export function treeReducer(
   { ids, entities }: Entity<TreeItem>,
   { type, payload }: TreeAction<TreeItem>,
 ): Entity<TreeItem> {
+  let e;
   switch (type) {
     case "add":
       payload.forEach((item) => (entities[item.id] = item));
       return { ids: Object.keys(entities), entities };
     case "set":
-      payload.forEach((item) => (entities[item.id] = item));
-      return { ids: Object.keys(entities), entities };
+      e = payload.reduce((acc, item) => ({ ...acc, [item.id]: item }), {});
+      // payload.forEach((item) => (entities[item.id] = item));
+      return { ids: Object.keys(e), entities: e };
     case "update":
       entities[payload.id] = payload;
       return { ids, entities };

packages/ui/src/components/custom/tree/tree-prodiver.tsx

@@ -15,7 +15,7 @@ import { TreeContext } from "./tree-context";
 
 interface TreeProviderProps extends PropsWithChildren {
   className?: string;
-  fetchItems: () => Promise<ActionState<never, TreeItem[]>>;
+  fetchItems?: () => Promise<ActionState<never, TreeItem[]>>;
   isItemActive?: (id: string) => boolean;
   onClickItem?: (id: string) => void;
 }
@@ -29,10 +29,14 @@ export function TreeProvider({
 }: TreeProviderProps) {
   const $initialItems = { ids: [], entities: {} };
   const [state, dispatch] = useReducer<TreeReducer>(treeReducer, $initialItems);
-  const { data, isLoading } = useFetch<TreeItem[]>(fetchItems, {
-    onSuccess: (data) => dispatch({ type: "set", payload: data }),
-    onError: (e) => toast.error(e),
-  });
+  const { data, isLoading } = useFetch<TreeItem[]>(
+    fetchItems!,
+    {
+      onSuccess: (data) => dispatch({ type: "set", payload: data }),
+      onError: (e) => toast.error(e),
+    },
+    [fetchItems],
+  );
 
   const treeItems = Object.values(state.entities);
   console.log(`treeItems:`, treeItems);