github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

tidy(ui): extract util in dnd

Browse Source
This commit is contained in:
psychedelicious
2025-02-11 13:24:04 +11:00
parent 1104d2a00f
commit 330a0f0028
1 changed files with 28 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
invokeai/frontend/web/src/features/nodes/components/sidePanel/builder/dnd.ts
View File

@@ -87,6 +87,32 @@ const flashElement = (elementId: ElementId) => {
}
};
const getAllowedDropRegions = (element: FormElement): CenterOrEdge[] => {
const dropRegions: CenterOrEdge[] = [];
if (isContainerElement(element) && element.data.children.length === 0) {
dropRegions.push('center');
}
// Parent is a container
if (element.parentId !== undefined) {
const parentContainer = getElement(element.parentId, isContainerElement);
if (parentContainer.data.layout === 'row') {
dropRegions.push('left', 'right');
} else {
// parentContainer.data.layout === 'row'
dropRegions.push('top', 'bottom');
}
}
// Parent is the root
if (element.parentId === undefined) {
dropRegions.push('top', 'bottom');
}
return dropRegions;
};
export const useBuilderDndMonitor = () => {
useAssertSingleton('useBuilderDndMonitor');
const dispatch = useAppDispatch();
@@ -538,29 +564,10 @@ export const useFormElementDnd = (
isFormElementDndData(source.data) && source.data.element.id !== getElement(elementId).parentId,
getData: ({ input }) => {
const element = getElement(elementId);
const targetData = buildFormElementDndData(element);
const allowedCenterOrEdge: CenterOrEdge[] = [];
if (isContainerElement(element) && element.data.children.length === 0) {
allowedCenterOrEdge.push('center');
}
// Parent is a container
if (element.parentId !== undefined) {
const parentContainer = getElement(element.parentId, isContainerElement);
if (parentContainer.data.layout === 'row') {
allowedCenterOrEdge.push('left', 'right');
} else {
// parentContainer.data.layout === 'row'
allowedCenterOrEdge.push('top', 'bottom');
}
}
// Parent is the root
if (element.parentId === undefined) {
allowedCenterOrEdge.push('top', 'bottom');
}
const allowedCenterOrEdge = getAllowedDropRegions(element);
return attachClosestCenterOrEdge(targetData, {
element: draggableElement,