From c046d08be967845bd4191cf02dd9308dfe8f6574 Mon Sep 17 00:00:00 2001
From: benekoehler <benedikt.koehler@peerigon.com>
Date: Sun, 16 Oct 2022 13:48:31 +0200
Subject: [PATCH] feat: add enabled prop to PresentationControls (#1096)

---
 .../stories/PresentationControls.stories.tsx  | 33 +++++++++++++++++++
 README.md                                     |  1 +
 src/web/PresentationControls.tsx              | 15 ++++++---
 3 files changed, 44 insertions(+), 5 deletions(-)
 create mode 100644 .storybook/stories/PresentationControls.stories.tsx

diff --git a/.storybook/stories/PresentationControls.stories.tsx b/.storybook/stories/PresentationControls.stories.tsx
new file mode 100644
index 000000000..13128f908
--- /dev/null
+++ b/.storybook/stories/PresentationControls.stories.tsx
@@ -0,0 +1,33 @@
+import * as React from 'react'
+
+import { Setup } from '../Setup'
+
+import { Box, PresentationControlProps, PresentationControls } from '../../src'
+
+export function PresentationControlStory({ enabled, ...rest }: PresentationControlProps) {
+  return (
+    <PresentationControls global snap enabled={enabled} {...rest}>
+      <Box args={[1, 1, 1]}>
+        <meshBasicMaterial wireframe />
+        <axesHelper args={[100]} />
+      </Box>
+    </PresentationControls>
+  )
+}
+
+PresentationControlStory.storyName = 'Default'
+PresentationControlStory.args = {
+  enabled: true,
+}
+
+export default {
+  title: 'Controls/PresentationControls',
+  component: PresentationControls,
+  decorators: [
+    (storyFn) => (
+      <Setup camera={{ near: 1, far: 1100, fov: 75 }} controls={false}>
+        {storyFn()}
+      </Setup>
+    ),
+  ],
+}
diff --git a/README.md b/README.md
index 5bb41f7b9..e005cf2f4 100644
--- a/README.md
+++ b/README.md
@@ -372,6 +372,7 @@ Semi-OrbitControls with spring-physics, polar zoom and snap-back, for presentati
 
 ```jsx
 <PresentationControls
+  enabled={true} // the controls can be disabled by setting this to false
   global={false} // Spin globally or by dragging the model
   cursor={true} // Whether to toggle cursor style on drag
   snap={false} // Snap-back to center (can also be a spring config)
diff --git a/src/web/PresentationControls.tsx b/src/web/PresentationControls.tsx
index 8dacebcec..cd43bba69 100644
--- a/src/web/PresentationControls.tsx
+++ b/src/web/PresentationControls.tsx
@@ -5,7 +5,7 @@ import { a, useSpring } from '@react-spring/three'
 import { useGesture } from '@use-gesture/react'
 import { SpringConfig } from '@react-spring/core'
 
-type Props = {
+export type PresentationControlProps = {
   snap?: Boolean | SpringConfig
   global?: boolean
   cursor?: boolean
@@ -15,10 +15,12 @@ type Props = {
   polar?: [number, number]
   azimuth?: [number, number]
   config?: any
+  enabled?: boolean
   children?: React.ReactNode
 }
 
 export function PresentationControls({
+  enabled = true,
   snap,
   global,
   cursor = true,
@@ -29,7 +31,7 @@ export function PresentationControls({
   polar = [0, Math.PI / 2],
   azimuth = [-Infinity, Infinity],
   config = { mass: 1, tension: 170, friction: 26 },
-}: Props) {
+}: PresentationControlProps) {
   const { size, gl } = useThree()
   const rPolar = React.useMemo(
     () => [rotation[0] + polar[0], rotation[0] + polar[1]],
@@ -46,14 +48,17 @@ export function PresentationControls({
   const [spring, api] = useSpring(() => ({ scale: 1, rotation: rInitial, config }))
   React.useEffect(() => void api.start({ scale: 1, rotation: rInitial, config }), [rInitial])
   React.useEffect(() => {
-    if (global && cursor) gl.domElement.style.cursor = 'grab'
-  }, [global, cursor, gl.domElement])
+    if (global && cursor && enabled) gl.domElement.style.cursor = 'grab'
+
+    return () => void (gl.domElement.style.cursor = 'default')
+  }, [global, cursor, gl.domElement, enabled])
   const bind = useGesture(
     {
       onHover: ({ last }) => {
-        if (cursor && !global) gl.domElement.style.cursor = last ? 'auto' : 'grab'
+        if (cursor && !global && enabled) gl.domElement.style.cursor = last ? 'auto' : 'grab'
       },
       onDrag: ({ down, delta: [x, y], memo: [oldY, oldX] = spring.rotation.animation.to || rInitial }) => {
+        if (!enabled) return [y, x]
         if (cursor) gl.domElement.style.cursor = down ? 'grabbing' : 'grab'
         x = MathUtils.clamp(oldX + (x / size.width) * Math.PI * speed, ...rAzimuth)
         y = MathUtils.clamp(oldY + (y / size.height) * Math.PI * speed, ...rPolar)