fix: improve row selection handling and clean up code, fixes: #1504

Author: kevenleone

ui/studio/grid/DataGrid.tsx

@@ -927,7 +927,7 @@ export function DataGrid(props: DataGridProps) {
       });
       const shouldEnablePreview = Boolean(
         dragDropTarget.compatibleOverId &&
-        dragDropTarget.compatibleOverId !== activeId,
+          dragDropTarget.compatibleOverId !== activeId,
       );
       setIsColumnReorderPreviewEnabled((current) =>
         current === shouldEnablePreview ? current : shouldEnablePreview,
@@ -1470,8 +1470,8 @@ export function DataGrid(props: DataGridProps) {
       const text = hasRowSelection
         ? getSelectedRowClipboardText()
         : hasCellSelection
-          ? getSelectedClipboardText()
-          : getFocusedCellClipboardText();
+        ? getSelectedClipboardText()
+        : getFocusedCellClipboardText();
 
       if (!text) {
         return;
@@ -2064,12 +2064,26 @@ export function DataGrid(props: DataGridProps) {
         return;
       }
 
-      selectSingleRowMode({ rowId, rowIndex, drag: isPrimaryButton });
+      clearNativeTextSelection();
+      clearCellSelectionState();
+
+      const nextSelection = { ...rowSelectionState };
+
+      if (nextSelection[rowId]) {
+        delete nextSelection[rowId];
+      } else {
+        nextSelection[rowId] = true;
+      }
+
+      setRowSelection(nextSelection);
+
+      rowSelectionDragRef.current = isPrimaryButton;
+      rowSelectionAnchorRef.current = rowIndex;
     },
     [
       clearCellSelectionState,
+      clearNativeTextSelection,
       rowSelectionState,
-      selectSingleRowMode,
       setRowSelection,
     ],
   );