@onehat/ui 0.3.227 → 0.3.231

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@onehat/ui",
-	"version": "0.3.227",
+	"version": "0.3.231",
 	"description": "Base UI for OneHat apps",
 	"main": "src/index.js",
 	"type": "module",
@@ -26,7 +26,7 @@
 	"license": "UNLICENSED",
 	"dependencies": {
 		"@gluestack-style/react": "^0.2.38",
-		"@gluestack-ui/themed": "^0.1.53",
+		"@gluestack-ui/themed": "^1.1.22",
 		"@hookform/resolvers": "^3.3.1",
 		"@k-renwick/colour-mixer": "^1.2.1",
 		"@onehat/data": "^1.20.0",

package/src/Components/Grid/Grid.js

@@ -50,12 +50,12 @@ import getIconButtonFromConfig from '../../Functions/getIconButtonFromConfig.js'
 import testProps from '../../Functions/testProps.js';
 import nbToRgb from '../../Functions/nbToRgb.js';
 import Loading from '../Messages/Loading.js';
-import GridHeaderRow from './GridHeaderRow.js';
-import GridRow from './GridRow.js';
+import GridHeaderRow from '../Grid/GridHeaderRow.js';
+import GridRow, { DragSourceDropTargetGridRow, DragSourceGridRow, DropTargetGridRow } from './GridRow.js';
 import IconButton from '../Buttons/IconButton.js';
 import ExpandButton from '../Buttons/ExpandButton.js';
 import PaginationToolbar from '../Toolbar/PaginationToolbar.js';
-import NoRecordsFound from './NoRecordsFound.js';
+import NoRecordsFound from '../Grid/NoRecordsFound.js';
 import Toolbar from '../Toolbar/Toolbar.js';
 import NoReorderRows from '../Icons/NoReorderRows.js';
 import ReorderRows from '../Icons/ReorderRows.js';
@@ -204,12 +204,13 @@ function GridComponent(props) {
 		gridContainerRef = useRef(),
 		isAddingRef = useRef(),
 		expandedRowsRef = useRef({}),
+		cachedDragElements = useRef(),
+		idsRef = useRef([]),
 		[isInited, setIsInited] = useState(false),
 		[isReady, setIsReady] = useState(false),
 		[isLoading, setIsLoading] = useState(false),
 		[localColumnsConfig, setLocalColumnsConfigRaw] = useState([]),
 		[isDragMode, setIsDragMode] = useState(false),
-		[cachedDragElements, setCachedDragElements] = useState(null),
 		[dragRow, setDragRow] = useState(),
 		getIsExpanded = (index) => {
 			return !!expandedRowsRef.current[index];
@@ -303,8 +304,7 @@ function GridComponent(props) {
 		},
 		getFooterToolbarItems = () => {
 			const items = _.map(additionalToolbarButtons, (config, ix) => getIconButtonFromConfig(config, ix, self));
-
-			if (canRowsReorder) {
+			if (canRowsReorder && CURRENT_MODE === UI_MODE_WEB) { // DND is currently web-only  TODO: implement for RN
 				items.unshift(<IconButton
 					key="reorderBtn"
 					parent={self}
@@ -332,159 +332,184 @@ function GridComponent(props) {
 				rowProps = getRowProps && !isHeaderRow ? getRowProps(item) : {},
 				isSelected = !isHeaderRow && !disableWithSelection && isInSelection(item);
 
-			let rowComponent = <Pressable
-						// {...testProps(Repository ? Repository.schema.name + '-' + item.id : item.id)}
-						onPress={(e) => {
-							if (e.preventDefault && e.cancelable) {
-								e.preventDefault();
-							}
-							if (isHeaderRow || isDragMode) {
-								return
-							}
-							if (CURRENT_MODE === UI_MODE_WEB) {
-								switch (e.detail) {
-									case 1: // single click
-										onRowClick(item, e); // sets selection
-										if (onEditorRowClick) {
-											onEditorRowClick(item, index, e);
-										}
-										break;
-									case 2: // double click
-										if (!isSelected) { // If a row was already selected when double-clicked, the first click will deselect it,
-											onRowClick(item, e); // so reselect it
+			let rowComponent =
+				<Pressable
+					// {...testProps(Repository ? Repository.schema.name + '-' + item.id : item.id)}
+					onPress={(e) => {
+						if (e.preventDefault && e.cancelable) {
+							e.preventDefault();
+						}
+						if (isHeaderRow || isDragMode) {
+							return
+						}
+						if (CURRENT_MODE === UI_MODE_WEB) {
+							switch (e.detail) {
+								case 1: // single click
+									onRowClick(item, e); // sets selection
+									if (onEditorRowClick) {
+										onEditorRowClick(item, index, e);
+									}
+									break;
+								case 2: // double click
+									if (!isSelected) { // If a row was already selected when double-clicked, the first click will deselect it,
+										onRowClick(item, e); // so reselect it
+									}
+
+									if (UiGlobals.doubleClickingGridRowOpensEditorInViewMode) { // global setting
+										if (onView) {
+											onView(true);
 										}
-
-										if (UiGlobals.doubleClickingGridRowOpensEditorInViewMode) { // global setting
-											if (onView) {
-												onView(true);
-											}
-										} else {
-											if (onEdit) {
-												if (verifyCanEdit && !verifyCanEdit(selection)) {
-													return;
-												}
-												onEdit();
+									} else {
+										if (onEdit) {
+											if (verifyCanEdit && !verifyCanEdit(selection)) {
+												return;
 											}
+											onEdit();
 										}
-										break;
-									case 3: // triple click
-										break;
-									default:
-								}
-							} else if (CURRENT_MODE === UI_MODE_REACT_NATIVE) {
-								onRowClick(item, e); // sets selection
-								if (onEditorRowClick) {
-									onEditorRowClick(item, index, e);
-								}
-							}
-						}}
-						onLongPress={(e) => {
-							if (e.preventDefault && e.cancelable) {
-								e.preventDefault();
-							}
-							if (isHeaderRow || isDragMode) {
-								return
-							}
-							
-							// context menu
-							const selection = [item];
-							if (!disableWithSelection) {
-								setSelection(selection);
+									}
+									break;
+								case 3: // triple click
+									break;
+								default:
 							}
-							if (onEditorRowClick) { // e.g. inline editor
+						} else if (CURRENT_MODE === UI_MODE_REACT_NATIVE) {
+							onRowClick(item, e); // sets selection
+							if (onEditorRowClick) {
 								onEditorRowClick(item, index, e);
 							}
-							if (onContextMenu) {
-								onContextMenu(item, e, selection, setSelection);
-							}
-						}}
-						flexDirection="row"
-						flexGrow={1}
-					>
-						{({
-							isHovered,
-							isFocused,
-							isPressed,
-						}) => {
-							if (isHeaderRow) {
-								return <GridHeaderRow
-											Repository={Repository}
-											columnsConfig={localColumnsConfig}
-											setColumnsConfig={setLocalColumnsConfig}
-											hideNavColumn={hideNavColumn}
-											canColumnsSort={canColumnsSort}
-											canColumnsReorder={canColumnsReorder}
-											canColumnsResize={canColumnsResize}
-											setSelection={setSelection}
-											gridRef={gridRef}
-											isHovered={isHovered}
-											isInlineEditorShown={isInlineEditorShown}
-										/>;
-							}
+						}
+					}}
+					onLongPress={(e) => {
+						if (e.preventDefault && e.cancelable) {
+							e.preventDefault();
+						}
+						if (isHeaderRow || isDragMode) {
+							return
+						}
+						
+						// context menu
+						const selection = [item];
+						if (!disableWithSelection) {
+							setSelection(selection);
+						}
+						if (onEditorRowClick) { // e.g. inline editor
+							onEditorRowClick(item, index, e);
+						}
+						if (onContextMenu) {
+							onContextMenu(item, e, selection, setSelection);
+						}
+					}}
+					flexDirection="row"
+					flexGrow={1}
+				>
+					{({
+						isHovered,
+						isFocused,
+						isPressed,
+					}) => {
+						if (isHeaderRow) {
+							return <GridHeaderRow
+										Repository={Repository}
+										columnsConfig={localColumnsConfig}
+										setColumnsConfig={setLocalColumnsConfig}
+										hideNavColumn={hideNavColumn}
+										canColumnsSort={canColumnsSort}
+										canColumnsReorder={canColumnsReorder}
+										canColumnsResize={canColumnsResize}
+										setSelection={setSelection}
+										gridRef={gridRef}
+										isHovered={isHovered}
+										isInlineEditorShown={isInlineEditorShown}
+									/>;
+						}
 
-							let bg = rowProps.bg || styles.GRID_ROW_BG,
-								mixWith;
-							if (isSelected) {
-								if (showHovers && isHovered) {
-									mixWith = styles.GRID_ROW_SELECTED_HOVER_BG;
-								} else {
-									mixWith = styles.GRID_ROW_SELECTED_BG;
-								}
-							} else if (showHovers && isHovered) {
-								mixWith = styles.GRID_ROW_HOVER_BG;
-							} else if (alternateRowBackgrounds && index % alternatingInterval === 0) { // i.e. every second line, or every third line
-								mixWith = styles.GRID_ROW_ALTERNATE_BG;
-							}
-							if (mixWith) {
-								const
-									mixWithObj = nbToRgb(mixWith),
-									ratio = mixWithObj.alpha ? 1 - mixWithObj.alpha : 0.5;
-								bg = colourMixer.blend(bg, ratio, mixWithObj.color);
+						let bg = rowProps.bg || styles.GRID_ROW_BG,
+							mixWith;
+						if (isSelected) {
+							if (showHovers && isHovered) {
+								mixWith = styles.GRID_ROW_SELECTED_HOVER_BG;
+							} else {
+								mixWith = styles.GRID_ROW_SELECTED_BG;
 							}
+						} else if (showHovers && isHovered) {
+							mixWith = styles.GRID_ROW_HOVER_BG;
+						} else if (alternateRowBackgrounds && index % alternatingInterval === 0) { // i.e. every second line, or every third line
+							mixWith = styles.GRID_ROW_ALTERNATE_BG;
+						}
+						if (mixWith) {
 							const
-								rowReorderProps = {},
-								rowDragProps = {};
+								mixWithObj = nbToRgb(mixWith),
+								ratio = mixWithObj.alpha ? 1 - mixWithObj.alpha : 0.5;
+							bg = colourMixer.blend(bg, ratio, mixWithObj.color);
+						}
+						const
+							rowReorderProps = {},
+							rowDragProps = {};
+						let WhichRow = GridRow;
+						if (CURRENT_MODE === UI_MODE_WEB) { // DND is currrently web-only  TODO: implement for RN
+							// Create a method that gets an always-current copy of the selection ids
+							const ids = _.map(selection, (item) => item.id);
+							idsRef.current = ids;
+							const getIds = () => idsRef.current;
+
+							// assign event handlers
 							if (canRowsReorder && isDragMode) {
-								rowReorderProps.isDraggable = true;
-								rowReorderProps.mode = VERTICAL;
-								rowReorderProps.onDragStart = onRowReorderDragStart;
-								rowReorderProps.onDrag = onRowReorderDrag;
-								rowReorderProps.onDragStop = onRowReorderDragStop;
-								rowReorderProps.proxyParent = gridRef.current?.getScrollableNode().children[0];
-								rowReorderProps.proxyPositionRelativeToParent = true;
-								rowReorderProps.getParentNode = (node) => node.parentElement.parentElement.parentElement;
-								rowReorderProps.getProxy = getReorderProxy;
+								WhichRow = DragSourceGridRow;
+								rowReorderProps.isDragSource = true;
+								rowReorderProps.dragSourceType = 'row';
+								const dragIx = showHeaders ? index - 1 : index;
+								rowReorderProps.dragSourceItem = {
+									id: item.id,
+									getIds,
+									onDrag: (dragState) => {
+										onRowReorderDrag(dragState, dragIx);
+									},
+								};
+								rowReorderProps.onDragEnd = onRowReorderEnd;
 							} else {
 								// Don't allow drag/drop from withDnd while reordering
 								if (areRowsDragSource) {
+									WhichRow = DragSourceGridRow;
 									rowDragProps.isDragSource = true;
 									rowDragProps.dragSourceType = rowDragSourceType;
-									rowDragProps.dragSourceItem = getRowDragSourceItem ? getRowDragSourceItem(item) : { id: item.id };
+									if (getRowDragSourceItem) {
+										rowDragProps.dragSourceItem = getRowDragSourceItem(item, getIds);
+									} else {
+										rowDragProps.dragSourceItem = {
+											id: item.id,
+											getIds,
+										};
+									}
 								}
 								if (areRowsDropTarget) {
+									WhichRow = DropTargetGridRow;
 									rowDragProps.isDropTarget = true;
 									rowDragProps.dropTargetAccept = dropTargetAccept;
 									rowDragProps.onDrop = (droppedItem) => {
-										// TODO: the item is somehow getting stale
-										// might have something to do with memoization
-										onRowDrop(item, droppedItem);
+										onRowDrop(item, droppedItem); // item is what it was dropped on; droppedItem is the dragSourceItem defined above
 									};
 								}
+								if (areRowsDragSource && areRowsDropTarget) {
+									WhichRow = DragSourceDropTargetGridRow;
+								}
 							}
-							return <GridRow
-										columnsConfig={localColumnsConfig}
-										columnProps={columnProps}
-										fields={fields}
-										rowProps={rowProps}
-										hideNavColumn={hideNavColumn}
-										bg={bg}
-										item={item}
-										isInlineEditorShown={isInlineEditorShown}
-										{...rowReorderProps}
-										{...rowDragProps}
-									/>;
-						}}
-					</Pressable>;
+						}
+						return <WhichRow
+									columnsConfig={localColumnsConfig}
+									columnProps={columnProps}
+									fields={fields}
+									rowProps={rowProps}
+									hideNavColumn={hideNavColumn}
+									bg={bg}
+									item={item}
+									isInlineEditorShown={isInlineEditorShown}
+									{...rowReorderProps}
+									{...rowDragProps}
+
+									key1={item.id}
+								/>;
+					}}
+				</Pressable>;
 
 			if (showRowExpander && !isHeaderRow) {
 				const isExpanded = getIsExpanded(index);
@@ -505,31 +530,10 @@ function GridComponent(props) {
 			}
 			return rowComponent;
 		},
-		getReorderProxy = (node) => {
-			const
-				row = node.parentElement.parentElement,
-				rowRect = row.getBoundingClientRect(),
-				parent = row.parentElement,
-				parentRect = parent.getBoundingClientRect(),
-				proxy = row.cloneNode(true),
-				top = rowRect.top - parentRect.top;
-			
-			setDragRow(row);
-
-			proxy.style.top = top + 'px';
-			proxy.style.left = '20px';
-			proxy.style.height = rowRect.height + 'px';
-			proxy.style.width = rowRect.width + 'px';
-			proxy.style.display = 'flex';
-			// proxy.style.backgroundColor = '#ccc';
-			proxy.style.position = 'absolute';
-			proxy.style.border = '1px solid #000';
-			return proxy;
-		},
 		getOverState = (rows, currentY, mouseX) => {
 			// determines which row the mouse is over
 			// and whether the marker should be moved to the top or bottom of the row
-			let newIx = 0,
+			let newIx = -1,
 				useBottom = false;
 			_.each(rows, (row, ix) => {
 				const
@@ -562,71 +566,89 @@ function GridComponent(props) {
 				useBottom,
 			};
 		},
-		onRowReorderDragStart = (info, e, proxy, node) => {
-			// console.log('onRowReorderDragStart', info, e, proxy, node);
-
+		buildCachedDragElements = (dragState) => {
 			const
-				proxyRect = proxy.getBoundingClientRect(),
-				row = node.parentElement.parentElement,
-				flatlist = row.parentElement,
+				{
+					canDrag,
+					isDragging,
+					clientOffset,
+					sourceClientOffset,
+				} = dragState,
+
+				scrollRef = gridRef.current._listRef._scrollRef,
+				isUsingScroll = scrollRef.childNodes[0].style.overflow === 'auto',
+				flatlist = isUsingScroll ? scrollRef.childNodes[0] : scrollRef,
 				flatlistRect = flatlist.getBoundingClientRect(),
 				rows = _.filter(flatlist.childNodes, (childNode, ix) => {
 					const
 						isZeroHeight = childNode.getBoundingClientRect().height === 0,
-						isProxy = childNode === proxy,
 						isHeader = showHeaders && ix === 0;
-					return !isZeroHeight && !isProxy && !isHeader;
+					return !isZeroHeight && !isHeader;
 				}),
-				currentY = proxyRect.top - flatlistRect.top, // top position of pointer, relative to page
-				{ ix, useBottom } = getOverState(rows, currentY, e.clientX);
+				{ ix, useBottom } = getOverState(rows, clientOffset.y, clientOffset.x);
 
 
 			// Render marker showing destination location
-			const
-				rowContainerRect = rows[ix].getBoundingClientRect(),
-				top = (useBottom ? rowContainerRect.bottom : rowContainerRect.top) 
-						- flatlistRect.top 
-						- (flatlist.style.borderWidth ? parseInt(flatlist.style.borderWidth) : 0), // get relative Y position
-				marker = document.createElement('div');
+			const marker = document.createElement('div');
 			marker.style.position = 'absolute';
-			marker.style.top = top + 'px';
-			marker.style.height = '4px';
+			marker.style.top = '0px';
+			marker.style.height = '8px';
 			marker.style.width = flatlistRect.width + 'px';
-			marker.style.backgroundColor = '#f00';
+			marker.style.backgroundColor = '#ccc';
 			flatlist.appendChild(marker);
+			
+			if (ix !== -1) {
+				marker.style.visibility = 'visible';
+				const
+					rowContainerRect = rows[ix].getBoundingClientRect(),
+					top = (useBottom ? rowContainerRect.bottom : rowContainerRect.top) 
+							- flatlistRect.top 
+							- (flatlist.style.borderWidth ? parseInt(flatlist.style.borderWidth) : 0); // get relative Y position
+				marker.style.top = top + 'px';
+			} else {
+				marker.style.visibility = 'hidden';
+			}
 
-			setCachedDragElements({ ix, useBottom, marker, rows, });
+			return { ix, useBottom, marker, rows };
 		},
-		onRowReorderDrag = (info, e, proxy, node) => {
-			// console.log('onRowReorderDrag', info, e, proxy, node);
+		onRowReorderDrag = (dragState, dragIx) => {
+			// initial setup
+			if (!cachedDragElements.current) {
+				cachedDragElements.current = buildCachedDragElements(dragState);
+			}
+
 			const
-				{ marker, rows, } = cachedDragElements,
-				proxyRect = proxy.getBoundingClientRect(),
-				row = node.parentElement.parentElement,
-				flatlist = row.parentElement,
+				{
+					canDrag,
+					isDragging,
+					clientOffset,
+					sourceClientOffset,
+				} = dragState,
+				{ marker, rows, } = cachedDragElements.current,
+				flatlist = gridRef.current._listRef._scrollRef.childNodes[0],
 				flatlistRect = flatlist.getBoundingClientRect(),
-				currentY = proxyRect.top - flatlistRect.top, // top position of pointer, relative to page
-				{ ix, useBottom } = getOverState(rows, currentY, e.clientX);
-			
+				{ ix, useBottom } = getOverState(rows, clientOffset.y, clientOffset.x);
 
 			// move marker to new location
-			const
-				rowContainerRect = rows[ix].getBoundingClientRect(),
-				top = (useBottom ? rowContainerRect.bottom : rowContainerRect.top) 
-						- flatlistRect.top 
-						- (flatlist.style.borderWidth ? parseInt(flatlist.style.borderWidth) : 0); // get relative Y position
-			marker.style.top = top + 'px';
+			if (ix !== -1) {
+				marker.style.visibility = 'visible';
+				const
+					rowContainerRect = rows[ix].getBoundingClientRect(),
+					top = (useBottom ? rowContainerRect.bottom : rowContainerRect.top) 
+							- flatlistRect.top 
+							- (flatlist.style.borderWidth ? parseInt(flatlist.style.borderWidth) : 0); // get relative Y position
+				marker.style.top = top + 'px';
+			} else {
+				marker.style.visibility = 'hidden';
+			}
 
-			setCachedDragElements({ ix, useBottom, marker, rows, });
+			cachedDragElements.current = { ix, useBottom, marker, rows, dragIx };
 		},
-		onRowReorderDragStop = (delta, e, config) => {
-			// console.log('onRowReorderDragStop', delta, e, config);
+		onRowReorderEnd = (item, monitor) => {
 			const
-				{ ix: dropIx, useBottom, marker, rows, } = cachedDragElements,
-				dragIx = rows.indexOf(dragRow);
-
-			const shouldMove = dropIx !== dragIx;
-			if (shouldMove) {
+				{ ix: dropIx, useBottom, marker, rows, dragIx } = cachedDragElements.current,
+				shouldMove = dropIx !== dragIx;
+			if (shouldMove && dropIx !== -1) {
 				// Update the row with the new ix
 				let dragRecord,
 					dropRecord;
@@ -647,7 +669,7 @@ function GridComponent(props) {
 			}
 
 			marker.remove();
-			setCachedDragElements(null);
+			cachedDragElements.current = null;
 		},
 		calculatePageSize = (containerHeight) => {
 			const
@@ -934,7 +956,7 @@ function GridComponent(props) {
 					nestedScrollEnabled={true}
 					contentContainerStyle={{
 						overflow: 'auto',
-						// flex: 1,
+						height: '100%',
 					}}
 					refreshing={isLoading}
 					onRefresh={pullToRefresh ? onRefresh : null}
@@ -970,15 +992,21 @@ function GridComponent(props) {
 	}
 
 	const gridContainerBorderProps = {};
-	if (isLoading) {
-		gridContainerBorderProps.borderTopWidth = 2;
-		gridContainerBorderProps.borderTopColor = '#f00';
-	} else if (isDragMode) {
+	if (isDragMode) {
 		gridContainerBorderProps.borderWidth = styles.REORDER_BORDER_WIDTH;
 		gridContainerBorderProps.borderColor = styles.REORDER_BORDER_COLOR;
 		gridContainerBorderProps.borderStyle = styles.REORDER_BORDER_STYLE;
+		gridContainerBorderProps.borderTopWidth = null;
+		if (isLoading) {
+			gridContainerBorderProps.borderTopColor = '#f00';
+		} else {
+			gridContainerBorderProps.borderTopColor = null;
+		}
+	} else if (isLoading) {
+		gridContainerBorderProps.borderTopWidth = 4;
+		gridContainerBorderProps.borderTopColor = '#f00';
 	} else {
-		gridContainerBorderProps.borderTopWidth = 1
+		gridContainerBorderProps.borderTopWidth = 1;
 		gridContainerBorderProps.borderTopColor = 'trueGray.300';
 	}
 

package/src/Components/Grid/GridRow.js

@@ -1,4 +1,4 @@
-import { useMemo, } from 'react';
+import { useEffect, useMemo, useRef, } from 'react';
 import {
 	Box,
 	Row,
@@ -10,7 +10,6 @@ import {
 import getComponentFromType from '../../Functions/getComponentFromType.js';
 import UiGlobals from '../../UiGlobals.js';
 import { withDragSource, withDropTarget } from '../Hoc/withDnd.js';
-import withDraggable from '../Hoc/withDraggable.js';
 import AngleRight from '../Icons/AngleRight.js';
 import RowDragHandle from './RowDragHandle.js';
 import _ from 'lodash';
@@ -30,6 +29,8 @@ function GridRow(props) {
 			isDraggable = false, // withDraggable
 			isDragSource = false, // withDnd
 			isOver = false,
+			dragSourceRef,
+			dropTargetRef,
 		} = props,
 		styles = UiGlobals.styles;
 
@@ -199,11 +200,11 @@ function GridRow(props) {
 												/>}
 						</>;
 
-		if (props.dragSourceRef) {
-			rowContents = <Row flexGrow={1} flex={1} w="100%" bg={bg} ref={props.dragSourceRef}>{rowContents}</Row>;
+		if (dragSourceRef) {
+			rowContents = <Row flexGrow={1} flex={1} w="100%" bg={bg} ref={dragSourceRef}>{rowContents}</Row>;
 		}
-		if (props.dropTargetRef) {
-			rowContents = <Row flexGrow={1} flex={1} w="100%" bg={bg} ref={props.dropTargetRef}>{rowContents}</Row>;
+		if (dropTargetRef) {
+			rowContents = <Row flexGrow={1} flex={1} w="100%" bg={bg} ref={dropTargetRef}>{rowContents}</Row>;
 		}
 
 		return <Row
@@ -225,7 +226,14 @@ function GridRow(props) {
 		hash, // this is an easy way to determine if the data has changed and the item needs to be rerendered
 		isInlineEditorShown,
 		isOver,
+		dragSourceRef,
+		dropTargetRef,
 	]);
 }
 
-export default withDraggable(withDragSource(withDropTarget(GridRow)));
+// export default withDraggable(withDragSource(withDropTarget(GridRow)));
+export default GridRow;
+
+export const DragSourceGridRow = withDragSource(GridRow);
+export const DropTargetGridRow = withDropTarget(GridRow);
+export const DragSourceDropTargetGridRow = withDragSource(withDropTarget(GridRow));

package/src/Components/Hoc/withDnd.js

@@ -1,7 +1,12 @@
-import { useDrag, useDrop } from 'react-dnd'; // https://react-dnd.github.io/react-dnd/about don't forget the wrapping <DndProvider /> as shown here: https://react-dnd.github.io/react-dnd/docs/api/dnd-provider
+import { useEffect, useRef, } from 'react';
+import { useDrag, useDrop, useDragLayer } from 'react-dnd'; // https://react-dnd.github.io/react-dnd/about don't forget the wrapping <DndProvider /> as shown here: https://react-dnd.github.io/react-dnd/docs/api/dnd-provider
+
 
 // This HOC allows components to be dragged and dropped onto another component.
-// It doesn't contraint the moment of the preview item.
+// It can't contrain the movement of the preview item, because react-dnd uses 
+// a native drag layer for its proxy, and there are no contraints on that provided by OS.
+// If you need constraints, you can potentially use a CustomDragLayer (see code at bottom)
+// but it will lag behind, compared to what the native drag layer can do
 
 export function withDragSource(WrappedComponent) {
 	return (props) => {
@@ -16,47 +21,71 @@ export function withDragSource(WrappedComponent) {
 		if (!props.dragSourceItem) {
 			throw Error('dragSourceItem not defined');
 		}
-
+		
 		const {
 				dragSourceType,
 				dragSourceItem,
+				dragPreviewOptions = null,
+				dragOptions = null,
+				dropEffect = null,
+				// onDrag,
+				onDragEnd = null,
+				canDrag = null,
+				isDragging = null,
+				dragCollect = (monitor, props2) => { // Optional. The collecting function. It should return a plain object of the props to return for injection into your component. It receives two parameters, monitor and props. Read the overview for an introduction to the monitors and the collecting function. See the collecting function described in detail in the next section.
+					// monitor fn determines which props from dnd state get passed
+					return {
+						// canDrag: !!monitor.canDrag(), // Returns trueif no drag operation is in progress, and the owner's canDrag() returns true or is not defined.
+						isDragging: !!monitor.isDragging(), // Returns trueif a drag operation is in progress, and either the owner initiated the drag, or its isDragging() is defined and returns true.
+						type: monitor.getItemType(), // Returns a string or a symbol identifying the type of the current dragged item. Returns null if no item is being dragged.
+						item: monitor.getItem(), // Returns a plain object representing the currently dragged item. Every drag source must specify it by returning an object from its beginDrag() method. Returns nullif no item is being dragged.
+						dropResult: monitor.getDropResult(), // Returns a plain object representing the last recorded drop result. The drop targets may optionally specify it by returning an object from their drop()methods. When a chain of drop()is dispatched for the nested targets, bottom up, any parent that explicitly returns its own result from drop()overrides the child drop result previously set by the child. Returns nullif called outside endDrag().
+						didDrop: !!monitor.didDrop(), // Returns trueif some drop target has handled the drop event, falseotherwise. Even if a target did not return a drop result, didDrop() returns true. Use it inside endDrag()to test whether any drop target has handled the drop. Returns falseif called outside endDrag().
+						initialClientOffset: monitor.getInitialClientOffset(), // Returns the { x, y }client offset of the pointer at the time when the current drag operation has started. Returns nullif no item is being dragged.
+						initialSourceClientOffset: monitor.getInitialSourceClientOffset(), // Returns the { x, y }client offset of the drag source component's root DOM node at the time when the current drag operation has started. Returns nullif no item is being dragged.
+						clientOffset: monitor.getClientOffset(), // Returns the last recorded { x, y }client offset of the pointer while a drag operation is in progress. Returns nullif no item is being dragged.
+						differenceFromInitialOffset: monitor.getDifferenceFromInitialOffset(), // Returns the { x, y }difference between the last recorded client offset of the pointer and the client offset when the current drag operation has started. Returns nullif no item is being dragged.
+						sourceClientOffset: monitor.getSourceClientOffset(), // Returns the projected { x, y }client offset of the drag source component's root DOM node, based on its position at the time when the current drag operation has started, and the movement difference. Returns nullif no item is being dragged.
+					};
+				},
 			} = props,
 			[dragState, dragSourceRef, dragPreviewRef] = useDrag(() => { // A specification object or a function that creates a specification object.
 				// The useDrag hook provides a way to wire your component into the DnD system as a drag source. By passing in a specification into useDrag, you declaratively describe the typeof draggable being generated, the itemobject representing the drag source, what props to collect, and more. The useDraghooks returns a few key items: a set of collected props, and refs that may be attached to drag source and drag preview elements
+
 				return {
 					type: dragSourceType, // Required. This must be either a string or a symbol. Only the drop targets registered for the same type will react to this item.
 					item: dragSourceItem, // Required (object or function).
 						// When an object, it is a plain JavaScript object describing the data being dragged. This is the only information available to the drop targets about the drag source so it's important to pick the minimal data they need to know. You may be tempted to put a complex reference here, but you should try very hard to avoid doing this because it couples the drag sources and drop targets. It's a good idea to use something like { id }.
 						// When a function, it is fired at the beginning of the drag operation and returns an object representing the drag operation (see first bullet). If null is returned, the drag operation is cancelled.
-					previewOptions: null, // Optional. A plain JavaScript object describing drag preview options.
-					options: null, // Optional. A plain object optionally containing any of the following properties:
-						// dropEffect: Optional: The type of drop effect to use on this drag. ("move" or "copy" are valid values.)
-					end: null, // (item, monitor) Optional. When the dragging stops, endis called. For every begin call, a corresponding end call is guaranteed. You may call monitor.didDrop() to check whether or not the drop was handled by a compatible drop target. If it was handled, and the drop target specified a drop result by returning a plain object from its drop()method, it will be available as monitor.getDropResult(). This method is a good place to fire a Flux action. Note: If the component is unmounted while dragging, componentparameter is set to be null.
-					canDrag: null, // (monitor): Optional. Use it to specify whether the dragging is currently allowed. If you want to always allow it, just omit this method. Specifying it is handy if you'd like to disable dragging based on some predicate over props. Note: You may not call monitor.canDrag()inside this method.
-					isDragging: null, // (monitor): Optional. By default, only the drag source that initiated the drag operation is considered to be dragging. You can override this behavior by defining a custom isDraggingmethod. It might return something like props.id === monitor.getItem().id. Do this if the original component may be unmounted during the dragging and later “resurrected” with a different parent. For example, when moving a card across the lists in a Kanban board, you want it to retain the dragged appearance—even though technically, the component gets unmounted and a different one gets mounted every time you move it to another list. Note: You may not call monitor.isDragging()inside this method.
-					collect: (monitor, props) => { // Optional. The collecting function. It should return a plain object of the props to return for injection into your component. It receives two parameters, monitor and props. Read the overview for an introduction to the monitors and the collecting function. See the collecting function described in detail in the next section.
+					previewOptions: dragPreviewOptions, // Optional. A plain JavaScript object describing drag preview options.
+					options: dragOptions, // Optional. A plain object optionally containing any of the following properties:
+					dropEffect, // Optional: The type of drop effect to use on this drag. ("move" or "copy" are valid values.)
+					end: onDragEnd, // (item, monitor) Optional. When the dragging stops, endis called. For every begin call, a corresponding end call is guaranteed. You may call monitor.didDrop() to check whether or not the drop was handled by a compatible drop target. If it was handled, and the drop target specified a drop result by returning a plain object from its drop()method, it will be available as monitor.getDropResult(). This method is a good place to fire a Flux action. Note: If the component is unmounted while dragging, componentparameter is set to be null.
+					canDrag, // (monitor): Optional. Use it to specify whether the dragging is currently allowed. If you want to always allow it, just omit this method. Specifying it is handy if you'd like to disable dragging based on some predicate over props. Note: You may not call monitor.canDrag()inside this method.
+					isDragging, // (monitor): Optional. By default, only the drag source that initiated the drag operation is considered to be dragging. You can override this behavior by defining a custom isDraggingmethod. It might return something like props.id === monitor.getItem().id. Do this if the original component may be unmounted during the dragging and later “resurrected” with a different parent. For example, when moving a card across the lists in a Kanban board, you want it to retain the dragged appearance—even though technically, the component gets unmounted and a different one gets mounted every time you move it to another list. Note: You may not call monitor.isDragging()inside this method.
+					collect: dragCollect, /* (monitor, props) => { // Optional. The collecting function. It should return a plain object of the props to return for injection into your component. It receives two parameters, monitor and props. Read the overview for an introduction to the monitors and the collecting function. See the collecting function described in detail in the next section.
 						// monitor fn determines which props from dnd state get passed
 						return {
 							canDrag: !!monitor.canDrag(), // Returns trueif no drag operation is in progress, and the owner's canDrag() returns true or is not defined.
 							isDragging: !!monitor.isDragging(), // Returns trueif a drag operation is in progress, and either the owner initiated the drag, or its isDragging() is defined and returns true.
 							// type: monitor.getItemType(), // Returns a string or a symbol identifying the type of the current dragged item. Returns null if no item is being dragged.
-							// item: monitor.getItem(), // Returns a plain object representing the currently dragged item. Every drag source must specify it by returning an object from its beginDrag()method. Returns nullif no item is being dragged.
+							// item: monitor.getItem(), // Returns a plain object representing the currently dragged item. Every drag source must specify it by returning an object from its beginDrag() method. Returns nullif no item is being dragged.
 							// dropResult: monitor.getDropResult(), // Returns a plain object representing the last recorded drop result. The drop targets may optionally specify it by returning an object from their drop()methods. When a chain of drop()is dispatched for the nested targets, bottom up, any parent that explicitly returns its own result from drop()overrides the child drop result previously set by the child. Returns nullif called outside endDrag().
-							// didDrop: !!monitor.didDrop(), // Returns trueif some drop target has handled the drop event, falseotherwise. Even if a target did not return a drop result, didDrop()returns true. Use it inside endDrag()to test whether any drop target has handled the drop. Returns falseif called outside endDrag().
+							// didDrop: !!monitor.didDrop(), // Returns trueif some drop target has handled the drop event, falseotherwise. Even if a target did not return a drop result, didDrop() returns true. Use it inside endDrag()to test whether any drop target has handled the drop. Returns falseif called outside endDrag().
 							// initialClientOffset: monitor.getInitialClientOffset(), // Returns the { x, y }client offset of the pointer at the time when the current drag operation has started. Returns nullif no item is being dragged.
 							// initialSourceClientOffset: monitor.getInitialSourceClientOffset(), // Returns the { x, y }client offset of the drag source component's root DOM node at the time when the current drag operation has started. Returns nullif no item is being dragged.
 							// clientOffset: monitor.getClientOffset(), // Returns the last recorded { x, y }client offset of the pointer while a drag operation is in progress. Returns nullif no item is being dragged.
 							// differenceFromInitialOffset: monitor.getDifferenceFromInitialOffset(), // Returns the { x, y }difference between the last recorded client offset of the pointer and the client offset when the current drag operation has started. Returns nullif no item is being dragged.
 							// sourceClientOffset: monitor.getSourceClientOffset(), // Returns the projected { x, y }client offset of the drag source component's root DOM node, based on its position at the time when the current drag operation has started, and the movement difference. Returns nullif no item is being dragged.
 						};
-					},
+					},*/
 				};
 			}),
 			{
-				canDrag,
-				isDragging,
+				canDrag: stateCanDrag,
+				isDragging: stateIsDragging,
 				// type,
-				// item,
+				// item: stateItem,
 				// dropResult,
 				// didDrop,
 				// initialClientOffset,
@@ -65,12 +94,24 @@ export function withDragSource(WrappedComponent) {
 				// differenceFromInitialOffset,
 				// sourceClientOffset,
 			} = dragState;
+		
+		if (dragSourceItem.onDrag) {
+			// This will enable the onDrag callback, as useDragLayer will cause withDragSource to re-render any time state changes. useDrag will not
+			const layer = useDragLayer(dragCollect);
+			useEffect(() => {
+				if (layer.isDragging && dragSourceItem?.id === layer.item.id) {
+					dragSourceItem.onDrag(layer);
+				}
+			}, [layer]);
+		}
 
 		return <WrappedComponent 
-					canDrag={canDrag}
-					isDragging={isDragging}
-					dragSourceRef={dragSourceRef}
 					{...props}
+					canDrag={stateCanDrag}
+					isDragging={stateIsDragging}
+					dragSourceRef={dragSourceRef}
+					dragPreviewRef={dragPreviewRef}
+					dragState={dragState}
 				/>;
 	};
 }
@@ -78,6 +119,7 @@ export function withDragSource(WrappedComponent) {
 
 export function withDropTarget(WrappedComponent) {
 	return (props) => {
+
 		if (!props.isDropTarget) {
 			return <WrappedComponent {...props} />;
 		}
@@ -88,17 +130,27 @@ export function withDropTarget(WrappedComponent) {
 
 		const {
 				dropTargetAccept,
+				dropOptions = null,
 				onDrop = null,
+				onDropHover = null,
+				canDrop = null,
+				dropCollect = (monitor, props) => {
+					return {
+						canDrop: !!monitor.canDrop(),
+						isOver: !!monitor.isOver(),
+					};
+				},
 			} = props,
+			localTargetRef = useRef(null),
 			[dropState, dropTargetRef] = useDrop(() => { // A specification object or a function that creates a specification object.
 				// The useDrophook provides a way for you to wire in your component into the DnD system as a drop target. By passing in a specification into the useDrophook, you can specify including what types of data items the drop-target will accept, what props to collect, and more. This function returns an array containing a ref to attach to the Drop Target node and the collected props.
 				return {
 					accept: dropTargetAccept, // Required. A string, a symbol, or an array of either. This drop target will only react to the items produced by the drag sources of the specified type or types. Read the overview to learn more about the items and types.
-					// options: null, // Optional. A plain object. If some of the props to your component are not scalar (that is, are not primitive values or functions), specifying a custom arePropsEqual(props, otherProps) function inside the options object can improve the performance. Unless you have performance problems, don't worry about it.
+					options: dropOptions, // Optional. A plain object. If some of the props to your component are not scalar (that is, are not primitive values or functions), specifying a custom arePropsEqual(props, otherProps) function inside the options object can improve the performance. Unless you have performance problems, don't worry about it.
 					drop: onDrop, // (item, monitor): Optional. Called when a compatible item is dropped on the target. You may either return undefined, or a plain object. If you return an object, it is going to become the drop result and will be available to the drag source in its endDragmethod as monitor.getDropResult(). This is useful in case you want to perform different actions depending on which target received the drop. If you have nested drop targets, you can test whether a nested target has already handled dropby checking monitor.didDrop()and monitor.getDropResult(). Both this method and the source's endDragmethod are good places to fire Flux actions. This method will not be called if canDrop()is defined and returns false.
-					// hover: null, // (item, monitor): Optional. Called when an item is hovered over the component. You can check monitor.isOver({ shallow: true })to test whether the hover happens over only the current target, or over a nested one. Unlike drop(), this method will be called even if canDrop()is defined and returns false. You can check monitor.canDrop()to test whether this is the case.
-					// canDrop: null, // (item, monitor): Optional. Use it to specify whether the drop target is able to accept the item. If you want to always allow it, omit this method. Specifying it is handy if you'd like to disable dropping based on some predicate over props or monitor.getItem(). Note: You may not call monitor.canDrop() inside this method.
-					collect: (monitor, props) => { // Optional. The collecting function. It should return a plain object of the props to return for injection into your component. It receives two parameters, monitorand props. Read the overview for an introduction to the monitors and the collecting function. See the collecting function described in detail in the next section.
+					hover: onDropHover, // (item, monitor): Optional. Called when an item is hovered over the component. You can check monitor.isOver({ shallow: true })to test whether the hover happens over only the current target, or over a nested one. Unlike drop(), this method will be called even if canDrop()is defined and returns false. You can check monitor.canDrop()to test whether this is the case.
+					canDrop, // (item, monitor): Optional. Use it to specify whether the drop target is able to accept the item. If you want to always allow it, omit this method. Specifying it is handy if you'd like to disable dropping based on some predicate over props or monitor.getItem(). Note: You may not call monitor.canDrop() inside this method.
+					collect: dropCollect, /*: (monitor, props) => { // Optional. The collecting function. It should return a plain object of the props to return for injection into your component. It receives two parameters, monitorand props. Read the overview for an introduction to the monitors and the collecting function. See the collecting function described in detail in the next section.
 						return {
 							canDrop: !!monitor.canDrop(),
 							isOver: !!monitor.isOver(),
@@ -112,11 +164,11 @@ export function withDropTarget(WrappedComponent) {
 							// receiveHandlerId
 							// subscribeToStateChange
 						};
-					},
+					},*/
 				};
 			}),
 			{
-				canDrop,
+				canDrop: stateCanDrop,
 				isOver,
 				// didDrop,
 				// clientOffset,
@@ -127,11 +179,70 @@ export function withDropTarget(WrappedComponent) {
 				// initialSourceClientOffset,
 			} = dropState;
 			
+		dropTargetRef(localTargetRef); // register DOM node with react-dnd
+
 		return <WrappedComponent 
-					canDrop={canDrop}
+					canDrop={stateCanDrop}
 					isOver={isOver}
-					dropTargetRef={dropTargetRef}
+					dropTargetRef={localTargetRef}
 					{...props}
 				/>;
 	};
 }
+
+
+// export function CustomDragLayer(props) {
+
+// 	// if (CURRENT_MODE !== UI_MODE_WEB) {
+// 	// 	throw Error('CustomDragLayer only works in web mode');
+// 	// }
+
+// 	const
+// 		{
+// 			onDrag,
+// 			axis = null,
+// 		} = props,
+// 		layer = useDragLayer((monitor) => ({
+// 			isDragging: monitor.isDragging(),
+// 			item: monitor.getItem(),
+// 			currentOffset: monitor.getSourceClientOffset(),
+// 		})),
+// 		{ isDragging, item, currentOffset } = layer;
+
+// 	useEffect(() => {
+// 		if (layer.isDragging) {
+// 			onDrag(layer);
+// 		}
+// 	}, [layer]);
+
+// 	return null;
+
+// 	if (!isDragging || !currentOffset) {
+// 		return null;
+// 	}
+
+// 	let transform;
+// 	if (axis === 'x') {
+// 		transform = `translate(${currentOffset.x}px, 0)`;
+// 	} else if (axis === 'y') {
+// 		transform = `translate(0, ${currentOffset.y}px)`;
+// 	} else {
+// 		transform = `translate(${currentOffset.x}px, ${currentOffset.y}px)`;
+// 	}
+
+// 	return (
+// 		<div id="dragLayer" style={{
+// 			background: '#f00',
+// 			position: 'fixed',
+// 			pointerEvents: 'none',
+// 			zIndex: 10000,
+// 			width: '200px',
+// 			height: '10px',
+// 			left: 0,
+// 			top: 0,
+// 			transform,
+// 		}}>
+// 			{children}
+// 		</div>
+// 	);
+// }

package/src/Components/Hoc/withSelection.js

@@ -1,10 +1,11 @@
-import { useState, useEffect, } from 'react';
+import { useState, useEffect, useRef, } from 'react';
 import {
 	SELECTION_MODE_SINGLE,
 	SELECTION_MODE_MULTI,
 	SELECT_UP,
 	SELECT_DOWN,
 } from '../../Constants/Selection.js';
+import useForceUpdate from '../../Hooks/useForceUpdate.js';
 import inArray from '../../Functions/inArray.js';
 import _ from 'lodash';
 
@@ -46,20 +47,22 @@ export default function withSelection(WrappedComponent) {
 			} = props,
 			usesWithValue = !!setValue,
 			initialSelection = selection || defaultSelection || [],
-			[localSelection, setLocalSelection] = useState(initialSelection),
+			forceUpdate = useForceUpdate(),
+			localSelection = useRef(initialSelection),
 			[isReady, setIsReady] = useState(selection || false), // if selection is already defined, or value is not null and we don't need to load repository, it's ready
 			setSelection = (selection) => {
-				if (_.isEqual(selection, localSelection)) {
+				if (_.isEqual(selection, localSelection.current)) {
 					return;
 				}
 
-				setLocalSelection(selection);
+				localSelection.current = selection;
 				if (onChangeSelection) {
 					onChangeSelection(selection);
 				}
 				if (fireEvent) {
 					fireEvent('changeSelection', selection);
 				}
+				forceUpdate();
 			},
 			selectNext = () => {
 				selectDirection(SELECT_DOWN);
@@ -91,23 +94,21 @@ export default function withSelection(WrappedComponent) {
 			},
 			addToSelection = (item) => {
 				let newSelection = [];
-				newSelection = _.clone(localSelection); // so we get a new object, so descendants rerender
+				newSelection = _.clone(localSelection.current); // so we get a new object, so descendants rerender
 				newSelection.push(item);
 				setSelection(newSelection);
 			},
 			removeFromSelection = (item) => {
 				let newSelection = [];
 				if (Repository) {
-					newSelection = _.remove(localSelection, (sel) => sel !== item);
+					newSelection = _.remove(localSelection.current, (sel) => sel !== item);
 				} else {
-					newSelection = _.remove(localSelection, (sel) => sel[idIx] !== item[idIx]);
+					newSelection = _.remove(localSelection.current, (sel) => sel[idIx] !== item[idIx]);
 				}
 				setSelection(newSelection);
 			},
 			deselectAll = () => {
-				if (!_.isEmpty(localSelection)) {
-					setSelection([]);
-				}
+				setSelection([]);
 			},
 			getMaxMinSelectionIndices = () => {
 				let items,
@@ -134,9 +135,9 @@ export default function withSelection(WrappedComponent) {
 			selectRangeTo = (item) => {
 				// Select above max or below min to this one
 				const
-					currentSelectionLength = localSelection.length,
+					currentSelectionLength = localSelection.current.length,
 					index = getIndexOfSelectedItem(item);
-				let newSelection = _.clone(localSelection); // so we get a new object, so descendants rerender
+				let newSelection = _.clone(localSelection.current); // so we get a new object, so descendants rerender
 
 				if (currentSelectionLength) {
 					const { items, max, min, } = getMaxMinSelectionIndices();
@@ -163,10 +164,10 @@ export default function withSelection(WrappedComponent) {
 			},
 			isInSelection = (item) => {
 				if (Repository) {
-					return inArray(item, localSelection);
+					return inArray(item, localSelection.current);
 				}
 
-				const found = _.find(localSelection, (selectedItem) => {
+				const found = _.find(localSelection.current, (selectedItem) => {
 						return selectedItem[idIx] === item[idIx];
 					});
 				return !!found;
@@ -188,10 +189,10 @@ export default function withSelection(WrappedComponent) {
 				return found;
 			},
 			getIdsFromLocalSelection = () => {
-				if (!localSelection[0]) {
+				if (!localSelection.current[0]) {
 					return null;
 				}
-				const values = _.map(localSelection, (item) => {
+				const values = _.map(localSelection.current, (item) => {
 					if (Repository) {
 						return item.id;
 					}
@@ -202,7 +203,7 @@ export default function withSelection(WrappedComponent) {
 				}
 				return values;
 			},
-			getDisplayValuesFromLocalSelection = (selection) => {
+			getDisplayValuesFromSelection = (selection) => {
 				if (!selection[0]) {
 					return '';
 				}
@@ -275,11 +276,21 @@ export default function withSelection(WrappedComponent) {
 					}
 				}
 
-				if (!_.isEqual(newSelection, localSelection)) {
+				if (!_.isEqual(newSelection, localSelection.current)) {
 					setSelection(newSelection);
 				}
 			};
 
+		if (Repository) {
+			useEffect(() => {
+				// clear the selection when Repository loads
+				Repository.on('load', deselectAll);
+				return () => {
+					Repository.off('load', deselectAll);
+				};
+			}, []);
+		}
+
 		useEffect(() => {
 
 			(async () => {
@@ -317,7 +328,7 @@ export default function withSelection(WrappedComponent) {
 		}, [value]);
 
 		if (self) {
-			self.selection = localSelection;
+			self.selection = localSelection.current;
 			self.setSelection = setSelection;
 			self.selectNext = selectNext;
 			self.selectPrev = selectPrev;
@@ -327,7 +338,7 @@ export default function withSelection(WrappedComponent) {
 			self.selectRangeTo = selectRangeTo;
 			self.isInSelection = isInSelection;
 			self.getIdsFromLocalSelection = getIdsFromLocalSelection;
-			self.getDisplayValuesFromSelection = getDisplayValuesFromLocalSelection;
+			self.getDisplayValuesFromSelection = getDisplayValuesFromSelection;
 		}
 
 		if (usesWithValue) {
@@ -357,7 +368,7 @@ export default function withSelection(WrappedComponent) {
 		return <WrappedComponent
 					{...props}
 					disableWithSelection={false}
-					selection={localSelection}
+					selection={localSelection.current}
 					setSelection={setSelection}
 					selectionMode={selectionMode}
 					selectNext={selectNext}
@@ -368,7 +379,7 @@ export default function withSelection(WrappedComponent) {
 					selectRangeTo={selectRangeTo}
 					isInSelection={isInSelection}
 					getIdsFromSelection={getIdsFromLocalSelection}
-					getDisplayValuesFromSelection={getDisplayValuesFromLocalSelection}
+					getDisplayValuesFromSelection={getDisplayValuesFromSelection}
 				/>;
 	};
 }