From afbf0cb01edf09f4169fbef7878412e619b1d078 Mon Sep 17 00:00:00 2001
From: Jan Dockal <jan.dockal@metservice.com>
Date: Wed, 16 Oct 2019 11:58:28 +1300
Subject: [PATCH] feat(DraggableCore): apply scale property while dragging an
 element

---
 lib/DraggableCore.js     |  7 ++++
 lib/utils/domFns.js      |  6 ++--
 lib/utils/positionFns.js |  2 +-
 specs/draggable.spec.jsx | 76 ++++++++++++++++++++++++++++++++++++++++
 4 files changed, 87 insertions(+), 4 deletions(-)

diff --git a/lib/DraggableCore.js b/lib/DraggableCore.js
index 4fbabfd2..2429a612 100644
--- a/lib/DraggableCore.js
+++ b/lib/DraggableCore.js
@@ -67,6 +67,7 @@ export type DraggableCoreProps = {
   onDrag: DraggableEventHandler,
   onStop: DraggableEventHandler,
   onMouseDown: (e: MouseEvent) => void,
+  scale: number,
 };
 
 //
@@ -185,6 +186,11 @@ export default class DraggableCore extends React.Component<DraggableCoreProps, D
      */
     onMouseDown: PropTypes.func,
 
+    /**
+     * `scale`, if set, applies scaling while dragging an element
+     */
+    scale: PropTypes.number,
+
     /**
      * These properties should be defined on the child, not here.
      */
@@ -206,6 +212,7 @@ export default class DraggableCore extends React.Component<DraggableCoreProps, D
     onDrag: function(){},
     onStop: function(){},
     onMouseDown: function(){},
+    scale: 1,
   };
 
   state = {
diff --git a/lib/utils/domFns.js b/lib/utils/domFns.js
index 3c9ab03c..6963da72 100644
--- a/lib/utils/domFns.js
+++ b/lib/utils/domFns.js
@@ -99,12 +99,12 @@ export function innerWidth(node: HTMLElement): number {
 }
 
 // Get from offsetParent
-export function offsetXYFromParent(evt: {clientX: number, clientY: number}, offsetParent: HTMLElement): ControlPosition {
+export function offsetXYFromParent(evt: {clientX: number, clientY: number}, offsetParent: HTMLElement, scale: number): ControlPosition {
   const isBody = offsetParent === offsetParent.ownerDocument.body;
   const offsetParentRect = isBody ? {left: 0, top: 0} : offsetParent.getBoundingClientRect();
 
-  const x = evt.clientX + offsetParent.scrollLeft - offsetParentRect.left;
-  const y = evt.clientY + offsetParent.scrollTop - offsetParentRect.top;
+  const x = (evt.clientX + offsetParent.scrollLeft - offsetParentRect.left) / scale;
+  const y = (evt.clientY + offsetParent.scrollTop - offsetParentRect.top) / scale;
 
   return {x, y};
 }
diff --git a/lib/utils/positionFns.js b/lib/utils/positionFns.js
index e080c86d..3b77b12f 100644
--- a/lib/utils/positionFns.js
+++ b/lib/utils/positionFns.js
@@ -73,7 +73,7 @@ export function getControlPosition(e: MouseTouchEvent, touchIdentifier: ?number,
   const node = findDOMNode(draggableCore);
   // User can provide an offsetParent if desired.
   const offsetParent = draggableCore.props.offsetParent || node.offsetParent || node.ownerDocument.body;
-  return offsetXYFromParent(touchObj || e, offsetParent);
+  return offsetXYFromParent(touchObj || e, offsetParent, draggableCore.props.scale);
 }
 
 // Create an data object exposed by <DraggableCore>'s events
diff --git a/specs/draggable.spec.jsx b/specs/draggable.spec.jsx
index 83725154..9946ad27 100644
--- a/specs/draggable.spec.jsx
+++ b/specs/draggable.spec.jsx
@@ -683,6 +683,44 @@ describe('react-draggable', function () {
 
       simulateMovementFromTo(drag, 200, 200, 300, 300);
     });
+
+    it('should call back with correct position when parent element is 2x scaled', function() {
+      function onDrag(event, data) {
+        // visually it will look like 100, because parent is 2x scaled
+        assert(data.x === 50);
+        assert(data.y === 50);
+        assert(data.deltaX === 50);
+        assert(data.deltaY === 50);
+        assert(data.node === ReactDOM.findDOMNode(drag));
+      }
+      drag = TestUtils.renderIntoDocument(
+        <Draggable onDrag={onDrag} scale={2}>
+          <div />
+        </Draggable>
+      );
+
+      // (element, fromX, fromY, toX, toY)
+      simulateMovementFromTo(drag, 0, 0, 100, 100);
+    });
+
+    it('should call back with correct position when parent element is 0.5x scaled', function() {
+      function onDrag(event, data) {
+        // visually it will look like 100, because parent is 0.5x scaled
+        assert(data.x === 200);
+        assert(data.y === 200);
+        assert(data.deltaX === 200);
+        assert(data.deltaY === 200);
+        assert(data.node === ReactDOM.findDOMNode(drag));
+      }
+      drag = TestUtils.renderIntoDocument(
+        <Draggable onDrag={onDrag} scale={0.5}>
+          <div />
+        </Draggable>
+      );
+
+      // (element, fromX, fromY, toX, toY)
+      simulateMovementFromTo(drag, 0, 0, 100, 100);
+    });
   });
 
   describe('DraggableCore callbacks', function () {
@@ -703,6 +741,44 @@ describe('react-draggable', function () {
       // (element, fromX, fromY, toX, toY)
       simulateMovementFromTo(drag, 0, 0, 100, 100);
     });
+
+    it('should call back with correct position when parent element is 2x scaled', function() {
+      function onDrag(event, data) {
+        // visually it will look like 100, because parent is 2x scaled
+        assert(data.x === 50);
+        assert(data.y === 50);
+        assert(data.deltaX === 50);
+        assert(data.deltaY === 50);
+        assert(data.node === ReactDOM.findDOMNode(drag));
+      }
+      drag = TestUtils.renderIntoDocument(
+        <DraggableCore onDrag={onDrag} scale={2}>
+          <div />
+        </DraggableCore>
+      );
+
+      // (element, fromX, fromY, toX, toY)
+      simulateMovementFromTo(drag, 0, 0, 100, 100);
+    });
+
+    it('should call back with correct position when parent element is 0.5x scaled', function() {
+      function onDrag(event, data) {
+        // visually it will look like 100, because parent is  0.5x scaled
+        assert(data.x === 200);
+        assert(data.y === 200);
+        assert(data.deltaX === 200);
+        assert(data.deltaY === 200);
+        assert(data.node === ReactDOM.findDOMNode(drag));
+      }
+      drag = TestUtils.renderIntoDocument(
+        <DraggableCore onDrag={onDrag} scale={0.5}>
+          <div />
+        </DraggableCore>
+      );
+
+      // (element, fromX, fromY, toX, toY)
+      simulateMovementFromTo(drag, 0, 0, 100, 100);
+    });
   });