Refactor:리액트 윈도우 추가 드래그 앤 드랍 추가
This commit is contained in:
정현조
parent
6b4946b273
commit
dacc7e357e
@ -1,4 +1,4 @@
|
||||
import { useEffect, useState, useCallback } from 'react';
|
||||
import { useEffect, useState, useCallback, useMemo, useRef } from 'react';
|
||||
import { TreeNode } from 'react-treebeard';
|
||||
import useProjectStore from '@/stores/useProjectStore';
|
||||
import useCanvasStore from '@/stores/useCanvasStore';
|
||||
@ -9,7 +9,22 @@ import AutoLabelButton from './AutoLabelButton';
|
||||
import useMoveImageQuery from '@/queries/images/useMoveImageQuery';
|
||||
import { Folder, Image as ImageIcon, Minus, Loader, ArrowDownToLine, Send, CircleSlash, Check } from 'lucide-react';
|
||||
import { Spinner } from '../ui/spinner';
|
||||
import { DragDropContext, Droppable, Draggable, DropResult } from 'react-beautiful-dnd';
|
||||
|
||||
import { FixedSizeList as List, ListChildComponentProps } from 'react-window';
|
||||
|
||||
import { DndProvider, useDrag, useDrop } from 'react-dnd';
|
||||
import { HTML5Backend } from 'react-dnd-html5-backend';
|
||||
|
||||
interface FlatNode extends TreeNode {
|
||||
depth: number;
|
||||
isLeaf: boolean;
|
||||
parent?: FlatNode;
|
||||
index?: number;
|
||||
}
|
||||
|
||||
const ItemTypes = {
|
||||
NODE: 'node',
|
||||
};
|
||||
|
||||
export default function ProjectStructure({ project }: { project: Project }) {
|
||||
const { setProject } = useProjectStore();
|
||||
@ -18,11 +33,20 @@ export default function ProjectStructure({ project }: { project: Project }) {
|
||||
const [cursor, setCursor] = useState<TreeNode | null>(null);
|
||||
const moveImageMutation = useMoveImageQuery();
|
||||
|
||||
const containerRef = useRef<HTMLDivElement>(null);
|
||||
const [containerHeight, setContainerHeight] = useState<number>(400);
|
||||
|
||||
useEffect(() => {
|
||||
setProject(project);
|
||||
initializeTree();
|
||||
}, [project, setProject, initializeTree]);
|
||||
|
||||
useEffect(() => {
|
||||
if (containerRef.current) {
|
||||
setContainerHeight(containerRef.current.clientHeight);
|
||||
}
|
||||
}, [containerRef, treeData, isLoading]);
|
||||
|
||||
const onToggle = useCallback(
|
||||
async (node: TreeNode, toggled: boolean) => {
|
||||
if (cursor) {
|
||||
@ -37,15 +61,27 @@ export default function ProjectStructure({ project }: { project: Project }) {
|
||||
if (toggled && (!node.children || node.children.length === 0)) {
|
||||
await fetchNodeData(node);
|
||||
}
|
||||
node.toggled = toggled;
|
||||
}
|
||||
|
||||
setTreeData((prevData) => ({ ...prevData! }));
|
||||
const updateNode = (currentNode: TreeNode): TreeNode => {
|
||||
if (currentNode.id === node.id) {
|
||||
return { ...currentNode, toggled };
|
||||
}
|
||||
if (currentNode.children) {
|
||||
return {
|
||||
...currentNode,
|
||||
children: currentNode.children.map(updateNode),
|
||||
};
|
||||
}
|
||||
return currentNode;
|
||||
};
|
||||
|
||||
setTreeData((prevData) => prevData && updateNode(prevData));
|
||||
}
|
||||
},
|
||||
[cursor, fetchNodeData, setImage, setTreeData]
|
||||
);
|
||||
|
||||
const renderStatusIcon = (status: ImageStatus) => {
|
||||
const renderStatusIcon = useCallback((status: ImageStatus) => {
|
||||
switch (status) {
|
||||
case 'PENDING':
|
||||
return (
|
||||
@ -91,32 +127,119 @@ export default function ProjectStructure({ project }: { project: Project }) {
|
||||
/>
|
||||
);
|
||||
}
|
||||
};
|
||||
}, []);
|
||||
|
||||
const renderTree = useCallback(
|
||||
(nodes: TreeNode[], _parentId: string, level: number = 0) => {
|
||||
return nodes.map((node, index) => (
|
||||
<Draggable
|
||||
draggableId={node.id!}
|
||||
index={index}
|
||||
key={node.id}
|
||||
>
|
||||
{(provided) => (
|
||||
const flattenTree = useCallback((nodes: TreeNode[], depth: number = 0, parent?: FlatNode): FlatNode[] => {
|
||||
let flatList: FlatNode[] = [];
|
||||
|
||||
nodes.forEach((node, index) => {
|
||||
const flatNode: FlatNode = {
|
||||
...node,
|
||||
depth,
|
||||
isLeaf: !node.children || node.children.length === 0,
|
||||
parent,
|
||||
index,
|
||||
};
|
||||
flatList.push(flatNode);
|
||||
|
||||
if (node.toggled && node.children) {
|
||||
flatList = flatList.concat(flattenTree(node.children, depth + 1, flatNode));
|
||||
}
|
||||
});
|
||||
|
||||
return flatList;
|
||||
}, []);
|
||||
|
||||
const flatData = useMemo(() => {
|
||||
if (!treeData) return [];
|
||||
return flattenTree([treeData]);
|
||||
}, [treeData, flattenTree]);
|
||||
|
||||
const getItemKey = (index: number, data: FlatNode[]) => data[index].id!;
|
||||
|
||||
const moveNode = useCallback(
|
||||
(dragItem: FlatNode, hoverItem: FlatNode) => {
|
||||
const updatedTreeData = (function moveNodeInTree(node: TreeNode): TreeNode {
|
||||
if (node.id === dragItem.parent?.id) {
|
||||
const newChildren = node.children?.filter((child) => child.id !== dragItem.id) || [];
|
||||
return { ...node, children: newChildren };
|
||||
}
|
||||
|
||||
if (node.id === hoverItem.parent?.id) {
|
||||
const newChildren = [...(node.children || [])];
|
||||
const hoverIndex = newChildren.findIndex((child) => child.id === hoverItem.id);
|
||||
newChildren.splice(hoverIndex, 0, { ...dragItem, parent: hoverItem.parent } as FlatNode);
|
||||
return { ...node, children: newChildren };
|
||||
}
|
||||
|
||||
if (node.children) {
|
||||
return {
|
||||
...node,
|
||||
children: node.children.map(moveNodeInTree),
|
||||
};
|
||||
}
|
||||
|
||||
return node;
|
||||
})(treeData!);
|
||||
|
||||
setTreeData(updatedTreeData);
|
||||
|
||||
if (dragItem.imageData) {
|
||||
const moveFolderId = Number(hoverItem.parent?.id) || 0;
|
||||
const folderId = Number(dragItem.parent?.id) || 0;
|
||||
const projectId = Number(project.id);
|
||||
|
||||
moveImageMutation.mutate({
|
||||
projectId,
|
||||
folderId,
|
||||
imageId: dragItem.imageData.id,
|
||||
moveRequest: {
|
||||
moveFolderId,
|
||||
},
|
||||
});
|
||||
}
|
||||
},
|
||||
[treeData, setTreeData, moveImageMutation, project.id]
|
||||
);
|
||||
|
||||
const Row = ({ index, style, data }: ListChildComponentProps<FlatNode[]>) => {
|
||||
const node = data[index];
|
||||
const ref = useRef<HTMLDivElement>(null);
|
||||
|
||||
const [, drop] = useDrop({
|
||||
accept: ItemTypes.NODE,
|
||||
drop(item: FlatNode) {
|
||||
const dragItem = item;
|
||||
const hoverItem = node;
|
||||
|
||||
if (dragItem.id === hoverItem.id) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 드래그가 끝났을 때 노드 이동 처리
|
||||
moveNode(dragItem, hoverItem);
|
||||
},
|
||||
});
|
||||
|
||||
const [{ isDragging }, drag] = useDrag({
|
||||
type: ItemTypes.NODE,
|
||||
item: () => ({ ...node, index }),
|
||||
collect: (monitor) => ({
|
||||
isDragging: monitor.isDragging(),
|
||||
}),
|
||||
});
|
||||
|
||||
drag(drop(ref));
|
||||
|
||||
return (
|
||||
<div
|
||||
ref={provided.innerRef}
|
||||
{...provided.draggableProps}
|
||||
style={{
|
||||
...provided.draggableProps.style,
|
||||
display: 'flex',
|
||||
flexDirection: 'column',
|
||||
}}
|
||||
>
|
||||
<div
|
||||
{...provided.dragHandleProps}
|
||||
ref={ref}
|
||||
style={{
|
||||
...style,
|
||||
opacity: isDragging ? 0.5 : 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
paddingLeft: `${level * 20}px`,
|
||||
paddingLeft: `${node.depth * 20}px`,
|
||||
cursor: 'pointer',
|
||||
}}
|
||||
onClick={() => onToggle(node, !node.toggled)}
|
||||
@ -137,89 +260,16 @@ export default function ProjectStructure({ project }: { project: Project }) {
|
||||
<span style={{ color: '#4a4a4a', flexGrow: 1 }}>{node.name}</span>
|
||||
{node.imageData && <div style={{ marginLeft: '10px' }}>{renderStatusIcon(node.imageData.status)}</div>}
|
||||
</div>
|
||||
{node.toggled && node.children && node.children.length > 0 && (
|
||||
<Droppable
|
||||
droppableId={node.id!}
|
||||
type="TREE"
|
||||
>
|
||||
{(provided) => (
|
||||
<div
|
||||
ref={provided.innerRef}
|
||||
{...provided.droppableProps}
|
||||
style={{ paddingLeft: `${level * 20}px` }}
|
||||
>
|
||||
{renderTree(node.children!, node.id!, level + 1)}
|
||||
{provided.placeholder}
|
||||
</div>
|
||||
)}
|
||||
</Droppable>
|
||||
)}
|
||||
</div>
|
||||
)}
|
||||
</Draggable>
|
||||
));
|
||||
},
|
||||
[onToggle]
|
||||
);
|
||||
|
||||
const onDragEnd = useCallback(
|
||||
(result: DropResult) => {
|
||||
if (!result.destination || !treeData) {
|
||||
return;
|
||||
}
|
||||
|
||||
const sourceDroppableId = result.source.droppableId;
|
||||
const destinationDroppableId = result.destination.droppableId;
|
||||
const sourceIndex = result.source.index;
|
||||
const destinationIndex = result.destination.index;
|
||||
|
||||
const findNodeById = (nodes: TreeNode[], id: string): TreeNode | null => {
|
||||
for (const node of nodes) {
|
||||
if (node.id === id) {
|
||||
return node;
|
||||
}
|
||||
if (node.children) {
|
||||
const found = findNodeById(node.children, id);
|
||||
if (found) return found;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
};
|
||||
|
||||
const sourceParentNode = sourceDroppableId === 'root' ? treeData : findNodeById([treeData], sourceDroppableId);
|
||||
const destinationParentNode =
|
||||
destinationDroppableId === 'root' ? treeData : findNodeById([treeData], destinationDroppableId);
|
||||
|
||||
if (!sourceParentNode || !destinationParentNode) {
|
||||
return;
|
||||
}
|
||||
|
||||
const [movedItem] = sourceParentNode.children!.splice(sourceIndex, 1);
|
||||
|
||||
destinationParentNode.children!.splice(destinationIndex, 0, movedItem);
|
||||
|
||||
setTreeData({ ...treeData });
|
||||
|
||||
if (movedItem && movedItem.imageData) {
|
||||
const moveFolderId = Number(destinationParentNode.id) || 0;
|
||||
const folderId = Number(sourceParentNode.id) || 0;
|
||||
const projectId = Number(project.id);
|
||||
|
||||
moveImageMutation.mutate({
|
||||
projectId,
|
||||
folderId,
|
||||
imageId: movedItem.imageData.id,
|
||||
moveRequest: {
|
||||
moveFolderId,
|
||||
},
|
||||
});
|
||||
}
|
||||
},
|
||||
[treeData, setTreeData, moveImageMutation, project.id]
|
||||
);
|
||||
|
||||
return (
|
||||
<div className="box-border flex h-full min-h-0 flex-col bg-gray-50 p-2">
|
||||
<DndProvider backend={HTML5Backend}>
|
||||
<div
|
||||
className="box-border flex h-full min-h-0 flex-col bg-gray-50 p-2"
|
||||
style={{ overflowX: 'hidden' }}
|
||||
ref={containerRef}
|
||||
>
|
||||
<div className="flex h-full flex-col gap-2 overflow-hidden px-1 pb-2">
|
||||
<header className="flex w-full items-center gap-2 rounded-md bg-white p-2 shadow-sm">
|
||||
<div className="flex w-full min-w-0 items-center gap-1 pr-1">
|
||||
@ -236,31 +286,28 @@ export default function ProjectStructure({ project }: { project: Project }) {
|
||||
<Spinner />
|
||||
</div>
|
||||
) : !treeData ? (
|
||||
<div className="body-small flex h-full select-none items-center justify-center text-gray-400">Loading...</div>
|
||||
<div className="body-small flex h-full select-none items-center justify-center text-gray-400">
|
||||
Loading...
|
||||
</div>
|
||||
) : (
|
||||
<DragDropContext onDragEnd={onDragEnd}>
|
||||
<Droppable
|
||||
droppableId="root"
|
||||
type="TREE"
|
||||
>
|
||||
{(provided) => (
|
||||
<div
|
||||
ref={provided.innerRef}
|
||||
{...provided.droppableProps}
|
||||
<List
|
||||
height={Math.min(flatData.length * 40, containerHeight)}
|
||||
itemCount={flatData.length}
|
||||
itemSize={40}
|
||||
width={'100%'}
|
||||
itemData={flatData}
|
||||
itemKey={getItemKey}
|
||||
className="flex-1 overflow-auto"
|
||||
style={{ overflowX: 'hidden' }}
|
||||
>
|
||||
{renderTree(treeData.children!, 'root')}
|
||||
{provided.placeholder}
|
||||
</div>
|
||||
)}
|
||||
</Droppable>
|
||||
</DragDropContext>
|
||||
{Row}
|
||||
</List>
|
||||
)}
|
||||
</div>
|
||||
<div className="flex">
|
||||
<AutoLabelButton projectId={project.id} />
|
||||
</div>
|
||||
</div>
|
||||
</DndProvider>
|
||||
);
|
||||
}
|
||||
|
||||
@ -31,7 +31,25 @@ export default function useTreeData(projectId: string, initialFolderId: number)
|
||||
node.children = [...childFolders, ...images];
|
||||
node.loading = false;
|
||||
node.toggled = true;
|
||||
setTreeData((prevData) => ({ ...prevData! }));
|
||||
|
||||
setTreeData((prevData) => {
|
||||
if (!prevData) return null;
|
||||
|
||||
const updateNode = (currentNode: TreeNode): TreeNode => {
|
||||
if (currentNode.id === node.id) {
|
||||
return { ...node };
|
||||
}
|
||||
if (currentNode.children) {
|
||||
return {
|
||||
...currentNode,
|
||||
children: currentNode.children.map(updateNode),
|
||||
};
|
||||
}
|
||||
return currentNode;
|
||||
};
|
||||
|
||||
return updateNode(prevData);
|
||||
});
|
||||
} catch (error) {
|
||||
node.loading = false;
|
||||
console.error(`Error fetching data for node ${node.id}:`, error);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user