Refactor:리액트 윈도우 추가 드래그 앤 드랍 추가

This commit is contained in:
정현조 2024-10-04 02:55:06 +09:00
parent 6b4946b273
commit dacc7e357e
2 changed files with 217 additions and 152 deletions

View File

@ -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,124 +127,72 @@ 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) => (
<div
ref={provided.innerRef}
{...provided.draggableProps}
style={{
...provided.draggableProps.style,
display: 'flex',
flexDirection: 'column',
}}
>
<div
{...provided.dragHandleProps}
style={{
display: 'flex',
alignItems: 'center',
paddingLeft: `${level * 20}px`,
cursor: 'pointer',
}}
onClick={() => onToggle(node, !node.toggled)}
>
<div style={{ marginRight: '5px' }}>
{!node.imageData ? (
<Folder
size={16}
className="stroke-gray-500"
/>
) : (
<ImageIcon
size={16}
className="stroke-gray-500"
/>
)}
</div>
<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 flattenTree = useCallback((nodes: TreeNode[], depth: number = 0, parent?: FlatNode): FlatNode[] => {
let flatList: FlatNode[] = [];
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;
nodes.forEach((node, index) => {
const flatNode: FlatNode = {
...node,
depth,
isLeaf: !node.children || node.children.length === 0,
parent,
index,
};
flatList.push(flatNode);
const sourceParentNode = sourceDroppableId === 'root' ? treeData : findNodeById([treeData], sourceDroppableId);
const destinationParentNode =
destinationDroppableId === 'root' ? treeData : findNodeById([treeData], destinationDroppableId);
if (!sourceParentNode || !destinationParentNode) {
return;
if (node.toggled && node.children) {
flatList = flatList.concat(flattenTree(node.children, depth + 1, flatNode));
}
});
const [movedItem] = sourceParentNode.children!.splice(sourceIndex, 1);
return flatList;
}, []);
destinationParentNode.children!.splice(destinationIndex, 0, movedItem);
const flatData = useMemo(() => {
if (!treeData) return [];
return flattenTree([treeData]);
}, [treeData, flattenTree]);
setTreeData({ ...treeData });
const getItemKey = (index: number, data: FlatNode[]) => data[index].id!;
if (movedItem && movedItem.imageData) {
const moveFolderId = Number(destinationParentNode.id) || 0;
const folderId = Number(sourceParentNode.id) || 0;
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: movedItem.imageData.id,
imageId: dragItem.imageData.id,
moveRequest: {
moveFolderId,
},
@ -218,49 +202,112 @@ export default function ProjectStructure({ project }: { project: Project }) {
[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={ref}
style={{
...style,
opacity: isDragging ? 0.5 : 1,
display: 'flex',
alignItems: 'center',
paddingLeft: `${node.depth * 20}px`,
cursor: 'pointer',
}}
onClick={() => onToggle(node, !node.toggled)}
>
<div style={{ marginRight: '5px' }}>
{!node.imageData ? (
<Folder
size={16}
className="stroke-gray-500"
/>
) : (
<ImageIcon
size={16}
className="stroke-gray-500"
/>
)}
</div>
<span style={{ color: '#4a4a4a', flexGrow: 1 }}>{node.name}</span>
{node.imageData && <div style={{ marginLeft: '10px' }}>{renderStatusIcon(node.imageData.status)}</div>}
</div>
);
};
return (
<div className="box-border flex h-full min-h-0 flex-col bg-gray-50 p-2">
<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">
<h2 className="caption overflow-hidden text-ellipsis whitespace-nowrap text-gray-600">{project.type}</h2>
</div>
<WorkspaceDropdownMenu
projectId={project.id}
folderId={0}
onRefetch={() => {}}
/>
</header>
{isLoading ? (
<div className="flex h-full items-center justify-center">
<Spinner />
</div>
) : !treeData ? (
<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"
<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">
<h2 className="caption overflow-hidden text-ellipsis whitespace-nowrap text-gray-600">{project.type}</h2>
</div>
<WorkspaceDropdownMenu
projectId={project.id}
folderId={0}
onRefetch={() => {}}
/>
</header>
{isLoading ? (
<div className="flex h-full items-center justify-center">
<Spinner />
</div>
) : !treeData ? (
<div className="body-small flex h-full select-none items-center justify-center text-gray-400">
Loading...
</div>
) : (
<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' }}
>
{(provided) => (
<div
ref={provided.innerRef}
{...provided.droppableProps}
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>
<div className="flex">
<AutoLabelButton projectId={project.id} />
</div>
</div>
</DndProvider>
);
}

View File

@ -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);