fix: fix the error of image rotating too many times after reseting im…

…age (#3108)

src/image-viewer/hooks/useRotate.ts

@@ -1,14 +1,31 @@
 // 旋转控制
-import { useCallback, useState } from 'react';
+import { useCallback, useRef, useState } from 'react';
 
 const useRotate = () => {
+  // There is an useEffect in the line 472 of ImageViewerModal.tsx, so we need to use a ref to store the rotateZ value.
+  const rotRef = useRef(0);
   const [rotateZ, setRotateZ] = useState(0);
 
   const onRotate = useCallback((ROTATE_COUNT: number) => {
-    setRotateZ((rotateZ) => rotateZ + ROTATE_COUNT);
+    setRotateZ((rotateZ) => {
+      rotRef.current = rotateZ + ROTATE_COUNT;
+      return rotateZ + ROTATE_COUNT;
+    });
   }, []);
 
-  const onResetRotate = useCallback(() => setRotateZ(0), []);
+  const onResetRotate = useCallback(() => {
+    let degreeToRotate = rotRef.current % 360;
+    // make sure we always rotate to the shortest direction
+    if (Math.abs(degreeToRotate) > 180) {
+      degreeToRotate = (degreeToRotate + 360) % 360;
+    }
+    if (degreeToRotate !== 0) {
+      setRotateZ((rotateZ) => {
+        rotRef.current = rotateZ - degreeToRotate;
+        return rotateZ - degreeToRotate;
+      });
+    }
+  }, []);
 
   return {
     rotateZ,