From f974181924177a46b81cdc46fda6ccfc92b26cf9 Mon Sep 17 00:00:00 2001
From: mkosir <kosir.mar@gmail.com>
Date: Sat, 23 Nov 2019 11:27:57 +0100
Subject: [PATCH] add wheel event

---
 src/react-hook-mighty-mouse/types.ts           | 10 +++++++++-
 src/react-hook-mighty-mouse/useMightyMouse.tsx |  7 +++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/react-hook-mighty-mouse/types.ts b/src/react-hook-mighty-mouse/types.ts
index e7d4d42..8a83c09 100644
--- a/src/react-hook-mighty-mouse/types.ts
+++ b/src/react-hook-mighty-mouse/types.ts
@@ -23,6 +23,8 @@ export type Mouse = {
     left: boolean | null;
     middle: boolean | null;
     right: boolean | null;
+    wheelDown: boolean | null;
+    wheelUp: boolean | null;
   };
   /*
    * Pressed keys
@@ -60,4 +62,10 @@ type SelectedElement = {
   isHover: boolean;
 };
 
-export type EventType = 'mousemove' | 'mousedown' | 'mouseup' | 'touchmove' | 'touchstart';
+export type EventType =
+  | 'mousemove'
+  | 'mousedown'
+  | 'mouseup'
+  | 'touchmove'
+  | 'touchstart'
+  | 'wheel';
diff --git a/src/react-hook-mighty-mouse/useMightyMouse.tsx b/src/react-hook-mighty-mouse/useMightyMouse.tsx
index e32b1c8..32c7d6f 100644
--- a/src/react-hook-mighty-mouse/useMightyMouse.tsx
+++ b/src/react-hook-mighty-mouse/useMightyMouse.tsx
@@ -11,6 +11,8 @@ const initMouse: Mouse = {
     left: null,
     middle: null,
     right: null,
+    wheelDown: null,
+    wheelUp: null,
   },
   keyboard: {
     ctrl: null,
@@ -55,6 +57,7 @@ const useMightyMouse = (
       case 'mousemove':
       case 'mousedown':
       case 'mouseup':
+      case 'wheel':
         const mouseEvent = event as MouseEvent;
         clientX = mouseEvent.clientX;
         clientY = mouseEvent.clientY;
@@ -65,6 +68,8 @@ const useMightyMouse = (
         buttons.left = [1, 3, 5, 7].indexOf(mouseEvent.buttons) > -1;
         buttons.right = [2, 3, 6, 7].indexOf(mouseEvent.buttons) > -1;
         buttons.middle = [4, 5, 6, 7].indexOf(mouseEvent.buttons) > -1;
+        buttons.wheelDown = (event as WheelEvent).deltaY > 0;
+        buttons.wheelUp = (event as WheelEvent).deltaY < 0;
         break;
       case 'touchmove':
       case 'touchstart':
@@ -160,6 +165,7 @@ const useMightyMouse = (
     document.addEventListener('mousemove', onMouseTouchEvent);
     document.addEventListener('mousedown', onMouseTouchEvent);
     document.addEventListener('mouseup', onMouseTouchEvent);
+    document.addEventListener('wheel', onMouseTouchEvent);
     document.addEventListener('mouseleave', onLeave);
     document.addEventListener('keydown', onKeyEvent);
     document.addEventListener('keyup', onKeyEvent);
@@ -173,6 +179,7 @@ const useMightyMouse = (
       document.removeEventListener('mousemove', onMouseTouchEvent);
       document.removeEventListener('mousedown', onMouseTouchEvent);
       document.removeEventListener('mouseup', onMouseTouchEvent);
+      document.removeEventListener('wheel', onMouseTouchEvent);
       document.removeEventListener('mouseleave', onLeave);
       document.removeEventListener('keydown', onKeyEvent);
       document.removeEventListener('keyup', onKeyEvent);