From 2a2a298b73f7beb9a2a61c309e649be3d2527473 Mon Sep 17 00:00:00 2001
From: streamich <streamich@gmail.com>
Date: Sat, 16 May 2020 14:26:51 +0200
Subject: [PATCH] =?UTF-8?q?feat:=20=F0=9F=8E=B8=20add=20useScratch=20hook?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/useScratch.ts | 180 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 180 insertions(+)
 create mode 100644 src/useScratch.ts

diff --git a/src/useScratch.ts b/src/useScratch.ts
new file mode 100644
index 0000000000..436502727f
--- /dev/null
+++ b/src/useScratch.ts
@@ -0,0 +1,180 @@
+import { useState, useEffect, useRef, FC, cloneElement } from 'react';
+import { render } from 'react-universal-interface';
+
+const noop = () => {};
+
+export interface ScratchSensorParams {
+  disabled?: boolean;
+  onScratch?: (state: ScratchSensorState) => void;
+  onScratchStart?: (state: ScratchSensorState) => void;
+  onScratchEnd?: (state: ScratchSensorState) => void;
+}
+
+export interface ScratchSensorState {
+  isScratching: boolean;
+  start?: number;
+  end?: number;
+  x?: number;
+  y?: number;
+  dx?: number;
+  dy?: number;
+  docX?: number;
+  docY?: number;
+  posX?: number;
+  posY?: number;
+  elH?: number;
+  elW?: number;
+  elX?: number;
+  elY?: number;
+}
+
+const useScratch = ({
+  disabled,
+  onScratch = noop,
+  onScratchStart = noop,
+  onScratchEnd = noop,
+}: ScratchSensorParams = {}): [ScratchSensorState, (el: HTMLElement | null) => void] => {
+  const [state, setState] = useState<ScratchSensorState>({ isScratching: false });
+  const refState = useRef<ScratchSensorState>(state);
+  const refScratching = useRef<boolean>(false);
+  const refAnimationFrame = useRef<any>(null);
+  const [el, setEl] = useState<HTMLElement | null>(null);
+  useEffect(() => {
+    if (disabled) return;
+    if (!el) return;
+
+    const onMoveEvent = (docX, docY) => {
+      cancelAnimationFrame(refAnimationFrame.current);
+      refAnimationFrame.current = requestAnimationFrame(() => {
+        const { left, top } = el.getBoundingClientRect();
+        const elX = left + window.scrollX;
+        const elY = top + window.scrollY;
+        const x = docX - elX;
+        const y = docY - elY;
+        setState(oldState => {
+          const newState = {
+            ...oldState,
+            dx: x - (oldState.x || 0),
+            dy: y - (oldState.y || 0),
+            end: Date.now(),
+            isScratching: true,
+          };
+          refState.current = newState;
+          onScratch(newState);
+          return newState;
+        });
+      });
+    };
+
+    const onMouseMove = event => {
+      onMoveEvent(event.pageX, event.pageY);
+    };
+
+    const onTouchMove = event => {
+      onMoveEvent(event.changedTouches[0].pageX, event.changedTouches[0].pageY);
+    };
+
+    let onMouseUp;
+    let onTouchEnd;
+
+    const stopScratching = () => {
+      if (!refScratching.current) return;
+      refScratching.current = false;
+      refState.current = { ...refState.current, isScratching: false };
+      onScratchEnd(refState.current);
+      setState({ isScratching: false });
+      window.removeEventListener('mousemove', onMouseMove);
+      window.removeEventListener('touchmove', onTouchMove);
+      window.removeEventListener('mouseup', onMouseUp);
+      window.removeEventListener('touchend', onTouchEnd);
+    };
+
+    onMouseUp = stopScratching;
+    onTouchEnd = stopScratching;
+
+    const startScratching = (docX, docY) => {
+      if (!refScratching.current) return;
+      const { left, top } = el.getBoundingClientRect();
+      const elX = left + window.scrollX;
+      const elY = top + window.scrollY;
+      const x = docX - elX;
+      const y = docY - elY;
+      const time = Date.now();
+      const newState = {
+        isScratching: true,
+        start: time,
+        end: time,
+        docX,
+        docY,
+        x,
+        y,
+        dx: 0,
+        dy: 0,
+        elH: el.offsetHeight,
+        elW: el.offsetWidth,
+        elX,
+        elY,
+      };
+      refState.current = newState;
+      onScratchStart(newState);
+      setState(newState);
+      window.addEventListener('mousemove', onMouseMove);
+      window.addEventListener('touchmove', onTouchMove);
+      window.addEventListener('mouseup', onMouseUp);
+      window.addEventListener('touchend', onTouchEnd);
+    };
+
+    const onMouseDown = event => {
+      refScratching.current = true;
+      startScratching(event.pageX, event.pageY);
+    };
+
+    const onTouchStart = event => {
+      refScratching.current = true;
+      startScratching(event.changedTouches[0].pageX, event.changedTouches[0].pageY);
+    };
+
+    el.addEventListener('mousedown', onMouseDown);
+    el.addEventListener('touchstart', onTouchStart);
+
+    return () => {
+      el.removeEventListener('mousedown', onMouseDown);
+      el.removeEventListener('touchstart', onTouchStart);
+      window.removeEventListener('mousemove', onMouseMove);
+      window.removeEventListener('touchmove', onTouchMove);
+      window.removeEventListener('mouseup', onMouseUp);
+      window.removeEventListener('touchend', onTouchEnd);
+
+      if (refAnimationFrame.current) cancelAnimationFrame(refAnimationFrame.current);
+      refAnimationFrame.current = null;
+
+      refScratching.current = false;
+      refState.current = { isScratching: false };
+      setState(refState.current);
+    };
+  }, [el, disabled, onScratchStart, onScratch, onScratchEnd]);
+
+  return [state, setEl];
+};
+
+export interface ScratchSensorProps extends ScratchSensorParams {
+  children: (state: ScratchSensorState, ref: (el: HTMLElement | null) => void) => React.ReactElement<any>;
+}
+
+export const ScratchSensor: FC<ScratchSensorProps> = props => {
+  const { children, ...params } = props;
+  const [state, ref] = useScratch(params);
+  const element = render(props, state);
+  return cloneElement(element, {
+    ...element.props,
+    ref: el => {
+      if (element.props.ref) {
+        if (typeof element.props.ref === 'object') element.props.ref.current = el;
+        if (typeof element.props.ref === 'function') element.props.ref(el);
+      }
+      ref(el);
+    },
+  });
+};
+
+export default useScratch;